From 7c0a63c7da11721a80096230bf75d97d8dbd644d Mon Sep 17 00:00:00 2001
From: Andrey Antukh <niwi@niwi.nz>
Date: Tue, 30 Jan 2024 13:30:48 +0100
Subject: [PATCH] :sparkles: Add minor improvements to comp-v2 migration srepl
 helper

---
 backend/src/app/srepl/components_v2.clj | 40 +++++++++++++++++--------
 1 file changed, 28 insertions(+), 12 deletions(-)

diff --git a/backend/src/app/srepl/components_v2.clj b/backend/src/app/srepl/components_v2.clj
index 3db774600..ffbd01d66 100644
--- a/backend/src/app/srepl/components_v2.clj
+++ b/backend/src/app/srepl/components_v2.clj
@@ -188,17 +188,27 @@
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
 (defn migrate-file!
-  [file-id & {:keys [rollback? validate? label] :or {rollback? true validate? false}}]
+  [file-id & {:keys [rollback? validate? label cache skip-on-graphic-error?]
+              :or {rollback? true
+                   validate? false
+                   skip-on-graphic-error? true}}]
   (l/dbg :hint "migrate:start" :rollback rollback?)
-  (let [tpoint (dt/tpoint)
+  (let [tpoint  (dt/tpoint)
         file-id (if (string? file-id)
                   (parse-uuid file-id)
-                  file-id)]
-    (binding [feat/*stats* (atom {})]
+                  file-id)
+        cache   (if (int? cache)
+                  (cache/create :executor (::wrk/executor main/system)
+                                :max-items cache)
+                  nil)]
+
+    (binding [feat/*stats* (atom {})
+              feat/*cache* cache]
       (try
         (-> (assoc main/system ::db/rollback rollback?)
             (feat/migrate-file! file-id
                                 :validate? validate?
+                                :skip-on-graphic-error? skip-on-graphic-error?
                                 :label label))
 
         (-> (deref feat/*stats*)
@@ -212,22 +222,28 @@
             (l/dbg :hint "migrate:end" :rollback rollback? :elapsed elapsed)))))))
 
 (defn migrate-team!
-  [team-id & {:keys [rollback? skip-on-graphic-error? validate? label]
+  [team-id & {:keys [rollback? skip-on-graphic-error? validate? label cache]
               :or {rollback? true
                    validate? true
-                   skip-on-graphic-error? false}}]
+                   skip-on-graphic-error? true}}]
 
   (l/dbg :hint "migrate:start" :rollback rollback?)
 
-  (let [team-id   (if (string? team-id)
-                    (parse-uuid team-id)
-                    team-id)
-        stats     (atom {})
-        tpoint    (dt/tpoint)]
+  (let [team-id (if (string? team-id)
+                  (parse-uuid team-id)
+                  team-id)
+        stats   (atom {})
+        tpoint  (dt/tpoint)
+
+        cache   (if (int? cache)
+                  (cache/create :executor (::wrk/executor main/system)
+                                :max-items cache)
+                  nil)]
 
     (add-watch stats :progress-report (report-progress-files tpoint))
 
-    (binding [feat/*stats* stats]
+    (binding [feat/*stats* stats
+              feat/*cache* cache]
       (try
         (-> (assoc main/system ::db/rollback rollback?)
             (feat/migrate-team! team-id