fix(import): remove ::byte-array to prevent excessive file content from being stored in memory

causing an out-of-memory error
This commit is contained in:
rcmerci
2025-09-26 17:03:02 +08:00
committed by Tienson Qin
parent 9f8089c406
commit bf78faa6e7

View File

@@ -55,7 +55,7 @@
(defn- roam-import-handler
[e]
(let [file (first (array-seq (.-files (.-target e))))
(let [file (first (array-seq (.-files (.-target e))))
file-name (gobj/get file "name")]
(if (string/ends-with? file-name ".json")
(do
@@ -75,7 +75,7 @@
(defn- lsq-import-handler
[e & {:keys [sqlite? debug-transit? graph-name db-edn?]}]
(let [file (first (array-seq (.-files (.-target e))))
(let [file (first (array-seq (.-files (.-target e))))
file-name (some-> (gobj/get file "name")
(string/lower-case))
edn? (string/ends-with? file-name ".edn")
@@ -155,7 +155,7 @@
(defn- opml-import-handler
[e]
(let [file (first (array-seq (.-files (.-target e))))
(let [file (first (array-seq (.-files (.-target e))))
file-name (gobj/get file "name")]
(if (string/ends-with? file-name ".opml")
(do
@@ -358,23 +358,25 @@
{:size (.-size (:file-object file))
:checksum checksum
:type (db-asset/asset-path->type (:path file))
:path (:path file)
;; Save array to avoid reading asset twice
::byte-array byte-array})
:path (:path file)})
byte-array)))))
(defn- copy-asset [repo repo-dir asset-m]
(-> (::byte-array asset-m)
(p/then (fn [content]
(let [assets-dir (path/path-join repo-dir common-config/local-assets-dir)]
(p/do!
(fs/mkdir-if-not-exists assets-dir)
(if (:block/uuid asset-m)
(fs/write-plain-text-file! repo assets-dir (str (:block/uuid asset-m) "." (:type asset-m)) content {:skip-transact? true})
(when-not (:pdf-annotation? asset-m)
(println "Copied asset" (pr-str (node-path/basename (:path asset-m)))
"by its name since it was unused.")
(fs/write-plain-text-file! repo assets-dir (node-path/basename (:path asset-m)) content {:skip-transact? true})))))))))
(defn- copy-asset [repo repo-dir asset-m path->file-object]
(let [assets-dir (path/path-join repo-dir common-config/local-assets-dir)
file-obj (get path->file-object (:path asset-m))]
(if-not file-obj
(do (log/error :copy-asset-error {:msg "file-obj not found" :asset-m asset-m})
(p/rejected (js/Error. "file-obj not found")))
(-> (.arrayBuffer file-obj)
(p/then (fn [content]
(p/do!
(fs/mkdir-if-not-exists assets-dir)
(if (:block/uuid asset-m)
(fs/write-plain-text-file! repo assets-dir (str (:block/uuid asset-m) "." (:type asset-m)) content {:skip-transact? true})
(when-not (:pdf-annotation? asset-m)
(println "Copied asset" (pr-str (node-path/basename (:path asset-m)))
"by its name since it was unused.")
(fs/write-plain-text-file! repo assets-dir (node-path/basename (:path asset-m)) content {:skip-transact? true}))))))))))
(defn- import-file-graph
[*files
@@ -386,13 +388,15 @@
_ (repo-handler/new-db! graph-name {:file-graph-import? true})
repo (state/get-current-repo)
db-conn (db/get-db repo false)
path->file-object (reduce (fn [acc file] (assoc acc (:path file) (:file-object file))) {} *files)
options {:user-options
(merge
(dissoc user-options :graph-name)
{:tag-classes (some-> tag-classes string/trim not-empty (string/split #",\s*") set)
:property-classes (some-> property-classes string/trim not-empty (string/split #",\s*") set)
:property-parent-classes (some-> property-parent-classes string/trim not-empty (string/split #",\s*") set)})
{:tag-classes (some-> tag-classes string/trim not-empty (string/split #",\s*") set)
:property-classes (some-> property-classes string/trim not-empty (string/split #",\s*") set)
:property-parent-classes (some-> property-parent-classes string/trim not-empty (string/split #",\s*") set)})
;; common options
:path->file-object path->file-object
:notify-user show-notification
:set-ui-state state/set-state!
:<read-file (fn <read-file [file] (.text (:file-object file)))
@@ -405,7 +409,7 @@
(db-editor-handler/save-file! path content))
;; asset file options
:<read-asset read-asset
:<copy-asset #(copy-asset repo (config/get-repo-dir repo) %)
:<copy-asset #(copy-asset repo (config/get-repo-dir repo) % path->file-object)
;; doc file options
;; Write to frontend first as writing to worker first is poor ux with slow streaming changes
:export-file (fn export-file [conn m opts]