Implementação de gravações em massa usando o Spring Boot para MongoDB

1
<dependencies>
2
   <dependency>
3
       <groupId>org.springframework.boot</groupId>
4
       <artifactId>spring-boot-starter-data-mongodb</artifactId>
5
       <version>2.7.2</version>
6
   </dependency>

1
@SpringBootApplication
2
public class SpringDataBulkInsertApplication implements CommandLineRunner {
3
4
   @Value("${documentCount}")
5
   private int count;
6
   private static final Logger LOG = LoggerFactory
7
           .getLogger(SpringDataBulkInsertApplication.class);
8
9
   @Autowired
10
   private CustomProductsRepository repository;
11
12
   public static void main(String[] args) {
13
       SpringApplication.run(SpringDataBulkInsertApplication.class, args);
14
   }
15
16
17
   @Override
18
   public void run(String... args) {
19
20
       repository.bulkInsertProducts(count);
21
       LOG.info("End run");
22
   }
23
}

1
@Configuration
2
public class MongoConfig   {
3
   @Value("${mongodb.uri}")
4
   private String uri;
5
6
   @Value("${mongodb.database}")
7
   private String databaseName;
8
9
   @Value("${truststore.path}")
10
   private String trustStorePath;
11
   @Value("${truststore.pwd}")
12
   private String trustStorePwd;
13
14
   @Value("${mongodb.atlas}")
15
   private boolean atlas;
16
17
   @Bean
18
   public MongoClient mongo() {
19
       ConnectionString connectionString = new ConnectionString(uri);
20
           MongoClientSettings mongoClientSettings = MongoClientSettings.builder()
21
                   .applyConnectionString(connectionString)
22
                   .applyToSslSettings(builder -> {
23
                       if (!atlas) {
24
                           // Use SSLContext if a trustStore has been provided
25
                           if (!trustStorePath.isEmpty()) {
26
                               SSLFactory sslFactory = SSLFactory.builder()
27
                                       .withTrustMaterial(Paths.get(trustStorePath), trustStorePwd.toCharArray())
28
                                       .build();
29
                               SSLContext sslContext = sslFactory.getSslContext();
30
                               builder.context(sslContext);
31
                               builder.invalidHostNameAllowed(true);
32
                           }
33
                       }
34
                       builder.enabled(true);
35
                   })
36
                   .build();
37
       return MongoClients.create(mongoClientSettings);
38
   }
39
40
   @Bean
41
   public MongoTemplate mongoTemplate() throws Exception {
42
       return new MongoTemplate(mongo(), databaseName);
43
   }
44
}

1
@Document("products")
2
public class Products {
3
4
   private static final Logger LOG = LoggerFactory
5
           .getLogger(Products.class);
6
   @Id
7
   private String id;
8
   private String name;
9
   private int qty;
10
   private double price;
11
   private Date available;
12
   private Date  unavailable;
13
   private String skuId;

1
@Component
2
public class CustomProductsRepositoryImpl implements CustomProductsRepository {
3
4
   private static final Logger LOG = LoggerFactory
5
           .getLogger(CustomProductsRepository.class);
6
7
   @Autowired
8
   MongoTemplate mongoTemplate;
9
10
   public int bulkInsertProducts(int count) {
11
12
       LOG.info("Dropping collection...");
13
       mongoTemplate.dropCollection(Products.class);
14
       LOG.info("Dropped!");
15
16
       Instant start = Instant.now();
17
       mongoTemplate.setWriteConcern(WriteConcern.W1.withJournal(true));
18
19
       Products [] productList = Products.RandomProducts(count);
20
       BulkOperations bulkInsertion = mongoTemplate.bulkOps(BulkOperations.BulkMode.UNORDERED, Products.class);
21
22
       for (int i=0; i<productList.length; ++i)
23
           bulkInsertion.insert(productList[i]);
24
25
       BulkWriteResult bulkWriteResult = bulkInsertion.execute();
26
27
       LOG.info("Bulk insert of "+bulkWriteResult.getInsertedCount()+" documents completed in "+ Duration.between(start, Instant.now()).toMillis() + " milliseconds");
28
       return bulkWriteResult.getInsertedCount();
29
   }
30
}