CosmosTemplate Class

Package:

com.azure.spring.data.cosmos.core

Maven Artifact:

com.azure:azure-spring-data-cosmos:3.47.0

• java.lang.Object
• • com.azure.spring.data.cosmos.core.CosmosTemplate

Implements

CosmosOperations org.springframework.context.ApplicationContextAware

public class CosmosTemplate
implements CosmosOperations, org.springframework.context.ApplicationContextAware

Template class for cosmos db

Constructor Summary

Constructor	Description
CosmosTemplate(CosmosAsyncClient client, String databaseName, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter)	Initialization
CosmosTemplate(CosmosAsyncClient client, String databaseName, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter, IsNewAwareAuditingHandler cosmosAuditingHandler)	Initialization
CosmosTemplate(CosmosFactory cosmosFactory, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter)	Initialization
CosmosTemplate(CosmosFactory cosmosFactory, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter, IsNewAwareAuditingHandler cosmosAuditingHandler)	Initialization

Method Summary

Modifier and Type	Method and Description
void	<S,T>deleteEntities(CosmosEntityInformation<T,?> information, Iterable<S> entities) Deletes the entities using bulk
Iterable<S>	<S,T>insertAll(CosmosEntityInformation<T,?> information, Iterable<S> entities) Insert all items with bulk.
Iterable<T>	<T,ID>findByIds(Iterable<ID> ids, Class<T> domainType, String containerName) Find by ids
long	count(CosmosQuery query, String containerName) Count
long	count(SqlQuerySpec querySpec, String containerName) Count
long	count(String containerName) Count
CosmosContainerProperties	createContainerIfNotExists(CosmosEntityInformation<?,?> information) Creates container if not exists
Iterable<T>	delete(CosmosQuery query, Class<T> domainType, String containerName) Delete the DocumentQuery, need to query the domains at first, then delete the item from the result.
void	deleteAll(String containerName, Class<?> domainType) Delete the DocumentQuery, delete all the items in the given container.
void	deleteById(String containerName, Object id, PartitionKey partitionKey) Deletes the item by id and partition key.
void	deleteContainer(String containerName) Delete container
void	deleteEntity(String containerName, T entity) Deletes the entity
Boolean	exists(CosmosQuery query, Class<T> domainType, String containerName) Checks if document query items exist
Iterable<T>	find(CosmosQuery query, Class<T> domainType, String containerName) Finds the document query items
Iterable<T>	findAll(Class<T> domainType) Find the DocumentQuery, find all the items specified by domain type.
org.springframework.data.domain.Page<T>	findAll(Pageable pageable, Class<T> domainType, String containerName) Find all items in a given container with partition key
Iterable<T>	findAll(PartitionKey partitionKey, Class<T> domainType) Find the DocumentQuery, find all the items specified by domain type in the given container.
Iterable<T>	findAll(String containerName, Class<T> domainType) Find the DocumentQuery, find all the items specified by domain type in the given container.
T	findById(Object id, Class<T> domainType) Finds item by id
T	findById(Object id, Class<T> domainType, PartitionKey partitionKey) Finds item by id
T	findById(String containerName, Object id, Class<T> domainType) Finds item by id
String	getContainerName(Class<?> domainType) To get container name by domainType
String	getContainerNameOverride(String containerName) Check if an overridden version of containerName is present, and if so, return it
CosmosContainerProperties	getContainerProperties(String containerName) Get properties for specified container
MappingCosmosConverter	getConverter() To get converter
T	insert(String containerName, T objectToSave) Inserts item into the given container
T	insert(String containerName, T objectToSave, PartitionKey partitionKey) Inserts item into the given container
T	insert(T objectToSave, PartitionKey partitionKey) Inserts item
org.springframework.data.domain.Page<T>	paginationQuery(CosmosQuery query, Class<T> domainType, String containerName) Pagination query
T	patch(Object id, PartitionKey partitionKey, Class<T> domainType, CosmosPatchOperations patchOperations) Patches item applies partial update (patch) to an item
T	patch(Object id, PartitionKey partitionKey, Class<T> domainType, CosmosPatchOperations patchOperations, CosmosPatchItemRequestOptions options) applies partial update (patch) to an item with CosmosPatchItemRequestOptions
CosmosContainerProperties	replaceContainerProperties(String containerName, CosmosContainerProperties properties) Replace container properties for the specified container
org.springframework.data.domain.Page<T>	runPaginationQuery(SqlQuerySpec querySpec, Pageable pageable, Class<?> domainType, Class<T> returnType) Run the query.
Iterable<T>	runQuery(SqlQuerySpec querySpec, Class<?> domainType, Class<T> returnType) Run the query.
Iterable<T>	runQuery(SqlQuerySpec querySpec, Sort sort, Class<?> domainType, Class<T> returnType) Run the query.
org.springframework.data.domain.Slice<T>	runSliceQuery(SqlQuerySpec querySpec, Pageable pageable, Class<?> domainType, Class<T> returnType) Run custom SQL query
void	setApplicationContext(ApplicationContext applicationContext) Sets the application context
org.springframework.data.domain.Slice<T>	sliceQuery(CosmosQuery query, Class<T> domainType, String containerName) Slice query
void	upsert(String containerName, T object) Upserts an item into container with partition key
void	upsert(T object) Upserts an item with partition key
T	upsertAndReturnEntity(String containerName, T object) Upserts an item and return item properties

Methods inherited from java.lang.Object

clone equals finalize getClass hashCode notify notifyAll toString wait wait wait

Constructor Details

CosmosTemplate

public CosmosTemplate(CosmosAsyncClient client, String databaseName, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter)

Initialization

Parameters:

client - must not be null

databaseName - must not be null

cosmosConfig - must not be null

mappingCosmosConverter - must not be null

CosmosTemplate

public CosmosTemplate(CosmosAsyncClient client, String databaseName, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter, IsNewAwareAuditingHandler cosmosAuditingHandler)

Initialization

Parameters:

client - must not be null

databaseName - must not be null

cosmosConfig - must not be null

mappingCosmosConverter - must not be null

cosmosAuditingHandler - can be null

CosmosTemplate

public CosmosTemplate(CosmosFactory cosmosFactory, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter)

Initialization

Parameters:

cosmosFactory - must not be null

cosmosConfig - must not be null

mappingCosmosConverter - must not be null

CosmosTemplate

public CosmosTemplate(CosmosFactory cosmosFactory, CosmosConfig cosmosConfig, MappingCosmosConverter mappingCosmosConverter, IsNewAwareAuditingHandler cosmosAuditingHandler)

Initialization

Parameters:

cosmosFactory - must not be null

cosmosConfig - must not be null

mappingCosmosConverter - must not be null

cosmosAuditingHandler - can be null

Method Details

<S,T>deleteEntities

public void <S,T>deleteEntities(CosmosEntityInformation<T,?> information, Iterable entities)

Deletes the entities using bulk

Parameters:

information - the CosmosEntityInformation

entities - the Iterable entities to be inserted

<S,T>insertAll

public Iterable <S,T>insertAll(CosmosEntityInformation<T,?> information, Iterable entities)

Insert all items with bulk.

Parameters:

information - the CosmosEntityInformation

entities - the Iterable entities to be inserted

Returns:

Flux of result

<T,ID>findByIds

public Iterable<T> <T,ID>findByIds(Iterable<ID> ids, Class<T> domainType, String containerName)

Find by ids

Parameters:

ids

domainType

containerName

count

public long <T>count(CosmosQuery query, String containerName)

Count

Parameters:

query

containerName

count

public long <T>count(SqlQuerySpec querySpec, String containerName)

Count

Parameters:

querySpec

containerName

count

public long count(String containerName)

Count

Parameters:

containerName

createContainerIfNotExists

public CosmosContainerProperties createContainerIfNotExists(CosmosEntityInformation information)

Creates container if not exists

Parameters:

information

delete

public Iterable<T> <T>delete(CosmosQuery query, Class<T> domainType, String containerName)

Delete the DocumentQuery, need to query the domains at first, then delete the item from the result. The cosmos db Sql API do _NOT_ support DELETE query, we cannot add one DeleteQueryGenerator. Uses bulk if possible.

Parameters:

query - The representation for query method.

domainType - Class of domain

containerName - Container name of database

Returns:

All the deleted items as List.

deleteAll

public void deleteAll(String containerName, Class<?> domainType)

Delete the DocumentQuery, delete all the items in the given container.

Parameters:

containerName - Container name of database

domainType - the domain type

deleteById

public void deleteById(String containerName, Object id, PartitionKey partitionKey)

Deletes the item by id and partition key.

Parameters:

containerName - Container name of database

id - item id

partitionKey - the partition key

deleteContainer

public void deleteContainer(String containerName)

Delete container

Parameters:

containerName

deleteEntity

public void <T>deleteEntity(String containerName, T entity)

Deletes the entity

Parameters:

containerName - the container name

entity - the entity object

exists

public Boolean <T>exists(CosmosQuery query, Class<T> domainType, String containerName)

Checks if document query items exist

Parameters:

query - The representation for query method.

domainType - Class of domain

containerName - Container name of database

Returns:

if items exist

find

public Iterable<T> <T>find(CosmosQuery query, Class<T> domainType, String containerName)

Finds the document query items

Parameters:

query - The representation for query method.

domainType - Class of domain

containerName - Container name of database

Returns:

All the found items as List.

findAll

public Iterable<T> <T>findAll(Class<T> domainType)

Find the DocumentQuery, find all the items specified by domain type.

Parameters:

domainType - the domain type

Returns:

found results in a List

findAll

public Page<T> <T>findAll(Pageable pageable, Class<T> domainType, String containerName)

Find all items in a given container with partition key

Parameters:

pageable

domainType

containerName

findAll

public Iterable<T> <T>findAll(PartitionKey partitionKey, Class<T> domainType)

Find the DocumentQuery, find all the items specified by domain type in the given container.

Parameters:

partitionKey

domainType

findAll

public Iterable<T> <T>findAll(String containerName, Class<T> domainType)

Find the DocumentQuery, find all the items specified by domain type in the given container.

Parameters:

containerName - the container name

domainType - the domain type

Returns:

found results in a List

findById

public T <T>findById(Object id, Class<T> domainType)

Finds item by id

Parameters:

id - must not be null

domainType - must not be null

Returns:

found item

findById

public T <T>findById(Object id, Class<T> domainType, PartitionKey partitionKey)

Finds item by id

Parameters:

id

domainType

partitionKey

findById

public T <T>findById(String containerName, Object id, Class<T> domainType)

Finds item by id

Parameters:

containerName - must not be null

id - must not be null

domainType - must not be null

Returns:

found item

getContainerName

public String getContainerName(Class<?> domainType)

To get container name by domainType

Parameters:

domainType

getContainerNameOverride

public String getContainerNameOverride(String containerName)

Check if an overridden version of containerName is present, and if so, return it

Parameters:

containerName - Container name of database

Returns:

containerName

getContainerProperties

public CosmosContainerProperties getContainerProperties(String containerName)

Get properties for specified container

Parameters:

containerName

getConverter

public MappingCosmosConverter getConverter()

To get converter

insert

public T <T>insert(String containerName, T objectToSave)

Inserts item into the given container

Parameters:

containerName - must not be null

objectToSave - must not be null

Returns:

the inserted item

insert

public T <T>insert(String containerName, T objectToSave, PartitionKey partitionKey)

Inserts item into the given container

Parameters:

containerName - must not be null

objectToSave - must not be null

partitionKey - must not be null

Returns:

the inserted item

insert

public T <T>insert(T objectToSave, PartitionKey partitionKey)

Inserts item

Parameters:

objectToSave - must not be null

partitionKey - must not be null

Returns:

the inserted item

paginationQuery

public Page<T> <T>paginationQuery(CosmosQuery query, Class<T> domainType, String containerName)

Pagination query

Parameters:

query

domainType

containerName

patch

public T <T>patch(Object id, PartitionKey partitionKey, Class<T> domainType, CosmosPatchOperations patchOperations)

Patches item applies partial update (patch) to an item

Parameters:

id - must not be null

partitionKey - must not be null

domainType

patchOperations - must not be null

Returns:

the patched item

patch

public T <T>patch(Object id, PartitionKey partitionKey, Class<T> domainType, CosmosPatchOperations patchOperations, CosmosPatchItemRequestOptions options)

applies partial update (patch) to an item with CosmosPatchItemRequestOptions

Parameters:

id - must not be null

partitionKey - must not be null

domainType - must not be null

patchOperations - must not be null

options - Optional CosmosPatchItemRequestOptions, e.g. options.setFilterPredicate("FROM products p WHERE p.used = false");

Returns:

the patched item

replaceContainerProperties

public CosmosContainerProperties replaceContainerProperties(String containerName, CosmosContainerProperties properties)

Replace container properties for the specified container

Parameters:

containerName

properties

runPaginationQuery

public Page<T> <T>runPaginationQuery(SqlQuerySpec querySpec, Pageable pageable, Class<?> domainType, Class<T> returnType)

Run the query.

Parameters:

querySpec

pageable

domainType

returnType

runQuery

public Iterable<T> <T>runQuery(SqlQuerySpec querySpec, Class<?> domainType, Class<T> returnType)

Run the query.

Parameters:

querySpec

domainType

returnType

runQuery

public Iterable<T> <T>runQuery(SqlQuerySpec querySpec, Sort sort, Class<?> domainType, Class<T> returnType)

Run the query.

Parameters:

querySpec

sort

domainType

returnType

runSliceQuery

public Slice<T> <T>runSliceQuery(SqlQuerySpec querySpec, Pageable pageable, Class<?> domainType, Class<T> returnType)

Run custom SQL query

Parameters:

querySpec

pageable

domainType

returnType

setApplicationContext

public void setApplicationContext(ApplicationContext applicationContext)

Sets the application context

Parameters:

applicationContext - must not be null

Throws:

org.springframework.beans.BeansException - the bean exception

sliceQuery

public Slice<T> <T>sliceQuery(CosmosQuery query, Class<T> domainType, String containerName)

Slice query

Parameters:

query

domainType

containerName

upsert

public void <T>upsert(String containerName, T object)

Upserts an item into container with partition key

Parameters:

containerName - the container name

object - upsert object

upsert

public void <T>upsert(T object)

Upserts an item with partition key

Parameters:

object - upsert object

upsertAndReturnEntity

public T <T>upsertAndReturnEntity(String containerName, T object)

Upserts an item and return item properties

Parameters:

containerName - the container name

object - upsert object

Returns:

upsert object entity