3pp-mirror/penpot
1
0
Fork 0
mirror of https://github.com/penpot/penpot.git synced 2025-05-25 05:26:10 +02:00
Code Issues Projects Releases Packages Wiki Activity

✨ Improve srelp.helper analyze-files usability

Browse source
This commit is contained in:
Andrey Antukh 2023-07-27 11:49:10 +02:00
parent 456b604937
commit 960ae66cbd
1 changed files with 18 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

33
backend/src/app/srepl/helpers.clj
View file

@ -110,7 +110,7 @@
The `on-file` parameter should be a function that receives the file The `on-file` parameter should be a function that receives the file
and the previous state and returns the new state." and the previous state and returns the new state."
[system & {:keys [chunk-size max-items start-at on-file on-error on-end] [system & {:keys [chunk-size max-items start-at on-file on-error on-end on-init]
:or {chunk-size 10 max-items Long/MAX_VALUE}}] :or {chunk-size 10 max-items Long/MAX_VALUE}}]
(letfn [(get-chunk [conn cursor] (letfn [(get-chunk [conn cursor]
(let [rows (db/exec! conn [sql:retrieve-files-chunk cursor chunk-size])] (let [rows (db/exec! conn [sql:retrieve-files-chunk cursor chunk-size])]
@ -124,24 +124,27 @@
(take max-items) (take max-items)
(map #(update % :data blob/decode)))) (map #(update % :data blob/decode))))
(on-error* [file cause] (on-error* [cause file]
(println "unexpected exception happened on processing file: " (:id file)) (println "unexpected exception happened on processing file: " (:id file))
(strace/print-stack-trace cause))] (strace/print-stack-trace cause))]
(db/with-atomic [conn (:app.db/pool system)] (when (fn? on-init) (on-init))
(loop [state {}
files (get-candidates conn)]
(if-let [file (first files)]
(let [state' (try
(on-file file state)
(catch Throwable cause
(let [on-error (or on-error on-error*)]
(on-error file cause))))]
(recur (or state' state) (rest files)))
(if (fn? on-end) (db/with-atomic [conn (:app.db/pool system)]
(on-end state) (doseq [file (get-candidates conn)]
state)))))) (binding [*conn* conn
pmap/*tracked* (atom {})
pmap/*load-fn* (partial files/load-pointer conn (:id file))
ffeat/*wrap-with-pointer-map-fn*
(if (contains? (:features file) "storage/pointer-map") pmap/wrap identity)
ffeat/*wrap-with-objects-map-fn*
(if (contains? (:features file) "storage/objectd-map") omap/wrap identity)]
(try
(on-file file)
(catch Throwable cause
((or on-error on-error*) cause file))))))
(when (fn? on-end) (on-end))))
(defn update-pages (defn update-pages
"Apply a function to all pages of one file. The function receives a page and returns an updated page." "Apply a function to all pages of one file. The function receives a page and returns an updated page."