private Builder order(String fieldName, TypeName fieldType, SortOrder sortOrder) { QueryTerm sortTerm = new QueryTerm(); sortTerm.propertyName = fieldName; sortTerm.propertyType = fieldType; if (this.querySpec.sortTerm == null) { this.querySpec.sortTerm = sortTerm; this.querySpec.sortOrder = sortOrder; } else { sortTerm.sortOrder = sortOrder; if (this.querySpec.additionalSortTerms == null) { this.querySpec.additionalSortTerms = new ArrayList<>(); } this.querySpec.additionalSortTerms.add(sortTerm); } addOption(QueryOption.SORT); return this; }
@Override public Builder adjustBuilder(Builder builder) { builder.addOption(QueryOption.EXPAND_SELECTED_FIELDS); fields.forEach((field) -> { builder.addSelectTerm(field); }); return builder; }
@Override public Builder adjustBuilder(Builder builder) { builder.addOption(QueryOption.EXPAND_SELECTED_FIELDS); fields.forEach((field) -> { builder.addSelectTerm(field); }); return builder; }
private void verifyNumericGroupQueryPaginatedPerGroup(VerificationHost targetHost, List<Long> groups) throws Throwable { Query query = Query.Builder.create() .addKindFieldClause(ExampleServiceState.class) .build(); QueryTask queryTask = QueryTask.Builder.create() .addOption(QueryOption.GROUP_BY) .addOption(QueryOption.EXPAND_CONTENT) .setResultLimit(this.serviceCount / 5) .orderAscending(ExampleServiceState.FIELD_NAME_NAME, TypeName.STRING) .groupOrder(ExampleServiceState.FIELD_NAME_SORTED_COUNTER, TypeName.LONG, SortOrder.ASC) .setQuery(query).build(); URI queryTaskURI = this.host.createQueryTaskService(queryTask); QueryTask finalState = this.host.waitForQueryTask(queryTaskURI, TaskStage.FINISHED); int expectedCountPerPage = queryTask.querySpec.resultLimit; validateNumericGroupByResults(targetHost, groups, null, finalState, null, expectedCountPerPage); }
private static Operation buildLatestInTransactionQueryTaskOp(StatefulService s, String txid, AuthorizationContext authContext) { Query.Builder queryBuilder = Query.Builder.create(); queryBuilder.addFieldClause(ServiceDocument.FIELD_NAME_SELF_LINK, s.getSelfLink()); queryBuilder.addFieldClause(ServiceDocument.FIELD_NAME_TRANSACTION_ID, txid); QueryTask.Builder queryTaskBuilder = QueryTask.Builder.createDirectTask() .setQuery(queryBuilder.build()); queryTaskBuilder.addOption(QueryOption.EXPAND_CONTENT); queryTaskBuilder.addOption(QueryOption.INCLUDE_ALL_VERSIONS); queryTaskBuilder.orderDescending(ServiceDocument.FIELD_NAME_VERSION, TypeName.LONG); QueryTask task = queryTaskBuilder.build(); Operation returnOp = Operation .createPost(s.getHost(), ServiceUriPaths.CORE_QUERY_TASKS) .setBody(task); returnOp.setAuthorizationContext(authContext); return returnOp; }
private Operation createEpzQueryOperation(String resourcePoolLink, boolean expandContent) { Query epzQuery = Query.Builder.create() .addKindFieldClause(ElasticPlacementZoneState.class) .addFieldClause(ElasticPlacementZoneState.FIELD_NAME_RESOURCE_POOL_LINK, resourcePoolLink) .build(); QueryTask.Builder epzQueryTaskBuilder = QueryTask.Builder.createDirectTask() .setQuery(epzQuery); if (expandContent) { epzQueryTaskBuilder.addOption(QueryOption.EXPAND_CONTENT); } Operation queryEpzOp = Operation .createPost(getHost(), ServiceUriPaths.CORE_QUERY_TASKS) .setBody(epzQueryTaskBuilder.build()) .setReferer(getUri()); return queryEpzOp; }
private static QueryTask getQueryTask() { Query computeQuery = Query.Builder.create() .addKindFieldClause(SslTrustCertificateState.class) .build(); QueryTask task = QueryTask.Builder .create() .addOption(QueryOption.CONTINUOUS) .addOption(QueryOption.EXPAND_CONTENT) .setQuery(computeQuery).build(); task.documentSelfLink = QUERY_TASK_SELF_LINK_PREFIX; task.documentExpirationTimeMicros = Utils.fromNowMicrosUtc( TimeUnit.DAYS.toMicros(QUERY_TASK_EXPIRATION_DAYS)); return task; }
public QueryTask createTagsQueryTask(boolean deletedTags) { Query.Builder query = Query.Builder.create() .addKindFieldClause(TagState.class); if (deletedTags) { query.addFieldClause(TagState.FIELD_NAME_DELETED, Boolean.TRUE); } else { query.addFieldClause(TagState.FIELD_NAME_DELETED, Boolean.FALSE); } return QueryTask.Builder.createDirectTask() .addOption(QueryOption.EXPAND_CONTENT) .setQuery(query.build()) .build(); }
@Override public Builder adjustBuilder(Builder builder) { return builder.addOption(QueryOption.EXPAND_CONTENT); }
private QueryTask createGraphQueryStage(int stageIndex) { QueryTask stage = QueryTask.Builder.create() .addOption(QueryOption.SELECT_LINKS) .addLinkTerm(QueryValidationServiceState.FIELD_NAME_SERVICE_LINKS) .setQuery(Query.Builder.create() .addRangeClause(QueryValidationServiceState.FIELD_NAME_LONG_VALUE, NumericRange.createLongRange((long) stageIndex, (long) stageIndex, true, true)) .addKindFieldClause(QueryValidationServiceState.class) .build()) .build(); return stage; }
private Long getDocumentCount(VerificationHost host, String factoryLink) { QueryTask queryTask = QueryTask.Builder .createDirectTask() .addOption(QueryOption.INCLUDE_ALL_VERSIONS) .addOption(QueryOption.COUNT) .setQuery( Query.Builder.create().addFieldClause(ServiceDocument.FIELD_NAME_SELF_LINK, factoryLink, MatchType.PREFIX).build()) .build(); host.createQueryTaskService(queryTask, false, true, queryTask, null); return queryTask.results.documentCount; } }
@Override public Builder adjustBuilder(Builder builder) { return builder.addOption(QueryOption.EXPAND_CONTENT); }
private QueryTask createGraphQueryStage(int stageIndex) { QueryTask stage = QueryTask.Builder.create() .addOption(QueryOption.SELECT_LINKS) .addLinkTerm(QueryValidationServiceState.FIELD_NAME_SERVICE_LINKS) .setQuery(Query.Builder.create() .addRangeClause(QueryValidationServiceState.FIELD_NAME_LONG_VALUE, NumericRange.createLongRange((long) stageIndex, (long) stageIndex, true, true)) .addKindFieldClause(QueryValidationServiceState.class) .build()) .build(); return stage; }
/** * Add {@link QueryOption#TOP_RESULTS} since this is 'query TOP documents/links'. */ @Override protected QueryTask.Builder newQueryTaskBuilder() { return super.newQueryTaskBuilder().addOption(QueryOption.TOP_RESULTS); }
@Override public Builder adjustBuilder(Builder builder) { builder.addOption(QueryOption.EXPAND_SELECTED_FIELDS); fields.forEach((field) -> { builder.addSelectTerm(field); }); return builder; }
private void assertDocumentCount(long expectedCount, Collection<String> documentLinks) throws Throwable { Query query = Query.Builder.create() .addInClause(ServiceDocument.FIELD_NAME_SELF_LINK, documentLinks).build(); QueryTask queryTask = QueryTask.Builder.createDirectTask().setQuery(query) .addOption(QueryOption.COUNT).build(); QueryTask returnedTask = querySynchronously(queryTask); assertThat(returnedTask.results.documentCount, is(expectedCount)); }
@Override public DeferredResult<Void> queryDocuments(Consumer<T> documentConsumer) { QueryTask.Builder queryTaskBuilder = newQueryTaskBuilder() .addOption(QueryOption.EXPAND_CONTENT); return queryImpl(queryTaskBuilder, documentConsumer); }
/** * Add {@link QueryOption#TOP_RESULTS} since this is 'query TOP documents/links'. */ @Override protected QueryTask.Builder newQueryTaskBuilder() { return super.newQueryTaskBuilder() .addOption(QueryOption.TOP_RESULTS); }
@Override public DeferredResult<Void> queryDocuments(Consumer<T> documentConsumer) { QueryTask.Builder queryTaskBuilder = newQueryTaskBuilder() .addOption(QueryOption.EXPAND_CONTENT); return queryImpl(queryTaskBuilder, documentConsumer); }
@Override public Builder adjustBuilder(Builder builder) { builder.addOption(QueryOption.EXPAND_SELECTED_FIELDS); fields.forEach((field) -> { builder.addSelectTerm(field); }); return builder; }