Search in sources :

Example 1 with BulkOperation

use of co.elastic.clients.elasticsearch.core.bulk.BulkOperation in project core-ng-project by neowu.

the class ElasticSearchTypeImpl method bulkIndex.

@Override
public void bulkIndex(BulkIndexRequest<T> request) {
    var watch = new StopWatch();
    if (request.sources == null || request.sources.isEmpty())
        throw new Error("request.sources must not be empty");
    String index = request.index == null ? this.index : request.index;
    List<BulkOperation> operations = new ArrayList<>(request.sources.size());
    for (Map.Entry<String, T> entry : request.sources.entrySet()) {
        String id = entry.getKey();
        T source = entry.getValue();
        validator.validate(source, false);
        operations.add(BulkOperation.of(builder -> builder.index(i -> i.index(index).id(id).document(source))));
    }
    long esTook = 0;
    try {
        BulkResponse response = elasticSearch.client.bulk(builder -> builder.operations(operations));
        // mills to nano
        esTook = response.took() * 1_000_000;
        validate(response);
    } catch (IOException e) {
        throw new UncheckedIOException(e);
    } catch (ElasticsearchException e) {
        throw elasticSearch.searchException(e);
    } finally {
        long elapsed = watch.elapsed();
        ActionLogContext.track("elasticsearch", elapsed, 0, request.sources.size());
        logger.debug("bulkIndex, index={}, size={}, esTook={}, elapsed={}", index, request.sources.size(), esTook, elapsed);
        checkSlowOperation(elapsed);
    }
}
Also used : AnalyzeResponse(co.elastic.clients.elasticsearch.indices.AnalyzeResponse) BulkOperation(co.elastic.clients.elasticsearch.core.bulk.BulkOperation) CompletionSuggestOption(co.elastic.clients.elasticsearch.core.search.CompletionSuggestOption) GetRequest(core.framework.search.GetRequest) CodeBuilder(core.framework.internal.asm.CodeBuilder) ActionLogContext(core.framework.log.ActionLogContext) LoggerFactory(org.slf4j.LoggerFactory) UpdateRequest(core.framework.search.UpdateRequest) Index(core.framework.search.Index) BulkIndexRequest(core.framework.search.BulkIndexRequest) IndexRequest(core.framework.search.IndexRequest) StopWatch(core.framework.util.StopWatch) ArrayList(java.util.ArrayList) DeleteResponse(co.elastic.clients.elasticsearch.core.DeleteResponse) BulkDeleteRequest(core.framework.search.BulkDeleteRequest) Markers.errorCode(core.framework.log.Markers.errorCode) CompleteRequest(core.framework.search.CompleteRequest) Duration(java.time.Duration) Map(java.util.Map) SearchException(core.framework.search.SearchException) SearchRequest(core.framework.search.SearchRequest) SearchResponse(core.framework.search.SearchResponse) ErrorCause(co.elastic.clients.elasticsearch._types.ErrorCause) Time(co.elastic.clients.elasticsearch._types.Time) ElasticsearchException(co.elastic.clients.elasticsearch._types.ElasticsearchException) Validator(core.framework.internal.validate.Validator) Logger(org.slf4j.Logger) JsonData(co.elastic.clients.json.JsonData) GetResponse(co.elastic.clients.elasticsearch.core.GetResponse) Collection(java.util.Collection) Suggester(co.elastic.clients.elasticsearch.core.search.Suggester) IOException(java.io.IOException) AnalyzeToken(co.elastic.clients.elasticsearch.indices.analyze.AnalyzeToken) ForEach(core.framework.search.ForEach) Collectors(java.util.stream.Collectors) Result(co.elastic.clients.elasticsearch._types.Result) UncheckedIOException(java.io.UncheckedIOException) DeleteRequest(core.framework.search.DeleteRequest) Hit(co.elastic.clients.elasticsearch.core.search.Hit) ElasticSearchType(core.framework.search.ElasticSearchType) List(java.util.List) AnalyzeRequest(core.framework.search.AnalyzeRequest) BulkResponse(co.elastic.clients.elasticsearch.core.BulkResponse) BulkResponseItem(co.elastic.clients.elasticsearch.core.bulk.BulkResponseItem) Strings(core.framework.util.Strings) Optional(java.util.Optional) BulkOperation(co.elastic.clients.elasticsearch.core.bulk.BulkOperation) ArrayList(java.util.ArrayList) BulkResponse(co.elastic.clients.elasticsearch.core.BulkResponse) UncheckedIOException(java.io.UncheckedIOException) IOException(java.io.IOException) UncheckedIOException(java.io.UncheckedIOException) ElasticsearchException(co.elastic.clients.elasticsearch._types.ElasticsearchException) StopWatch(core.framework.util.StopWatch) Map(java.util.Map)

Example 2 with BulkOperation

use of co.elastic.clients.elasticsearch.core.bulk.BulkOperation in project core-ng-project by neowu.

the class ElasticSearchTypeImpl method bulkDelete.

@Override
public void bulkDelete(BulkDeleteRequest request) {
    var watch = new StopWatch();
    if (request.ids == null || request.ids.isEmpty())
        throw new Error("request.ids must not be empty");
    String index = request.index == null ? this.index : request.index;
    List<BulkOperation> operations = new ArrayList<>(request.ids.size());
    for (String id : request.ids) {
        operations.add(BulkOperation.of(builder -> builder.delete(r -> r.index(index).id(id))));
    }
    long esTook = 0;
    try {
        BulkResponse response = elasticSearch.client.bulk(builder -> builder.operations(operations));
        // mills to nano
        esTook = response.took() * 1_000_000;
        validate(response);
    } catch (IOException e) {
        throw new UncheckedIOException(e);
    } catch (ElasticsearchException e) {
        throw elasticSearch.searchException(e);
    } finally {
        long elapsed = watch.elapsed();
        int size = request.ids.size();
        ActionLogContext.track("elasticsearch", elapsed, 0, size);
        logger.debug("bulkDelete, index={}, ids={}, size={}, esTook={}, elapsed={}", index, request.ids, size, esTook, elapsed);
        checkSlowOperation(elapsed);
    }
}
Also used : AnalyzeResponse(co.elastic.clients.elasticsearch.indices.AnalyzeResponse) BulkOperation(co.elastic.clients.elasticsearch.core.bulk.BulkOperation) CompletionSuggestOption(co.elastic.clients.elasticsearch.core.search.CompletionSuggestOption) GetRequest(core.framework.search.GetRequest) CodeBuilder(core.framework.internal.asm.CodeBuilder) ActionLogContext(core.framework.log.ActionLogContext) LoggerFactory(org.slf4j.LoggerFactory) UpdateRequest(core.framework.search.UpdateRequest) Index(core.framework.search.Index) BulkIndexRequest(core.framework.search.BulkIndexRequest) IndexRequest(core.framework.search.IndexRequest) StopWatch(core.framework.util.StopWatch) ArrayList(java.util.ArrayList) DeleteResponse(co.elastic.clients.elasticsearch.core.DeleteResponse) BulkDeleteRequest(core.framework.search.BulkDeleteRequest) Markers.errorCode(core.framework.log.Markers.errorCode) CompleteRequest(core.framework.search.CompleteRequest) Duration(java.time.Duration) Map(java.util.Map) SearchException(core.framework.search.SearchException) SearchRequest(core.framework.search.SearchRequest) SearchResponse(core.framework.search.SearchResponse) ErrorCause(co.elastic.clients.elasticsearch._types.ErrorCause) Time(co.elastic.clients.elasticsearch._types.Time) ElasticsearchException(co.elastic.clients.elasticsearch._types.ElasticsearchException) Validator(core.framework.internal.validate.Validator) Logger(org.slf4j.Logger) JsonData(co.elastic.clients.json.JsonData) GetResponse(co.elastic.clients.elasticsearch.core.GetResponse) Collection(java.util.Collection) Suggester(co.elastic.clients.elasticsearch.core.search.Suggester) IOException(java.io.IOException) AnalyzeToken(co.elastic.clients.elasticsearch.indices.analyze.AnalyzeToken) ForEach(core.framework.search.ForEach) Collectors(java.util.stream.Collectors) Result(co.elastic.clients.elasticsearch._types.Result) UncheckedIOException(java.io.UncheckedIOException) DeleteRequest(core.framework.search.DeleteRequest) Hit(co.elastic.clients.elasticsearch.core.search.Hit) ElasticSearchType(core.framework.search.ElasticSearchType) List(java.util.List) AnalyzeRequest(core.framework.search.AnalyzeRequest) BulkResponse(co.elastic.clients.elasticsearch.core.BulkResponse) BulkResponseItem(co.elastic.clients.elasticsearch.core.bulk.BulkResponseItem) Strings(core.framework.util.Strings) Optional(java.util.Optional) BulkOperation(co.elastic.clients.elasticsearch.core.bulk.BulkOperation) ArrayList(java.util.ArrayList) BulkResponse(co.elastic.clients.elasticsearch.core.BulkResponse) UncheckedIOException(java.io.UncheckedIOException) IOException(java.io.IOException) UncheckedIOException(java.io.UncheckedIOException) ElasticsearchException(co.elastic.clients.elasticsearch._types.ElasticsearchException) StopWatch(core.framework.util.StopWatch)

Aggregations

ElasticsearchException (co.elastic.clients.elasticsearch._types.ElasticsearchException)2 ErrorCause (co.elastic.clients.elasticsearch._types.ErrorCause)2 Result (co.elastic.clients.elasticsearch._types.Result)2 Time (co.elastic.clients.elasticsearch._types.Time)2 BulkResponse (co.elastic.clients.elasticsearch.core.BulkResponse)2 DeleteResponse (co.elastic.clients.elasticsearch.core.DeleteResponse)2 GetResponse (co.elastic.clients.elasticsearch.core.GetResponse)2 BulkOperation (co.elastic.clients.elasticsearch.core.bulk.BulkOperation)2 BulkResponseItem (co.elastic.clients.elasticsearch.core.bulk.BulkResponseItem)2 CompletionSuggestOption (co.elastic.clients.elasticsearch.core.search.CompletionSuggestOption)2 Hit (co.elastic.clients.elasticsearch.core.search.Hit)2 Suggester (co.elastic.clients.elasticsearch.core.search.Suggester)2 AnalyzeResponse (co.elastic.clients.elasticsearch.indices.AnalyzeResponse)2 AnalyzeToken (co.elastic.clients.elasticsearch.indices.analyze.AnalyzeToken)2 JsonData (co.elastic.clients.json.JsonData)2 CodeBuilder (core.framework.internal.asm.CodeBuilder)2 Validator (core.framework.internal.validate.Validator)2 ActionLogContext (core.framework.log.ActionLogContext)2 Markers.errorCode (core.framework.log.Markers.errorCode)2 AnalyzeRequest (core.framework.search.AnalyzeRequest)2