@iann29/rastro 0.1.0-alpha.0 → 0.1.0-alpha.10

package/src/component/live.ts

@@ -1,6 +1,9 @@
 import { v } from "convex/values";
 import { internal } from "./_generated/api.js";
 import { internalMutation } from "./_generated/server.js";
+import { LIVE_SWEEP_INTERVAL_MS } from "./constants.js";
+
+const SWEEP_BATCH_SIZE = 500;
 
 export const expire = internalMutation({
   args: {
@@ -20,17 +23,95 @@ export const expire = internalMutation({
     if (liveSession.expiryChainDeadline !== args.scheduledDeadline) {
       return { expired: false, rescheduled: false, stale: true };
     }
-    if (liveSession.expiresAt > Date.now()) {
-      await ctx.db.patch("liveSessions", liveSession._id, {
-        expiryChainDeadline: liveSession.expiresAt,
-      });
-      await ctx.scheduler.runAt(liveSession.expiresAt, internal.live.expire, {
-        liveSessionId: liveSession._id,
-        scheduledDeadline: liveSession.expiresAt,
-      });
-      return { expired: false, rescheduled: true, stale: false };
+    const now = Date.now();
+    if (liveSession.expiresAt > now) {
+      const state = await ctx.db
+        .query("liveSweepStates")
+        .withIndex("by_siteId", (range) =>
+          range.eq("siteId", liveSession.siteId),
+        )
+        .unique();
+      const scheduledAt = Math.min(
+        liveSession.expiresAt,
+        now + LIVE_SWEEP_INTERVAL_MS,
+      );
+      if (
+        !state ||
+        state.scheduledAt <= now ||
+        state.scheduledAt > scheduledAt
+      ) {
+        if (state) {
+          await ctx.db.patch("liveSweepStates", state._id, {
+            scheduledAt,
+          });
+        } else {
+          await ctx.db.insert("liveSweepStates", {
+            siteId: liveSession.siteId,
+            scheduledAt,
+          });
+        }
+        await ctx.scheduler.runAt(scheduledAt, internal.live.sweep, {
+          siteId: liveSession.siteId,
+          scheduledAt,
+        });
+        return { expired: false, rescheduled: true, stale: false };
+      }
+      return { expired: false, rescheduled: false, stale: false };
     }
     await ctx.db.delete("liveSessions", liveSession._id);
     return { expired: true, rescheduled: false, stale: false };
   },
 });
+
+export const sweep = internalMutation({
+  args: {
+    siteId: v.id("sites"),
+    scheduledAt: v.number(),
+  },
+  returns: v.object({
+    expired: v.number(),
+    rescheduled: v.boolean(),
+    stale: v.boolean(),
+  }),
+  handler: async (ctx, args) => {
+    const state = await ctx.db
+      .query("liveSweepStates")
+      .withIndex("by_siteId", (range) => range.eq("siteId", args.siteId))
+      .unique();
+    if (!state || state.scheduledAt !== args.scheduledAt) {
+      return { expired: 0, rescheduled: false, stale: true };
+    }
+
+    const now = Date.now();
+    const expired = await ctx.db
+      .query("liveSessions")
+      .withIndex("by_siteId_and_expiresAt", (range) =>
+        range.eq("siteId", args.siteId).lte("expiresAt", now),
+      )
+      .take(SWEEP_BATCH_SIZE);
+    for (const session of expired) {
+      await ctx.db.delete("liveSessions", session._id);
+    }
+
+    const next = await ctx.db
+      .query("liveSessions")
+      .withIndex("by_siteId_and_expiresAt", (range) =>
+        range.eq("siteId", args.siteId),
+      )
+      .order("asc")
+      .first();
+    if (!next) {
+      await ctx.db.delete("liveSweepStates", state._id);
+      return { expired: expired.length, rescheduled: false, stale: false };
+    }
+
+    const scheduledAt =
+      expired.length === SWEEP_BATCH_SIZE ? now : now + LIVE_SWEEP_INTERVAL_MS;
+    await ctx.db.patch("liveSweepStates", state._id, { scheduledAt });
+    await ctx.scheduler.runAt(scheduledAt, internal.live.sweep, {
+      siteId: args.siteId,
+      scheduledAt,
+    });
+    return { expired: expired.length, rescheduled: true, stale: false };
+  },
+});