iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BAEL-2972 : MongoDB Aggregations Using Java (#8718)

This commit is contained in:
Sampada 2020-02-13 23:31:19 +05:30 committed by GitHub
parent 7f271902a3
commit 2816111e98
1 changed files with 111 additions and 111 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

222
persistence-modules/java-mongodb/src/test/java/com/baeldung/aggregation/AggregationLiveTest.java
View File

@ -1,111 +1,111 @@
package com.baeldung.aggregation; package com.baeldung.aggregation;
import static com.mongodb.client.model.Aggregates.count; import static com.mongodb.client.model.Aggregates.count;
import static com.mongodb.client.model.Aggregates.group; import static com.mongodb.client.model.Aggregates.group;
import static com.mongodb.client.model.Aggregates.limit; import static com.mongodb.client.model.Aggregates.limit;
import static com.mongodb.client.model.Aggregates.match; import static com.mongodb.client.model.Aggregates.match;
import static com.mongodb.client.model.Aggregates.out; import static com.mongodb.client.model.Aggregates.out;
import static com.mongodb.client.model.Aggregates.project; import static com.mongodb.client.model.Aggregates.project;
import static com.mongodb.client.model.Aggregates.sort; import static com.mongodb.client.model.Aggregates.sort;
import static org.junit.Assert.assertTrue; import static org.junit.Assert.assertTrue;
import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertNotNull;
import java.io.BufferedReader; import java.io.BufferedReader;
import java.io.IOException; import java.io.IOException;
import java.io.InputStream; import java.io.InputStream;
import java.io.InputStreamReader; import java.io.InputStreamReader;
import java.util.Arrays; import java.util.Arrays;
import org.bson.Document; import org.bson.Document;
import org.bson.conversions.Bson; import org.bson.conversions.Bson;
import org.junit.AfterClass; import org.junit.AfterClass;
import org.junit.BeforeClass; import org.junit.BeforeClass;
import org.junit.Test; import org.junit.Test;
import com.mongodb.client.MongoClient; import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients; import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoCollection; import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase; import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Accumulators; import com.mongodb.client.model.Accumulators;
import com.mongodb.client.model.Filters; import com.mongodb.client.model.Filters;
import com.mongodb.client.model.Projections; import com.mongodb.client.model.Projections;
import com.mongodb.client.model.Sorts; import com.mongodb.client.model.Sorts;
public class AggregationLiveTest { public class AggregationLiveTest {
private static final String DATABASE = "world"; private static final String DATABASE = "world";
private static final String COLLECTION = "countries"; private static final String COLLECTION = "countries";
private static final String DATASET_JSON = "/countrydata.json"; private static final String DATASET_JSON = "/countrydata.json";
private static MongoClient mongoClient; private static MongoClient mongoClient;
private static MongoDatabase database; private static MongoDatabase database;
private static MongoCollection<Document> collection; private static MongoCollection<Document> collection;
@BeforeClass @BeforeClass
public static void setUpDB() throws IOException { public static void setUpDB() throws IOException {
mongoClient = MongoClients.create(); mongoClient = MongoClients.create();
database = mongoClient.getDatabase(DATABASE); database = mongoClient.getDatabase(DATABASE);
collection = database.getCollection(COLLECTION); collection = database.getCollection(COLLECTION);
collection.drop(); collection.drop();
InputStream is = AggregationLiveTest.class.getResourceAsStream(DATASET_JSON); InputStream is = AggregationLiveTest.class.getResourceAsStream(DATASET_JSON);
BufferedReader reader = new BufferedReader(new InputStreamReader(is)); BufferedReader reader = new BufferedReader(new InputStreamReader(is));
reader.lines() reader.lines()
.forEach(line -> collection.insertOne(Document.parse(line))); .forEach(line -> collection.insertOne(Document.parse(line)));
reader.close(); reader.close();
} }
@Test @Test
public void givenCountryCollection_whenNAFTACountriesCounted_thenThree() { public void givenCountryCollection_whenEnglishSpeakingCountriesCounted_thenNinetyOne() {
Document naftaCountries = collection.aggregate(Arrays.asList(match(Filters.eq("regionalBlocs.acronym", "NAFTA")), count())) Document englishSpeakingCountries = collection.aggregate(Arrays.asList(match(Filters.eq("languages.name", "English")), count()))
.first(); .first();
assertEquals(3, naftaCountries.get("count")); assertEquals(91, englishSpeakingCountries.get("count"));
} }
@Test @Test
public void givenCountryCollection_whenAreaSortedDescending_thenSuccess() { public void givenCountryCollection_whenAreaSortedDescending_thenSuccess() {
collection.aggregate(Arrays.asList(sort(Sorts.descending("area")), limit(7), out("largest_seven"))) collection.aggregate(Arrays.asList(sort(Sorts.descending("area")), limit(7), out("largest_seven")))
.toCollection(); .toCollection();
MongoCollection<Document> largestSeven = database.getCollection("largest_seven"); MongoCollection<Document> largestSeven = database.getCollection("largest_seven");
assertEquals(7, largestSeven.countDocuments()); assertEquals(7, largestSeven.countDocuments());
Document usa = largestSeven.find(Filters.eq("alpha3Code", "USA")) Document usa = largestSeven.find(Filters.eq("alpha3Code", "USA"))
.first(); .first();
assertNotNull(usa); assertNotNull(usa);
} }
@Test @Test
public void givenCountryCollection_whenCountedRegionWise_thenMaxInAfrica() { public void givenCountryCollection_whenCountedRegionWise_thenMaxInAfrica() {
Document maxCountriedRegion = collection.aggregate(Arrays.asList(group("$region", Accumulators.sum("tally", 1)), sort(Sorts.descending("tally")))) Document maxCountriedRegion = collection.aggregate(Arrays.asList(group("$region", Accumulators.sum("tally", 1)), sort(Sorts.descending("tally"))))
.first(); .first();
assertTrue(maxCountriedRegion.containsValue("Africa")); assertTrue(maxCountriedRegion.containsValue("Africa"));
} }
@Test @Test
public void givenCountryCollection_whenNeighborsCalculated_thenMaxIsFifteenInChina() { public void givenCountryCollection_whenNeighborsCalculated_thenMaxIsFifteenInChina() {
Bson borderingCountriesCollection = project(Projections.fields(Projections.excludeId(), Projections.include("name"), Projections.computed("borderingCountries", Projections.computed("$size", "$borders")))); Bson borderingCountriesCollection = project(Projections.fields(Projections.excludeId(), Projections.include("name"), Projections.computed("borderingCountries", Projections.computed("$size", "$borders"))));
int maxValue = collection.aggregate(Arrays.asList(borderingCountriesCollection, group(null, Accumulators.max("max", "$borderingCountries")))) int maxValue = collection.aggregate(Arrays.asList(borderingCountriesCollection, group(null, Accumulators.max("max", "$borderingCountries"))))
.first() .first()
.getInteger("max"); .getInteger("max");
assertEquals(15, maxValue); assertEquals(15, maxValue);
Document maxNeighboredCountry = collection.aggregate(Arrays.asList(borderingCountriesCollection, match(Filters.eq("borderingCountries", maxValue)))) Document maxNeighboredCountry = collection.aggregate(Arrays.asList(borderingCountriesCollection, match(Filters.eq("borderingCountries", maxValue))))
.first(); .first();
assertTrue(maxNeighboredCountry.containsValue("China")); assertTrue(maxNeighboredCountry.containsValue("China"));
} }
@AfterClass @AfterClass
public static void cleanUp() { public static void cleanUp() {
mongoClient.close(); mongoClient.close();
} }
} }