com.azure.cosmos.CosmosDatabase Maven / Gradle / Ivy
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos;
import reactor.core.Exceptions;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import java.util.Iterator;
/**
* Perform read and delete databases, update database throughput, and perform operations on child resources in
* a synchronous way
*/
public class CosmosDatabase {
private final CosmosAsyncDatabase databaseWrapper;
private final CosmosClient client;
private final String id;
/**
* Instantiates a new Cosmos sync database.
*
* @param id the id
* @param client the client
* @param database the database
*/
CosmosDatabase(String id, CosmosClient client, CosmosAsyncDatabase database) {
this.id = id;
this.client = client;
this.databaseWrapper = database;
}
/**
* Get the id of the CosmosAsyncDatabase
*
* @return the id of the database
*/
public String getId() {
return id;
}
/**
* Reads a database
*
* @return the {@link CosmosDatabaseResponse}
* @throws CosmosClientException the cosmos client exception
*/
public CosmosDatabaseResponse read() throws CosmosClientException {
return client.mapDatabaseResponseAndBlock((databaseWrapper.read()));
}
/**
* Reads a database.
*
* @param options the {@link CosmosDatabaseRequestOptions} request options.
* @return the {@link CosmosDatabaseResponse}
* @throws CosmosClientException the cosmos client exception
*/
public CosmosDatabaseResponse read(CosmosDatabaseRequestOptions options) throws CosmosClientException {
return client.mapDatabaseResponseAndBlock(databaseWrapper.read(options));
}
/**
* Delete a database.
*
* @return the {@link CosmosDatabaseResponse}
* @throws CosmosClientException the cosmos client exception
*/
public CosmosDatabaseResponse delete() throws CosmosClientException {
return client.mapDatabaseResponseAndBlock(databaseWrapper.delete());
}
/**
* Delete a database.
*
* @param options the {@link CosmosDatabaseRequestOptions} request options.
* @return the {@link CosmosDatabaseResponse}
* @throws CosmosClientException the cosmos client exception
*/
public CosmosDatabaseResponse delete(CosmosDatabaseRequestOptions options) throws CosmosClientException {
return client.mapDatabaseResponseAndBlock(databaseWrapper.delete(options));
}
/* CosmosAsyncContainer operations */
/**
* Creates a cosmos container.
*
* @param containerProperties the {@link CosmosContainerProperties}
* @return the {@link CosmosContainerResponse} with the created container.
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainer(CosmosContainerProperties containerProperties) throws
CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainer(containerProperties));
}
/**
* Creates a cosmos container.
*
* @param containerProperties the {@link CosmosContainerProperties}
* @param throughput the throughput
* @return the {@link CosmosContainerResponse} with the created container.
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainer(
CosmosContainerProperties containerProperties,
int throughput) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainer(containerProperties, throughput));
}
/**
* Creates a cosmos container.
*
* @param containerProperties the {@link CosmosContainerProperties}
* @param options the {@link CosmosContainerProperties}
* @return the {@link CosmosContainerResponse} with the created container.
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainer(
CosmosContainerProperties containerProperties,
CosmosContainerRequestOptions options) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainer(containerProperties, options));
}
/**
* Creates a cosmos container.
*
* @param containerProperties the {@link CosmosContainerProperties}
* @param throughput the throughput
* @param options the {@link CosmosContainerProperties}
* @return the {@link CosmosContainerResponse} with the created container.
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainer(
CosmosContainerProperties containerProperties,
int throughput,
CosmosContainerRequestOptions options) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainer(containerProperties,
throughput,
options));
}
/**
* Create container cosmos sync container response.
*
* @param id the id
* @param partitionKeyPath the partition key path
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainer(String id, String partitionKeyPath) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainer(id, partitionKeyPath));
}
/**
* Create container cosmos sync container response.
*
* @param id the id
* @param partitionKeyPath the partition key path
* @param throughput the throughput
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainer(String id, String partitionKeyPath, int throughput) throws
CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainer(id, partitionKeyPath, throughput));
}
/**
* Create container if not exists cosmos sync container response.
*
* @param containerProperties the container properties
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainerIfNotExists(CosmosContainerProperties containerProperties) throws
CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainerIfNotExists(containerProperties));
}
/**
* Create container if not exists cosmos sync container response.
*
* @param containerProperties the container properties
* @param throughput the throughput
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainerIfNotExists(
CosmosContainerProperties containerProperties,
int throughput) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainerIfNotExists(containerProperties,
throughput));
}
/**
* Create container if not exists cosmos sync container response.
*
* @param id the id
* @param partitionKeyPath the partition key path
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainerIfNotExists(
String id,
String partitionKeyPath) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainerIfNotExists(id, partitionKeyPath));
}
/**
* Create container if not exists cosmos sync container response.
*
* @param id the id
* @param partitionKeyPath the partition key path
* @param throughput the throughput
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosContainerResponse createContainerIfNotExists(
String id, String partitionKeyPath,
int throughput) throws CosmosClientException {
return this.mapContainerResponseAndBlock(databaseWrapper.createContainerIfNotExists(id,
partitionKeyPath,
throughput));
}
/**
* Map container response and block cosmos sync container response.
*
* @param containerMono the container mono
* @return the cosmos sync container response
* @throws CosmosClientException the cosmos client exception
*/
CosmosContainerResponse mapContainerResponseAndBlock(Mono containerMono) throws
CosmosClientException {
try {
return containerMono
.map(this::convertResponse)
.block();
} catch (Exception ex) {
final Throwable throwable = Exceptions.unwrap(ex);
if (throwable instanceof CosmosClientException) {
throw (CosmosClientException) throwable;
} else {
throw Exceptions.propagate(ex);
}
}
}
/**
* Read all containers iterator.
*
* @param options the options
* @return the iterator
*/
public Iterator> readAllContainers(FeedOptions options) {
return databaseWrapper.readAllContainers(options)
.toIterable()
.iterator();
}
/**
* Read all containers iterator.
*
* @return the iterator
*/
public Iterator> readAllContainers() {
return databaseWrapper.readAllContainers()
.toIterable()
.iterator();
}
/**
* Query containers iterator.
*
* @param query the query
* @return the iterator
*/
public Iterator> queryContainers(String query) {
return databaseWrapper.queryContainers(query)
.toIterable()
.iterator();
}
/**
* Query containers iterator.
*
* @param query the query
* @param options the options
* @return the iterator
*/
public Iterator> queryContainers(String query, FeedOptions options) {
return databaseWrapper.queryContainers(query, options)
.toIterable()
.iterator();
}
/**
* Query containers iterator.
*
* @param querySpec the query spec
* @return the iterator
*/
public Iterator> queryContainers(SqlQuerySpec querySpec) {
return databaseWrapper.queryContainers(querySpec)
.toIterable()
.iterator();
}
/**
* Query containers iterator.
*
* @param querySpec the query spec
* @param options the options
* @return the iterator
*/
public Iterator> queryContainers(
SqlQuerySpec querySpec,
FeedOptions options) {
return databaseWrapper.queryContainers(querySpec, options)
.toIterable()
.iterator();
}
/**
* Gets a CosmosContainer object without making a service call
*
* @param id id of the container
* @return Cosmos Container
*/
public CosmosContainer getContainer(String id) {
return new CosmosContainer(id, this, databaseWrapper.getContainer(id));
}
/**
* Convert response cosmos sync container response.
*
* @param response the response
* @return the cosmos sync container response
*/
CosmosContainerResponse convertResponse(CosmosAsyncContainerResponse response) {
return new CosmosContainerResponse(response, this, client);
}
/* Users */
/**
* Create user cosmos sync user response.
*
* @param settings the settings
* @return the cosmos sync user response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosUserResponse createUser(CosmosUserProperties settings) throws CosmosClientException {
return mapUserResponseAndBlock(databaseWrapper.createUser(settings));
}
/**
* Upsert user cosmos sync user response.
*
* @param settings the settings
* @return the cosmos sync user response
* @throws CosmosClientException the cosmos client exception
*/
public CosmosUserResponse upsertUser(CosmosUserProperties settings) throws CosmosClientException {
return mapUserResponseAndBlock(databaseWrapper.upsertUser(settings));
}
/**
* Read all users iterator.
*
* @return the iterator
*/
public Iterator> readAllUsers() {
return getFeedIterator(databaseWrapper.readAllUsers());
}
/**
* Read all users iterator.
*
* @param options the options
* @return the iterator
*/
public Iterator> readAllUsers(FeedOptions options) {
return getFeedIterator(databaseWrapper.readAllUsers(options));
}
/**
* Query users iterator.
*
* @param query the query
* @return the iterator
*/
public Iterator> queryUsers(String query) {
return getFeedIterator(databaseWrapper.queryUsers(query));
}
/**
* Query users iterator.
*
* @param query the query
* @param options the options
* @return the iterator
*/
public Iterator> queryUsers(String query, FeedOptions options) {
return getFeedIterator(databaseWrapper.queryUsers(query, options));
}
/**
* Query users iterator.
*
* @param querySpec the query spec
* @return the iterator
*/
public Iterator> queryUsers(SqlQuerySpec querySpec) {
return getFeedIterator(databaseWrapper.queryUsers(querySpec));
}
/**
* Query users iterator.
*
* @param querySpec the query spec
* @param options the options
* @return the iterator
*/
public Iterator> queryUsers(SqlQuerySpec querySpec, FeedOptions options) {
return getFeedIterator(databaseWrapper.queryUsers(querySpec, options));
}
/**
* Gets user.
*
* @param id the id
* @return the user
*/
public CosmosUser getUser(String id) {
return new CosmosUser(databaseWrapper.getUser(id), this, id);
}
CosmosUserResponse mapUserResponseAndBlock(Mono containerMono) throws
CosmosClientException {
try {
return containerMono.map(this::convertUserResponse).block();
} catch (Exception ex) {
final Throwable throwable = Exceptions.unwrap(ex);
if (throwable instanceof CosmosClientException) {
throw (CosmosClientException) throwable;
} else {
throw Exceptions.propagate(ex);
}
}
}
private CosmosUserResponse convertUserResponse(CosmosAsyncUserResponse response) {
return new CosmosUserResponse(response, this);
}
/**
* Read provisioned throughput integer.
*
* @return the integer. null response indicates database doesn't have any provisioned RUs
* @throws CosmosClientException the cosmos client exception
*/
public Integer readProvisionedThroughput() throws CosmosClientException {
return throughputResponseToBlock(databaseWrapper.readProvisionedThroughput());
}
/**
* Replace provisioned throughput integer.
*
* @param requestUnitsPerSecond the request units per second
* @return the integer
* @throws CosmosClientException the cosmos client exception
*/
public Integer replaceProvisionedThroughput(int requestUnitsPerSecond) throws CosmosClientException {
return throughputResponseToBlock(databaseWrapper.replaceProvisionedThroughput(requestUnitsPerSecond));
}
Integer throughputResponseToBlock(Mono throughputResponse) throws CosmosClientException {
try {
return throughputResponse.block();
} catch (Exception ex) {
final Throwable throwable = Exceptions.unwrap(ex);
if (throwable instanceof CosmosClientException) {
throw (CosmosClientException) throwable;
} else {
throw Exceptions.propagate(ex);
}
}
}
private Iterator> getFeedIterator(Flux> itemFlux) {
return itemFlux.toIterable(1).iterator();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy