add migration to fill folder->createdAt

Thomasr · Thomasr · commit f3ba6528df0d · 2024-08-01T13:22:40.000-04:00
diff --git a/server/api-service/lowcoder-server/src/main/java/org/lowcoder/runner/migrations/DatabaseChangelog.java b/server/api-service/lowcoder-server/src/main/java/org/lowcoder/runner/migrations/DatabaseChangelog.java
@@ -36,6 +36,7 @@
 import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.data.mongodb.core.query.Update;
 
+import java.time.Instant;
 import java.util.Set;
 
 import static org.lowcoder.domain.util.QueryDslUtils.fieldName;
@@ -243,6 +244,30 @@ public void processDocument(Document document) {
         });
     }
 
+    @ChangeSet(order = "024", id = "fill-create-at", author = "")
+    public void fillCreateAt(MongockTemplate mongoTemplate) {
+        // Create a query to match all documents
+        Query query = new Query();
+
+        // Use a DocumentCallbackHandler to iterate through each document
+        mongoTemplate.executeQuery(query, "folder", new DocumentCallbackHandler() {
+            @Override
+            public void processDocument(Document document) {
+                Object object = document.get("createdAt");
+                if(object != null) return;
+                // Create an update object to add the 'gid' field
+                Update update = new Update();
+                update.set("createdAt", Instant.now());
+
+                // Create a query to match the current document by its _id
+                Query idQuery = new Query(Criteria.where("_id").is(document.getObjectId("_id")));
+
+                // Update the document with the new 'gid' field
+                mongoTemplate.updateFirst(idQuery, update, "folder");
+            }
+        });
+    }
+
     private void addGidField(MongockTemplate mongoTemplate, String collectionName) {
         // Create a query to match all documents
         Query query = new Query();
