Spring/spring-data-examples
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add declarative MongoDB aggregation example.

Browse Source 
Original pull request: #600.
This commit is contained in:
divya_jnu08
2021-01-14 22:38:07 +05:30
committed by Mark Paluch
parent 3de30e1117
commit 20b0c7c490
2 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mongodb/aggregation/src/main/java/example/springdata/mongodb/aggregation/OrderRepository.java
View File

@@ -22,7 +22,8 @@ import org.springframework.data.mongodb.repository.Aggregation;
import org.springframework.data.repository.CrudRepository;
/**
* A repository interface assembling CRUD functionality as well as the API to invoke the methods implemented manually.
* A repository interface assembling CRUD functionality as well as the API to
* invoke the methods implemented manually.
*
* @author Thomas Darimont
* @author Oliver Gierke
@@ -35,4 +36,11 @@ public interface OrderRepository extends CrudRepository<Order, String>, OrderRep
@Aggregation(pipeline = { "{ $match : { customerId : ?0 } }", "{ $count : total }" })
Long totalOrdersForCustomer(String customerId);
@Aggregation(pipeline = { "{ $match : { id : ?0 } }", "{ $unwind : { path : '$items' } }",
"{ $project : { id : 1 , customerId : 1 , items : 1 , lineTotal : { $multiply: [ '$items.price' , '$items.quantity' ] } } }",
"{ $group : { '_id' : '$_id' , 'netAmount' : { $sum : '$lineTotal' } , 'items' : { $addToSet : '$items' } } }",
"{ $project : { 'orderId' : '$_id' , 'items' : 1 , 'netAmount' : 1 , 'taxAmount' : { $multiply: [ '$netAmount' , 0.19 ] } , 'totalAmount' : { $multiply: [ '$netAmount' , 1.19 ] } } }" })
Invoice getInvoiceForOrderDeclarative(String orderId);
}

19
mongodb/aggregation/src/test/java/example/springdata/mongodb/aggregation/OrderRepositoryIntegrationTests.java
View File

@@ -51,7 +51,7 @@ public class OrderRepositoryIntegrationTests {
}
@Test
public void createsInvoiceViaAggregation() {
public void createsInvoiceViaAggregationProgrammatic() {
Order order = new Order("c42", new Date()).//
addItem(product1).addItem(product2).addItem(product3);
@@ -65,6 +65,22 @@ public class OrderRepositoryIntegrationTests {
assertThat(invoice.getTaxAmount()).isCloseTo(1.577D, offset(0.00001));
assertThat(invoice.getTotalAmount()).isCloseTo(9.877, offset(0.00001));
}
@Test
public void createsInvoiceViaAggregationDeclarative() {
Order order = new Order("c42", new Date()).//
addItem(product1).addItem(product2).addItem(product3);
order = repository.save(order);
Invoice invoice = repository.getInvoiceForOrderDeclarative(order.getId());
assertThat(invoice).isNotNull();
assertThat(invoice.getOrderId()).isEqualTo(order.getId());
assertThat(invoice.getNetAmount()).isCloseTo(8.3D, offset(0.00001));
assertThat(invoice.getTaxAmount()).isCloseTo(1.577D, offset(0.00001));
assertThat(invoice.getTotalAmount()).isCloseTo(9.877, offset(0.00001));
}
@Test
public void declarativeAggregationWithSort() {
@@ -94,4 +110,5 @@ public class OrderRepositoryIntegrationTests {
assertThat(repository.totalOrdersForCustomer("c42")).isEqualTo(3);
}
}