SearchResponse runRequest(SearchRequest request) { logSearchRequest(request); SearchResponse response = esClient.search(request); logSearchResponse(response); return response; }
protected SearchResponse searchScroll(NxQueryBuilder query, SearchType searchType, long keepAlive) { try (Context ignored = searchTimer.time()) { SearchRequest request = buildEsSearchScrollRequest(query, searchType, keepAlive); logSearchRequest(request, query, searchType); SearchResponse response = esa.getClient().search(request); logSearchResponse(response); return response; } }
protected SearchResponse search(NxQueryBuilder query) { try (Context ignored = searchTimer.time()) { SearchType searchType = SearchType.DFS_QUERY_THEN_FETCH; SearchRequest request = buildEsSearchRequest(query, searchType); logSearchRequest(request, query, searchType); SearchResponse response = esa.getClient().search(request); logSearchResponse(response); return response; } }
@Override public Long getEventsCount(String eventId) { SearchResponse res = esClient.search( new SearchRequest(getESIndexName()).source( new SearchSourceBuilder().query( QueryBuilders.constantScoreQuery(QueryBuilders.termQuery("eventId", eventId))) .size(0))); return res.getHits().getTotalHits(); }
indexName, DOC_TYPE, keepAlive, query.toString())); for (SearchResponse response = esa.getClient().search(request); // response.getHits().getHits().length > 0; // response = runNextScroll(response, keepAlive)) {
/** * Ensures the audit sequence returns an UID greater or equal than the maximum log entry id. */ protected void ensureUIDSequencer(ESClient esClient) { boolean auditIndexExists = esClient.indexExists(getESIndexName()); if (!auditIndexExists) { return; } // Get max log entry id SearchRequest request = createSearchRequest(); request.source(new SearchSourceBuilder().query(QueryBuilders.matchAllQuery()) .aggregation(AggregationBuilders.max("maxAgg").field("id"))); SearchResponse searchResponse = esClient.search(request); Max agg = searchResponse.getAggregations().get("maxAgg"); long maxLogEntryId = (long) agg.getValue(); // Get next sequence id UIDGeneratorService uidGeneratorService = Framework.getService(UIDGeneratorService.class); UIDSequencer seq = uidGeneratorService.getSequencer(); seq.init(); long nextSequenceId = seq.getNextLong(SEQ_NAME); // Increment sequence to max log entry id if needed if (nextSequenceId < maxLogEntryId) { log.info(String.format("Next UID returned by %s sequence is %d, initializing sequence to %d", SEQ_NAME, nextSequenceId, maxLogEntryId)); seq.initSequence(SEQ_NAME, maxLogEntryId); } }