mirror of
https://github.com/logseq/logseq.git
synced 2026-06-01 19:01:22 +00:00
integrate worker sync
This commit is contained in:
@@ -4,8 +4,8 @@
|
||||
[frontend.common.missionary :as c.m]
|
||||
[frontend.config :as config]
|
||||
[frontend.db :as db]
|
||||
[frontend.handler.db-based.rtc :as rtc-handler]
|
||||
[frontend.handler.db-based.rtc-flows :as rtc-flows]
|
||||
[frontend.handler.db-based.sync :as rtc-handler]
|
||||
[frontend.handler.notification :as notification]
|
||||
[frontend.state :as state]
|
||||
[lambdaisland.glogi :as log]
|
||||
|
||||
51
src/main/frontend/handler/db_based/sync.cljs
Normal file
51
src/main/frontend/handler/db_based/sync.cljs
Normal file
@@ -0,0 +1,51 @@
|
||||
(ns frontend.handler.db-based.sync
|
||||
"Dispatch RTC calls between legacy RTC and worker-sync implementations."
|
||||
(:require [frontend.config :as config]
|
||||
[frontend.handler.db-based.rtc :as rtc-handler]
|
||||
[frontend.handler.db-based.worker-sync :as worker-sync-handler]))
|
||||
|
||||
(defn- worker-sync-enabled? []
|
||||
config/worker-sync-enabled?)
|
||||
|
||||
(defn <rtc-create-graph! [repo]
|
||||
(if (worker-sync-enabled?)
|
||||
(worker-sync-handler/<rtc-create-graph! repo)
|
||||
(rtc-handler/<rtc-create-graph! repo)))
|
||||
|
||||
(defn <rtc-delete-graph! [graph-uuid schema-version]
|
||||
(if (worker-sync-enabled?)
|
||||
(worker-sync-handler/<rtc-delete-graph! graph-uuid schema-version)
|
||||
(rtc-handler/<rtc-delete-graph! graph-uuid schema-version)))
|
||||
|
||||
(defn <rtc-download-graph! [graph-name graph-uuid graph-schema-version timeout-ms]
|
||||
(rtc-handler/<rtc-download-graph! graph-name graph-uuid graph-schema-version timeout-ms))
|
||||
|
||||
(defn <rtc-stop! []
|
||||
(if (worker-sync-enabled?)
|
||||
(worker-sync-handler/<rtc-stop!)
|
||||
(rtc-handler/<rtc-stop!)))
|
||||
|
||||
(defn <rtc-branch-graph! [repo]
|
||||
(rtc-handler/<rtc-branch-graph! repo))
|
||||
|
||||
(defn notification-download-higher-schema-graph! [graph-name graph-uuid schema-version]
|
||||
(rtc-handler/notification-download-higher-schema-graph! graph-name graph-uuid schema-version))
|
||||
|
||||
(defn <rtc-get-users-info []
|
||||
(if (worker-sync-enabled?)
|
||||
(worker-sync-handler/<rtc-get-users-info)
|
||||
(rtc-handler/<rtc-get-users-info)))
|
||||
|
||||
(defn <rtc-start!
|
||||
[repo & {:keys [stop-before-start?] :or {stop-before-start? true}}]
|
||||
(if (worker-sync-enabled?)
|
||||
(worker-sync-handler/<rtc-start! repo :stop-before-start? stop-before-start?)
|
||||
(rtc-handler/<rtc-start! repo :stop-before-start? stop-before-start?)))
|
||||
|
||||
(defn <get-remote-graphs []
|
||||
(if (worker-sync-enabled?)
|
||||
(worker-sync-handler/<get-remote-graphs)
|
||||
(rtc-handler/<get-remote-graphs)))
|
||||
|
||||
(defn <rtc-invite-email [graph-uuid email]
|
||||
(rtc-handler/<rtc-invite-email graph-uuid email))
|
||||
113
src/main/frontend/handler/db_based/worker_sync.cljs
Normal file
113
src/main/frontend/handler/db_based/worker_sync.cljs
Normal file
@@ -0,0 +1,113 @@
|
||||
(ns frontend.handler.db-based.worker-sync
|
||||
"Worker-sync handler based on Cloudflare Durable Objects."
|
||||
(:require [clojure.string :as string]
|
||||
[frontend.config :as config]
|
||||
[frontend.db :as db]
|
||||
[frontend.handler.repo :as repo-handler]
|
||||
[frontend.state :as state]
|
||||
[lambdaisland.glogi :as log]
|
||||
[logseq.db :as ldb]
|
||||
[promesa.core :as p]))
|
||||
|
||||
(defn- ws->http-base [ws-url]
|
||||
(when (string? ws-url)
|
||||
(let [base (cond
|
||||
(string/starts-with? ws-url "wss://")
|
||||
(str "https://" (subs ws-url (count "wss://")))
|
||||
|
||||
(string/starts-with? ws-url "ws://")
|
||||
(str "http://" (subs ws-url (count "ws://")))
|
||||
|
||||
:else ws-url)
|
||||
base (string/replace base #"/sync/%s$" "")]
|
||||
base)))
|
||||
|
||||
(defn- http-base []
|
||||
(or config/worker-sync-http-base
|
||||
(ws->http-base config/worker-sync-ws-url)))
|
||||
|
||||
(defn- get-graph-id [repo]
|
||||
(let [db (db/get-db repo)]
|
||||
(or (ldb/get-graph-rtc-uuid db)
|
||||
;; FIXME: only for testing
|
||||
(random-uuid))))
|
||||
|
||||
(defn- fetch-json
|
||||
[url opts]
|
||||
(p/let [resp (js/fetch url (clj->js opts))
|
||||
text (.text resp)
|
||||
data (when (seq text) (js/JSON.parse text))]
|
||||
(if (.-ok resp)
|
||||
data
|
||||
(throw (ex-info "worker-sync request failed"
|
||||
{:status (.-status resp)
|
||||
:url url
|
||||
:body data})))))
|
||||
|
||||
(defn <rtc-start!
|
||||
[repo & {:keys [_stop-before-start?] :as _opts}]
|
||||
(log/info :worker-sync/start {:repo repo})
|
||||
(state/<invoke-db-worker :thread-api/worker-sync-start repo))
|
||||
|
||||
(defn <rtc-stop!
|
||||
[]
|
||||
(log/info :worker-sync/stop true)
|
||||
(state/<invoke-db-worker :thread-api/worker-sync-stop))
|
||||
|
||||
(defn <rtc-get-users-info
|
||||
[]
|
||||
(p/resolved nil))
|
||||
|
||||
(defn <rtc-create-graph!
|
||||
[repo]
|
||||
(let [graph-id (get-graph-id repo)
|
||||
schema-version (some-> (ldb/get-graph-schema-version (db/get-db)) :major str)
|
||||
base (http-base)]
|
||||
(if (and graph-id base)
|
||||
(p/let [result (fetch-json (str base "/graphs")
|
||||
{:method "POST"
|
||||
:headers {"content-type" "application/json"}
|
||||
:body (js/JSON.stringify
|
||||
#js {:graph_id (str graph-id)
|
||||
:graph_name repo
|
||||
:schema_version schema-version})})]
|
||||
(ldb/transact! repo [{:logseq.kv/graph-uuid graph-id}])
|
||||
result)
|
||||
(p/rejected (ex-info "worker-sync missing graph info"
|
||||
{:type :worker-sync/invalid-graph
|
||||
:graph-id graph-id
|
||||
:base base})))))
|
||||
|
||||
(defn <rtc-delete-graph!
|
||||
[graph-uuid _schema-version]
|
||||
(let [base (http-base)]
|
||||
(if (and graph-uuid base)
|
||||
(fetch-json (str base "/graphs/" graph-uuid) {:method "DELETE"})
|
||||
(p/rejected (ex-info "worker-sync missing graph id"
|
||||
{:type :worker-sync/invalid-graph
|
||||
:graph-uuid graph-uuid
|
||||
:base base})))))
|
||||
|
||||
(defn <get-remote-graphs
|
||||
[]
|
||||
(let [base (http-base)]
|
||||
(if-not base
|
||||
(p/resolved [])
|
||||
(-> (p/let [_ (state/set-state! :rtc/loading-graphs? true)
|
||||
resp (fetch-json (str base "/graphs") {:method "GET"})
|
||||
graphs (js->clj (aget resp "graphs") :keywordize-keys true)
|
||||
result (mapv (fn [graph]
|
||||
(merge
|
||||
{:url (str config/db-version-prefix (:graph_name graph))
|
||||
:GraphName (:graph_name graph)
|
||||
:GraphSchemaVersion (:schema_version graph)
|
||||
:GraphUUID (:graph_id graph)
|
||||
:rtc-graph? true}
|
||||
(dissoc graph :graph_id :graph_name :schema_version)))
|
||||
graphs)]
|
||||
(state/set-state! :rtc/graphs result)
|
||||
(repo-handler/refresh-repos!)
|
||||
result)
|
||||
(p/finally
|
||||
(fn []
|
||||
(state/set-state! :rtc/loading-graphs? false)))))))
|
||||
Reference in New Issue
Block a user