Java CountRequestBuilder类代码示例

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
@Override
public Long count(Class clazz, @Nullable QueryBuilder queryBuilder) {
    CountRequestBuilder countRequestBuilder = client.prepareCount(getIndexName());
    if (queryBuilder != null) {
        countRequestBuilder.setQuery(queryBuilder);
    }


    String typeName = MappingProcessor.getIndexTypeName(clazz);
    countRequestBuilder.setTypes(typeName);

    if (logger.isDebugEnabled()) {
        String queryStr = "";
        if (queryBuilder != null) {
            queryStr = queryBuilder.buildAsBytes().toUtf8();
        }
        logger.debug("Count for class: {}, query: {}", clazz.getSimpleName(), queryStr);
    }

    return countRequestBuilder.get().getCount();
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
/**
 * With retry
 */
public static long countRequest(final String index, final QueryBuilder query, final String... types) {
	CountResponse response = withRetry(new ElasticsearchWithRetry<CountResponse>() {
		public CountResponse call(Client client) throws NoNodeAvailableException {
			CountRequestBuilder request_builder = new CountRequestBuilder(client);
			request_builder.setIndices(index);
			request_builder.setTypes(types);
			request_builder.setQuery(query);
			return request_builder.execute().actionGet();
		}
	});
	if (response == null) {
		return 0;
	}
	return response.getCount();
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
/**
 * http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/search-count.html
 * @param query
 * @param type
 * @return
 */
public <T> long count(SearchCriteria query, String type) throws SearchException {
       Timer.Context timer = searchMetrics.indexTimer(type, "count");
	try {
		CountRequestBuilder request = client.prepareCount(indexName).setTypes(type);
		if(query!=null && query.getQuery()!=null){
			request.setQuery(query.getQuery());
		}
           searchMetrics.indexIncr(type, "count");
		return request.execute().actionGet().getCount();
	} catch (Exception e) {
           searchMetrics.failedIncr(type, "count", e);
           logger.error("count Error. ", e);
           throw new SearchException("count Error. typeName="+type, e);
	}finally {
           timer.stop();
       }
   }
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
@SneakyThrows({InterruptedException.class})
protected void waitForUpdateCompletion(int inv) {
  while(true){
    CountRequestBuilder countRequestBuilder = client.prepareCount (dummyIndexName (getNUMROWS(), getNUMCOLS(), inv)).setTypes (getDocumentType ());
    CountResponse countResponse = countRequestBuilder.execute ().actionGet ();
    if(countResponse.getCount ()==getNUMROWS ())
      break;
    Thread.sleep (500);
  }
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
@SneakyThrows({InterruptedException.class})
protected void waitForUpdateCompletion() {
  while(true){
    CountRequestBuilder countRequestBuilder = client.prepareCount (adminHelper.dummyName (getNUMROWS(), getNUMCOLS())).setTypes (getDocumentType ());
    CountResponse countResponse = countRequestBuilder.execute ().actionGet ();
    if(countResponse.getCount ()==getNUMROWS ())
      break;
    Thread.sleep (500);
  }
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
@Override
public CountResponse count(String[] indices, String... types) {
    CountRequestBuilder countRequestBuilder = esClient.getClient().prepareCount(indices);
    if (types != null && types.length > 0) {
        countRequestBuilder.setTypes(types);
    }
    countRequestBuilder.setQuery(this.queryBuilder);
    return countRequestBuilder.execute().actionGet();
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
@Override
public CountRequestBuilder prepareCount(String... indices) {
    deprecationLogger.deprecated("the count api is deprecated and will be removed from the java api in the next major version");
    return new CountRequestBuilder(this, CountAction.INSTANCE).setIndices(indices);
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
@Override
public long count(Set<String> types, String query, Set<String> filterIds, EzSecurityToken userToken)
        throws TException {
    final Stopwatch watch = LoggingUtils.createStopWatch();
    CountRequestBuilder requestBuilder = client.prepareCount(indexName);

    final VisibilityFilterConfig filterConfig =
            new VisibilityFilterConfig(VISIBILITY_FIELD, EnumSet.of(Permission.DISCOVER));

    final ScriptFilterBuilder visibilityFilter = getVisibilityFilter(userToken, filterConfig);

    if (!types.isEmpty()) {
        requestBuilder = requestBuilder.setTypes(types.toArray(new String[types.size()]));
    }
    if (!filterIds.isEmpty()) {
        if (StringUtils.isNotEmpty(query)) {
            requestBuilder = requestBuilder.setQuery(
                    QueryBuilders.filteredQuery(
                            QueryBuilders.boolQuery().must(parseQuery(query)).must(
                                    QueryBuilders.idsQuery().addIds(
                                            filterIds.toArray(new String[filterIds.size()]))), visibilityFilter));
        } else {
            requestBuilder = requestBuilder.setQuery(
                    QueryBuilders.filteredQuery(
                            QueryBuilders.idsQuery().addIds(filterIds.toArray(new String[filterIds.size()])),
                            visibilityFilter));
        }
    } else if (StringUtils.isNotEmpty(query)) {
        requestBuilder = requestBuilder.setQuery(QueryBuilders.filteredQuery(parseQuery(query), visibilityFilter));
    } else {
        requestBuilder = requestBuilder.setQuery(
                QueryBuilders.filteredQuery(
                        QueryBuilders.matchAllQuery(), visibilityFilter));
    }

    final CountResponse response = requestBuilder.get();

    LoggingUtils.stopAndLogStopWatch(logger, watch, "Count");

    return response.getCount();
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
private long count() throws InterruptedException, ExecutionException {
    return new CountRequestBuilder(esClient.client())
            .setIndices(EsSchema.CONTENT_INDEX)
            .execute().get().getCount();
}
 

import org.elasticsearch.action.count.CountRequestBuilder; //导入依赖的package包/类
/**
 * A count of all the documents matching a specific query.
 * @deprecated use {@link #prepareSearch(String...)} instead and set size to 0
  */
@Deprecated
CountRequestBuilder prepareCount(String... indices);