refactor linker actions into async functions

tayloraswift · Sep 16, 2024 · 15dd5d0 · 15dd5d0
1 parent d581634
commit 15dd5d0
Show file tree

Hide file tree

Showing 6 changed files with 118 additions and 120 deletions.
diff --git a/Sources/UnidocDB/Snapshots/Unidoc.DB.Snapshots.ClearAction.swift b/Sources/UnidocDB/Snapshots/Unidoc.DB.Snapshots.ClearAction.swift
diff --git a/Sources/UnidocDB/Snapshots/Unidoc.DB.Snapshots.QueueAction.swift b/Sources/UnidocDB/Snapshots/Unidoc.DB.Snapshots.QueueAction.swift
diff --git a/Sources/UnidocDB/Snapshots/Unidoc.DB.Snapshots.swift b/Sources/UnidocDB/Snapshots/Unidoc.DB.Snapshots.swift
@@ -169,6 +169,107 @@ extension Unidoc.DB.Snapshots
     }
 }
 
+extension Unidoc.DB.Snapshots
+{
+    @discardableResult
+    public
+    func queueAll(for action:Unidoc.LinkerAction) async throws -> Int
+    {
+        try await self.updateMany
+        {
+            $0
+            {
+                $0[.multi] = true
+                $0[.q]
+                {
+                    $0[Unidoc.Snapshot[.action]] { $0[.exists] = false }
+                }
+                $0[.u]
+                {
+                    $0[.set]
+                    {
+                        $0[Unidoc.Snapshot[.action]] = action
+                    }
+                }
+            }
+        }
+    }
+
+    @discardableResult
+    public
+    func queue(id:Unidoc.Edition, for action:Unidoc.LinkerAction) async throws -> Bool?
+    {
+        try await self.update
+        {
+            $0
+            {
+                $0[.q]
+                {
+                    $0[Unidoc.Snapshot[.action]] { $0[.exists] = false }
+                    $0[Unidoc.Snapshot[.id]] = id
+                }
+                $0[.u]
+                {
+                    $0[.set]
+                    {
+                        $0[Unidoc.Snapshot[.action]] = action
+                    }
+                }
+            }
+        }
+    }
+
+    /// Clears the **queued action** for a single snapshot. Does not delete the snapshot!
+    @discardableResult
+    public
+    func clear(id:Unidoc.Edition) async throws -> Bool?
+    {
+        try await self.update
+        {
+            $0
+            {
+                $0[.q] { $0[Unidoc.Snapshot[.id]] = id }
+                $0[.u]
+                {
+                    $0[.unset]
+                    {
+                        $0[Unidoc.Snapshot[.action]] = ()
+                    }
+                }
+            }
+        }
+    }
+
+    @discardableResult
+    public
+    func mark(id:Unidoc.Edition, vintage:Bool) async throws -> Bool?
+    {
+        try await self.update
+        {
+            $0
+            {
+                $0[.q] { $0[Unidoc.Snapshot[.id]] = id }
+                $0[.u]
+                {
+                    if  vintage
+                    {
+                        $0[.set]
+                        {
+                            $0[Unidoc.Snapshot[.vintage]] = true
+                        }
+                    }
+                    else
+                    {
+                        $0[.unset]
+                        {
+                            $0[Unidoc.Snapshot[.vintage]] = ()
+                        }
+                    }
+                }
+            }
+        }
+    }
+}
 extension Unidoc.DB.Snapshots
 {
     /// Returns a single batch of symbol graphs that are queued for linking.

diff --git a/Sources/UnidocLinkerPlugin/Unidoc.GraphLinker.swift b/Sources/UnidocLinkerPlugin/Unidoc.GraphLinker.swift
@@ -136,8 +136,7 @@ extension Unidoc.GraphLinker
             }
         }
 
-        try await context.db.update(
-            with: Unidoc.DB.Snapshots.ClearAction.one(operation.edition))
+        try await context.db.snapshots.clear(id: operation.edition)
 
         if  let event:Event
         {

diff --git a/Sources/UnidocServer/Operations/Interactions/Unidoc.LinkerOperation.swift b/Sources/UnidocServer/Operations/Interactions/Unidoc.LinkerOperation.swift
@@ -10,12 +10,14 @@ extension Unidoc
     /// Queues one or more editions for uplinking. The uplinking process itself is asynchronous.
     struct LinkerOperation:Sendable
     {
-        let queue:DB.Snapshots.QueueAction
+        let action:LinkerAction
+        let scope:Edition?
         let back:String?
 
-        init(queue:DB.Snapshots.QueueAction, back:String? = nil)
+        init(action:Unidoc.LinkerAction, scope:Unidoc.Edition?, back:String? = nil)
         {
-            self.queue = queue
+            self.action = action
+            self.scope = scope
             self.back = back
         }
     }
@@ -24,7 +26,7 @@ extension Unidoc.LinkerOperation
 {
     init(action:Unidoc.LinkerAction, form:Unidoc.LinkerForm)
     {
-        self.init(queue: .one(form.edition, action: action), back: form.back)
+        self.init(action: action, scope: form.edition, back: form.back)
     }
 }
 extension Unidoc.LinkerOperation:Unidoc.AdministrativeOperation
@@ -33,7 +35,15 @@ extension Unidoc.LinkerOperation:Unidoc.AdministrativeOperation
         db:Unidoc.DB,
         as _:Unidoc.RenderFormat) async throws -> HTTP.ServerResponse?
     {
-        try await db.update(with: self.queue)
+        if  let scope:Unidoc.Edition = self.scope
+        {
+            try await db.snapshots.queue(id: scope, for: self.action)
+        }
+        else
+        {
+            try await db.snapshots.queueAll(for: self.action)
+        }
+
         return .redirect(.seeOther(self.back ?? "/admin"))
     }
 }
diff --git a/Sources/UnidocServer/Requests/Unidoc.Router.swift b/Sources/UnidocServer/Requests/Unidoc.Router.swift
@@ -646,7 +646,7 @@ extension Unidoc.Router
             }
 
         case .uplinkAll:
-            return .unordered(Unidoc.LinkerOperation.init(queue: .all))
+            return .unordered(Unidoc.LinkerOperation.init(action: .uplinkRefresh, scope: nil))
 
         case .userConfig:
             if  let account:Unidoc.Account = self.authorization.account,