rivetkit 2.3.11-rc.8 → 2.3.11-rc.9

package/dist/tsup/test/mod.cjs

@@ -1,13 +1,13 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-var _chunkZVM35ICHcjs = require('../chunk-ZVM35ICH.cjs');
+var _chunkV5CVATRUcjs = require('../chunk-V5CVATRU.cjs');
 
 
-var _chunkA55CGZIVcjs = require('../chunk-A55CGZIV.cjs');
+var _chunkBZWKY6WOcjs = require('../chunk-BZWKY6WO.cjs');
 require('../chunk-MYWZCDML.cjs');
+require('../chunk-T5XVTVQY.cjs');
 require('../chunk-UEULFXUI.cjs');
-require('../chunk-CCUO5OKD.cjs');
-require('../chunk-2ROUKH7W.cjs');
+require('../chunk-T56IZ42M.cjs');
 require('../chunk-4FC7TVS6.cjs');
 
 // src/test/ready.ts
@@ -36,8 +36,8 @@ async function setupTest(c, registry) {
   registry.config.noWelcome = true;
   registry.start();
   await waitForRegistryReady(registry);
-  const client = _chunkZVM35ICHcjs.createClient.call(void 0, {
-    ..._chunkA55CGZIVcjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
+  const client = _chunkV5CVATRUcjs.createClient.call(void 0, {
+    ..._chunkBZWKY6WOcjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
     disableMetadataLookup: false
   });
   c.onTestFinished(async () => {

package/dist/tsup/test/mod.js

@@ -1,13 +1,13 @@
 import {
   createClient
-} from "../chunk-WRRB24UY.js";
+} from "../chunk-IOLZ3ONU.js";
 import {
   convertRegistryConfigToClientConfig
-} from "../chunk-JBQ63WSM.js";
+} from "../chunk-HOZ3ZLZ2.js";
 import "../chunk-JI6GZ2C2.js";
+import "../chunk-XOOESJY7.js";
 import "../chunk-7AFZMFPQ.js";
-import "../chunk-V6ANE6JM.js";
-import "../chunk-AGNBLCNJ.js";
+import "../chunk-LZ4TSBIP.js";
 import "../chunk-ZZ3WBRPD.js";
 
 // src/test/ready.ts

package/dist/tsup/utils.cjs

@@ -19,7 +19,7 @@
 
 
 
-var _chunk2ROUKH7Wcjs = require('./chunk-2ROUKH7W.cjs');
+var _chunkT56IZ42Mcjs = require('./chunk-T56IZ42M.cjs');
 require('./chunk-4FC7TVS6.cjs');
 
 
@@ -42,5 +42,5 @@ require('./chunk-4FC7TVS6.cjs');
 
 
 
-exports.EXTRA_ERROR_LOG = _chunk2ROUKH7Wcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunk2ROUKH7Wcjs.SinglePromiseQueue; exports.VERSION = _chunk2ROUKH7Wcjs.VERSION; exports.arrayBuffersEqual = _chunk2ROUKH7Wcjs.arrayBuffersEqual; exports.assertUnreachable = _chunk2ROUKH7Wcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunk2ROUKH7Wcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunk2ROUKH7Wcjs.combineUrlPath; exports.dbg = _chunk2ROUKH7Wcjs.dbg; exports.detectRuntime = _chunk2ROUKH7Wcjs.detectRuntime; exports.getEnvUniversal = _chunk2ROUKH7Wcjs.getEnvUniversal; exports.httpUserAgent = _chunk2ROUKH7Wcjs.httpUserAgent; exports.interval = _chunk2ROUKH7Wcjs.interval; exports.joinSignals = _chunk2ROUKH7Wcjs.joinSignals; exports.jsonParseCompat = _chunk2ROUKH7Wcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunk2ROUKH7Wcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunk2ROUKH7Wcjs.promiseWithResolvers; exports.setLongTimeout = _chunk2ROUKH7Wcjs.setLongTimeout; exports.sleep = _chunk2ROUKH7Wcjs.sleep; exports.stringifyError = _chunk2ROUKH7Wcjs.stringifyError; exports.toUint8Array = _chunk2ROUKH7Wcjs.toUint8Array;
+exports.EXTRA_ERROR_LOG = _chunkT56IZ42Mcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunkT56IZ42Mcjs.SinglePromiseQueue; exports.VERSION = _chunkT56IZ42Mcjs.VERSION; exports.arrayBuffersEqual = _chunkT56IZ42Mcjs.arrayBuffersEqual; exports.assertUnreachable = _chunkT56IZ42Mcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkT56IZ42Mcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkT56IZ42Mcjs.combineUrlPath; exports.dbg = _chunkT56IZ42Mcjs.dbg; exports.detectRuntime = _chunkT56IZ42Mcjs.detectRuntime; exports.getEnvUniversal = _chunkT56IZ42Mcjs.getEnvUniversal; exports.httpUserAgent = _chunkT56IZ42Mcjs.httpUserAgent; exports.interval = _chunkT56IZ42Mcjs.interval; exports.joinSignals = _chunkT56IZ42Mcjs.joinSignals; exports.jsonParseCompat = _chunkT56IZ42Mcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunkT56IZ42Mcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunkT56IZ42Mcjs.promiseWithResolvers; exports.setLongTimeout = _chunkT56IZ42Mcjs.setLongTimeout; exports.sleep = _chunkT56IZ42Mcjs.sleep; exports.stringifyError = _chunkT56IZ42Mcjs.stringifyError; exports.toUint8Array = _chunkT56IZ42Mcjs.toUint8Array;
 //# sourceMappingURL=utils.cjs.map

package/dist/tsup/utils.js

@@ -19,7 +19,7 @@ import {
   sleep,
   stringifyError,
   toUint8Array
-} from "./chunk-AGNBLCNJ.js";
+} from "./chunk-LZ4TSBIP.js";
 import "./chunk-ZZ3WBRPD.js";
 export {
   EXTRA_ERROR_LOG,

package/dist/tsup/workflow/mod.cjs

@@ -1,7 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2;
 
-var _chunkE6WB2BLOcjs = require('../chunk-E6WB2BLO.cjs');
-require('../chunk-N6MBTYGX.cjs');
+var _chunk6SBPWTXJcjs = require('../chunk-6SBPWTXJ.cjs');
+require('../chunk-IMRDHZM2.cjs');
 
 
 
@@ -12,7 +12,7 @@ var _chunkHNHLBM3Ycjs = require('../chunk-HNHLBM3Y.cjs');
 var _chunkMYWZCDMLcjs = require('../chunk-MYWZCDML.cjs');
 
 
-var _chunk2ROUKH7Wcjs = require('../chunk-2ROUKH7W.cjs');
+var _chunkT56IZ42Mcjs = require('../chunk-T56IZ42M.cjs');
 
 
 
@@ -716,7 +716,7 @@ function workflow(fn, options = {}) {
   function getWorkflowInspector(actorId) {
     let workflowInspector = workflowInspectors.get(actorId);
     if (!workflowInspector) {
-      workflowInspector = _chunkE6WB2BLOcjs.createWorkflowInspectorAdapter.call(void 0, );
+      workflowInspector = _chunk6SBPWTXJcjs.createWorkflowInspectorAdapter.call(void 0, );
       workflowInspectors.set(actorId, workflowInspector);
     }
     return workflowInspector;
@@ -784,13 +784,13 @@ function workflow(fn, options = {}) {
       if (shouldRethrowWorkflowError(error)) {
         runCtx.log.error({
           msg: "workflow run failed",
-          error: _chunk2ROUKH7Wcjs.stringifyError.call(void 0, error)
+          error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
         });
         throw error;
       }
       runCtx.log.warn({
         msg: "workflow failed and will sleep until woken",
-        error: _chunk2ROUKH7Wcjs.stringifyError.call(void 0, error)
+        error: _chunkT56IZ42Mcjs.stringifyError.call(void 0, error)
       });
     } finally {
       runCtx.abortSignal.removeEventListener("abort", onAbort);

package/dist/tsup/workflow/mod.js

@@ -1,7 +1,7 @@
 import {
   createWorkflowInspectorAdapter
-} from "../chunk-KVD6ZWNM.js";
-import "../chunk-4EDGOGCB.js";
+} from "../chunk-QVCO6DTZ.js";
+import "../chunk-C5TORAVP.js";
 import {
   ACTOR_CONTEXT_INTERNAL_SYMBOL,
   RUN_FUNCTION_CONFIG_SYMBOL
@@ -12,7 +12,7 @@ import {
 } from "../chunk-JI6GZ2C2.js";
 import {
   stringifyError
-} from "../chunk-AGNBLCNJ.js";
+} from "../chunk-LZ4TSBIP.js";
 import {
   RivetError,
   isActorAbortedError

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "rivetkit",
-	"version": "2.3.11-rc.8",
+	"version": "2.3.11-rc.9",
 	"description": "Lightweight libraries for building stateful actors on edge platforms",
 	"license": "Apache-2.0",
 	"keywords": [
@@ -201,14 +201,14 @@
 		"@hono/zod-openapi": "^1.1.5",
 		"@rivet-dev/agent-os-core": "^0.1.1",
 		"@rivetkit/bare-ts": "^0.6.2",
-		"@rivetkit/engine-cli": "2.3.11-rc.8",
-		"@rivetkit/engine-envoy-protocol": "2.3.11-rc.8",
+		"@rivetkit/engine-cli": "2.3.11-rc.9",
+		"@rivetkit/engine-envoy-protocol": "2.3.11-rc.9",
 		"@rivetkit/on-change": "6.0.1",
-		"@rivetkit/rivetkit-napi": "2.3.11-rc.8",
-		"@rivetkit/rivetkit-wasm": "2.3.11-rc.8",
-		"@rivetkit/traces": "2.3.11-rc.8",
-		"@rivetkit/virtual-websocket": "2.3.11-rc.8",
-		"@rivetkit/workflow-engine": "2.3.11-rc.8",
+		"@rivetkit/rivetkit-napi": "2.3.11-rc.9",
+		"@rivetkit/rivetkit-wasm": "2.3.11-rc.9",
+		"@rivetkit/traces": "2.3.11-rc.9",
+		"@rivetkit/virtual-websocket": "2.3.11-rc.9",
+		"@rivetkit/workflow-engine": "2.3.11-rc.9",
 		"cbor-x": "^1.6.0",
 		"drizzle-orm": "^0.44.2",
 		"hono": "^4.7.0",

package/src/registry/native.ts

@@ -104,7 +104,10 @@ import type {
 	WebSocketHandle,
 } from "./runtime";
 import { loadWasmRuntime } from "./wasm-runtime";
-import { createWriteThroughProxy } from "./write-through-proxy";
+import {
+	createWriteThroughProxy,
+	unwrapWriteThroughProxy,
+} from "./write-through-proxy";
 
 const textEncoder = new TextEncoder();
 const textDecoder = new TextDecoder();
@@ -126,6 +129,10 @@ type NativeOnStateChangeHandler = (
 ) => void | Promise<void>;
 type NativePersistConnState = {
 	state: unknown;
+	// Memoized deep write-through proxy and the state object it wraps. Rebuilt
+	// only when the underlying state object identity changes.
+	stateProxy?: unknown;
+	stateProxyTarget?: unknown;
 };
 
 const defaultRuntimeLoaders: RuntimeLoaders = {
@@ -1270,8 +1277,32 @@ class NativeConnAdapter {
 
 	get state(): unknown {
 		const nextState = this.#readState();
+		if (!this.#ctx) {
+			return this.#createStateProxy(nextState);
+		}
+
+		// Reading `conn.state` rebuilds the deep write-through proxy, which
+		// allocates fresh on-change caches and rewraps the whole tree. Memoize
+		// the proxy keyed on the underlying state object so repeated reads and
+		// deep read cascades reuse a single proxy.
+		const connState = getNativeConnPersistState(
+			this.#runtime,
+			this.#ctx,
+			this.#conn,
+		);
+		if (
+			connState.stateProxy === undefined ||
+			connState.stateProxyTarget !== nextState
+		) {
+			connState.stateProxyTarget = nextState;
+			connState.stateProxy = this.#createStateProxy(nextState);
+		}
+		return connState.stateProxy;
+	}
+
+	#createStateProxy(state: unknown): unknown {
 		return createWriteThroughProxy(
-			nextState,
+			state,
 			(nextValue) => {
 				this.#writeState(nextValue, { writeNative: true });
 			},
@@ -1282,12 +1313,15 @@ class NativeConnAdapter {
 	}
 
 	set state(value: unknown) {
-		assertJsonCompatValue(value);
-		this.#writeState(value, { writeNative: true });
+		const nextValue = unwrapWriteThroughProxy(value);
+		assertJsonCompatValue(nextValue);
+		this.#writeState(nextValue, { writeNative: true });
 	}
 
 	initializeState(value: unknown): void {
-		this.#writeState(value, { writeNative: false });
+		this.#writeState(unwrapWriteThroughProxy(value), {
+			writeNative: false,
+		});
 	}
 
 	get isHibernatable(): boolean {
@@ -2693,15 +2727,18 @@ export class ActorContextHandleAdapter {
 			throw stateNotEnabledError();
 		}
 		this.#assertCanMutateState();
-		assertJsonCompatValue(value);
-		this.#writeState(value, { scheduleSave: true });
+		const nextValue = unwrapWriteThroughProxy(value);
+		assertJsonCompatValue(nextValue);
+		this.#writeState(nextValue, { scheduleSave: true });
 	}
 
 	initializeState(value: unknown): void {
 		if (!this.#stateEnabled) {
 			return;
 		}
-		this.#writeState(value, { scheduleSave: false });
+		this.#writeState(unwrapWriteThroughProxy(value), {
+			scheduleSave: false,
+		});
 	}
 
 	get vars(): unknown {

package/src/registry/write-through-proxy.ts

@@ -38,3 +38,104 @@ export function createWriteThroughProxy<T>(
 		},
 	) as T;
 }
+
+/**
+ * Returns the raw target behind an `@rivetkit/on-change` proxy, following
+ * chains of proxies wrapping proxies until reaching a plain value.
+ */
+function unwrapProxy(value: unknown): unknown {
+	let current = value;
+	while (current !== null && typeof current === "object") {
+		const target = onChange.target(current as Record<string, any>);
+		if (target === current) {
+			break;
+		}
+		current = target;
+	}
+	return current;
+}
+
+function isPlainObject(value: unknown): value is Record<string, unknown> {
+	const proto = Object.getPrototypeOf(value as object);
+	return proto === Object.prototype || proto === null;
+}
+
+function unwrapDeep(value: unknown, seen: Set<object>): unknown {
+	const unwrapped = unwrapProxy(value);
+	if (!unwrapped || typeof unwrapped !== "object") {
+		return unwrapped;
+	}
+	if (seen.has(unwrapped)) {
+		return unwrapped;
+	}
+	seen.add(unwrapped);
+
+	if (Array.isArray(unwrapped)) {
+		for (let i = 0; i < unwrapped.length; i++) {
+			const child = unwrapDeep(unwrapped[i], seen);
+			if (child !== unwrapped[i]) {
+				unwrapped[i] = child;
+			}
+		}
+		return unwrapped;
+	}
+
+	if (unwrapped instanceof Map) {
+		const replacements: [unknown, unknown, unknown][] = [];
+		for (const [key, child] of unwrapped.entries()) {
+			const nextKey = unwrapDeep(key, seen);
+			const nextChild = unwrapDeep(child, seen);
+			if (nextKey !== key || nextChild !== child) {
+				replacements.push([key, nextKey, nextChild]);
+			}
+		}
+		for (const [key, nextKey, nextChild] of replacements) {
+			if (nextKey !== key) {
+				unwrapped.delete(key);
+			}
+			unwrapped.set(nextKey, nextChild);
+		}
+		return unwrapped;
+	}
+
+	if (unwrapped instanceof Set) {
+		const replacements: [unknown, unknown][] = [];
+		for (const child of unwrapped.values()) {
+			const next = unwrapDeep(child, seen);
+			if (next !== child) {
+				replacements.push([child, next]);
+			}
+		}
+		for (const [child, next] of replacements) {
+			unwrapped.delete(child);
+			unwrapped.add(next);
+		}
+		return unwrapped;
+	}
+
+	if (isPlainObject(unwrapped)) {
+		for (const key of Object.keys(unwrapped)) {
+			const child = unwrapDeep(unwrapped[key], seen);
+			if (child !== unwrapped[key]) {
+				unwrapped[key] = child;
+			}
+		}
+	}
+
+	return unwrapped;
+}
+
+/**
+ * Strips every `@rivetkit/on-change` proxy out of a value in place, including
+ * proxies nested inside plain objects, arrays, `Map`s, and `Set`s.
+ *
+ * A read of `c.state` or `conn.state` hands back a deep write-through proxy,
+ * so an update written as `c.state = { ...c.state, foo }` produces a plain root
+ * object whose children are still proxies. Persisting that value as-is makes
+ * the next read wrap proxies in another proxy layer, and each layer multiplies
+ * the work of traversing the state, so repeated spread updates degrade
+ * exponentially. Unwrapping before persisting keeps stored state proxy-free.
+ */
+export function unwrapWriteThroughProxy<T>(value: T): T {
+	return unwrapDeep(value, new Set()) as T;
+}