Add new scheduling methods to API

Author: zephraph

packages/actor-core/src/actor/driver.ts

@@ -14,6 +14,8 @@ export interface ActorDriver {
 
 	// Schedule
 	setAlarm(actor: AnyActorInstance, timestamp: number): Promise<void>;
+	getAlarm(actor: AnyActorInstance): Promise<number | null>;
+	deleteAlarm(actor: AnyActorInstance): Promise<void>;
 
 	// TODO:
 	//destroy(): Promise<void>;

packages/actor-core/src/test/driver/actor.ts

@@ -27,4 +27,12 @@ export class TestActorDriver implements ActorDriver {
 			actor.onAlarm();
 		}, timestamp - Date.now());
 	}
+
+	async getAlarm(actor: AnyActorInstance): Promise<number | null> {
+		throw new Error("Method not implemented.");
+	}
+
+	async deleteAlarm(actor: AnyActorInstance): Promise<void> {
+		throw new Error("Method not implemented.");
+	}
 }

packages/drivers/file-system/src/actor.ts

@@ -41,4 +41,12 @@ export class FileSystemActorDriver implements ActorDriver {
             actor.onAlarm();
         }, delay);
     }
-}
+
+    async getAlarm(actor: AnyActorInstance): Promise<number | null> {
+        throw new Error("Method not implemented.");
+    }
+
+    async deleteAlarm(actor: AnyActorInstance): Promise<void> {
+        throw new Error("Method not implemented.");
+    }
+}

packages/drivers/memory/src/actor.ts

@@ -28,4 +28,12 @@ export class MemoryActorDriver implements ActorDriver {
 			actor.onAlarm();
 		}, delay);
 	}
+
+	async getAlarm(actor: AnyActorInstance): Promise<number | null> {
+		throw new Error("Method not implemented.");
+	}
+
+	async deleteAlarm(actor: AnyActorInstance): Promise<void> {
+		throw new Error("Method not implemented.");
+	}
 }

packages/drivers/redis/src/actor.ts

@@ -36,4 +36,12 @@ export class RedisActorDriver implements ActorDriver {
 			actor.onAlarm();
 		}, delay);
 	}
+
+	getAlarm(actor: AnyActorInstance): Promise<number | null> {
+		throw new Error("Method not implemented.");
+	}
+
+	deleteAlarm(actor: AnyActorInstance): Promise<void> {
+		throw new Error("Method not implemented.");
+	}
 }

packages/platforms/cloudflare-workers/src/actor_driver.ts

@@ -61,4 +61,12 @@ export class CloudflareWorkersActorDriver implements ActorDriver {
 	async setAlarm(actor: AnyActorInstance, timestamp: number): Promise<void> {
 		await this.#getDOCtx(actor.id).storage.setAlarm(timestamp);
 	}
+
+	async getAlarm(actor: AnyActorInstance): Promise<number | null> {
+		return await this.#getDOCtx(actor.id).storage.getAlarm();
+	}
+
+	async deleteAlarm(actor: AnyActorInstance): Promise<void> {
+		await this.#getDOCtx(actor.id).storage.deleteAlarm();
+	}
 }

packages/platforms/rivet/src/actor_driver.ts

@@ -32,4 +32,12 @@ export class RivetActorDriver implements ActorDriver {
 			actor.onAlarm();
 		}, timeout);
 	}
+
+	getAlarm(actor: AnyActorInstance): Promise<number | null> {
+		throw new Error("Method not implemented.");
+	}
+
+	deleteAlarm(actor: AnyActorInstance): Promise<void> {
+		throw new Error("Method not implemented.");
+	}
 }