rivetkit 2.3.11-rc.9 → 2.3.11

package/dist/tsup/mod.cjs

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2; var _class3;
 
-var _chunkIMRDHZM2cjs = require('./chunk-IMRDHZM2.cjs');
+var _chunkVBJQIBXIcjs = require('./chunk-VBJQIBXI.cjs');
 
 
 
@@ -12,28 +12,34 @@ var _chunkIMRDHZM2cjs = require('./chunk-IMRDHZM2.cjs');
 
 
 
-var _chunkHNHLBM3Ycjs = require('./chunk-HNHLBM3Y.cjs');
 
 
 
 
+var _chunk4KGXFC6Mcjs = require('./chunk-4KGXFC6M.cjs');
 
 
 
 
 
 
-var _chunkBZWKY6WOcjs = require('./chunk-BZWKY6WO.cjs');
+
+
+
+
+var _chunkRXXDNJXBcjs = require('./chunk-RXXDNJXB.cjs');
 
 
 
 var _chunkMYWZCDMLcjs = require('./chunk-MYWZCDML.cjs');
-require('./chunk-T5XVTVQY.cjs');
-require('./chunk-UEULFXUI.cjs');
 
 
 
 
+var _chunkLWQFXGKScjs = require('./chunk-LWQFXGKS.cjs');
+require('./chunk-KVFIONKQ.cjs');
+require('./chunk-GN2EUMEW.cjs');
+
 
 
 
@@ -62,7 +68,10 @@ require('./chunk-UEULFXUI.cjs');
 
 
 
-var _chunkT56IZ42Mcjs = require('./chunk-T56IZ42M.cjs');
+
+
+
+var _chunkGU7ML3KEcjs = require('./chunk-GU7ML3KE.cjs');
 
 
 
@@ -78,7 +87,7 @@ var _chunk4FC7TVS6cjs = require('./chunk-4FC7TVS6.cjs');
 
 // src/actor/log.ts
 function loggerWithoutContext() {
-  return _chunkT56IZ42Mcjs.getLogger.call(void 0, "actor-runtime");
+  return _chunkGU7ML3KEcjs.getLogger.call(void 0, "actor-runtime");
 }
 
 // src/actor/definition.ts
@@ -117,9 +126,9 @@ function actor(input) {
   warnDeprecatedOptionKeys(
     input == null ? void 0 : input.options
   );
-  const config = _chunkHNHLBM3Ycjs.ActorConfigSchema.parse(input);
-  _chunkHNHLBM3Ycjs.flattenActionHandlers.call(void 0, config.actions);
-  _chunkHNHLBM3Ycjs.flattenActionInputSchemas.call(void 0, config.actions, config.actionInputSchemas);
+  const config = _chunk4KGXFC6Mcjs.ActorConfigSchema.parse(input);
+  _chunk4KGXFC6Mcjs.flattenActionHandlers.call(void 0, config.actions);
+  _chunk4KGXFC6Mcjs.flattenActionInputSchemas.call(void 0, config.actions, config.actionInputSchemas);
   return new ActorDefinition(config);
 }
 function isStaticActorDefinition(definition) {
@@ -218,7 +227,7 @@ function validateSchemaSync(schemas, key, data) {
 var _virtualwebsocket = require('@rivetkit/virtual-websocket');
 var _ws = require('hono/ws');
 function logger() {
-  return _chunkT56IZ42Mcjs.getLogger.call(void 0, "inline-websocket-adapter");
+  return _chunkGU7ML3KEcjs.getLogger.call(void 0, "inline-websocket-adapter");
 }
 var InlineWebSocketAdapter = class {
   #handler;
@@ -389,7 +398,7 @@ function isLocalEngineEndpoint(endpoint) {
 
 // src/registry/log.ts
 function logger2() {
-  return _chunkT56IZ42Mcjs.getLogger.call(void 0, "registry");
+  return _chunkGU7ML3KEcjs.getLogger.call(void 0, "registry");
 }
 
 // src/serverless/configure.ts
@@ -435,8 +444,8 @@ async function configureServerlessPool(config) {
         );
       }
       const customConfig = config.configurePool;
-      const clientConfig = _chunkBZWKY6WOcjs.convertRegistryConfigToClientConfig.call(void 0, config);
-      const dcsRes = await _chunkBZWKY6WOcjs.getDatacenters.call(void 0, clientConfig);
+      const clientConfig = _chunkRXXDNJXBcjs.convertRegistryConfigToClientConfig.call(void 0, config);
+      const dcsRes = await _chunkRXXDNJXBcjs.getDatacenters.call(void 0, clientConfig);
       const poolName = _nullishCoalesce(customConfig.name, () => ( "default"));
       const serverlessToken = _nullishCoalesce(config.token, () => ( config.publicToken));
       const headers = {
@@ -458,7 +467,7 @@ async function configureServerlessPool(config) {
         metadata: _nullishCoalesce(customConfig.metadata, () => ( {})),
         drain_on_version_upgrade: _nullishCoalesce(customConfig.drainOnVersionUpgrade, () => ( true))
       };
-      await _chunkBZWKY6WOcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
+      await _chunkRXXDNJXBcjs.updateRunnerConfig.call(void 0, clientConfig, poolName, {
         datacenters: Object.fromEntries(
           dcsRes.datacenters.map((dc) => [dc.name, serverlessConfig])
         )
@@ -475,7 +484,7 @@ async function configureServerlessPool(config) {
       logger2().warn({
         msg: "serverless pool configuration attempt failed",
         attempts,
-        error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
+        error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, error)
       });
       await sleep(CONFIGURE_RETRY_DELAY_MS);
     }
@@ -483,7 +492,7 @@ async function configureServerlessPool(config) {
   logger2().error({
     msg: "failed to configure serverless pool, validate endpoint is configured correctly then restart this process",
     attempts,
-    error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, lastError)
+    error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, lastError)
   });
   throw lastError;
 }
@@ -495,20 +504,20 @@ var _zod = require('zod');
 var _v4 = require('zod/v4');
 var warnedMissingVersion = false;
 var EnvoyConfigSchema = _v4.z.object({
-  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunkT56IZ42Mcjs.getRivetPool.call(void 0, ), () => ( "default"))),
+  poolName: _v4.z.string().default(() => _nullishCoalesce(_chunkGU7ML3KEcjs.getRivetPool.call(void 0, ), () => ( "default"))),
   version: _v4.z.number().default(() => {
-    const version = _chunkT56IZ42Mcjs.getRivetEnvoyVersion.call(void 0, );
+    const version = _chunkGU7ML3KEcjs.getRivetEnvoyVersion.call(void 0, );
     if (version !== void 0) return version;
-    if (_chunkT56IZ42Mcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
+    if (_chunkGU7ML3KEcjs.getNodeEnv.call(void 0, ) === "production" && !warnedMissingVersion) {
       warnedMissingVersion = true;
-      _chunkT56IZ42Mcjs.getLogger.call(void 0, "rivetkit").error(
+      _chunkGU7ML3KEcjs.getLogger.call(void 0, "rivetkit").error(
         "RIVET_ENVOY_VERSION is not set. Actors will not be versioned, which means they won't be drained on deploy. This is only needed when self-hosting or using a custom envoy (not needed for Rivet Compute). Set this as a build arg in your Dockerfile. See https://rivet.dev/docs/actors/versions"
       );
     }
     return 1;
   }),
   // Deprecated.
-  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunkT56IZ42Mcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
+  totalSlots: _v4.z.number().default(() => _nullishCoalesce(_chunkGU7ML3KEcjs.getRivetTotalSlots.call(void 0, ), () => ( 1e5))),
   envoyKey: _v4.z.string().optional()
 });
 
@@ -546,7 +555,7 @@ var ServerlessConfigSchema = _v4.z.object({
    *
    * Can also be set via RIVET_PUBLIC_ENDPOINT environment variable.
    */
-  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkT56IZ42Mcjs.getRivetPublicEndpoint.call(void 0, )))),
+  publicEndpoint: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkGU7ML3KEcjs.getRivetPublicEndpoint.call(void 0, )))),
   /**
    * Token that clients should use when connecting via the public endpoint.
    *
@@ -554,7 +563,7 @@ var ServerlessConfigSchema = _v4.z.object({
    *
    * Can also be specified in the publicEndpoint URL as `https://namespace:token@host`.
    */
-  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkT56IZ42Mcjs.getRivetPublicToken.call(void 0, ))))
+  publicToken: _v4.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkGU7ML3KEcjs.getRivetPublicToken.call(void 0, ))))
   // There is no publicNamespace config option because the frontend and backend
   // cannot use different namespaces. The namespace is extracted from the
   // publicEndpoint URL auth syntax if provided.
@@ -608,7 +617,7 @@ var RegistryConfigSchema = _zod.z.object({
    * Runtime binding to use for RivetKit core.
    * */
   runtime: RuntimeKindSchema.optional().transform((val, ctx) => {
-    const rawRuntime = _nullishCoalesce(val, () => ( _chunkT56IZ42Mcjs.getRivetkitRuntime.call(void 0, )));
+    const rawRuntime = _nullishCoalesce(val, () => ( _chunkGU7ML3KEcjs.getRivetkitRuntime.call(void 0, )));
     if (rawRuntime === void 0) {
       return "auto";
     }
@@ -645,7 +654,7 @@ var RegistryConfigSchema = _zod.z.object({
    * */
   logging: _zod.z.object({
     baseLogger: _zod.z.custom().optional(),
-    level: _chunkT56IZ42Mcjs.LogLevelSchema.optional()
+    level: _chunkGU7ML3KEcjs.LogLevelSchema.optional()
   }).optional().default(() => ({})),
   // MARK: Routing
   // // This is a function to allow for lazy configuration of upgradeWebSocket on the
@@ -663,9 +672,9 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Can also be set via RIVET_ENDPOINT environment variables.
    */
-  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunkT56IZ42Mcjs.getRivetEngine.call(void 0, ))), () => ( _chunkT56IZ42Mcjs.getRivetEndpoint.call(void 0, )))),
-  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkT56IZ42Mcjs.getRivetToken.call(void 0, )))),
-  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkT56IZ42Mcjs.getRivetNamespace.call(void 0, )))),
+  endpoint: _zod.z.string().optional().transform((val) => _nullishCoalesce(_nullishCoalesce(val, () => ( _chunkGU7ML3KEcjs.getRivetEngine.call(void 0, ))), () => ( _chunkGU7ML3KEcjs.getRivetEndpoint.call(void 0, )))),
+  token: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkGU7ML3KEcjs.getRivetToken.call(void 0, )))),
+  namespace: _zod.z.string().optional().transform((val) => _nullishCoalesce(val, () => ( _chunkGU7ML3KEcjs.getRivetNamespace.call(void 0, )))),
   headers: _zod.z.record(_zod.z.string(), _zod.z.string()).optional().default({}),
   // MARK: Client
   // TODO:
@@ -705,21 +714,21 @@ var RegistryConfigSchema = _zod.z.object({
    *
    * Starts the full Rust engine process locally.
    */
-  startEngine: _zod.z.boolean().default(() => _chunkT56IZ42Mcjs.getRivetRunEngine.call(void 0, )),
+  startEngine: _zod.z.boolean().default(() => _chunkGU7ML3KEcjs.getRivetRunEngine.call(void 0, )),
   /**
    * @experimental
    *
    * Host to bind the spawned local engine process to.
    */
-  engineHost: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkT56IZ42Mcjs.getRivetRunEngineHost.call(void 0, ), () => ( ENGINE_HOST))),
+  engineHost: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkGU7ML3KEcjs.getRivetRunEngineHost.call(void 0, ), () => ( ENGINE_HOST))),
   /**
    * @experimental
    *
    * Port to bind the spawned local engine process to.
    */
-  enginePort: _zod.z.number().int().min(1).max(65535).optional().default(() => _nullishCoalesce(_chunkT56IZ42Mcjs.getRivetRunEnginePort.call(void 0, ), () => ( ENGINE_PORT))),
+  enginePort: _zod.z.number().int().min(1).max(65535).optional().default(() => _nullishCoalesce(_chunkGU7ML3KEcjs.getRivetRunEnginePort.call(void 0, ), () => ( ENGINE_PORT))),
   /** @experimental */
-  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkT56IZ42Mcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkT56IZ42Mcjs.VERSION))),
+  engineVersion: _zod.z.string().optional().default(() => _nullishCoalesce(_chunkGU7ML3KEcjs.getRivetRunEngineVersion.call(void 0, ), () => ( _chunkGU7ML3KEcjs.VERSION))),
   /**
    * @experimental
    *
@@ -767,7 +776,7 @@ var RegistryConfigSchema = _zod.z.object({
   }))
 }).transform((config, ctx) => {
   var _a, _b, _c;
-  const isDevEnv = _chunkT56IZ42Mcjs.isDev.call(void 0, );
+  const isDevEnv = _chunkGU7ML3KEcjs.isDev.call(void 0, );
   const sqliteBackend = _nullishCoalesce(((_a = config.sqlite) == null ? void 0 : _a.backend), () => ( ((_b = config.test) == null ? void 0 : _b.sqliteBackend)));
   if (config.runtime === "wasm" && sqliteBackend === "local") {
     ctx.addIssue({
@@ -777,7 +786,7 @@ var RegistryConfigSchema = _zod.z.object({
     });
   }
   const sqlite = config.runtime === "wasm" && config.sqlite === void 0 ? { backend: "remote" } : config.sqlite;
-  const parsedEndpoint = config.endpoint ? _chunkBZWKY6WOcjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedEndpoint = config.endpoint ? _chunkRXXDNJXBcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.endpoint,
     path: ["endpoint"],
     namespace: config.namespace,
@@ -805,7 +814,7 @@ var RegistryConfigSchema = _zod.z.object({
   );
   const namespace = _nullishCoalesce(_nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.namespace), () => ( config.namespace)), () => ( "default"));
   const token = _nullishCoalesce((parsedEndpoint == null ? void 0 : parsedEndpoint.token), () => ( config.token));
-  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunkBZWKY6WOcjs.tryParseEndpoint.call(void 0, ctx, {
+  const parsedPublicEndpoint = config.serverless.publicEndpoint ? _chunkRXXDNJXBcjs.tryParseEndpoint.call(void 0, ctx, {
     endpoint: config.serverless.publicEndpoint,
     path: ["serverless", "publicEndpoint"]
   }) : void 0;
@@ -840,7 +849,7 @@ function buildActorNames(config) {
     Object.keys(config.use).map((actorName) => {
       const definition = config.use[actorName];
       const options = _nullishCoalesce(definition.config.options, () => ( {}));
-      const runMeta = _chunkHNHLBM3Ycjs.getRunMetadata.call(void 0, definition.config.run);
+      const runMeta = _chunk4KGXFC6Mcjs.getRunMetadata.call(void 0, definition.config.run);
       const metadata = {};
       metadata.icon = _nullishCoalesce(options.icon, () => ( runMeta.icon));
       metadata.name = _nullishCoalesce(options.name, () => ( runMeta.name));
@@ -908,7 +917,7 @@ var DocRegistryConfigSchema = _zod.z.object({
   noWelcome: _zod.z.boolean().optional().describe("Disable the welcome message on startup. Default: false"),
   sqlite: DocSqliteConfigSchema,
   logging: _zod.z.object({
-    level: _chunkT56IZ42Mcjs.LogLevelSchema.optional().describe(
+    level: _chunkGU7ML3KEcjs.LogLevelSchema.optional().describe(
       "Log level for RivetKit. Default: 'warn'"
     )
   }).optional().describe("Logging configuration."),
@@ -1193,11 +1202,33 @@ function wrapJsNativeDatabase(database) {
         release();
       }
     },
-    async beginTransaction(timeoutMs) {
+    async beginTransaction(timeoutMs, name) {
       const release = gate.enter();
       let transaction;
       try {
-        transaction = await database.beginTransaction(timeoutMs);
+        transaction = await database.beginTransaction(timeoutMs, name);
+      } catch (error) {
+        enrichNativeDatabaseError(database, error);
+      } finally {
+        release();
+      }
+      return wrapTransaction(database, transaction, gate, (result) => {
+        if (result.lastInsertRowId !== void 0) {
+          lastInsertRowId = result.lastInsertRowId;
+        }
+      });
+    },
+    async beginStateTransaction(timeoutMs, context) {
+      if (!database.beginStateTransaction) {
+        throw new Error("actor state transactions are not configured");
+      }
+      const release = gate.enter();
+      let transaction;
+      try {
+        transaction = await database.beginStateTransaction(
+          timeoutMs,
+          context
+        );
       } catch (error) {
         enrichNativeDatabaseError(database, error);
       } finally {
@@ -1307,45 +1338,37 @@ var nodeChildProcess;
 var nodeStream;
 var nodeUrl;
 var hasImportedDependencies = false;
-function getRequireFn() {
-  return _module.createRequire.call(void 0, import.meta.url);
+var fallbackRequire;
+function loadNodeBuiltin(id) {
+  var _a;
+  const nodeProcess = globalThis.process;
+  const builtin = (_a = nodeProcess == null ? void 0 : nodeProcess.getBuiltinModule) == null ? void 0 : _a.call(nodeProcess, id);
+  if (builtin !== void 0) {
+    return builtin;
+  }
+  if (!(nodeProcess == null ? void 0 : nodeProcess.execPath)) {
+    throw new Error("Node.js built-ins are not available");
+  }
+  fallbackRequire ??= _module.createRequire.call(void 0, nodeProcess.execPath);
+  return fallbackRequire(id);
 }
 function importNodeDependencies() {
   if (hasImportedDependencies) return;
   try {
-    const requireFn = getRequireFn();
-    nodeCrypto = requireFn(
-      /* webpackIgnore: true */
-      "node:crypto"
-    );
-    nodeFsSync = requireFn(
-      /* webpackIgnore: true */
-      "node:fs"
-    );
-    nodeFs = requireFn(
-      /* webpackIgnore: true */
+    nodeCrypto = loadNodeBuiltin("node:crypto");
+    nodeFsSync = loadNodeBuiltin("node:fs");
+    nodeFs = loadNodeBuiltin(
       "node:fs/promises"
     );
-    nodePath = requireFn(
-      /* webpackIgnore: true */
-      "node:path"
-    );
-    nodeOs = requireFn(
-      /* webpackIgnore: true */
-      "node:os"
-    );
-    nodeChildProcess = requireFn(
-      /* webpackIgnore: true */
+    nodePath = loadNodeBuiltin("node:path");
+    nodeOs = loadNodeBuiltin("node:os");
+    nodeChildProcess = loadNodeBuiltin(
       "node:child_process"
     );
-    nodeStream = requireFn(
-      /* webpackIgnore: true */
+    nodeStream = loadNodeBuiltin(
       "node:stream/promises"
     );
-    nodeUrl = requireFn(
-      /* webpackIgnore: true */
-      "node:url"
-    );
+    nodeUrl = loadNodeBuiltin("node:url");
     hasImportedDependencies = true;
   } catch (err) {
     console.warn(
@@ -1396,6 +1419,9 @@ function asNativeWebSocket(handle) {
 function asNativeSqlTransaction(handle) {
   return handle;
 }
+function asNativeActorStateTransaction(handle) {
+  return handle;
+}
 function asNativeCancellationToken(handle) {
   return handle;
 }
@@ -1619,6 +1645,9 @@ var NapiCoreRuntime = (_class = class {
   actorSetAlarm(ctx, timestampMs) {
     asNativeActorContext(ctx).setAlarm(timestampMs);
   }
+  async actorSetRunWakeAt(ctx, timestampMs) {
+    await asNativeActorContext(ctx).setRunWakeAt(timestampMs);
+  }
   actorRequestSave(ctx, opts) {
     asNativeActorContext(ctx).requestSave(opts);
   }
@@ -1779,9 +1808,10 @@ var NapiCoreRuntime = (_class = class {
     );
     return results.map(normalizeRuntimeSqlExecuteResult);
   }
-  async actorSqlBeginTransaction(ctx, timeoutMs) {
+  async actorSqlBeginTransaction(ctx, timeoutMs, name) {
     return await this.#actorSql(ctx).beginTransaction(
-      timeoutMs
+      timeoutMs,
+      name
     );
   }
   async actorSqlTransactionExec(transaction, sql) {
@@ -1800,6 +1830,26 @@ var NapiCoreRuntime = (_class = class {
   async actorSqlTransactionRollback(transaction) {
     await asNativeSqlTransaction(transaction).rollback();
   }
+  async actorBeginStateTransaction(ctx, timeoutMs) {
+    return await asNativeActorContext(ctx).beginStateTransaction(
+      timeoutMs
+    );
+  }
+  async actorStateTransactionExecute(transaction, sql, params) {
+    const result = await asNativeActorStateTransaction(transaction).execute(
+      sql,
+      toNapiSqlBindParams(params)
+    );
+    return normalizeRuntimeSqlExecuteResult(result);
+  }
+  async actorStateTransactionCommit(transaction, payload) {
+    await asNativeActorStateTransaction(transaction).commit(
+      toNapiStateDeltaPayload(payload)
+    );
+  }
+  async actorStateTransactionRollback(transaction) {
+    await asNativeActorStateTransaction(transaction).rollback();
+  }
   async actorSqlQuery(ctx, sql, params) {
     return await this.#actorSql(ctx).query(
       sql,
@@ -1857,6 +1907,13 @@ var NapiCoreRuntime = (_class = class {
       signal ? asNativeCancellationToken(signal) : signal
     );
   }
+  async actorQueueCompletePersisted(ctx, messageId, expectedName, response) {
+    return await asNativeActorContext(ctx).queue().completePersisted(
+      messageId,
+      expectedName,
+      response == null ? response : toNapiBuffer(response)
+    );
+  }
   async actorQueueEnqueueAndWait(ctx, name, body, options, signal) {
     return await asNativeActorContext(ctx).queue().enqueueAndWait(
       name,
@@ -2046,7 +2103,167 @@ function validationError(target, issues) {
   });
 }
 
+// src/registry/run-handler-coordinator.ts
+function runUnavailable(actorId, reason) {
+  return new (0, _chunk4FC7TVS6cjs.RivetError)(
+    "actor",
+    "run_handler_unavailable",
+    `Run handler control is unavailable for actor ${actorId}: ${reason}.`,
+    {
+      public: true,
+      statusCode: 409,
+      metadata: { actorId, reason }
+    }
+  );
+}
+function notifyStateChanged(state) {
+  const waiters = [...state.waiters];
+  state.waiters.clear();
+  for (const waiter of waiters) {
+    waiter();
+  }
+}
+function waitForStateChange(state) {
+  return new Promise((resolve) => {
+    state.waiters.add(resolve);
+  });
+}
+var RunHandlerCoordinator = class {
+  #run;
+  #states = /* @__PURE__ */ new Map();
+  constructor(run) {
+    this.#run = run;
+  }
+  get inspectorKind() {
+    return _chunk4KGXFC6Mcjs.getRunInspectorKind.call(void 0, this.#run);
+  }
+  async run(actorId, restart, callback) {
+    const state = this.#getOrCreate(actorId);
+    state.restart = restart;
+    this.#initializeInspector(actorId, state);
+    state.queued += 1;
+    const blockedGeneration = state.exclusive ? state.exclusiveGeneration : void 0;
+    try {
+      while (!state.closed && (state.exclusive || state.active)) {
+        await waitForStateChange(state);
+      }
+    } finally {
+      state.queued -= 1;
+    }
+    if (state.closed) {
+      return "closed";
+    }
+    if (blockedGeneration !== void 0 && state.exclusiveGeneration === blockedGeneration && state.exclusiveOutcome === "failure") {
+      return "suppressed";
+    }
+    state.active = true;
+    try {
+      await callback();
+      return "ran";
+    } finally {
+      state.active = false;
+      notifyStateChanged(state);
+    }
+  }
+  getInspector(actorId, restart) {
+    var _a;
+    const state = this.#getOrCreate(actorId);
+    state.restart = restart;
+    this.#initializeInspector(actorId, state);
+    return (_a = state.inspector) == null ? void 0 : _a.inspector;
+  }
+  destroy(actorId) {
+    var _a, _b;
+    const state = this.#states.get(actorId);
+    if (!state) return;
+    state.closed = true;
+    (_b = (_a = state.inspector) == null ? void 0 : _a.dispose) == null ? void 0 : _b.call(_a);
+    state.inspector = void 0;
+    notifyStateChanged(state);
+    this.#states.delete(actorId);
+  }
+  #control(actorId, state) {
+    return {
+      run: {
+        withInactive: async (options, callback) => {
+          if (state.closed) {
+            throw runUnavailable(actorId, "actor is destroyed");
+          }
+          if (state.active || state.queued > 0 || state.exclusive) {
+            throw runUnavailable(
+              actorId,
+              "the run handler is active or waiting to start"
+            );
+          }
+          state.exclusive = true;
+          state.exclusiveGeneration += 1;
+          state.exclusiveOutcome = void 0;
+          try {
+            const result = await callback();
+            if (state.closed) {
+              throw runUnavailable(actorId, "actor is destroyed");
+            }
+            state.exclusiveOutcome = "success";
+            if (options.restartOnSuccess && state.queued === 0) {
+              if (!state.restart) {
+                throw runUnavailable(
+                  actorId,
+                  "the runtime restart callback is not available"
+                );
+              }
+              await state.restart();
+            }
+            return result;
+          } catch (error) {
+            state.exclusiveOutcome = "failure";
+            throw error;
+          } finally {
+            state.exclusive = false;
+            notifyStateChanged(state);
+          }
+        }
+      }
+    };
+  }
+  #getOrCreate(actorId) {
+    let state = this.#states.get(actorId);
+    if (!state) {
+      state = {
+        active: false,
+        closed: false,
+        exclusive: false,
+        exclusiveGeneration: 0,
+        inspectorInitialized: false,
+        queued: 0,
+        waiters: /* @__PURE__ */ new Set()
+      };
+      this.#states.set(actorId, state);
+    }
+    return state;
+  }
+  #initializeInspector(actorId, state) {
+    if (state.inspectorInitialized) return;
+    const inspector = _chunk4KGXFC6Mcjs.createRunInspector.call(void 0, this.#run, {
+      actorId,
+      control: this.#control(actorId, state)
+    });
+    if (this.inspectorKind === "workflow") {
+      const workflow = inspector === void 0 ? void 0 : inspector.inspector.workflow;
+      if (!workflow || typeof workflow.getHistory !== "function" || typeof workflow.getState !== "function" || typeof workflow.onHistoryUpdated !== "function" || typeof workflow.replayFromStep !== "function") {
+        throw new TypeError(
+          `defineRunHandler createInspector returned an invalid workflow adapter for actor ${actorId}`
+        );
+      }
+    }
+    state.inspector = inspector;
+    state.inspectorInitialized = true;
+  }
+};
+
 // src/registry/wasm-runtime.ts
+function asWasmConstructors(bindings) {
+  return bindings;
+}
 function asWasmRegistry(handle) {
   return handle;
 }
@@ -2171,10 +2388,10 @@ function childHandle(handle, name) {
 }
 var WasmCoreRuntime = (_class2 = class {
   __init2() {this.kind = "wasm"}
-  #bindings;
+  #constructors;
   #sql = /* @__PURE__ */ new WeakMap();
   constructor(bindings) {;_class2.prototype.__init2.call(this);
-    this.#bindings = bindings;
+    this.#constructors = asWasmConstructors(bindings);
   }
   #actorSql(ctx) {
     const wasmCtx = asWasmActorContext(ctx);
@@ -2187,7 +2404,7 @@ var WasmCoreRuntime = (_class2 = class {
   }
   createRegistry() {
     return callWasmSync(
-      () => asRegistryHandle2(new this.#bindings.CoreRegistry())
+      () => asRegistryHandle2(new this.#constructors.CoreRegistry())
     );
   }
   registerActor(registry, name, factory) {
@@ -2242,13 +2459,13 @@ var WasmCoreRuntime = (_class2 = class {
   createActorFactory(callbacks, config) {
     return callWasmSync(
       () => asActorFactoryHandle2(
-        new this.#bindings.ActorFactory(callbacks, config)
+        new this.#constructors.ActorFactory(callbacks, config)
       )
     );
   }
   createCancellationToken() {
     return callWasmSync(
-      () => new this.#bindings.CancellationToken()
+      () => new this.#constructors.CancellationToken()
     );
   }
   cancellationTokenAborted(token) {
@@ -2283,6 +2500,13 @@ var WasmCoreRuntime = (_class2 = class {
       optionalWasmNumber(timestampMs)
     );
   }
+  async actorSetRunWakeAt(ctx, timestampMs) {
+    await callHandleAsync(
+      asWasmActorContext(ctx),
+      "setRunWakeAt",
+      optionalWasmNumber(timestampMs)
+    );
+  }
   actorRequestSave(ctx, opts) {
     callHandle(asWasmActorContext(ctx), "requestSave", opts);
   }
@@ -2497,9 +2721,9 @@ var WasmCoreRuntime = (_class2 = class {
     );
     return results.map(normalizeRuntimeSqlExecuteResult);
   }
-  async actorSqlBeginTransaction(ctx, timeoutMs) {
+  async actorSqlBeginTransaction(ctx, timeoutMs, name) {
     return await callWasm(
-      () => this.#actorSql(ctx).beginTransaction(timeoutMs)
+      () => this.#actorSql(ctx).beginTransaction(timeoutMs, name)
     );
   }
   async actorSqlTransactionExec(transaction, sql) {
@@ -2523,6 +2747,27 @@ var WasmCoreRuntime = (_class2 = class {
       () => transaction.rollback()
     );
   }
+  async actorBeginStateTransaction(ctx, timeoutMs) {
+    return await callWasm(
+      () => asWasmActorContext(ctx).beginStateTransaction(timeoutMs)
+    );
+  }
+  async actorStateTransactionExecute(transaction, sql, params) {
+    const result = await callWasm(
+      () => transaction.execute(sql, params)
+    );
+    return normalizeRuntimeSqlExecuteResult(result);
+  }
+  async actorStateTransactionCommit(transaction, payload) {
+    await callWasm(
+      () => transaction.commit(payload)
+    );
+  }
+  async actorStateTransactionRollback(transaction) {
+    await callWasm(
+      () => transaction.rollback()
+    );
+  }
   async actorSqlQuery(ctx, sql, params) {
     return await callWasm(() => this.#actorSql(ctx).query(sql, params));
   }
@@ -2595,6 +2840,16 @@ var WasmCoreRuntime = (_class2 = class {
       signal
     );
   }
+  async actorQueueCompletePersisted(ctx, messageId, expectedName, response) {
+    const queue2 = childHandle(asWasmActorContext(ctx), "queue");
+    return await callHandleAsync(
+      queue2,
+      "completePersisted",
+      messageId,
+      expectedName,
+      response
+    );
+  }
   async actorQueueEnqueueAndWait(ctx, name, body, options, signal) {
     const queue2 = childHandle(asWasmActorContext(ctx), "queue");
     return optionalBytes(
@@ -3018,6 +3273,14 @@ function stateMutationReentrantError() {
     "State mutations are not allowed inside onStateChange."
   );
 }
+function stateTransactionConflictError() {
+  return new (0, _chunk4FC7TVS6cjs.RivetError)(
+    "actor",
+    "state_transaction_conflict",
+    "State cannot be mutated by another action while an experimental state-aware database transaction is active.",
+    { public: true, statusCode: 409 }
+  );
+}
 function databaseNotConfiguredError() {
   return new (0, _chunk4FC7TVS6cjs.RivetError)(
     "actor",
@@ -3034,6 +3297,14 @@ function databaseClientNotReadyError() {
     { public: true }
   );
 }
+function runHandlerNotConfiguredError() {
+  return new (0, _chunk4FC7TVS6cjs.RivetError)(
+    "actor",
+    "run_handler_unavailable",
+    "Cannot schedule a run wake because this actor does not define a run handler.",
+    { public: true, statusCode: 409 }
+  );
+}
 function stateNotEnabledError() {
   return new (0, _chunk4FC7TVS6cjs.RivetError)(
     "actor",
@@ -3096,7 +3367,7 @@ async function cleanupNativeSleepRuntimeState(runtime, ctx, afterTrackedWorkDrai
     }).catch((error) => {
       logger2().warn({
         msg: "deferred native sleep cleanup failed",
-        error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
+        error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, error)
       });
     });
     return;
@@ -3136,10 +3407,11 @@ function getOrCreateNativeSqlDatabase(runtime, ctx) {
     exec: (sql) => runtime.actorSqlExec(ctx, sql),
     execute: (sql, params) => runtime.actorSqlExecute(ctx, sql, params),
     executeBatch: (statements) => runtime.actorSqlExecuteBatch(ctx, statements),
-    beginTransaction: async (timeoutMs) => {
+    beginTransaction: async (timeoutMs, name) => {
       const transaction = await runtime.actorSqlBeginTransaction(
         ctx,
-        timeoutMs
+        timeoutMs,
+        name
       );
       return {
         exec: (sql) => runtime.actorSqlTransactionExec(transaction, sql),
@@ -3152,6 +3424,38 @@ function getOrCreateNativeSqlDatabase(runtime, ctx) {
         rollback: () => runtime.actorSqlTransactionRollback(transaction)
       };
     },
+    beginStateTransaction: async (timeoutMs, context) => {
+      const scope = context;
+      if (!scope) {
+        throw new Error(
+          "actor state transaction context is not configured"
+        );
+      }
+      const transaction = await runtime.actorBeginStateTransaction(
+        ctx,
+        timeoutMs
+      );
+      return {
+        exec: async () => {
+          throw new Error(
+            "actor state transactions only support single-statement execute calls"
+          );
+        },
+        execute: (sql, params) => runtime.actorStateTransactionExecute(
+          transaction,
+          sql,
+          params
+        ),
+        commit: async () => {
+          await runtime.actorStateTransactionCommit(
+            transaction,
+            scope.actorContext.serializeForTick("save")
+          );
+          scope.committed = true;
+        },
+        rollback: () => runtime.actorStateTransactionRollback(transaction)
+      };
+    },
     query: (sql, params) => runtime.actorSqlQuery(ctx, sql, params),
     run: (sql, params) => runtime.actorSqlRun(ctx, sql, params),
     metrics: () => runtime.actorSqlMetrics(ctx),
@@ -3223,10 +3527,10 @@ function decodeValue(value) {
   if (!value || value.length === 0) {
     return void 0;
   }
-  return _chunkT56IZ42Mcjs.decodeCborCompat.call(void 0, value);
+  return _chunkGU7ML3KEcjs.decodeCborCompat.call(void 0, value);
 }
 function encodeValue(value) {
-  return _chunkT56IZ42Mcjs.encodeCborCompat.call(void 0, value);
+  return _chunkGU7ML3KEcjs.encodeCborCompat.call(void 0, value);
 }
 function normalizeArgs(value) {
   return Array.isArray(value) ? value : value === void 0 || value === null ? [] : [value];
@@ -3262,7 +3566,7 @@ function isStructuredBridgeError(error) {
   return _chunk4FC7TVS6cjs.isRivetErrorLike.call(void 0, error) && "__type" in error && (error.__type === "RivetError" || error.__type === "ActorError");
 }
 function encodeNativeCallbackError(error) {
-  const structuredError = isStructuredBridgeError(error) ? error : _chunkT56IZ42Mcjs.deconstructError.call(void 0, error, true);
+  const structuredError = isStructuredBridgeError(error) ? error : _chunkGU7ML3KEcjs.deconstructError.call(void 0, error, true);
   const bridgeError = new Error(_chunk4FC7TVS6cjs.encodeBridgeRivetError.call(void 0, structuredError), {
     cause: error instanceof Error ? error : void 0
   });
@@ -3312,7 +3616,7 @@ function decodeWorkflowCbor(data) {
     return null;
   }
   try {
-    return _chunkT56IZ42Mcjs.decodeCborCompat.call(void 0, new Uint8Array(data));
+    return _chunkGU7ML3KEcjs.decodeCborCompat.call(void 0, new Uint8Array(data));
   } catch (e5) {
     return null;
   }
@@ -3425,7 +3729,7 @@ function serializeWorkflowHistoryForJson(data) {
   if (data === null) {
     return null;
   }
-  const history = _chunkIMRDHZM2cjs.decodeWorkflowHistoryTransport.call(void 0, data);
+  const history = _chunkVBJQIBXIcjs.decodeWorkflowHistoryTransport.call(void 0, data);
   return jsonSafe({
     nameRegistry: [...history.nameRegistry],
     entries: history.entries.map((entry) => ({
@@ -3532,13 +3836,17 @@ var NativeConnAdapter = class {
   #schemas;
   #ctx;
   #queueHibernationRemoval;
-  constructor(runtime, conn, schemas = {}, ctx, queueHibernationRemoval) {
+  #assertCanMutateState;
+  #stateProxy;
+  #stateProxyTarget;
+  constructor(runtime, conn, schemas = {}, ctx, queueHibernationRemoval, assertCanMutateState) {
     this.#runtime = runtime;
     this.#conn = conn;
     this.#schemas = schemas;
     this.#ctx = ctx;
     this.#queueHibernationRemoval = queueHibernationRemoval;
-    this[_chunkHNHLBM3Ycjs.CONN_STATE_MANAGER_SYMBOL] = {
+    this.#assertCanMutateState = assertCanMutateState;
+    this[_chunk4KGXFC6Mcjs.CONN_STATE_MANAGER_SYMBOL] = {
       stateEnabled: true,
       get state() {
         return thisConn.state;
@@ -3555,7 +3863,7 @@ var NativeConnAdapter = class {
       decodeValue(this.#runtime.connParams(this.#conn))
     );
   }
-  [_chunkHNHLBM3Ycjs.RAW_STATE_SYMBOL]() {
+  [_chunk4KGXFC6Mcjs.RAW_STATE_SYMBOL]() {
     return this.#readState();
   }
   get state() {
@@ -3563,31 +3871,37 @@ var NativeConnAdapter = class {
     if (!this.#ctx) {
       return this.#createStateProxy(nextState);
     }
-    const connState = getNativeConnPersistState(
-      this.#runtime,
-      this.#ctx,
-      this.#conn
-    );
-    if (connState.stateProxy === void 0 || connState.stateProxyTarget !== nextState) {
-      connState.stateProxyTarget = nextState;
-      connState.stateProxy = this.#createStateProxy(nextState);
+    if (this.#stateProxy === void 0 || this.#stateProxyTarget !== nextState) {
+      this.#stateProxyTarget = nextState;
+      this.#stateProxy = this.#createStateProxy(nextState);
     }
-    return connState.stateProxy;
+    return this.#stateProxy;
   }
   #createStateProxy(state) {
+    const assertCurrentState = () => {
+      var _a;
+      (_a = this.#assertCanMutateState) == null ? void 0 : _a.call(this);
+      if (this.#ctx && this.#readState() !== state) {
+        throw stateTransactionConflictError();
+      }
+    };
     return createWriteThroughProxy(
       state,
       (nextValue) => {
+        assertCurrentState();
         this.#writeState(nextValue, { writeNative: true });
       },
       (newValue) => {
-        _chunkT56IZ42Mcjs.assertJsonCompatValue.call(void 0, newValue);
+        assertCurrentState();
+        _chunkGU7ML3KEcjs.assertJsonCompatValue.call(void 0, newValue);
       }
     );
   }
   set state(value) {
+    var _a;
+    (_a = this.#assertCanMutateState) == null ? void 0 : _a.call(this);
     const nextValue = unwrapWriteThroughProxy(value);
-    _chunkT56IZ42Mcjs.assertJsonCompatValue.call(void 0, nextValue);
+    _chunkGU7ML3KEcjs.assertJsonCompatValue.call(void 0, nextValue);
     this.#writeState(nextValue, { writeNative: true });
   }
   initializeState(value) {
@@ -3911,8 +4225,9 @@ var NativeKvAdapter = class {
 };
 function wrapQueueMessage(message, schemas) {
   const name = callNativeSync(() => message.name());
+  const id = callNativeSync(() => message.id());
   return {
-    id: Number(callNativeSync(() => message.id())),
+    id: id <= BigInt(Number.MAX_SAFE_INTEGER) ? Number(id) : id,
     name,
     body: validateQueueBody(
       schemas,
@@ -4050,6 +4365,25 @@ var NativeQueueAdapter = class {
       cleanup == null ? void 0 : cleanup();
     }
   }
+  async waitForAvailable(names, options) {
+    await this.waitForNamesAvailable(_nullishCoalesce(names, () => ( [])), options);
+  }
+  async complete(message, response) {
+    const validatedResponse = validateQueueComplete(
+      this.#schemas,
+      message.name,
+      response
+    );
+    await callNative(
+      () => this.#runtime.actorQueueCompletePersisted(
+        this.#ctx,
+        BigInt(message.id),
+        message.name,
+        encodeValue(validatedResponse)
+      )
+    );
+    this.#pendingCompletableMessageIds.delete(message.id.toString());
+  }
   async enqueueAndWait(name, body, options) {
     const validatedBody = validateQueueBody(this.#schemas, name, body);
     const { token, cleanup } = await createCancellationTokenHandle(
@@ -4486,10 +4820,12 @@ var NativeConnectionMap = (_class3 = class {
   #runtime;
   #ctx;
   #schemas;
-  constructor(runtime, ctx, schemas) {;_class3.prototype.__init3.call(this);
+  #assertCanMutateState;
+  constructor(runtime, ctx, schemas, assertCanMutateState) {;_class3.prototype.__init3.call(this);
     this.#runtime = runtime;
     this.#ctx = ctx;
     this.#schemas = schemas;
+    this.#assertCanMutateState = assertCanMutateState;
   }
   #connToAdapter(conn) {
     return new NativeConnAdapter(
@@ -4502,7 +4838,8 @@ var NativeConnectionMap = (_class3 = class {
           this.#ctx,
           connId
         )
-      )
+      ),
+      this.#assertCanMutateState
     );
   }
   get size() {
@@ -4561,24 +4898,27 @@ var ActorContextHandleAdapter = class {
   #queue;
   #request;
   #schedule;
-  #sql;
-  #runHandlerActiveProvider;
+  #run;
+  #runHandlerConfigured;
   #onStateChange;
   #stateEnabled;
-  constructor(runtime, ctx, clientFactory, schemas = {}, databaseProvider, request, stateEnabled = true, runHandlerActiveProvider, onStateChange, dispatchCancelToken) {
+  #stateProxy;
+  #stateProxyTarget;
+  #stateTransactionOwner = /* @__PURE__ */ Symbol("actor-state-transaction-owner");
+  constructor(runtime, ctx, clientFactory, schemas = {}, databaseProvider, request, stateEnabled = true, onStateChange, dispatchCancelToken, runHandlerConfigured = false) {
     this.#runtime = runtime;
     this.#ctx = ctx;
     this.#clientFactory = clientFactory;
     this.#schemas = schemas;
     this.#dispatchCancelToken = dispatchCancelToken;
-    this.#runHandlerActiveProvider = runHandlerActiveProvider;
+    this.#runHandlerConfigured = runHandlerConfigured;
     this.#onStateChange = onStateChange;
     this.#stateEnabled = stateEnabled;
     if (databaseProvider) {
       this.#databaseProvider = databaseProvider;
     }
     this.#request = request;
-    this[_chunkHNHLBM3Ycjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
+    this[_chunk4KGXFC6Mcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL] = new NativeWorkflowRuntimeAdapter(
       this
     );
   }
@@ -4588,12 +4928,6 @@ var ActorContextHandleAdapter = class {
     }
     return this.#kv;
   }
-  get sql() {
-    if (!this.#sql) {
-      this.#sql = getOrCreateNativeSqlDatabase(this.#runtime, this.#ctx);
-    }
-    return this.#sql;
-  }
   async actorRuntimeSocket() {
     return await callNative(
       () => this.#runtime.actorRuntimeSocketProvision(this.#ctx)
@@ -4609,12 +4943,12 @@ var ActorContextHandleAdapter = class {
     const runtimeState = getNativeRuntimeState(this.#runtime, this.#ctx);
     const cachedClient = runtimeState.databaseClient;
     if (cachedClient) {
-      this.#db = cachedClient.client;
+      this.#db = this.#bindDatabaseClient(cachedClient.client);
       return this.#db;
     }
     throw databaseClientNotReadyError();
   }
-  [_chunkHNHLBM3Ycjs.RAW_STATE_SYMBOL]() {
+  [_chunk4KGXFC6Mcjs.RAW_STATE_SYMBOL]() {
     if (!this.#stateEnabled) {
       throw stateNotEnabledError();
     }
@@ -4624,22 +4958,28 @@ var ActorContextHandleAdapter = class {
     if (!this.#stateEnabled) {
       throw stateNotEnabledError();
     }
-    const actorState = getNativePersistState(this.#runtime, this.#ctx);
     const nextState = this.#readState();
-    if (actorState.stateProxy === void 0 || actorState.stateProxyTarget !== nextState) {
-      actorState.stateProxyTarget = nextState;
-      actorState.stateProxy = createWriteThroughProxy(
+    if (this.#stateProxy === void 0 || this.#stateProxyTarget !== nextState) {
+      const assertCurrentState = () => {
+        this.#assertCanMutateState();
+        if (this.#readState() !== nextState) {
+          throw stateTransactionConflictError();
+        }
+      };
+      this.#stateProxyTarget = nextState;
+      this.#stateProxy = createWriteThroughProxy(
         nextState,
         (nextValue) => {
+          assertCurrentState();
           this.#writeState(nextValue, { scheduleSave: true });
         },
         (newValue) => {
-          this.#assertCanMutateState();
-          _chunkT56IZ42Mcjs.assertJsonCompatValue.call(void 0, newValue);
+          assertCurrentState();
+          _chunkGU7ML3KEcjs.assertJsonCompatValue.call(void 0, newValue);
         }
       );
     }
-    return actorState.stateProxy;
+    return this.#stateProxy;
   }
   set state(value) {
     if (!this.#stateEnabled) {
@@ -4647,7 +4987,7 @@ var ActorContextHandleAdapter = class {
     }
     this.#assertCanMutateState();
     const nextValue = unwrapWriteThroughProxy(value);
-    _chunkT56IZ42Mcjs.assertJsonCompatValue.call(void 0, nextValue);
+    _chunkGU7ML3KEcjs.assertJsonCompatValue.call(void 0, nextValue);
     this.#writeState(nextValue, { scheduleSave: true });
   }
   initializeState(value) {
@@ -4682,6 +5022,26 @@ var ActorContextHandleAdapter = class {
     }
     return this.#queue;
   }
+  get run() {
+    if (!this.#run) {
+      this.#run = {
+        setWakeAt: async (timestamp) => {
+          if (timestamp !== null && !this.#runHandlerConfigured) {
+            throw runHandlerNotConfiguredError();
+          }
+          if (timestamp !== null && (!Number.isSafeInteger(timestamp) || timestamp < 0)) {
+            throw new TypeError(
+              "Run wake timestamp must be a non-negative safe integer or null"
+            );
+          }
+          await callNative(
+            () => this.#runtime.actorSetRunWakeAt(this.#ctx, timestamp)
+          );
+        }
+      };
+    }
+    return this.#run;
+  }
   get schedule() {
     if (!this.#schedule) {
       this.#schedule = new NativeScheduleAdapter(
@@ -4716,7 +5076,8 @@ var ActorContextHandleAdapter = class {
       this.#connMap = new NativeConnectionMap(
         this.#runtime,
         this.#ctx,
-        this.#schemas
+        this.#schemas,
+        () => this.#assertCanMutateState()
       );
     }
     return this.#connMap;
@@ -4785,11 +5146,11 @@ var ActorContextHandleAdapter = class {
     const runtimeState = getNativeRuntimeState(this.#runtime, this.#ctx);
     const cachedClient = runtimeState.databaseClient;
     if (cachedClient) {
-      this.#db = cachedClient.client;
+      this.#db = this.#bindDatabaseClient(cachedClient.client);
       return this.#db;
     }
     const actorId = this.actorId;
-    const client = await this.#databaseProvider.createClient({
+    const createdClient = await this.#databaseProvider.createClient({
       actorId,
       kv: {
         batchPut: async (entries) => {
@@ -4810,21 +5171,126 @@ var ActorContextHandleAdapter = class {
       log: {
         debug: (obj) => logger2().debug(obj)
       },
-      nativeDatabaseProvider: {
-        open: async (requestedActorId) => {
-          void requestedActorId;
-          return getOrCreateNativeSqlDatabase(
-            this.#runtime,
-            this.#ctx
-          );
-        }
-      }
+      nativeDatabaseProvider: _chunkLWQFXGKScjs.registerNativeStateTransactionOpener.call(void 0, 
+        {
+          open: async (requestedActorId) => {
+            void requestedActorId;
+            return getOrCreateNativeSqlDatabase(
+              this.#runtime,
+              this.#ctx
+            );
+          }
+        },
+        async (timeoutMs, context) => await getOrCreateNativeSqlDatabase(
+          this.#runtime,
+          this.#ctx
+        ).beginStateTransaction(timeoutMs, context)
+      )
     });
     runtimeState.databaseClient = {
-      client
+      client: createdClient
     };
-    this.#db = client;
-    return client;
+    this.#db = this.#bindDatabaseClient(createdClient);
+    return this.#db;
+  }
+  #bindDatabaseClient(client) {
+    return _chunkLWQFXGKScjs.bindNativeStateTransactionContext.call(void 0, client, {
+      enter: async () => await this.#enterStateTransaction(),
+      exit: (scope) => this.#exitStateTransaction(
+        scope
+      )
+    });
+  }
+  async #enterStateTransaction() {
+    const actorState = getNativePersistState(this.#runtime, this.#ctx);
+    let pendingOwners = actorState.pendingStateTransactionOwners;
+    if (!pendingOwners) {
+      pendingOwners = /* @__PURE__ */ new Set();
+      actorState.pendingStateTransactionOwners = pendingOwners;
+    }
+    if (actorState.activeStateTransactionOwner === this.#stateTransactionOwner || pendingOwners.has(this.#stateTransactionOwner)) {
+      throw stateTransactionConflictError();
+    }
+    pendingOwners.add(this.#stateTransactionOwner);
+    const predecessor = _nullishCoalesce(actorState.stateTransactionTail, () => ( Promise.resolve()));
+    let release;
+    const hold = new Promise((resolve) => {
+      release = resolve;
+    });
+    actorState.stateTransactionTail = predecessor.then(
+      async () => await hold
+    );
+    try {
+      await predecessor;
+      pendingOwners.delete(this.#stateTransactionOwner);
+      actorState.activeStateTransactionOwner = this.#stateTransactionOwner;
+      return {
+        actorContext: this,
+        actorStateBaseline: this.#stateEnabled ? structuredClone(this.#readState()) : void 0,
+        connectionStateBaselines: new Map(
+          callNativeSync(
+            () => this.#runtime.actorConns(this.#ctx)
+          ).map((conn) => [
+            callNativeSync(() => this.#runtime.connId(conn)),
+            new Uint8Array(
+              callNativeSync(() => this.#runtime.connState(conn))
+            )
+          ])
+        ),
+        committed: false,
+        release
+      };
+    } catch (error) {
+      pendingOwners.delete(this.#stateTransactionOwner);
+      if (actorState.activeStateTransactionOwner === this.#stateTransactionOwner) {
+        actorState.activeStateTransactionOwner = void 0;
+      }
+      release();
+      throw error;
+    }
+  }
+  #exitStateTransaction(scope) {
+    const actorState = getNativePersistState(this.#runtime, this.#ctx);
+    try {
+      if (!scope.committed) {
+        this.#restoreStateTransactionBaseline(scope);
+      }
+    } finally {
+      if (actorState.activeStateTransactionOwner === this.#stateTransactionOwner) {
+        actorState.activeStateTransactionOwner = void 0;
+      }
+      if (actorState.stateTransactionSaveDeferred) {
+        actorState.stateTransactionSaveDeferred = false;
+        this.#scheduleSave();
+      }
+      scope.release();
+    }
+  }
+  #restoreStateTransactionBaseline(scope) {
+    const actorState = getNativePersistState(this.#runtime, this.#ctx);
+    if (actorState.pendingSaveHandle !== void 0) {
+      clearImmediate(actorState.pendingSaveHandle);
+    }
+    actorState.pendingSaveHandle = void 0;
+    actorState.saveScheduled = false;
+    if (this.#stateEnabled) {
+      actorState.state = structuredClone(scope.actorStateBaseline);
+    }
+    for (const conn of callNativeSync(
+      () => this.#runtime.actorConns(this.#ctx)
+    )) {
+      const connId = callNativeSync(() => this.#runtime.connId(conn));
+      const baseline = scope.connectionStateBaselines.get(connId);
+      if (baseline === void 0) {
+        continue;
+      }
+      callNativeSync(() => this.#runtime.connSetState(conn, baseline));
+      const cached = actorState.connStates.get(connId);
+      if (cached) {
+        cached.state = decodeValue(baseline);
+      }
+    }
+    this.#scheduleSave();
   }
   async prepare() {
     if (!this.#databaseProvider) {
@@ -4842,7 +5308,6 @@ var ActorContextHandleAdapter = class {
   }
   async closeDatabase() {
     this.#db = void 0;
-    this.#sql = void 0;
     await closeNativeDatabaseClient(this.#runtime, this.#ctx);
     await closeNativeSqlDatabase(this.#runtime, this.#ctx);
   }
@@ -4862,6 +5327,7 @@ var ActorContextHandleAdapter = class {
   }
   async saveState(opts) {
     if (opts == null ? void 0 : opts.immediate) {
+      this.#assertCanAwaitStateSave();
       await callNative(
         () => this.#runtime.actorRequestSaveAndWait(this.#ctx, {
           immediate: true
@@ -4882,10 +5348,17 @@ var ActorContextHandleAdapter = class {
     );
   }
   async saveStateAndWorkflowBatch(writes) {
+    this.#assertCanAwaitStateSave();
     await callNative(
       () => this.#runtime.actorSaveStateAndWorkflowBatch(this.#ctx, writes)
     );
   }
+  #assertCanAwaitStateSave() {
+    const actorState = getNativePersistState(this.#runtime, this.#ctx);
+    if (actorState.activeStateTransactionOwner === this.#stateTransactionOwner) {
+      throw stateTransactionConflictError();
+    }
+  }
   serializeForTick(reason) {
     void reason;
     const actorState = getNativePersistState(this.#runtime, this.#ctx);
@@ -4925,7 +5398,7 @@ var ActorContextHandleAdapter = class {
     const trackedPromise = Promise.resolve(promise).catch((error) => {
       logger2().warn({
         msg: "keepAwake promise rejected",
-        error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
+        error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, error)
       });
     }).then(() => null);
     try {
@@ -4939,10 +5412,6 @@ var ActorContextHandleAdapter = class {
     }
     return promise;
   }
-  runHandlerActive() {
-    var _a;
-    return _nullishCoalesce(((_a = this.#runHandlerActiveProvider) == null ? void 0 : _a.call(this)), () => ( false));
-  }
   internalKeepAwake(run) {
     const promise = typeof run === "function" ? run() : run;
     const trackedPromise = promise.then(
@@ -5024,7 +5493,6 @@ var ActorContextHandleAdapter = class {
     var _a;
     this.#flushStateChange();
     (_a = this.#abortSignalCleanup) == null ? void 0 : _a.call(this);
-    this.#sql = void 0;
   }
   #createActorAbortSignal() {
     const nativeSignal = callNativeSync(
@@ -5058,10 +5526,18 @@ var ActorContextHandleAdapter = class {
     this.#scheduleSave();
   }
   #assertCanMutateState() {
+    var _a;
     const actorState = getNativePersistState(this.#runtime, this.#ctx);
     if (actorState.isInOnStateChange) {
       throw stateMutationReentrantError();
     }
+    if (actorState.activeStateTransactionOwner !== this.#stateTransactionOwner && (actorState.activeStateTransactionOwner !== void 0 || (_nullishCoalesce(((_a = actorState.pendingStateTransactionOwners) == null ? void 0 : _a.size), () => ( 0))) > 0)) {
+      throw stateTransactionConflictError();
+    }
+  }
+  /** @internal */
+  assertCanMutateState() {
+    this.#assertCanMutateState();
   }
   // Coalesce the request-save and onStateChange work to once per event loop
   // tick. A synchronous burst of mutations (for example
@@ -5083,6 +5559,15 @@ var ActorContextHandleAdapter = class {
     if (!actorState.saveScheduled) {
       return;
     }
+    if (actorState.activeStateTransactionOwner !== void 0) {
+      actorState.saveScheduled = false;
+      if (actorState.pendingSaveHandle !== void 0) {
+        clearImmediate(actorState.pendingSaveHandle);
+        actorState.pendingSaveHandle = void 0;
+      }
+      actorState.stateTransactionSaveDeferred = true;
+      return;
+    }
     actorState.saveScheduled = false;
     if (actorState.pendingSaveHandle !== void 0) {
       clearImmediate(actorState.pendingSaveHandle);
@@ -5206,9 +5691,6 @@ var NativeWorkflowRuntimeAdapter = class {
       }
     };
   }
-  isRunHandlerActive() {
-    return this.#ctx.runHandlerActive();
-  }
   async restartRunHandler() {
     await this.#ctx.restartRunHandler();
   }
@@ -5241,43 +5723,45 @@ var NativeWorkflowRuntimeAdapter = class {
     };
   }
 };
-function withConnContext(runtime, ctx, conn, clientFactory, schemas = {}, databaseProvider, request, stateEnabled = true, onStateChange, dispatchCancelToken) {
-  return Object.assign(
-    new ActorContextHandleAdapter(
+function withConnContext(runtime, ctx, conn, clientFactory, schemas = {}, databaseProvider, request, stateEnabled = true, onStateChange, dispatchCancelToken, runHandlerConfigured = false) {
+  const actorContext = new ActorContextHandleAdapter(
+    runtime,
+    ctx,
+    clientFactory,
+    schemas,
+    databaseProvider,
+    request,
+    stateEnabled,
+    onStateChange,
+    dispatchCancelToken,
+    runHandlerConfigured
+  );
+  return Object.assign(actorContext, {
+    conn: new NativeConnAdapter(
       runtime,
-      ctx,
-      clientFactory,
+      conn,
       schemas,
-      databaseProvider,
-      request,
-      stateEnabled,
-      void 0,
-      onStateChange,
-      dispatchCancelToken
-    ),
-    {
-      conn: new NativeConnAdapter(
-        runtime,
-        conn,
-        schemas,
-        ctx,
-        (connId) => callNativeSync(
-          () => runtime.actorQueueHibernationRemoval(ctx, connId)
-        )
-      )
-    }
-  );
+      ctx,
+      (connId) => callNativeSync(
+        () => runtime.actorQueueHibernationRemoval(ctx, connId)
+      ),
+      () => actorContext.assertCanMutateState()
+    )
+  });
 }
 function buildActorConfig(definition, registryConfig, runtimeKind) {
+  var _a;
   const config = definition.config;
   const options = _nullishCoalesce(config.options, () => ( {}));
   const canHibernate = options.canHibernateWebSocket;
   const usesRemoteSqlite = sqliteBackendForConfig(registryConfig) === "remote";
+  const sqliteProfiling = (_a = config.db) == null ? void 0 : _a.sqliteProfiling;
   return {
     name: options.name,
     icon: options.icon,
-    hasDatabase: config.db !== void 0 || usesRemoteSqlite,
+    hasDatabase: true,
     remoteSqlite: usesRemoteSqlite,
+    sqliteProfiling,
     enableActorRuntimeSocket: options.enableActorRuntimeSocket === true,
     hasState: config.state !== void 0 || typeof config.createState === "function",
     canHibernateWebsocket: typeof canHibernate === "boolean" ? canHibernate : void 0,
@@ -5300,7 +5784,7 @@ function buildActorConfig(definition, registryConfig, runtimeKind) {
     maxOutgoingMessageSize: registryConfig.maxOutgoingMessageSize,
     preloadMaxWorkflowBytes: options.preloadMaxWorkflowBytes,
     preloadMaxConnectionsBytes: options.preloadMaxConnectionsBytes,
-    actions: Object.keys(_chunkHNHLBM3Ycjs.flattenActionHandlers.call(void 0, config.actions)).sort().map((name) => ({ name })),
+    actions: Object.keys(_chunk4KGXFC6Mcjs.flattenActionHandlers.call(void 0, config.actions)).sort().map((name) => ({ name })),
     inspectorTabs: buildInspectorTabs(config.inspector, runtimeKind)
   };
 }
@@ -5376,10 +5860,10 @@ function validateInspectorTabSource(tabId, resolved) {
 function buildNativeFactory(runtime, registryConfig, definition) {
   var _a;
   const config = definition.config;
-  const databaseProvider = config.db;
-  const actionHandlers = _chunkHNHLBM3Ycjs.flattenActionHandlers.call(void 0, config.actions);
+  const databaseProvider = _nullishCoalesce(config.db, () => ( _chunkLWQFXGKScjs.db.call(void 0, )));
+  const actionHandlers = _chunk4KGXFC6Mcjs.flattenActionHandlers.call(void 0, config.actions);
   const schemaConfig = {
-    actionInputSchemas: _chunkHNHLBM3Ycjs.flattenActionInputSchemas.call(void 0, 
+    actionInputSchemas: _chunk4KGXFC6Mcjs.flattenActionInputSchemas.call(void 0, 
       config.actions,
       config.actionInputSchemas
     ),
@@ -5387,22 +5871,19 @@ function buildNativeFactory(runtime, registryConfig, definition) {
     events: config.events,
     queues: config.queues
   };
-  const createClient = () => _chunkBZWKY6WOcjs.createClientWithDriver.call(void 0, 
-    new (0, _chunkBZWKY6WOcjs.RemoteEngineControlClient)(
-      _chunkBZWKY6WOcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
+  const createClient = () => _chunkRXXDNJXBcjs.createClientWithDriver.call(void 0, 
+    new (0, _chunkRXXDNJXBcjs.RemoteEngineControlClient)(
+      _chunkRXXDNJXBcjs.convertRegistryConfigToClientConfig.call(void 0, registryConfig)
     ),
     { encoding: "bare" }
   );
-  const nativeRunHandlerActiveByActorId = /* @__PURE__ */ new Map();
-  const isNativeRunHandlerActive = (ctx) => _nullishCoalesce(nativeRunHandlerActiveByActorId.get(
-    callNativeSync(() => runtime.actorId(ctx))
-  ), () => ( false));
+  const run = _chunk4KGXFC6Mcjs.getRunFunction.call(void 0, config.run);
+  const runHandlerCoordinator = _chunk4KGXFC6Mcjs.getRunInspectorKind.call(void 0, config.run) === "workflow" ? new RunHandlerCoordinator(config.run) : void 0;
   const getNativeWorkflowInspector = (ctx) => {
-    var _a2;
-    return (_a2 = _chunkHNHLBM3Ycjs.getRunInspectorConfig.call(void 0, 
-      config.run,
-      callNativeSync(() => runtime.actorId(ctx))
-    )) == null ? void 0 : _a2.workflow;
+    var _a2, _b;
+    const actorId = callNativeSync(() => runtime.actorId(ctx));
+    const restart = () => callNativeSync(() => runtime.actorRestartRunHandler(ctx));
+    return _nullishCoalesce(((_a2 = runHandlerCoordinator == null ? void 0 : runHandlerCoordinator.getInspector(actorId, restart)) == null ? void 0 : _a2.workflow), () => ( ((_b = _chunk4KGXFC6Mcjs.getRunInspectorConfig.call(void 0, config.run, actorId)) == null ? void 0 : _b.workflow)));
   };
   const onStateChange = typeof config.onStateChange === "function" ? (actorCtx, nextState) => {
     config.onStateChange(actorCtx, nextState);
@@ -5422,9 +5903,9 @@ function buildNativeFactory(runtime, registryConfig, definition) {
     databaseProvider,
     request,
     stateEnabled,
-    () => isNativeRunHandlerActive(ctx),
     onStateChange,
-    cancelToken
+    cancelToken,
+    run !== void 0
   );
   const makeConnCtx = (ctx, conn, request, cancelToken) => withConnContext(
     runtime,
@@ -5436,7 +5917,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
     request,
     stateEnabled,
     onStateChange,
-    cancelToken
+    cancelToken,
+    run !== void 0
   );
   const maybeHandleNativeInspectorRequest = async (ctx, _rawRequest, jsRequest) => {
     var _a2, _b, _c, _d;
@@ -5487,6 +5969,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       return await _asyncNullishCoalesce(await ((_b2 = (_a3 = getNativeWorkflowInspector(ctx)) == null ? void 0 : _a3.getState) == null ? void 0 : _b2.call(_a3)), async () => ( null));
     };
     const actorCtx = makeActorCtx(ctx, jsRequest);
+    const sql = getOrCreateNativeSqlDatabase(runtime, ctx);
     try {
       if (url.pathname === "/inspector/state" && jsRequest.method === "GET") {
         return jsonResponse({
@@ -5572,7 +6055,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
             400
           );
         }
-        const cbor = _chunkT56IZ42Mcjs.encodeCborCompat.call(void 0, 
+        const cbor = _chunkGU7ML3KEcjs.encodeCborCompat.call(void 0, 
           _nullishCoalesce(body.body, () => ( null))
         );
         const message = await runtime.actorQueueSend(ctx, name, cbor);
@@ -5614,9 +6097,9 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         }
       }
       if (url.pathname === "/inspector/database/schema" && jsRequest.method === "GET") {
-        const db = actorCtx.sql;
+        const db2 = sql;
         const tables = queryRows(
-          await db.query(
+          await db2.query(
             "SELECT name, type FROM sqlite_master WHERE type IN ('table', 'view') AND name NOT LIKE 'sqlite_%' AND name NOT LIKE '__drizzle_%' ORDER BY name"
           )
         );
@@ -5624,13 +6107,13 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         for (const table of tables) {
           const quoted = `"${table.name.replace(/"/g, '""')}"`;
           const columns = queryRows(
-            await db.query(`PRAGMA table_info(${quoted})`)
+            await db2.query(`PRAGMA table_info(${quoted})`)
           );
           const foreignKeys = queryRows(
-            await db.query(`PRAGMA foreign_key_list(${quoted})`)
+            await db2.query(`PRAGMA foreign_key_list(${quoted})`)
           );
           const countResult = queryRows(
-            await db.query(
+            await db2.query(
               `SELECT COUNT(*) as count FROM ${quoted}`
             )
           );
@@ -5665,7 +6148,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         );
         const quoted = `"${table.replace(/"/g, '""')}"`;
         const rows = queryRows(
-          await actorCtx.sql.query(
+          await sql.query(
             `SELECT * FROM ${quoted} LIMIT ? OFFSET ?`,
             [
               Math.max(0, Math.min(limit, 500)),
@@ -5693,15 +6176,11 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           const bindings = normalizeSqlitePropertyBindings(
             body.properties
           );
-          const rows2 = queryRows(
-            await actorCtx.sql.query(body.sql, bindings)
-          );
+          const rows2 = queryRows(await sql.query(body.sql, bindings));
           return jsonResponse({ rows: jsonSafe(rows2) });
         }
         const args = Array.isArray(body.args) ? body.args : [];
-        const rows = queryRows(
-          await actorCtx.sql.query(body.sql, args)
-        );
+        const rows = queryRows(await sql.query(body.sql, args));
         return jsonResponse({ rows: jsonSafe(rows) });
       }
       if (url.pathname === "/inspector/summary" && jsRequest.method === "GET") {
@@ -5727,7 +6206,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           rpcs: Object.keys(actionHandlers).sort(),
           queueSize: inspectorSnapshot.queueSize,
           isStateEnabled: stateEnabled,
-          isDatabaseEnabled: databaseProvider !== void 0,
+          isDatabaseEnabled: true,
           isWorkflowEnabled: getNativeWorkflowInspector(ctx) !== void 0,
           workflowState: await workflowState(),
           workflowHistory: workflowHistory()
@@ -5917,6 +6396,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       async (error, payload) => {
         const { ctx } = unwrapTsfnPayload(error, payload);
         const actorCtx = makeActorCtx(ctx);
+        const actorId = callNativeSync(() => runtime.actorId(ctx));
         const saveActorState = async () => {
           if (runtime.kind === "wasm") {
             await runtime.actorSaveState(
@@ -5942,6 +6422,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               saveActorState
             );
           } finally {
+            runHandlerCoordinator == null ? void 0 : runHandlerCoordinator.destroy(actorId);
+            _chunk4KGXFC6Mcjs.disposeRunInspector.call(void 0, config.run, actorId);
             await actorCtx.dispose();
           }
         }
@@ -5951,14 +6433,14 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       async (error, payload) => {
         const { ctx } = unwrapTsfnPayload(error, payload);
         const actorCtx = makeActorCtx(ctx);
+        const actorId = callNativeSync(() => runtime.actorId(ctx));
+        runHandlerCoordinator == null ? void 0 : runHandlerCoordinator.destroy(actorId);
+        _chunk4KGXFC6Mcjs.disposeRunInspector.call(void 0, config.run, actorId);
         try {
           if (typeof config.onDestroy === "function") {
             await config.onDestroy(actorCtx);
           }
         } finally {
-          const actorId = callNativeSync(() => runtime.actorId(ctx));
-          nativeRunHandlerActiveByActorId.delete(actorId);
-          _chunkHNHLBM3Ycjs.disposeRunInspector.call(void 0, config.run, actorId);
           resolveNativeDestroy(runtime, ctx);
           await actorCtx.closeDatabase();
           clearNativeRuntimeState(runtime, ctx);
@@ -6006,7 +6488,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               ctx,
               connId
             )
-          )
+          ),
+          () => actorCtx.assertCanMutateState()
         );
         try {
           const nextConnState = hasStaticConnState ? structuredClone(config.connState) : await config.createConnState(
@@ -6043,7 +6526,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               ctx,
               connId
             )
-          )
+          ),
+          () => actorCtx.assertCanMutateState()
         );
         try {
           await config.onConnect(
@@ -6075,7 +6559,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
                     ctx,
                     connId
                   )
-                )
+                ),
+                () => actorCtx.assertCanMutateState()
               )
             );
           }
@@ -6154,7 +6639,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               new Response(null, { status: 404 })
             );
           }
-          const rawConnParams = jsRequest.headers.get(_chunkBZWKY6WOcjs.HEADER_CONN_PARAMS);
+          const rawConnParams = jsRequest.headers.get(_chunkRXXDNJXBcjs.HEADER_CONN_PARAMS);
           let requestCtx;
           let conn;
           try {
@@ -6179,7 +6664,7 @@ function buildNativeFactory(runtime, registryConfig, definition) {
               requestCtx,
               jsRequest
             );
-            if (!_chunkBZWKY6WOcjs.isResponseLike.call(void 0, response)) {
+            if (!_chunkRXXDNJXBcjs.isResponseLike.call(void 0, response)) {
               throw new Error(
                 "onRequest handler must return a Response"
               );
@@ -6219,7 +6704,6 @@ function buildNativeFactory(runtime, registryConfig, definition) {
       }
     ) : void 0,
     run: (() => {
-      const run = _chunkHNHLBM3Ycjs.getRunFunction.call(void 0, config.run);
       if (!run) {
         return void 0;
       }
@@ -6227,26 +6711,38 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         async (error, payload) => {
           const { ctx } = unwrapTsfnPayload(error, payload);
           const actorId = callNativeSync(() => runtime.actorId(ctx));
-          const actorCtx = makeActorCtx(ctx);
-          nativeRunHandlerActiveByActorId.set(actorId, true);
-          try {
-            await run(actorCtx);
-          } finally {
-            nativeRunHandlerActiveByActorId.delete(actorId);
-            await actorCtx.dispose();
+          const executeRun = async () => {
+            const actorCtx = makeActorCtx(ctx);
+            try {
+              await run(actorCtx);
+            } finally {
+              await actorCtx.dispose();
+            }
+          };
+          if (runHandlerCoordinator) {
+            const outcome = await runHandlerCoordinator.run(
+              actorId,
+              () => callNativeSync(
+                () => runtime.actorRestartRunHandler(ctx)
+              ),
+              executeRun
+            );
+            return outcome !== "suppressed";
           }
+          await executeRun();
+          return true;
         }
       );
     })(),
-    getWorkflowHistory: _chunkHNHLBM3Ycjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    getWorkflowHistory: _chunk4KGXFC6Mcjs.hasRunInspectorConfig.call(void 0, config.run) ? wrapNativeCallback(
       async (error, payload) => {
         var _a2;
         const { ctx } = unwrapTsfnPayload(error, payload);
         const history = (_a2 = getNativeWorkflowInspector(ctx)) == null ? void 0 : _a2.getHistory();
-        return history == null ? void 0 : _chunkT56IZ42Mcjs.toUint8Array.call(void 0, history);
+        return history == null ? void 0 : _chunkGU7ML3KEcjs.toUint8Array.call(void 0, history);
       }
     ) : void 0,
-    replayWorkflow: _chunkHNHLBM3Ycjs.getRunInspectorConfig.call(void 0, config.run) !== void 0 ? wrapNativeCallback(
+    replayWorkflow: _chunk4KGXFC6Mcjs.hasRunInspectorConfig.call(void 0, config.run) ? wrapNativeCallback(
       async (error, payload) => {
         const { ctx, entryId } = unwrapTsfnPayload(
           error,
@@ -6256,10 +6752,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
         if (!(workflowInspector == null ? void 0 : workflowInspector.replayFromStep)) {
           return void 0;
         }
-        const history = await _asyncNullishCoalesce(await workflowInspector.replayFromStep(
-          entryId
-        ), async () => ( null));
-        return history == null ? void 0 : _chunkT56IZ42Mcjs.toUint8Array.call(void 0, history);
+        const history = await _asyncNullishCoalesce(await workflowInspector.replayFromStep(entryId), async () => ( null));
+        return history == null ? void 0 : _chunkGU7ML3KEcjs.toUint8Array.call(void 0, history);
       }
     ) : void 0,
     actions: Object.fromEntries(
@@ -6311,7 +6805,8 @@ function buildNativeFactory(runtime, registryConfig, definition) {
           jsRequest,
           stateEnabled,
           onStateChange,
-          cancelToken
+          cancelToken,
+          run !== void 0
         );
         try {
           if (!schemaConfig.queues || !hasSchemaConfigKey(schemaConfig.queues, name)) {
@@ -6382,7 +6877,7 @@ async function buildServeConfig(config) {
     poolName: config.envoy.poolName,
     handleInspectorHttpInRuntime: true,
     serverlessBasePath: config.serverless.basePath,
-    serverlessPackageVersion: _chunkT56IZ42Mcjs.VERSION,
+    serverlessPackageVersion: _chunkGU7ML3KEcjs.VERSION,
     serverlessClientEndpoint: config.publicEndpoint,
     serverlessClientNamespace: config.publicNamespace,
     serverlessClientToken: config.publicToken,
@@ -6395,19 +6890,19 @@ async function buildServeConfig(config) {
   } catch (error) {
     logger2().warn({
       msg: "could not resolve a local engine binary; if a local engine must be spawned it will fail with engine.binary_unavailable \u2014 set RIVET_ENGINE_BINARY_PATH or install the @rivetkit/engine-cli platform package",
-      error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
+      error: _chunkGU7ML3KEcjs.stringifyError.call(void 0, error)
     });
   }
   serveConfig.engineHost = config.engineHost;
   serveConfig.enginePort = config.enginePort;
   if ((_a = config.test) == null ? void 0 : _a.enabled) {
-    serveConfig.inspectorTestToken = _nullishCoalesce(_chunkT56IZ42Mcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
+    serveConfig.inspectorTestToken = _nullishCoalesce(_chunkGU7ML3KEcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN"), () => ( "token"));
   }
   return serveConfig;
 }
 async function buildRegistryWithRuntime(config, runtime) {
   var _a;
-  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunkT56IZ42Mcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
+  if (((_a = config.test) == null ? void 0 : _a.enabled) && _chunkGU7ML3KEcjs.getEnvUniversal.call(void 0, "_RIVET_TEST_INSPECTOR_TOKEN") === void 0) {
     trySetProcessEnv("_RIVET_TEST_INSPECTOR_TOKEN", "token");
   }
   if (runtime.kind === "napi") {
@@ -6462,7 +6957,7 @@ function createApplicationFetch(application) {
         body
       })
     );
-    if (!_chunkBZWKY6WOcjs.isResponseLike.call(void 0, response)) {
+    if (!_chunkRXXDNJXBcjs.isResponseLike.call(void 0, response)) {
       throw new TypeError(
         "registry.listen() application fetch must return a Response"
       );
@@ -6499,7 +6994,7 @@ var Registry = class {
     var _a;
     this.#config = config;
     if ((_a = config.logging) == null ? void 0 : _a.baseLogger) {
-      _chunkT56IZ42Mcjs.configureBaseLogger.call(void 0, config.logging.baseLogger);
+      _chunkGU7ML3KEcjs.configureBaseLogger.call(void 0, config.logging.baseLogger);
     }
     this.#buildConfiguredRegistry = _nullishCoalesce((deps == null ? void 0 : deps.buildConfiguredRegistry), () => ( buildConfiguredRegistry));
     this.routes = {
@@ -6725,11 +7220,11 @@ var Registry = class {
    * ```
    */
   async listen(opts = {}) {
-    const port = _nullishCoalesce(_nullishCoalesce(opts.port, () => ( _chunkT56IZ42Mcjs.parsePortEnv.call(void 0, process.env.RIVET_PORT))), () => ( 3e3));
-    const publicDir = _nullishCoalesce(opts.publicDir, () => ( _chunkT56IZ42Mcjs.getRivetkitPublicDir.call(void 0, )));
+    const port = _nullishCoalesce(_nullishCoalesce(opts.port, () => ( _chunkGU7ML3KEcjs.parsePortEnv.call(void 0, process.env.RIVET_PORT))), () => ( 3e3));
+    const publicDir = _nullishCoalesce(opts.publicDir, () => ( _chunkGU7ML3KEcjs.getRivetkitPublicDir.call(void 0, )));
     const config = this.parseConfig();
     const application = opts.application ? createApplicationFetch(opts.application) : void 0;
-    if (application && _chunkT56IZ42Mcjs.getRivetkitRuntimeMode.call(void 0, ) !== "serverless") {
+    if (application && _chunkGU7ML3KEcjs.getRivetkitRuntimeMode.call(void 0, ) !== "serverless") {
       if (config.runtime === "wasm") {
         throw new Error(
           "registry.listen() requires the native runtime; use an application-owned HTTP server with WebAssembly"
@@ -6803,7 +7298,7 @@ var Registry = class {
       return jsonRouteResponse(503, {
         status: "not_started",
         runtime: "rivetkit",
-        version: _chunkT56IZ42Mcjs.VERSION
+        version: _chunkGU7ML3KEcjs.VERSION
       });
     }
     const { runtime, registry } = configured;
@@ -6811,7 +7306,7 @@ var Registry = class {
       return jsonRouteResponse(501, {
         status: "unsupported",
         runtime: "rivetkit",
-        version: _chunkT56IZ42Mcjs.VERSION
+        version: _chunkGU7ML3KEcjs.VERSION
       });
     }
     const response = await runtime.registryHealth(registry);
@@ -7058,7 +7553,7 @@ var Registry = class {
       );
     }
     if (this.#runtimeReadyPromise) return this.#runtimeReadyPromise;
-    if (_chunkT56IZ42Mcjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
+    if (_chunkGU7ML3KEcjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
       return Promise.reject(
         new Error(
           "registry.startAndWait() requires envoy runtime mode; serverless registries become ready per request"
@@ -7128,8 +7623,8 @@ var Registry = class {
    * ```
    */
   start() {
-    if (_chunkT56IZ42Mcjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
-      const publicDir = _nullishCoalesce(_chunkT56IZ42Mcjs.getRivetkitPublicDir.call(void 0, ), () => ( "/public"));
+    if (_chunkGU7ML3KEcjs.getRivetkitRuntimeMode.call(void 0, ) === "serverless") {
+      const publicDir = _nullishCoalesce(_chunkGU7ML3KEcjs.getRivetkitPublicDir.call(void 0, ), () => ( "/public"));
       this.listen({ publicDir }).catch((error) => {
         logger2().error({ error }, "auto-listen failed; exiting");
         if (typeof process !== "undefined" && typeof process.exit === "function") {
@@ -7150,7 +7645,7 @@ var Registry = class {
     };
     console.log();
     console.log(
-      `  RivetKit ${_chunkT56IZ42Mcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
+      `  RivetKit ${_chunkGU7ML3KEcjs.VERSION} (Engine - ${kind === "serverless" ? "Serverless" : "Serverful"})`
     );
     if (config.namespace !== "default") {
       logLine("Namespace", config.namespace);
@@ -7225,5 +7720,6 @@ function setup(input) {
 
 
 
-exports.ALLOWED_PUBLIC_HEADERS = _chunkBZWKY6WOcjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunk4FC7TVS6cjs.RivetError; exports.ActorsSchema = ActorsSchema; exports.DocConfigurePoolSchema = DocConfigurePoolSchema; exports.DocEnvoyConfigSchema = DocEnvoyConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocSqliteConfigSchema = DocSqliteConfigSchema; exports.InlineWebSocketAdapter = InlineWebSocketAdapter; exports.Registry = Registry; exports.RegistryConfigSchema = RegistryConfigSchema; exports.RivetError = _chunk4FC7TVS6cjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunk4FC7TVS6cjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunkBZWKY6WOcjs.createClientWithDriver; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunkT56IZ42Mcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunkT56IZ42Mcjs.toUint8Array;
+
+exports.ALLOWED_PUBLIC_HEADERS = _chunkRXXDNJXBcjs.ALLOWED_PUBLIC_HEADERS; exports.ActorDefinition = ActorDefinition; exports.ActorError = _chunk4FC7TVS6cjs.RivetError; exports.ActorsSchema = ActorsSchema; exports.DocConfigurePoolSchema = DocConfigurePoolSchema; exports.DocEnvoyConfigSchema = DocEnvoyConfigSchema; exports.DocRegistryConfigSchema = DocRegistryConfigSchema; exports.DocServerlessConfigSchema = DocServerlessConfigSchema; exports.DocSqliteConfigSchema = DocSqliteConfigSchema; exports.InlineWebSocketAdapter = InlineWebSocketAdapter; exports.Registry = Registry; exports.RegistryConfigSchema = RegistryConfigSchema; exports.RivetError = _chunk4FC7TVS6cjs.RivetError; exports.RuntimeKindSchema = RuntimeKindSchema; exports.SqliteBackendSchema = SqliteBackendSchema; exports.SqliteConfigSchema = SqliteConfigSchema; exports.TestConfigSchema = TestConfigSchema; exports.UserError = _chunk4FC7TVS6cjs.UserError; exports.WasmRuntimeConfigSchema = WasmRuntimeConfigSchema; exports.actor = actor; exports.buildActorNames = buildActorNames; exports.createClientWithDriver = _chunkRXXDNJXBcjs.createClientWithDriver; exports.defineRunHandler = _chunk4KGXFC6Mcjs.defineRunHandler; exports.event = event; exports.isStaticActorDefinition = isStaticActorDefinition; exports.isStaticActorInstance = isStaticActorInstance; exports.lookupInRegistry = lookupInRegistry; exports.noopNext = _chunkGU7ML3KEcjs.noopNext; exports.queue = queue; exports.setup = setup; exports.toUint8Array = _chunkGU7ML3KEcjs.toUint8Array;
 //# sourceMappingURL=mod.cjs.map