From 4f089a819df757570306e128850ce9f169756a01 Mon Sep 17 00:00:00 2001
From: Denis O <denis.o@linux.com>
Date: Mon, 5 Feb 2024 20:43:51 +0000
Subject: [PATCH] Mongodb internals cleanup (#31)

---
 .../polydata4/mongodb/PolydataMongodb.java    | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/polydata-mongodb/src/main/java/com/unidev/polydata4/mongodb/PolydataMongodb.java b/polydata-mongodb/src/main/java/com/unidev/polydata4/mongodb/PolydataMongodb.java
index 21e62bd..f354bf2 100644
--- a/polydata-mongodb/src/main/java/com/unidev/polydata4/mongodb/PolydataMongodb.java
+++ b/polydata-mongodb/src/main/java/com/unidev/polydata4/mongodb/PolydataMongodb.java
@@ -171,10 +171,8 @@ public Optional<BasicPoly> indexData(String dataset, String indexId) {
     public BasicPolyList insert(String dataset, InsertOptions insertOptions, Collection<InsertRequest> insertRequests) {
         BasicPolyList basicPolyList = new BasicPolyList();
 
-        Set<String> polyIds = new HashSet<>();
         for (InsertRequest insertRequest : insertRequests) {
             BasicPoly data = insertRequest.getData();
-            polyIds.add(data._id());
 
             Set<String> indexToPersist = insertRequest.getIndexToPersist();
             if (CollectionUtils.isEmpty(indexToPersist)) {
@@ -457,7 +455,12 @@ public void close() throws IOException {
         mongoClient.close();
     }
 
-    private MongoCollection<Document> collection(String collection) {
+    /**
+     * Fetch collection for dataset.
+     * @param collection
+     * @return
+     */
+    public MongoCollection<Document> collection(String collection) {
         return mongoClient.getDatabase(Objects.requireNonNull(mongoClientURI.getDatabase())).getCollection(collection);
     }
 
@@ -506,11 +509,17 @@ private void persistPolyToCollection(String dataset, String collection, BasicPol
         putIfCache(collection + "-poly-from-collection-" + dataset, data);
     }
 
-    private MongoCollection<Document> indexCollection(String dataset) {
+    /**
+     * Fetch dataset collection.
+     */
+    public MongoCollection<Document> indexCollection(String dataset) {
         return collection(INDEX_COLLECTION);
     }
 
-    private void recalculateIndex(String dataset) {
+    /**
+     * Recalculate index for dataset
+     */
+    public void recalculateIndex(String dataset) {
         MongoCollection<Document> collection = collection(dataset);
         AggregateIterable<Document> documents = collection.aggregate(
                 Arrays.asList(