feat(core): actor sleeping

Author: NathanFlurry

packages/core/fixtures/driver-test-suite/action-timeout.ts

@@ -5,9 +5,7 @@ export const shortTimeoutActor = actor({
 	onAuth: () => {},
 	state: { value: 0 },
 	options: {
-		action: {
-			timeout: 50, // 50ms timeout
-		},
+		actionTimeout: 50, // 50ms timeout
 	},
 	actions: {
 		quickAction: async (c) => {
@@ -26,9 +24,7 @@ export const longTimeoutActor = actor({
 	onAuth: () => {},
 	state: { value: 0 },
 	options: {
-		action: {
-			timeout: 200, // 200ms timeout
-		},
+		actionTimeout: 200, // 200ms timeout
 	},
 	actions: {
 		delayedAction: async (c) => {
@@ -56,9 +52,7 @@ export const syncTimeoutActor = actor({
 	onAuth: () => {},
 	state: { value: 0 },
 	options: {
-		action: {
-			timeout: 50, // 50ms timeout
-		},
+		actionTimeout: 50, // 50ms timeout
 	},
 	actions: {
 		syncAction: (c) => {

packages/core/fixtures/driver-test-suite/conn-liveness.ts

@@ -7,10 +7,8 @@ export const connLivenessActor = actor({
 		acceptingConnections: true,
 	},
 	options: {
-		lifecycle: {
-			connectionLivenessInterval: 5_000,
-			connectionLivenessTimeout: 2_500,
-		},
+		connectionLivenessInterval: 5_000,
+		connectionLivenessTimeout: 2_500,
 	},
 	onConnect: (c, conn) => {
 		if (!c.state.acceptingConnections) {

packages/core/fixtures/driver-test-suite/error-handling.ts

@@ -69,10 +69,7 @@ export const errorHandlingActor = actor({
 		},
 	},
 	options: {
-		// Set a short timeout for this actor's actions
-		action: {
-			timeout: 500, // 500ms timeout for actions
-		},
+		actionTimeout: 500, // 500ms timeout for actions
 	},
 });
 
@@ -90,8 +87,6 @@ export const customTimeoutActor = actor({
 		},
 	},
 	options: {
-		action: {
-			timeout: 200, // 200ms timeout
-		},
+		actionTimeout: 200, // 200ms timeout
 	},
 });

packages/core/package.json

@@ -167,7 +167,7 @@
     "on-change": "^5.0.1",
     "p-retry": "^6.2.1",
     "zod": "^3.25.76",
-    "@rivetkit/engine-runner": "https://pkg.pr.new/rivet-gg/rivet/@rivetkit/engine-runner@f1c054d"
+    "@rivetkit/engine-runner": "https://pkg.pr.new/rivet-gg/engine/@rivetkit/engine-runner@664a377"
   },
   "devDependencies": {
     "@hono/node-server": "^1.18.2",

packages/core/src/actor/config.ts

@@ -60,28 +60,15 @@ export const ActorConfigSchema = z
 		createVars: z.function().optional(),
 		options: z
 			.object({
-				lifecycle: z
-					.object({
-						createVarsTimeout: z.number().positive().default(5000),
-						createConnStateTimeout: z.number().positive().default(5000),
-						onConnectTimeout: z.number().positive().default(5000),
-						connectionLivenessTimeout: z.number().positive().default(2500),
-						connectionLivenessInterval: z.number().positive().default(5000),
-					})
-					.strict()
-					.default({}),
-				state: z
-					.object({
-						saveInterval: z.number().positive().default(10_000),
-					})
-					.strict()
-					.default({}),
-				action: z
-					.object({
-						timeout: z.number().positive().default(60_000),
-					})
-					.strict()
-					.default({}),
+				createVarsTimeout: z.number().positive().default(5000),
+				createConnStateTimeout: z.number().positive().default(5000),
+				onConnectTimeout: z.number().positive().default(5000),
+				stateSaveInterval: z.number().positive().default(10_000),
+				actionTimeout: z.number().positive().default(60_000),
+				connectionLivenessTimeout: z.number().positive().default(2500),
+				connectionLivenessInterval: z.number().positive().default(5000),
+				noSleep: z.boolean().default(false),
+				sleepTimeout: z.number().positive().default(30_000),
 			})
 			.strict()
 			.default({}),

packages/core/src/actor/connection.ts

@@ -209,10 +209,7 @@ export class Conn<S, CP, CS, V, I, AD, DB extends AnyDatabaseProvider> {
 	 * @internal
 	 */
 	[CONNECTION_CHECK_LIVENESS_SYMBOL]() {
-		const readyState = this.#driver.getConnectionReadyState?.(
-			this.#actor,
-			this,
-		);
+		const readyState = this.#driver.getConnectionReadyState(this.#actor, this);
 
 		const isConnectionClosed =
 			readyState === ConnectionReadyState.CLOSED ||

packages/core/src/actor/driver.ts

@@ -39,6 +39,8 @@ export interface ActorDriver {
 	 */
 	getDatabase(actorId: string): Promise<unknown | undefined>;
 
+	sleep?(actorId: string): void;
+
 	shutdown?(immediate: boolean): Promise<void>;
 }
 
@@ -72,7 +74,7 @@ export interface ConnDriver<ConnDriverState = unknown> {
 	 * Returns the ready state of the connection.
 	 * This is used to determine if the connection is ready to send messages, or if the connection is stale.
 	 */
-	getConnectionReadyState?(
+	getConnectionReadyState(
 		actor: AnyActorInstance,
 		conn: AnyConn,
 	): ConnectionReadyState | undefined;

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

@@ -160,7 +160,9 @@ export interface GenericSseDriverState {
 	encoding: Encoding;
 }
 
-export function createGenericSseDriver(globalState: GenericConnGlobalState) {
+export function createGenericSseDriver(
+	globalState: GenericConnGlobalState,
+): ConnDriver<GenericSseDriverState> {
 	return {
 		sendMessage: (
 			_actor: AnyActorInstance,
@@ -219,8 +221,12 @@ export function createGenericSseDriver(globalState: GenericConnGlobalState) {
 // MARK: HTTP
 export type GenericHttpDriverState = Record<never, never>;
 
-export function createGenericHttpDriver() {
+export function createGenericHttpDriver(): ConnDriver<GenericHttpDriverState> {
 	return {
+		getConnectionReadyState(_actor, conn) {
+			// TODO: This might not be the correct logic
+			return ConnectionReadyState.OPEN;
+		},
 		disconnect: async () => {
 			// Noop
 		},