APIs REST com Java, Spring Boot e MongoDB

1
git clone https://github.com/mongodb-developer/java-spring-boot-mongodb-starter

1
package com.mongodb.starter;
2
3
import [...]
4
5
import static org.bson.codecs.configuration.CodecRegistries.fromProviders;
6
import static org.bson.codecs.configuration.CodecRegistries.fromRegistries;
7
8
@Configuration
9
public class MongoDBConfiguration {
10
11
    @Value("${spring.data.mongodb.uri}")
12
    private String connectionString;
13
14
    @Bean
15
    public MongoClient mongoClient() {
16
        CodecRegistry pojoCodecRegistry = fromProviders(PojoCodecProvider.builder().automatic(true).build());
17
        CodecRegistry codecRegistry = fromRegistries(MongoClientSettings.getDefaultCodecRegistry(), pojoCodecRegistry);
18
        return MongoClients.create(MongoClientSettings.builder()
19
                                                      .applyConnectionString(new ConnectionString(connectionString))
20
                                                      .codecRegistry(codecRegistry)
21
                                                      .build());
22
    }
23
24
}

1
private static final TransactionOptions txnOptions = TransactionOptions.builder()
2
    .readPreference(ReadPreference.primary())
3
    .readConcern(ReadConcern.MAJORITY)
4
    .writeConcern(WriteConcern.MAJORITY)
5
    .build();
6
7
@Override
8
public List<PersonEntity> saveAll(List<PersonEntity> personEntities) {
9
    try (ClientSession clientSession = client.startSession()) {
10
        return clientSession.withTransaction(() -> {
11
            personEntities.forEach(p -> p.setId(new ObjectId()));
12
            personCollection.insertMany(clientSession, personEntities);
13
            return personEntities;
14
        }, txnOptions);
15
    }
16
}

1
@ExceptionHandler(RuntimeException.class)
2
public final ResponseEntity<Exception> handleAllExceptions(RuntimeException e) {
3
    logger.error("Internal server error.", e);
4
    return new ResponseEntity<>(e, HttpStatus.INTERNAL_SERVER_ERROR);
5
}

1
@Override
2
public double getAverageAge() {
3
    List<Bson> pipeline = List.of(group(new BsonNull(), avg("averageAge", "$age")), project(excludeId()));
4
    return personCollection.aggregate(pipeline, AverageAgeDTO.class).first().averageAge();
5
}

1
private MongoCollection<PersonEntity> personCollection;
2
3
@PostConstruct
4
void init() {
5
    personCollection = client.getDatabase("test").getCollection("persons", PersonEntity.class);
6
}