7.4 KiB
Spring Data for Azure Cosmos DB DocumentDB API
Azure Cosmos DB is a globally-distributed database service that allows developers to work with data using a variety of standard APIs, such as DocumentDB, MongoDB, Graph, and Table APIs. Azure Cosmos DB DocumentDB Spring Data provides initial Spring Data support for Azure Cosmos DB Document API based on Spring Data framework, the other 3 APIs are not supported in this package. Key functionalities supported are listed at below. More features will coming soon.
Sample Code
Please refer to sample project here.
Feature List
- Spring Data CRUDRepository basic CRUD functionality
- save
- findAll
- findOne by Id
- deleteAll
- delete by Id
- delete entity
- Spring Data @Id annotation.
There're 2 ways to map a field in domain class to
id
field of Azure Cosmos DB document.- annotate a field in domain class with
@Id
, this field will be mapped to documentid
in Cosmos DB. - set name of this field to
id
, this field will be mapped to documentid
in Cosmos DB.
- annotate a field in domain class with
- Custom collection Name.
By default, collection name will be class name of user domain class. To customize it, add annoataion
@Document(collection="myCustomCollectionName")
to domain class, that's all. - Supports Azure Cosmos DB partition. To specify a field of domain class to be partition key field, just annotate it with
@PartitionKey
. When you do CRUD operation, pls specify your partition value. For more sample on partition CRUD, pls refer to test here
Quick Start
Add the dependency
spring-data-documentdb
is published on Maven Central Repository.
If you are using Maven, add the following dependency.
<dependency>
<groupId>com.microsoft.azure</groupId>
<artifactId>spring-data-documentdb</artifactId>
<version>0.1.2</version>
</dependency>
Setup Configuration
Setup Azure Cosmos DB DocumentDB configuration class.
@Configuration
@EnableDocumentDbRepositories
public class AppConfiguration extends AbstractDocumentDbConfiguration {
@Value("${azure.documentdb.uri}")
private String uri;
@Value("${azure.documentdb.key}")
private String key;
@Value("${azure.documentdb.database}")
private String dbName;
public DocumentClient documentClient() {
return new DocumentClient(uri, key, ConnectionPolicy.GetDefault(), ConsistencyLevel.Session);
}
public String getDatabase() {
return dbName;
}
}
By default, @EnableDocumentDbRepositories
will scan the current package for any interfaces that extend one of Spring Data's repository interfaces. Using it to annotate your Configuration class to scan a different root package by type if your project layout has multiple projects and it's not finding your repositories.
@Configuration
@EnableDocumentDbRepositories(basePackageClass=UserRepository.class)
public class AppConfiguration extends AbstractDocumentDbConfiguration {
// configuration code
}
Define en entity
Define a simple entity as Document in DocumentDB.
@Document(collection = "mycollection")
public class User {
private String id;
private String firstName;
@PartitionKey
private String lastName;
... // setters and getters
public User(String id, String firstName, String lastName) {
this.id = id;
this.firstName = firstName;
this.lastName = lastName;
}
@Override
public String toString() {
return String.format("User: %s %s, %s", firstName, lastName);
}
}
id
field will be used as document id in Azure Cosmos DB. If you want use another field like emailAddress
as document id
, just annotate that field with @Id
annotation.
Annotation @Document(collection="mycollection")
is used to specify collection name in Azure Cosmos DB.
Annotation @PartitionKey
on lastName
field is used to specify this field be partition key in Azure Cosmos DB.
@Document(collection = "mycollection")
public class User {
@Id
private String emailAddress;
...
}
Create repositories
Extends DocumentDbRepository interface, which provides Spring Data repository support.
import DocumentDbRepository;
import org.springframework.stereotype.Repository;
@Repository
public interface UserRepository extends DocumentDbRepository<User, String> {
}
So far DocumentDbRepository provides basic save, delete, update and find operations. More operations will be supported later.
Create an Application class
Here create an application class with all the components
@SpringBootApplication
public class SampleApplication implements CommandLineRunner {
@Autowired
private UserRepository repository;
public static void main(String[] args) {
SpringApplication.run(SampleApplication.class, args);
}
public void run(String... var1) throws Exception {
final User testUser = new User("testId", "testFirstName", "testLastName");
repository.deleteAll();
repository.save(testUser);
// to find by Id, please specify partition key value if collection is partitioned
final User result = repository.findOne(testUser.getId(), testUser.getLastName);
// if emailAddress is mapped to id, then
// final User result = respository.findOne(testUser.getEmailAddress(), testUser.getLastName());
}
}
Autowired UserRepository interface, then can do save, delete and find operations. Azure Cosmos DB DocumentDB Spring Data uses the DocumentTemplate to execute the queries behind find, save methods. You can use the template yourself for more complex queries.
Filing Issues
If you encounter any bug, please file an issue here.
To suggest a new feature or changes that could be made, file an issue the same way you would for a bug.
Pull Requests
Pull requests are welcome. To open your own pull request, click here. When creating a pull request, make sure you are pointing to the fork and branch that your changes were made in.
Code of Conduct
This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact opencode@microsoft.com with any additional questions or comments.