Examples with AfterQueryEvent org.springframework.cloud.gcp.data.datastore.core.mapping.event.AfterQueryEvent used on opensource projects

Example 6 with AfterQueryEvent

use of org.springframework.cloud.gcp.data.datastore.core.mapping.event.AfterQueryEvent in project spring-cloud-gcp by spring-cloud.

the class DatastoreTemplate method findAll.

@Override
public <T> DatastoreResultsCollection<T> findAll(Class<T> entityClass, DatastoreQueryOptions queryOptions) {
    DatastorePersistentEntity<?> persistentEntity = this.datastoreMappingContext.getPersistentEntity(entityClass);
    EntityQuery.Builder builder = Query.newEntityQueryBuilder().setKind(persistentEntity.kindName());
    applyQueryOptions(builder, queryOptions, persistentEntity);
    Query query = builder.build();
    QueryResults queryResults = getDatastoreReadWriter().run(query);
    Collection<T> convertedResults = convertEntitiesForRead(queryResults, entityClass);
    maybeEmitEvent(new AfterQueryEvent(convertedResults, query));
    return new DatastoreResultsCollection<>(convertedResults, queryResults != null ? queryResults.getCursorAfter() : null);
}

Example 7 with AfterQueryEvent

use of org.springframework.cloud.gcp.data.datastore.core.mapping.event.AfterQueryEvent in project spring-cloud-gcp by spring-cloud.

the class DatastoreTemplate method queryKeysOrEntities.

@Override
public <T> DatastoreResultsIterable<?> queryKeysOrEntities(Query query, Class<T> entityClass) {
    QueryResults results = getDatastoreReadWriter().run(query);
    DatastoreResultsIterable resultsIterable;
    if (results.getResultClass() == Key.class) {
        resultsIterable = new DatastoreResultsIterable(results, results.getCursorAfter());
    } else {
        resultsIterable = new DatastoreResultsIterable<>(convertEntitiesForRead(results, entityClass), results.getCursorAfter());
    }
    maybeEmitEvent(new AfterQueryEvent(resultsIterable, query));
    return resultsIterable;
}