コード例 #1
0
  @Override
  public void handleRequest(final RestRequest request, final RestChannel channel) {
    MultiSearchRequest multiSearchRequest = new MultiSearchRequest();
    multiSearchRequest.listenerThreaded(false);

    String[] indices = Strings.splitStringByCommaToArray(request.param("index"));
    String[] types = Strings.splitStringByCommaToArray(request.param("type"));
    IndicesOptions indicesOptions =
        IndicesOptions.fromRequest(request, multiSearchRequest.indicesOptions());

    try {
      multiSearchRequest.add(
          request.content(),
          request.contentUnsafe(),
          indices,
          types,
          request.param("search_type"),
          request.param("routing"),
          indicesOptions,
          allowExplicitIndex);
    } catch (Exception e) {
      try {
        XContentBuilder builder = restContentBuilder(request);
        channel.sendResponse(
            new XContentRestResponse(
                request,
                BAD_REQUEST,
                builder.startObject().field("error", e.getMessage()).endObject()));
      } catch (IOException e1) {
        logger.error("Failed to send failure response", e1);
      }
      return;
    }

    client.multiSearch(
        multiSearchRequest,
        new ActionListener<MultiSearchResponse>() {
          @Override
          public void onResponse(MultiSearchResponse response) {
            try {
              XContentBuilder builder = restContentBuilder(request);
              builder.startObject();
              response.toXContent(builder, request);
              builder.endObject();
              channel.sendResponse(new XContentRestResponse(request, OK, builder));
            } catch (Throwable e) {
              onFailure(e);
            }
          }

          @Override
          public void onFailure(Throwable e) {
            try {
              channel.sendResponse(new XContentThrowableRestResponse(request, e));
            } catch (IOException e1) {
              logger.error("Failed to send failure response", e1);
            }
          }
        });
  }
コード例 #2
0
  @Override
  public void handleRequest(final RestRequest request, final RestChannel channel) {
    OptimizeRequest optimizeRequest =
        new OptimizeRequest(RestActions.splitIndices(request.param("index")));
    optimizeRequest.listenerThreaded(false);
    if (request.hasParam("ignore_indices")) {
      optimizeRequest.ignoreIndices(IgnoreIndices.fromString(request.param("ignore_indices")));
    }
    try {
      optimizeRequest.waitForMerge(
          request.paramAsBoolean("wait_for_merge", optimizeRequest.waitForMerge()));
      optimizeRequest.maxNumSegments(
          request.paramAsInt("max_num_segments", optimizeRequest.maxNumSegments()));
      optimizeRequest.onlyExpungeDeletes(
          request.paramAsBoolean("only_expunge_deletes", optimizeRequest.onlyExpungeDeletes()));
      optimizeRequest.flush(request.paramAsBoolean("flush", optimizeRequest.flush()));
      optimizeRequest.refresh(request.paramAsBoolean("refresh", optimizeRequest.refresh()));

      BroadcastOperationThreading operationThreading =
          BroadcastOperationThreading.fromString(
              request.param("operation_threading"), BroadcastOperationThreading.SINGLE_THREAD);
      if (operationThreading == BroadcastOperationThreading.NO_THREADS) {
        // since we don't spawn, don't allow no_threads, but change it to a single thread
        operationThreading = BroadcastOperationThreading.THREAD_PER_SHARD;
      }
      optimizeRequest.operationThreading(operationThreading);
    } catch (Exception e) {
      try {
        XContentBuilder builder = RestXContentBuilder.restContentBuilder(request);
        channel.sendResponse(
            new XContentRestResponse(
                request,
                BAD_REQUEST,
                builder.startObject().field("error", e.getMessage()).endObject()));
      } catch (IOException e1) {
        logger.error("Failed to send failure response", e1);
      }
      return;
    }
    client
        .admin()
        .indices()
        .optimize(
            optimizeRequest,
            new ActionListener<OptimizeResponse>() {
              @Override
              public void onResponse(OptimizeResponse response) {
                try {
                  XContentBuilder builder = RestXContentBuilder.restContentBuilder(request);
                  builder.startObject();
                  builder.field("ok", true);

                  buildBroadcastShardsHeader(builder, response);

                  builder.endObject();
                  channel.sendResponse(new XContentRestResponse(request, OK, builder));
                } catch (Throwable e) {
                  onFailure(e);
                }
              }

              @Override
              public void onFailure(Throwable e) {
                try {
                  channel.sendResponse(new XContentThrowableRestResponse(request, e));
                } catch (IOException e1) {
                  logger.error("Failed to send failure response", e1);
                }
              }
            });
  }