Java SqlQuerySpec類代碼示例

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
public static Database GetDatabase(DocumentClient client, String databaseId) {
    BackoffExponentialRetryPolicy retryPolicy = new BackoffExponentialRetryPolicy();
    QueryIterable<Database> dbIterable = client.queryDatabases(
            new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]",
                    new SqlParameterCollection(new SqlParameter("@id", databaseId))),
            null).getQueryIterable();
    
    List<Database> databases = null;
    while(retryPolicy.shouldRetry()){
        try {
            databases = dbIterable.toList();
            break;
        } catch (Exception e) {
            retryPolicy.errorOccured(e);
        }
    }
    
    if(databases.size() == 0) {
        return null;
    }
    
    return databases.get(0);
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
private void cleanUpGeneratedDatabases() throws DocumentClientException {
    LOGGER.info("cleanup databases invoked");

    String[] allDatabaseIds = { DATABASE_ID };

    for (String id : allDatabaseIds) {
        try {
            List<FeedResponsePage<Database>> feedResponsePages = asyncClient
                    .queryDatabases(new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]",
                            new SqlParameterCollection(new SqlParameter("@id", id))), null).toList().toBlocking().single();
            
            
            if (!feedResponsePages.get(0).getResults().isEmpty()) {
                Database res = feedResponsePages.get(0).getResults().get(0);
                LOGGER.info("deleting a database " + feedResponsePages.get(0));
                asyncClient.deleteDatabase(res.getSelfLink(), null).toBlocking().single();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
private void cleanUpGeneratedDatabases() throws DocumentClientException {
    LOGGER.info("cleanup databases invoked");

    String[] allDatabaseIds = { DATABASE_ID };

    for (String id : allDatabaseIds) {
        try {
            List<FeedResponsePage<Database>> feedResponsePages = asyncClient
                    .queryDatabases(new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]",
                            new SqlParameterCollection(new SqlParameter("@id", id))), null)
                    .toList().toBlocking().single();

            if (!feedResponsePages.get(0).getResults().isEmpty()) {
                Database res = feedResponsePages.get(0).getResults().get(0);
                LOGGER.info("deleting a database " + feedResponsePages.get(0));
                asyncClient.deleteDatabase(res.getSelfLink(), null).toBlocking().single();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
/**
 * This does the same thing as {@link #groupByInMemory_MoreDetail()} but with pedagogical details
 * @throws Exception
 */
@Test
public void groupByInMemory_MoreDetail() throws Exception {

    int requestPageSize = 3;
    FeedOptions options = new FeedOptions();
    options.setPageSize(requestPageSize);


    Observable<Document> documentsObservable = asyncClient
            .queryDocuments(createdCollection.getSelfLink(),
                    new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]_id",
                            new SqlParameterCollection(new SqlParameter("@site_id", "ABC"))),
                    options)
            .flatMap(page -> Observable.from(page.getResults()));

    final LocalDateTime now = LocalDateTime.now();

    Observable<GroupedObservable<Integer, Document>> groupedByPayerIdObservable = documentsObservable
            .filter(doc -> Math.abs(now.getSecond() - doc.getInt("created_time")) <= 90)
            .groupBy(doc -> doc.getInt("payer_id"));

    Observable<List<Document>> docsGroupedAsList = groupedByPayerIdObservable.flatMap(grouped -> {
        Observable<List<Document>> list = grouped.toList();
        return list;
    });

    List<List<Document>> resultsGroupedAsLists = docsGroupedAsList.toList().toBlocking().single();

    for(List<Document> resultsForEachPayer : resultsGroupedAsLists) {
        System.out.println("documents with payer_id : " + resultsForEachPayer.get(0).getInt("payer_id") + " are " + resultsForEachPayer);
    }
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
/**
 * Gets an output collection with the passed name ( if the collection already exists return it, otherwise create new one
 * @param client The DocumentClient instance.
 * @param databaseSelfLink the self link of the passed database.
 * @param collectionId The id of the output collection.
 */
public static DocumentCollection GetDocumentCollection(DocumentClient client, String databaseSelfLink, String collectionId) {
    BackoffExponentialRetryPolicy retryPolicy = new BackoffExponentialRetryPolicy();
    QueryIterable<DocumentCollection> collIterable = client.queryCollections(
            databaseSelfLink,
            new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]",
                    new SqlParameterCollection(new SqlParameter("@id", collectionId))),
            null).getQueryIterable();
    
    List<DocumentCollection> collections = null;
    while(retryPolicy.shouldRetry()){
        try {
            collections = collIterable.toList();
            break;
        } catch (Exception e) {
            retryPolicy.errorOccured(e);
        }
    }
    
    if(collections.size() == 0) {
        return null;
    }
    
    return collections.get(0);
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
public static Database getDatabase(DocumentClient client, String databaseId) {
    FeedResponse<Database> feedResponsePages = client
            .queryDatabases(new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]",
                    new SqlParameterCollection(new SqlParameter("@id", databaseId))), null);
    Iterator<Database> it = feedResponsePages.getQueryIterator();
    if (it == null || !it.hasNext()) {
        throw new RuntimeException("cannot find database " + databaseId);
    }
    return it.next();
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
public static DocumentCollection getCollection(DocumentClient client, String databaseLink, String collectionId) {
    FeedResponse<DocumentCollection> feedResponsePages = client
            .queryCollections(databaseLink, new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]",
                    new SqlParameterCollection(new SqlParameter("@id", collectionId))), null);
    Iterator<DocumentCollection> it = feedResponsePages.getQueryIterator();
    if (it == null || !it.hasNext()) {
        throw new RuntimeException("cannot find collection " + collectionId);
    }
    return it.next();
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
/**
 * If you want to understand the steps in more details see {@link #groupByInMemory_MoreDetail()}
 * @throws Exception
 */
@Test
public void groupByInMemory() throws Exception {

    // if you want to understand the steps in more details see groupByInMemoryMoreDetail()

    int requestPageSize = 3;
    FeedOptions options = new FeedOptions();
    options.setPageSize(requestPageSize);

    Observable<Document> documentsObservable = asyncClient
            .queryDocuments(createdCollection.getSelfLink(),
                    new SqlQuerySpec("SELECT * FROM root r WHERE [email protected]_id",
                            new SqlParameterCollection(new SqlParameter("@site_id", "ABC"))),
                    options)
            .flatMap(page -> Observable.from(page.getResults()));

    final LocalDateTime now = LocalDateTime.now();

    List<List<Document>> resultsGroupedAsLists = documentsObservable
            .filter(doc -> Math.abs(now.getSecond() - doc.getInt("created_time")) <= 90)
            .groupBy(doc -> doc.getInt("payer_id")).flatMap(grouped -> grouped.toList())
            .toList()
            .toBlocking()
            .single();

    for(List<Document> resultsForEachPayer :resultsGroupedAsLists) {
        System.out.println("documents with payer_id : " + resultsForEachPayer.get(0).getInt("payer_id") + " are " + resultsForEachPayer);
    }
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<Database>> queryDatabases(final SqlQuerySpec querySpec, final FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<Database>>() {
        @Override
        public FeedResponse<Database> invoke() throws Exception {
            return client.queryDatabases(querySpec, options);
        }
    });
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<DocumentCollection>> queryCollections(final String databaseLink, final
        SqlQuerySpec querySpec, final FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<DocumentCollection>>() {
        @Override
        public FeedResponse<DocumentCollection> invoke() throws Exception {
            return client.queryCollections(databaseLink, querySpec, options);
        }
    });
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<Document>> queryDocuments(final String collectionLink, final SqlQuerySpec querySpec, final
        FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<Document>>() {
        @Override
        public FeedResponse<Document> invoke() throws Exception {
            return client.queryDocuments(collectionLink, querySpec, options);
        }
    });
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<StoredProcedure>> queryStoredProcedures(final String collectionLink, final
        SqlQuerySpec querySpec, final FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<StoredProcedure>>() {
        @Override
        public FeedResponse<StoredProcedure> invoke() throws Exception {
            return client.queryStoredProcedures(collectionLink, querySpec, options);
        }
    });
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<Trigger>> queryTriggers(final String collectionLink, final SqlQuerySpec querySpec, final
        FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<Trigger>>() {
        @Override
        public FeedResponse<Trigger> invoke() throws Exception {
            return client.queryTriggers(collectionLink, querySpec, options);
        }
    });
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<UserDefinedFunction>> queryUserDefinedFunctions(final String collectionLink, final
        SqlQuerySpec querySpec, final FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<UserDefinedFunction>>() {
        @Override
        public FeedResponse<UserDefinedFunction> invoke() throws Exception {
            return client.queryUserDefinedFunctions(collectionLink, querySpec, options);
        }
    });
}
 

import com.microsoft.azure.documentdb.SqlQuerySpec; //導入依賴的package包/類
@Override
public Observable<FeedResponsePage<Attachment>> queryAttachments(final String documentLink, final SqlQuerySpec querySpec, final
        FeedOptions options) {
    return this.createFeedResponsePageObservable(new ImplFunc<FeedResponse<Attachment>>() {
        @Override
        public FeedResponse<Attachment> invoke() throws Exception {
            return client.queryAttachments(documentLink, querySpec, options);
        }
    });
}