Skip to content
This repository was archived by the owner on Oct 22, 2025. It is now read-only.
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions packages/core/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -166,13 +166,13 @@
"nanoevents": "^9.1.0",
"on-change": "^5.0.1",
"p-retry": "^6.2.1",
"zod": "^3.25.76"
"zod": "^3.25.76",
"@rivetkit/engine-runner": "https://pkg.pr.new/rivet-gg/engine/@rivetkit/engine-runner@f8860f3"
},
"devDependencies": {
"@hono/node-server": "^1.18.2",
"@hono/node-ws": "^1.1.1",
"@rivet-gg/actor-core": "^25.1.0",
"@rivetkit/engine-runner": "https://pkg.pr.new/rivet-gg/engine/@rivetkit/engine-runner@7f23f3a",
"@types/invariant": "^2",
"@types/node": "^22.13.1",
"@types/ws": "^8",
Expand Down
5 changes: 4 additions & 1 deletion packages/core/src/drivers/engine/actor-driver.ts
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,10 @@ export class EngineActorDriver implements ActorDriver {
// TODO: Use alarm on sleep
// TODO: Send alarm to runner

throw new Error("Alarms not implemented for engine driver");
const delay = Math.max(timestamp - Date.now(), 0);
setTimeout(() => {
actor.onAlarm();
}, delay);
}

async getDatabase(_actorId: string): Promise<unknown | undefined> {
Expand Down
17 changes: 5 additions & 12 deletions packages/core/src/drivers/engine/api-endpoints.ts
Original file line number Diff line number Diff line change
@@ -1,23 +1,16 @@
import { CreateRequest } from "@/client/mod";
import { apiCall } from "./api-utils";
import type { Config } from "./config";
import { serializeActorKey } from "./keys";

// MARK: Common types
export type RivetId = string;

export interface ActorLifecycle {
kill_timeout_ms: number;
durable: boolean;
}

export interface Actor {
actor_id: RivetId;
name: string;
keys: string[];
key: string;
namespace_id: RivetId;
runner_name_selector: string;
lifecycle: ActorLifecycle;
create_ts: number;
connectable_ts?: number | null;
destroy_ts?: number | null;
Expand Down Expand Up @@ -46,9 +39,9 @@ export interface ActorsGetOrCreateByIdResponse {
export interface ActorsCreateRequest {
name: string;
runner_name_selector: string;
keys?: string[] | null;
crash_policy: string;
key?: string | null;
input?: string | null;
durable?: boolean | null;
}

export interface ActorsCreateResponse {
Expand Down Expand Up @@ -86,10 +79,10 @@ export async function getActorById(
// MARK: Get or create actor by id
export interface ActorsGetOrCreateByIdRequest {
name: string;
keys: string[];
key: string;
runner_name_selector: string;
crash_policy: "restart";
input?: string | null;
durable?: boolean | null;
}

export async function getOrCreateActorById(
Expand Down
12 changes: 8 additions & 4 deletions packages/core/src/drivers/engine/config.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,18 +7,22 @@ export const ConfigSchema = z
app: z.custom<Hono>().optional(),
endpoint: z
.string()
.default(getEnvUniversal("RIVET_ENGINE") ?? "http://localhost:7080"),
.default(
() => getEnvUniversal("RIVET_ENGINE") ?? "http://localhost:7080",
),
pegboardEndpoint: z.string().optional(),
namespace: z
.string()
.default(getEnvUniversal("RIVET_NAMESPACE") ?? "default"),
.default(() => getEnvUniversal("RIVET_NAMESPACE") ?? "default"),
runnerName: z
.string()
.default(getEnvUniversal("RIVET_RUNNER") ?? "rivetkit"),
.default(() => getEnvUniversal("RIVET_RUNNER") ?? "rivetkit"),
// TODO: Automatically attempt ot determine key by common env vars (e.g. k8s pod name)
runnerKey: z
.string()
.default(getEnvUniversal("RIVET_RUNNER_KEY") ?? crypto.randomUUID()),
.default(
() => getEnvUniversal("RIVET_RUNNER_KEY") ?? crypto.randomUUID(),
),
totalSlots: z.number().default(100_000),
addresses: z
.record(
Expand Down
10 changes: 5 additions & 5 deletions packages/core/src/drivers/engine/manager-driver.ts
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,7 @@ export class EngineManagerDriver implements ManagerDriver {
return undefined;
}

const keyRaw = response.actor.keys[0];
const keyRaw = response.actor.key;
invariant(keyRaw, `actor ${actorId} should have key`);
const key = deserializeActorKey(keyRaw);

Expand Down Expand Up @@ -233,10 +233,10 @@ export class EngineManagerDriver implements ManagerDriver {

const response = await getOrCreateActorById(this.#config, {
name,
keys: [serializeActorKey(key)],
key: serializeActorKey(key),
runner_name_selector: this.#config.runnerName,
input: input ? cbor.encode(actorInput).toString("base64") : undefined,
durable: true,
crash_policy: "restart",
});

const actorId = response.actor_id;
Expand Down Expand Up @@ -273,9 +273,9 @@ export class EngineManagerDriver implements ManagerDriver {
const result = await createActor(this.#config, {
name,
runner_name_selector: this.#config.runnerName,
keys: [serializeActorKey(key)],
key: serializeActorKey(key),
input: input ? cbor.encode(input).toString("base64") : null,
durable: true,
crash_policy: "restart",
});
const actorId = result.actor.actor_id;

Expand Down
4 changes: 1 addition & 3 deletions packages/core/tsup.config.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
import { defineConfig } from "tsup";
import defaultConfig from "../../tsup.base.ts";

export default defineConfig({
...defaultConfig,
});
export default defineConfig(defaultConfig);
3 changes: 3 additions & 0 deletions packages/drivers/cloudflare-workers/src/actor-handler-do.ts
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ import {
} from "@rivetkit/core";
import { serializeEmptyPersistData } from "@rivetkit/core/driver-helpers";
import type { ExecutionContext } from "hono";
import invariant from "invariant";
import {
CloudflareDurableObjectGlobalState,
createCloudflareActorsActorDriverBuilder,
Expand Down Expand Up @@ -113,6 +114,7 @@ export function createActorDurableObject(
globalState.setDOState(actorId, { ctx: this.ctx, env: env });

// Configure actor driver
invariant(runConfig.driver, "runConfig.driver");
runConfig.driver.actor =
createCloudflareActorsActorDriverBuilder(globalState);

Expand Down Expand Up @@ -184,6 +186,7 @@ export function createActorDurableObject(
const actorId = this.ctx.id.toString();

// Get the actor driver
invariant(runConfig.driver, "runConfig.driver");
const managerDriver = runConfig.driver.manager(
registry.config,
runConfig,
Expand Down
2 changes: 1 addition & 1 deletion packages/drivers/redis/src/manager.ts
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ export class RedisManagerDriver implements ManagerDriver {
return undefined;
}

return this.getForId({ actorId });
return this.getForId({ name, actorId });
}

async getOrCreateWithKey(
Expand Down
20 changes: 10 additions & 10 deletions pnpm-lock.yaml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading