@restatedev/restate-sdk 0.7.3-worker → 0.8.1

package/dist/{restate_context_impl.d.ts → context_impl.d.ts}

@@ -1,10 +1,12 @@
 /// <reference types="node" />
 /// <reference types="node" />
-import { Rand, RestateGrpcChannel, RestateGrpcContext, RpcContext, RpcGateway, ServiceApi } from "./restate_context";
+/// <reference types="node" />
+import { CombineablePromise, KeyedContext, Rand, RestateGrpcChannel, ServiceApi } from "./context";
 import { StateMachine } from "./state_machine";
 import { AsyncLocalStorage } from "async_hooks";
 import { RetrySettings } from "./utils/public_utils";
 import { Client, SendClient } from "./types/router";
+import { WrappedPromise } from "./utils/promises";
 export declare enum CallContexType {
     None = 0,
     SideEffect = 1,
@@ -14,56 +16,50 @@ export interface CallContext {
     type: CallContexType;
     delay?: number;
 }
-export declare class RestateGrpcContextImpl implements RestateGrpcContext {
+export type InternalCombineablePromise<T> = CombineablePromise<T> & WrappedPromise<T> & {
+    journalIndex: number;
+};
+export declare class ContextImpl implements KeyedContext, RestateGrpcChannel {
     readonly id: Buffer;
     readonly serviceName: string;
+    readonly console: Console;
+    readonly keyedContext: boolean;
     private readonly stateMachine;
     readonly rand: Rand;
     static callContext: AsyncLocalStorage<CallContext>;
-    constructor(id: Buffer, serviceName: string, stateMachine: StateMachine<any, any>, rand?: Rand);
+    private executingSideEffect;
+    constructor(id: Buffer, serviceName: string, console: Console, keyedContext: boolean, stateMachine: StateMachine<any, any>, rand?: Rand);
     get<T>(name: string): Promise<T | null>;
+    stateKeys(): Promise<Array<string>>;
     set<T>(name: string, value: T): void;
     clear(name: string): void;
+    clearAll(): void;
     request(service: string, method: string, data: Uint8Array): Promise<Uint8Array>;
     private invoke;
     private invokeOneWay;
     oneWayCall(call: () => Promise<any>): Promise<void>;
     delayedCall(call: () => Promise<any>, delayMillis?: number): Promise<void>;
+    rpc<M>({ path }: ServiceApi<M>): Client<M>;
+    send<M>(options: ServiceApi): SendClient<M>;
+    sendDelayed<M>({ path }: ServiceApi, delayMillis: number): SendClient<M>;
+    grpcChannel(): RestateGrpcChannel;
     sideEffect<T>(fn: () => Promise<T>, retryPolicy?: RetrySettings): Promise<T>;
-    sleep(millis: number): Promise<void>;
+    sleep(millis: number): CombineablePromise<void>;
+    private sleepInternal;
     awakeable<T>(): {
         id: string;
-        promise: Promise<T>;
+        promise: CombineablePromise<T>;
     };
-    resolveAwakeable<T>(id: string, payload: T): void;
+    resolveAwakeable<T>(id: string, payload?: T): void;
     rejectAwakeable(id: string, reason: string): void;
     private completeAwakeable;
+    createCombinator<T extends readonly CombineablePromise<unknown>[]>(combinatorConstructor: (promises: PromiseLike<unknown>[]) => Promise<unknown>, promises: T): WrappedPromise<unknown>;
     private isInSideEffect;
     private isInOneWayCall;
     private getOneWayCallDelay;
+    private checkNotExecutingSideEffect;
     private checkState;
-    rpcGateway(): RpcGateway;
-}
-export declare class RpcContextImpl implements RpcContext {
-    private readonly ctx;
-    readonly id: Buffer;
-    readonly rand: Rand;
-    readonly serviceName: string;
-    constructor(ctx: RestateGrpcContext, id?: Buffer, rand?: Rand, serviceName?: string);
-    rpc<M>({ path }: ServiceApi): Client<M>;
-    send<M>(options: ServiceApi): SendClient<M>;
-    sendDelayed<M>({ path }: ServiceApi, delayMillis: number): SendClient<M>;
-    get<T>(name: string): Promise<T | null>;
-    set<T>(name: string, value: T): void;
-    clear(name: string): void;
-    sideEffect<T>(fn: () => Promise<T>): Promise<T>;
-    awakeable<T>(): {
-        id: string;
-        promise: Promise<T>;
-    };
-    resolveAwakeable<T>(id: string, payload: T): void;
-    rejectAwakeable(id: string, reason: string): void;
-    sleep(millis: number): Promise<void>;
-    grpcChannel(): RestateGrpcChannel;
+    private checkStateOperation;
+    private markCombineablePromise;
 }
-//# sourceMappingURL=restate_context_impl.d.ts.map
+//# sourceMappingURL=context_impl.d.ts.map

package/dist/context_impl.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context_impl.d.ts","sourceRoot":"","sources":["../src/context_impl.ts"],"names":[],"mappings":";;;AAWA,OAAO,EACL,kBAAkB,EAClB,YAAY,EACZ,IAAI,EACJ,kBAAkB,EAClB,UAAU,EACX,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AA2B/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAahD,OAAO,EAIL,aAAa,EACd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAKpD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,oBAAY,cAAc;IACxB,IAAI,IAAA;IACJ,UAAU,IAAA;IACV,UAAU,IAAA;CACX;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,MAAM,0BAA0B,CAAC,CAAC,IAAI,kBAAkB,CAAC,CAAC,CAAC,GAC/D,cAAc,CAAC,CAAC,CAAC,GAAG;IAClB,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AAEJ,qBAAa,WAAY,YAAW,YAAY,EAAE,kBAAkB;aAchD,EAAE,EAAE,MAAM;aACV,WAAW,EAAE,MAAM;aACnB,OAAO,EAAE,OAAO;aAChB,YAAY,EAAE,OAAO;IAErC,OAAO,CAAC,QAAQ,CAAC,YAAY;aACb,IAAI,EAAE,IAAI;IAb5B,MAAM,CAAC,WAAW,iCAAwC;IAI1D,OAAO,CAAC,mBAAmB,CAAS;gBAGlB,EAAE,EAAE,MAAM,EACV,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,OAAO,EAChB,YAAY,EAAE,OAAO,EAEpB,YAAY,EAAE,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,EACrC,IAAI,GAAE,IAAuB;IAIxC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAqCvC,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAuBnC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,IAAI;IAOpC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAWzB,QAAQ,IAAI,IAAI;IAYhB,OAAO,CACZ,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,UAAU,GACf,OAAO,CAAC,UAAU,CAAC;IActB,OAAO,CAAC,MAAM;YAmBA,YAAY;IAuBnB,UAAU,CAEf,IAAI,EAAE,MAAM,OAAO,CAAC,GAAG,CAAC,GACvB,OAAO,CAAC,IAAI,CAAC;IAUT,WAAW,CAEhB,IAAI,EAAE,MAAM,OAAO,CAAC,GAAG,CAAC,EACxB,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,IAAI,CAAC;IAUhB,GAAG,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;IAoBnC,IAAI,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,GAAG,UAAU,CAAC,CAAC,CAAC;IAI3C,WAAW,CAAC,CAAC,EAClB,EAAE,IAAI,EAAE,EAAE,UAAU,EACpB,WAAW,EAAE,MAAM,GAClB,UAAU,CAAC,CAAC,CAAC;IAoBhB,WAAW,IAAI,kBAAkB;IAQ1B,UAAU,CAAC,CAAC,EACjB,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,WAAW,GAAE,aAAoD,GAChE,OAAO,CAAC,CAAC,CAAC;IAqGN,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC;IAKtD,OAAO,CAAC,aAAa;IASd,SAAS,CAAC,CAAC,KAAK;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAA;KAAE;IAgC9D,gBAAgB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI;IAUlD,eAAe,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAOxD,OAAO,CAAC,iBAAiB;IAYlB,gBAAgB,CAAC,CAAC,SAAS,SAAS,kBAAkB,CAAC,OAAO,CAAC,EAAE,EACtE,qBAAqB,EAAE,CACrB,QAAQ,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,KAC7B,OAAO,CAAC,OAAO,CAAC,EACrB,QAAQ,EAAE,CAAC,GACV,cAAc,CAAC,OAAO,CAAC;IAyB1B,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,kBAAkB;IAK1B,OAAO,CAAC,2BAA2B;IAUnC,OAAO,CAAC,UAAU;IAwBlB,OAAO,CAAC,mBAAmB;IAS3B,OAAO,CAAC,sBAAsB;CAoC/B"}

package/dist/context_impl.js

@@ -0,0 +1,439 @@
+"use strict";
+/*
+ * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH
+ *
+ * This file is part of the Restate SDK for Node.js/TypeScript,
+ * which is released under the MIT license.
+ *
+ * You can find a copy of the license in file LICENSE in the root
+ * directory of this repository or package, or at
+ * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextImpl = exports.CallContexType = void 0;
+const protocol_1 = require("./generated/proto/protocol");
+const protocol_2 = require("./types/protocol");
+const javascript_1 = require("./generated/proto/javascript");
+const async_hooks_1 = require("async_hooks");
+const errors_1 = require("./types/errors");
+const utils_1 = require("./utils/utils");
+const public_utils_1 = require("./utils/public_utils");
+const dynrpc_1 = require("./generated/proto/dynrpc");
+const assumptions_1 = require("./utils/assumptions");
+const rand_1 = require("./utils/rand");
+const promise_combinator_tracker_1 = require("./promise_combinator_tracker");
+var CallContexType;
+(function (CallContexType) {
+    CallContexType[CallContexType["None"] = 0] = "None";
+    CallContexType[CallContexType["SideEffect"] = 1] = "SideEffect";
+    CallContexType[CallContexType["OneWayCall"] = 2] = "OneWayCall";
+})(CallContexType = exports.CallContexType || (exports.CallContexType = {}));
+class ContextImpl {
+    id;
+    serviceName;
+    console;
+    keyedContext;
+    stateMachine;
+    rand;
+    // here, we capture the context information for actions on the Restate context that
+    // are executed within other actions, such as
+    // ctx.oneWayCall( () => client.foo(bar) );
+    // we also use this information to ensure we check that only allowed operations are
+    // used. Within side-effects, no operations are allowed on the RestateContext.
+    // For example, this is illegal: 'ctx.sideEffect(() => {await ctx.get("my-state")})'
+    static callContext = new async_hooks_1.AsyncLocalStorage();
+    // This is used to guard users against calling ctx.sideEffect without awaiting it.
+    // See https://github.com/restatedev/sdk-typescript/issues/197 for more details.
+    executingSideEffect = false;
+    constructor(id, serviceName, console, keyedContext, 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    stateMachine, rand = new rand_1.RandImpl(id)) {
+        this.id = id;
+        this.serviceName = serviceName;
+        this.console = console;
+        this.keyedContext = keyedContext;
+        this.stateMachine = stateMachine;
+        this.rand = rand;
+    }
+    // DON'T make this function async!!! see sideEffect comment for details.
+    get(name) {
+        // Check if this is a valid action
+        this.checkState("get state");
+        this.checkStateOperation("get state");
+        // Create the message and let the state machine process it
+        const msg = protocol_1.GetStateEntryMessage.create({ key: Buffer.from(name) });
+        const completed = this.stateMachine.localStateStore.tryCompleteGet(name, msg);
+        const getState = async () => {
+            const result = await this.stateMachine.handleUserCodeMessage(protocol_2.GET_STATE_ENTRY_MESSAGE_TYPE, msg, completed);
+            // If the GetState message did not have a value or empty,
+            // then we went to the runtime to get the value.
+            // When we get the response, we set it in the localStateStore,
+            // to answer subsequent requests
+            if (!completed) {
+                this.stateMachine.localStateStore.add(name, result);
+            }
+            if (!(result instanceof Buffer)) {
+                return null;
+            }
+            return (0, utils_1.jsonDeserialize)(result.toString());
+        };
+        return getState();
+    }
+    // DON'T make this function async!!! see sideEffect comment for details.
+    stateKeys() {
+        // Check if this is a valid action
+        this.checkState("state keys");
+        // Create the message and let the state machine process it
+        const msg = protocol_1.GetStateKeysEntryMessage.create({});
+        const completed = this.stateMachine.localStateStore.tryCompletedGetStateKeys(msg);
+        const getStateKeys = async () => {
+            const result = await this.stateMachine.handleUserCodeMessage(protocol_2.GET_STATE_KEYS_ENTRY_MESSAGE_TYPE, msg, completed);
+            return result.keys.map((b) => b.toString());
+        };
+        return getStateKeys();
+    }
+    set(name, value) {
+        this.checkState("set state");
+        this.checkStateOperation("set state");
+        const msg = this.stateMachine.localStateStore.set(name, value);
+        this.stateMachine.handleUserCodeMessage(protocol_2.SET_STATE_ENTRY_MESSAGE_TYPE, msg);
+    }
+    clear(name) {
+        this.checkState("clear state");
+        this.checkStateOperation("clear state");
+        const msg = this.stateMachine.localStateStore.clear(name);
+        this.stateMachine.handleUserCodeMessage(protocol_2.CLEAR_STATE_ENTRY_MESSAGE_TYPE, msg);
+    }
+    clearAll() {
+        this.checkState("clear all state");
+        const msg = this.stateMachine.localStateStore.clearAll();
+        this.stateMachine.handleUserCodeMessage(protocol_2.CLEAR_ALL_STATE_ENTRY_MESSAGE_TYPE, msg);
+    }
+    // --- Calls, background calls, etc
+    request(service, method, data) {
+        if (this.isInOneWayCall()) {
+            return this.invokeOneWay(service, method, data, this.getOneWayCallDelay());
+        }
+        else {
+            return this.invoke(service, method, data);
+        }
+    }
+    // DON'T make this function async!!! see sideEffect comment for details.
+    invoke(service, method, data) {
+        this.checkState("invoke");
+        const msg = protocol_1.InvokeEntryMessage.create({
+            serviceName: service,
+            methodName: method,
+            parameter: Buffer.from(data),
+        });
+        return this.markCombineablePromise(this.stateMachine
+            .handleUserCodeMessage(protocol_2.INVOKE_ENTRY_MESSAGE_TYPE, msg)
+            .transform((v) => v));
+    }
+    async invokeOneWay(service, method, data, delay) {
+        const actualDelay = delay || 0;
+        const invokeTime = actualDelay > 0 ? Date.now() + actualDelay : undefined;
+        const msg = protocol_1.BackgroundInvokeEntryMessage.create({
+            serviceName: service,
+            methodName: method,
+            parameter: Buffer.from(data),
+            invokeTime: invokeTime,
+        });
+        await this.stateMachine.handleUserCodeMessage(protocol_2.BACKGROUND_INVOKE_ENTRY_MESSAGE_TYPE, msg);
+        return new Uint8Array();
+    }
+    // DON'T make this function async!!! see sideEffect comment for details.
+    oneWayCall(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    call) {
+        this.checkState("oneWayCall");
+        return ContextImpl.callContext.run({ type: CallContexType.OneWayCall }, call);
+    }
+    // DON'T make this function async!!! see sideEffect comment for details.
+    delayedCall(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    call, delayMillis) {
+        this.checkState("delayedCall");
+        // Delayed call is a one way call with a delay
+        return ContextImpl.callContext.run({ type: CallContexType.OneWayCall, delay: delayMillis }, call);
+    }
+    rpc({ path }) {
+        const clientProxy = new Proxy({}, {
+            get: (_target, prop) => {
+                const route = prop;
+                return async (...args) => {
+                    const request = (0, assumptions_1.requestFromArgs)(args);
+                    const requestBytes = dynrpc_1.RpcRequest.encode(request).finish();
+                    return this.invoke(path, route, requestBytes).transform((responseBytes) => dynrpc_1.RpcResponse.decode(responseBytes).response);
+                };
+            },
+        });
+        return clientProxy;
+    }
+    send(options) {
+        return this.sendDelayed(options, 0);
+    }
+    sendDelayed({ path }, delayMillis) {
+        const clientProxy = new Proxy({}, {
+            get: (_target, prop) => {
+                const route = prop;
+                return (...args) => {
+                    const request = (0, assumptions_1.requestFromArgs)(args);
+                    const requestBytes = dynrpc_1.RpcRequest.encode(request).finish();
+                    this.invokeOneWay(path, route, requestBytes, delayMillis);
+                };
+            },
+        });
+        return clientProxy;
+    }
+    // --- Methods exposed by respective interfaces to interact with other APIs
+    grpcChannel() {
+        return this;
+    }
+    // DON'T make this function async!!!
+    // The reason is that we want the erros thrown by the initial checks to be propagated in the caller context,
+    // and not in the promise context. To understand the semantic difference, make this function async and run the
+    // UnawaitedSideEffectShouldFailSubsequentContextCall test.
+    sideEffect(fn, retryPolicy = public_utils_1.DEFAULT_INFINITE_EXPONENTIAL_BACKOFF) {
+        if (this.isInSideEffect()) {
+            throw new errors_1.TerminalError("You cannot do sideEffect calls from within a side effect.", { errorCode: errors_1.ErrorCodes.INTERNAL });
+        }
+        else if (this.isInOneWayCall()) {
+            throw new errors_1.TerminalError("Cannot do a side effect from within ctx.oneWayCall(...). " +
+                "Context method ctx.oneWayCall() can only be used to invoke other services unidirectionally. " +
+                "e.g. ctx.oneWayCall(() => client.greet(my_request))", { errorCode: errors_1.ErrorCodes.INTERNAL });
+        }
+        this.checkNotExecutingSideEffect();
+        this.executingSideEffect = true;
+        const executeAndLogSideEffect = async () => {
+            // in replay mode, we directly return the value from the log
+            if (this.stateMachine.nextEntryWillBeReplayed()) {
+                const emptyMsg = javascript_1.SideEffectEntryMessage.create({});
+                return this.stateMachine.handleUserCodeMessage(protocol_2.SIDE_EFFECT_ENTRY_MESSAGE_TYPE, emptyMsg);
+            }
+            let sideEffectResult;
+            try {
+                sideEffectResult = await ContextImpl.callContext.run({ type: CallContexType.SideEffect }, fn);
+            }
+            catch (e) {
+                // we commit any error from the side effet to thr journal, and re-throw it into
+                // the function. that way, any catching by the user and reacting to it will be
+                // deterministic on replay
+                const error = (0, errors_1.ensureError)(e);
+                const failure = (0, errors_1.errorToFailureWithTerminal)(error);
+                const sideEffectMsg = javascript_1.SideEffectEntryMessage.create({
+                    failure: failure,
+                });
+                // this may throw an error from the SDK/runtime/connection side, in case the
+                // failure message cannot be committed to the journal. That error would then
+                // be returned from this function (replace the original error)
+                // that is acceptable, because in such a situation (failure to append to journal),
+                // the state machine closes anyways and no further operations will succeed and the
+                // the execution aborts
+                await this.stateMachine.handleUserCodeMessage(protocol_2.SIDE_EFFECT_ENTRY_MESSAGE_TYPE, sideEffectMsg, false, undefined, true);
+                throw e;
+            }
+            // we have this code outside the above try/catch block, to ensure that any error arising
+            // from here is not incorrectly attributed to the side-effect
+            const sideEffectMsg = sideEffectResult !== undefined
+                ? javascript_1.SideEffectEntryMessage.create({
+                    value: Buffer.from((0, utils_1.jsonSerialize)(sideEffectResult)),
+                })
+                : javascript_1.SideEffectEntryMessage.create();
+            // if an error arises from committing the side effect result, then this error will
+            // be thrown here (reject the returned promise) and the function will see that error,
+            // even if the side-effect function completed correctly
+            // that is acceptable, because in such a situation (failure to append to journal),
+            // the state machine closes anyways and reports an execution failure, meaning no further
+            // operations will succeed and the the execution will be retried.
+            // If the side-effect result did in fact not make it to the journal, then the side-effect
+            // re-executes, and if it made it to the journal after all (error happend inly during
+            // ack-back), then retries will use the journaled result.
+            // So all good in any case, due to the beauty of "the runtime log is the ground thruth" approach.
+            await this.stateMachine.handleUserCodeMessage(protocol_2.SIDE_EFFECT_ENTRY_MESSAGE_TYPE, sideEffectMsg, false, undefined, true);
+            return sideEffectResult;
+        };
+        const sleep = (millis) => this.sleepInternal(millis);
+        return executeWithRetries(this.console, retryPolicy, executeAndLogSideEffect, sleep).finally(() => {
+            this.executingSideEffect = false;
+        });
+    }
+    sleep(millis) {
+        this.checkState("sleep");
+        return this.markCombineablePromise(this.sleepInternal(millis));
+    }
+    sleepInternal(millis) {
+        return this.stateMachine.handleUserCodeMessage(protocol_2.SLEEP_ENTRY_MESSAGE_TYPE, protocol_1.SleepEntryMessage.create({ wakeUpTime: Date.now() + millis }));
+    }
+    // -- Awakeables
+    awakeable() {
+        this.checkState("awakeable");
+        const msg = protocol_1.AwakeableEntryMessage.create();
+        const promise = this.stateMachine
+            .handleUserCodeMessage(protocol_2.AWAKEABLE_ENTRY_MESSAGE_TYPE, msg)
+            .transform((result) => {
+            if (!(result instanceof Buffer)) {
+                // This should either be a filled buffer or an empty buffer but never anything else.
+                throw errors_1.RetryableError.internal("Awakeable was not resolved with a buffer payload");
+            }
+            return JSON.parse(result.toString());
+        });
+        // This needs to be done after handling the message in the state machine
+        // otherwise the index is not yet incremented.
+        const encodedEntryIndex = Buffer.alloc(4 /* Size of u32 */);
+        encodedEntryIndex.writeUInt32BE(this.stateMachine.getUserCodeJournalIndex());
+        return {
+            id: protocol_2.AWAKEABLE_IDENTIFIER_PREFIX +
+                Buffer.concat([this.id, encodedEntryIndex]).toString("base64url"),
+            promise: this.markCombineablePromise(promise),
+        };
+    }
+    resolveAwakeable(id, payload) {
+        // We coerce undefined to null as null can be stringified by JSON.stringify
+        const payloadToWrite = payload === undefined ? null : payload;
+        this.checkState("resolveAwakeable");
+        this.completeAwakeable(id, {
+            value: Buffer.from(JSON.stringify(payloadToWrite)),
+        });
+    }
+    rejectAwakeable(id, reason) {
+        this.checkState("rejectAwakeable");
+        this.completeAwakeable(id, {
+            failure: { code: errors_1.ErrorCodes.UNKNOWN, message: reason },
+        });
+    }
+    completeAwakeable(id, base) {
+        base.id = id;
+        this.stateMachine.handleUserCodeMessage(protocol_2.COMPLETE_AWAKEABLE_ENTRY_MESSAGE_TYPE, protocol_1.CompleteAwakeableEntryMessage.create(base));
+    }
+    // Used by static methods of CombineablePromise
+    createCombinator(combinatorConstructor, promises) {
+        const outPromises = [];
+        for (const promise of promises) {
+            if (promise.__restate_context !== this) {
+                throw errors_1.RetryableError.internal("You're mixing up CombineablePromises from different RestateContext. This is not supported.");
+            }
+            const index = promise
+                .journalIndex;
+            outPromises.push({
+                id: (0, promise_combinator_tracker_1.newJournalEntryPromiseId)(index),
+                promise: promise,
+            });
+        }
+        return this.stateMachine.createCombinator(combinatorConstructor, outPromises);
+    }
+    // -- Various private methods
+    isInSideEffect() {
+        const context = ContextImpl.callContext.getStore();
+        return context?.type === CallContexType.SideEffect;
+    }
+    isInOneWayCall() {
+        const context = ContextImpl.callContext.getStore();
+        return context?.type === CallContexType.OneWayCall;
+    }
+    getOneWayCallDelay() {
+        const context = ContextImpl.callContext.getStore();
+        return context?.delay;
+    }
+    checkNotExecutingSideEffect() {
+        if (this.executingSideEffect) {
+            throw new errors_1.TerminalError(`Invoked a RestateContext method while a side effect is still executing. 
+          Make sure you await the ctx.sideEffect call before using any other RestateContext method.`, { errorCode: errors_1.ErrorCodes.INTERNAL });
+        }
+    }
+    checkState(callType) {
+        const context = ContextImpl.callContext.getStore();
+        if (!context) {
+            this.checkNotExecutingSideEffect();
+            return;
+        }
+        if (context.type === CallContexType.SideEffect) {
+            throw new errors_1.TerminalError(`You cannot do ${callType} calls from within a side effect.`, { errorCode: errors_1.ErrorCodes.INTERNAL });
+        }
+        if (context.type === CallContexType.OneWayCall) {
+            throw new errors_1.TerminalError(`Cannot do a ${callType} from within ctx.oneWayCall(...).
+          Context method oneWayCall() can only be used to invoke other services in the background.
+          e.g. ctx.oneWayCall(() => client.greet(my_request))`, { errorCode: errors_1.ErrorCodes.INTERNAL });
+        }
+    }
+    checkStateOperation(callType) {
+        if (!this.keyedContext) {
+            throw new errors_1.TerminalError(`You can do ${callType} calls only from keyed services/routers.`, { errorCode: errors_1.ErrorCodes.INTERNAL });
+        }
+    }
+    markCombineablePromise(p) {
+        const journalIndex = this.stateMachine.getUserCodeJournalIndex();
+        const orTimeout = (millis) => {
+            const sleepPromise = this.sleepInternal(millis).transform(() => {
+                throw new errors_1.TimeoutError();
+            });
+            const sleepPromiseIndex = this.stateMachine.getUserCodeJournalIndex();
+            return this.stateMachine.createCombinator(Promise.race.bind(Promise), [
+                {
+                    id: (0, promise_combinator_tracker_1.newJournalEntryPromiseId)(journalIndex),
+                    promise: p,
+                },
+                {
+                    id: (0, promise_combinator_tracker_1.newJournalEntryPromiseId)(sleepPromiseIndex),
+                    promise: sleepPromise,
+                },
+            ]);
+        };
+        return Object.defineProperties(p, {
+            __restate_context: {
+                value: this,
+            },
+            journalIndex: {
+                value: journalIndex,
+            },
+            orTimeout: {
+                value: orTimeout.bind(this),
+            },
+        });
+    }
+}
+exports.ContextImpl = ContextImpl;
+async function executeWithRetries(console, retrySettings, executeAndLogSideEffect, sleep) {
+    const { initialDelayMs = public_utils_1.DEFAULT_INITIAL_DELAY_MS, maxDelayMs = Number.MAX_SAFE_INTEGER, maxRetries = Number.MAX_SAFE_INTEGER, policy = public_utils_1.EXPONENTIAL_BACKOFF, name = "side-effect", } = retrySettings;
+    let currentDelayMs = initialDelayMs;
+    let retriesLeft = maxRetries;
+    // eslint-disable-next-line no-constant-condition
+    while (true) {
+        try {
+            return await executeAndLogSideEffect();
+        }
+        catch (e) {
+            if (e instanceof errors_1.TerminalError) {
+                throw e;
+            }
+            // journal mismatch errors are special:
+            //  - they are not terminal errors, because we want to allow pushing new code so
+            //    that retries succeed later
+            //  - they are not retried within the service, because they will never succeed within this service,
+            //    but can only succeed within a new invocation going to service with fixed code
+            //  we hence break the retries here similar to terminal errors
+            if (e instanceof errors_1.RestateError &&
+                e.code == errors_1.RestateErrorCodes.JOURNAL_MISMATCH) {
+                throw e;
+            }
+            const error = (0, errors_1.ensureError)(e);
+            console.debug("Error while executing side effect '%s': %s - %s", name, error.name, error.message);
+            if (error.stack) {
+                console.debug(error.stack);
+            }
+            if (retriesLeft > 0) {
+                console.debug("Retrying in %d ms", currentDelayMs);
+            }
+            else {
+                console.debug("No retries left.");
+                throw new errors_1.TerminalError(`Retries exhausted for ${name}. Last error: ${error.name}: ${error.message}`, {
+                    errorCode: errors_1.ErrorCodes.INTERNAL,
+                });
+            }
+        }
+        await sleep(currentDelayMs);
+        retriesLeft -= 1;
+        currentDelayMs = Math.min(policy.computeNextDelay(currentDelayMs), maxDelayMs);
+    }
+}
+//# sourceMappingURL=context_impl.js.map

package/dist/context_impl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context_impl.js","sourceRoot":"","sources":["../src/context_impl.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAUH,yDAUoC;AACpC,+CAa0B;AAC1B,6DAAsE;AACtE,6CAAgD;AAChD,2CASwB;AACxB,yCAA+D;AAE/D,uDAK8B;AAE9B,qDAAmE;AACnE,qDAAsD;AACtD,uCAAwC;AACxC,6EAAwE;AAGxE,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,mDAAI,CAAA;IACJ,+DAAU,CAAA;IACV,+DAAU,CAAA;AACZ,CAAC,EAJW,cAAc,GAAd,sBAAc,KAAd,sBAAc,QAIzB;AAYD,MAAa,WAAW;IAcJ;IACA;IACA;IACA;IAEC;IACD;IAnBlB,mFAAmF;IACnF,6CAA6C;IAC7C,2CAA2C;IAC3C,mFAAmF;IACnF,8EAA8E;IAC9E,oFAAoF;IACpF,MAAM,CAAC,WAAW,GAAG,IAAI,+BAAiB,EAAe,CAAC;IAE1D,kFAAkF;IAClF,gFAAgF;IACxE,mBAAmB,GAAG,KAAK,CAAC;IAEpC,YACkB,EAAU,EACV,WAAmB,EACnB,OAAgB,EAChB,YAAqB;IACrC,8DAA8D;IAC7C,YAAoC,EACrC,OAAa,IAAI,eAAQ,CAAC,EAAE,CAAC;QAN7B,OAAE,GAAF,EAAE,CAAQ;QACV,gBAAW,GAAX,WAAW,CAAQ;QACnB,YAAO,GAAP,OAAO,CAAS;QAChB,iBAAY,GAAZ,YAAY,CAAS;QAEpB,iBAAY,GAAZ,YAAY,CAAwB;QACrC,SAAI,GAAJ,IAAI,CAAyB;IAC5C,CAAC;IAEJ,wEAAwE;IACjE,GAAG,CAAI,IAAY;QACxB,kCAAkC;QAClC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7B,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;QAEtC,0DAA0D;QAC1D,MAAM,GAAG,GAAG,+BAAoB,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpE,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,cAAc,CAChE,IAAI,EACJ,GAAG,CACJ,CAAC;QAEF,MAAM,QAAQ,GAAG,KAAK,IAAuB,EAAE;YAC7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC1D,uCAA4B,EAC5B,GAAG,EACH,SAAS,CACV,CAAC;YAEF,yDAAyD;YACzD,gDAAgD;YAChD,8DAA8D;YAC9D,gCAAgC;YAChC,IAAI,CAAC,SAAS,EAAE;gBACd,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,MAAwB,CAAC,CAAC;aACvE;YAED,IAAI,CAAC,CAAC,MAAM,YAAY,MAAM,CAAC,EAAE;gBAC/B,OAAO,IAAI,CAAC;aACb;YAED,OAAO,IAAA,uBAAe,EAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC;QACF,OAAO,QAAQ,EAAE,CAAC;IACpB,CAAC;IAED,wEAAwE;IACjE,SAAS;QACd,kCAAkC;QAClC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAE9B,0DAA0D;QAC1D,MAAM,GAAG,GAAG,mCAAwB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChD,MAAM,SAAS,GACb,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;QAElE,MAAM,YAAY,GAAG,KAAK,IAA4B,EAAE;YACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC1D,4CAAiC,EACjC,GAAG,EACH,SAAS,CACV,CAAC;YAEF,OAAQ,MAA6C,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACnE,CAAC,CAAC,QAAQ,EAAE,CACb,CAAC;QACJ,CAAC,CAAC;QACF,OAAO,YAAY,EAAE,CAAC;IACxB,CAAC;IAEM,GAAG,CAAI,IAAY,EAAE,KAAQ;QAClC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7B,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAAC,uCAA4B,EAAE,GAAG,CAAC,CAAC;IAC7E,CAAC;IAEM,KAAK,CAAC,IAAY;QACvB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAC/B,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;QAExC,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,YAAY,CAAC,qBAAqB,CACrC,yCAA8B,EAC9B,GAAG,CACJ,CAAC;IACJ,CAAC;IAEM,QAAQ;QACb,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;QAEnC,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;QACzD,IAAI,CAAC,YAAY,CAAC,qBAAqB,CACrC,6CAAkC,EAClC,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,mCAAmC;IAE5B,OAAO,CACZ,OAAe,EACf,MAAc,EACd,IAAgB;QAEhB,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE;YACzB,OAAO,IAAI,CAAC,YAAY,CACtB,OAAO,EACP,MAAM,EACN,IAAI,EACJ,IAAI,CAAC,kBAAkB,EAAE,CAC1B,CAAC;SACH;aAAM;YACL,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;SAC3C;IACH,CAAC;IAED,wEAAwE;IAChE,MAAM,CACZ,OAAe,EACf,MAAc,EACd,IAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAE1B,MAAM,GAAG,GAAG,6BAAkB,CAAC,MAAM,CAAC;YACpC,WAAW,EAAE,OAAO;YACpB,UAAU,EAAE,MAAM;YAClB,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;SAC7B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,sBAAsB,CAChC,IAAI,CAAC,YAAY;aACd,qBAAqB,CAAC,oCAAyB,EAAE,GAAG,CAAC;aACrD,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAe,CAAC,CACrC,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,OAAe,EACf,MAAc,EACd,IAAgB,EAChB,KAAc;QAEd,MAAM,WAAW,GAAG,KAAK,IAAI,CAAC,CAAC;QAC/B,MAAM,UAAU,GAAG,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1E,MAAM,GAAG,GAAG,uCAA4B,CAAC,MAAM,CAAC;YAC9C,WAAW,EAAE,OAAO;YACpB,UAAU,EAAE,MAAM;YAClB,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;YAC5B,UAAU,EAAE,UAAU;SACvB,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC3C,+CAAoC,EACpC,GAAG,CACJ,CAAC;QACF,OAAO,IAAI,UAAU,EAAE,CAAC;IAC1B,CAAC;IAED,wEAAwE;IACjE,UAAU;IACf,8DAA8D;IAC9D,IAAwB;QAExB,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAE9B,OAAO,WAAW,CAAC,WAAW,CAAC,GAAG,CAChC,EAAE,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,EACnC,IAAI,CACL,CAAC;IACJ,CAAC;IAED,wEAAwE;IACjE,WAAW;IAChB,8DAA8D;IAC9D,IAAwB,EACxB,WAAoB;QAEpB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAE/B,8CAA8C;QAC9C,OAAO,WAAW,CAAC,WAAW,CAAC,GAAG,CAChC,EAAE,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,EACvD,IAAI,CACL,CAAC;IACJ,CAAC;IAED,GAAG,CAAI,EAAE,IAAI,EAAiB;QAC5B,MAAM,WAAW,GAAG,IAAI,KAAK,CAC3B,EAAE,EACF;YACE,GAAG,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE;gBACrB,MAAM,KAAK,GAAG,IAAc,CAAC;gBAC7B,OAAO,KAAK,EAAE,GAAG,IAAe,EAAE,EAAE;oBAClC,MAAM,OAAO,GAAG,IAAA,6BAAe,EAAC,IAAI,CAAC,CAAC;oBACtC,MAAM,YAAY,GAAG,mBAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;oBACzD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,SAAS,CACrD,CAAC,aAAa,EAAE,EAAE,CAAC,oBAAW,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,QAAQ,CAC9D,CAAC;gBACJ,CAAC,CAAC;YACJ,CAAC;SACF,CACF,CAAC;QAEF,OAAO,WAAwB,CAAC;IAClC,CAAC;IAEM,IAAI,CAAI,OAAmB;QAChC,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;IAEM,WAAW,CAChB,EAAE,IAAI,EAAc,EACpB,WAAmB;QAEnB,MAAM,WAAW,GAAG,IAAI,KAAK,CAC3B,EAAE,EACF;YACE,GAAG,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE;gBACrB,MAAM,KAAK,GAAG,IAAc,CAAC;gBAC7B,OAAO,CAAC,GAAG,IAAe,EAAE,EAAE;oBAC5B,MAAM,OAAO,GAAG,IAAA,6BAAe,EAAC,IAAI,CAAC,CAAC;oBACtC,MAAM,YAAY,GAAG,mBAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;oBACzD,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC;gBAC5D,CAAC,CAAC;YACJ,CAAC;SACF,CACF,CAAC;QAEF,OAAO,WAA4B,CAAC;IACtC,CAAC;IAED,2EAA2E;IAE3E,WAAW;QACT,OAAO,IAAI,CAAC;IACd,CAAC;IAED,oCAAoC;IACpC,4GAA4G;IAC5G,8GAA8G;IAC9G,2DAA2D;IACpD,UAAU,CACf,EAAoB,EACpB,cAA6B,mDAAoC;QAEjE,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE;YACzB,MAAM,IAAI,sBAAa,CACrB,2DAA2D,EAC3D,EAAE,SAAS,EAAE,mBAAU,CAAC,QAAQ,EAAE,CACnC,CAAC;SACH;aAAM,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE;YAChC,MAAM,IAAI,sBAAa,CACrB,2DAA2D;gBACzD,8FAA8F;gBAC9F,qDAAqD,EACvD,EAAE,SAAS,EAAE,mBAAU,CAAC,QAAQ,EAAE,CACnC,CAAC;SACH;QACD,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACnC,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;QAEhC,MAAM,uBAAuB,GAAG,KAAK,IAAI,EAAE;YACzC,4DAA4D;YAC5D,IAAI,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,EAAE;gBAC/C,MAAM,QAAQ,GAAG,mCAAsB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACnD,OAAO,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC5C,yCAA8B,EAC9B,QAAQ,CACK,CAAC;aACjB;YAED,IAAI,gBAAmB,CAAC;YACxB,IAAI;gBACF,gBAAgB,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,GAAG,CAClD,EAAE,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,EACnC,EAAE,CACH,CAAC;aACH;YAAC,OAAO,CAAC,EAAE;gBACV,+EAA+E;gBAC/E,8EAA8E;gBAC9E,0BAA0B;gBAC1B,MAAM,KAAK,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC;gBAC7B,MAAM,OAAO,GAAG,IAAA,mCAA0B,EAAC,KAAK,CAAC,CAAC;gBAClD,MAAM,aAAa,GAAG,mCAAsB,CAAC,MAAM,CAAC;oBAClD,OAAO,EAAE,OAAO;iBACjB,CAAC,CAAC;gBAEH,4EAA4E;gBAC5E,4EAA4E;gBAC5E,8DAA8D;gBAC9D,kFAAkF;gBAClF,kFAAkF;gBAClF,uBAAuB;gBACvB,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC3C,yCAA8B,EAC9B,aAAa,EACb,KAAK,EACL,SAAS,EACT,IAAI,CACL,CAAC;gBAEF,MAAM,CAAC,CAAC;aACT;YAED,wFAAwF;YACxF,6DAA6D;YAC7D,MAAM,aAAa,GACjB,gBAAgB,KAAK,SAAS;gBAC5B,CAAC,CAAC,mCAAsB,CAAC,MAAM,CAAC;oBAC5B,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAA,qBAAa,EAAC,gBAAgB,CAAC,CAAC;iBACpD,CAAC;gBACJ,CAAC,CAAC,mCAAsB,CAAC,MAAM,EAAE,CAAC;YAEtC,kFAAkF;YAClF,qFAAqF;YACrF,uDAAuD;YACvD,kFAAkF;YAClF,wFAAwF;YACxF,iEAAiE;YACjE,yFAAyF;YACzF,qFAAqF;YACrF,yDAAyD;YACzD,iGAAiG;YACjG,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC3C,yCAA8B,EAC9B,aAAa,EACb,KAAK,EACL,SAAS,EACT,IAAI,CACL,CAAC;YAEF,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC;QAEF,MAAM,KAAK,GAAG,CAAC,MAAc,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC7D,OAAO,kBAAkB,CACvB,IAAI,CAAC,OAAO,EACZ,WAAW,EACX,uBAAuB,EACvB,KAAK,CACN,CAAC,OAAO,CAAC,GAAG,EAAE;YACb,IAAI,CAAC,mBAAmB,GAAG,KAAK,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,MAAc;QACzB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACzB,OAAO,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEO,aAAa,CAAC,MAAc;QAClC,OAAO,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC5C,mCAAwB,EACxB,4BAAiB,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC,CAC9D,CAAC;IACJ,CAAC;IAED,gBAAgB;IAET,SAAS;QACd,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAE7B,MAAM,GAAG,GAAG,gCAAqB,CAAC,MAAM,EAAE,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY;aAC9B,qBAAqB,CAAS,uCAA4B,EAAE,GAAG,CAAC;aAChE,SAAS,CAAC,CAAC,MAAqB,EAAE,EAAE;YACnC,IAAI,CAAC,CAAC,MAAM,YAAY,MAAM,CAAC,EAAE;gBAC/B,oFAAoF;gBACpF,MAAM,uBAAc,CAAC,QAAQ,CAC3B,kDAAkD,CACnD,CAAC;aACH;YAED,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAM,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEL,wEAAwE;QACxE,8CAA8C;QAC9C,MAAM,iBAAiB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;QAC5D,iBAAiB,CAAC,aAAa,CAC7B,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,CAC5C,CAAC;QAEF,OAAO;YACL,EAAE,EACA,sCAA2B;gBAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;YACnE,OAAO,EAAE,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC;SAC9C,CAAC;IACJ,CAAC;IAEM,gBAAgB,CAAI,EAAU,EAAE,OAAW;QAChD,2EAA2E;QAC3E,MAAM,cAAc,GAAG,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;QAE9D,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;QACpC,IAAI,CAAC,iBAAiB,CAAC,EAAE,EAAE;YACzB,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;SACnD,CAAC,CAAC;IACL,CAAC;IAEM,eAAe,CAAC,EAAU,EAAE,MAAc;QAC/C,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;QACnC,IAAI,CAAC,iBAAiB,CAAC,EAAE,EAAE;YACzB,OAAO,EAAE,EAAE,IAAI,EAAE,mBAAU,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE;SACvD,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CACvB,EAAU,EACV,IAAgD;QAEhD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,YAAY,CAAC,qBAAqB,CACrC,gDAAqC,EACrC,wCAA6B,CAAC,MAAM,CAAC,IAAI,CAAC,CAC3C,CAAC;IACJ,CAAC;IAED,+CAA+C;IACxC,gBAAgB,CACrB,qBAEqB,EACrB,QAAW;QAEX,MAAM,WAAW,GAAG,EAAE,CAAC;QAEvB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,IAAI,OAAO,CAAC,iBAAiB,KAAK,IAAI,EAAE;gBACtC,MAAM,uBAAc,CAAC,QAAQ,CAC3B,4FAA4F,CAC7F,CAAC;aACH;YACD,MAAM,KAAK,GAAI,OAA+C;iBAC3D,YAAY,CAAC;YAChB,WAAW,CAAC,IAAI,CAAC;gBACf,EAAE,EAAE,IAAA,qDAAwB,EAAC,KAAK,CAAC;gBACnC,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;SACJ;QAED,OAAO,IAAI,CAAC,YAAY,CAAC,gBAAgB,CACvC,qBAAqB,EACrB,WAAW,CACZ,CAAC;IACJ,CAAC;IAED,6BAA6B;IAErB,cAAc;QACpB,MAAM,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QACnD,OAAO,OAAO,EAAE,IAAI,KAAK,cAAc,CAAC,UAAU,CAAC;IACrD,CAAC;IAEO,cAAc;QACpB,MAAM,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QACnD,OAAO,OAAO,EAAE,IAAI,KAAK,cAAc,CAAC,UAAU,CAAC;IACrD,CAAC;IAEO,kBAAkB;QACxB,MAAM,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QACnD,OAAO,OAAO,EAAE,KAAK,CAAC;IACxB,CAAC;IAEO,2BAA2B;QACjC,IAAI,IAAI,CAAC,mBAAmB,EAAE;YAC5B,MAAM,IAAI,sBAAa,CACrB;oGAC4F,EAC5F,EAAE,SAAS,EAAE,mBAAU,CAAC,QAAQ,EAAE,CACnC,CAAC;SACH;IACH,CAAC;IAEO,UAAU,CAAC,QAAgB;QACjC,MAAM,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QACnD,IAAI,CAAC,OAAO,EAAE;YACZ,IAAI,CAAC,2BAA2B,EAAE,CAAC;YACnC,OAAO;SACR;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,cAAc,CAAC,UAAU,EAAE;YAC9C,MAAM,IAAI,sBAAa,CACrB,iBAAiB,QAAQ,mCAAmC,EAC5D,EAAE,SAAS,EAAE,mBAAU,CAAC,QAAQ,EAAE,CACnC,CAAC;SACH;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,cAAc,CAAC,UAAU,EAAE;YAC9C,MAAM,IAAI,sBAAa,CACrB,eAAe,QAAQ;;8DAE+B,EACtD,EAAE,SAAS,EAAE,mBAAU,CAAC,QAAQ,EAAE,CACnC,CAAC;SACH;IACH,CAAC;IAEO,mBAAmB,CAAC,QAAgB;QAC1C,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YACtB,MAAM,IAAI,sBAAa,CACrB,cAAc,QAAQ,0CAA0C,EAChE,EAAE,SAAS,EAAE,mBAAU,CAAC,QAAQ,EAAE,CACnC,CAAC;SACH;IACH,CAAC;IAEO,sBAAsB,CAC5B,CAAoB;QAEpB,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,CAAC;QACjE,MAAM,SAAS,GAAG,CAAC,MAAc,EAAc,EAAE;YAC/C,MAAM,YAAY,GAAe,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,SAAS,CACnE,GAAG,EAAE;gBACH,MAAM,IAAI,qBAAY,EAAE,CAAC;YAC3B,CAAC,CACF,CAAC;YACF,MAAM,iBAAiB,GAAG,IAAI,CAAC,YAAY,CAAC,uBAAuB,EAAE,CAAC;YAEtE,OAAO,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;gBACpE;oBACE,EAAE,EAAE,IAAA,qDAAwB,EAAC,YAAY,CAAC;oBAC1C,OAAO,EAAE,CAAC;iBACX;gBACD;oBACE,EAAE,EAAE,IAAA,qDAAwB,EAAC,iBAAiB,CAAC;oBAC/C,OAAO,EAAE,YAAY;iBACtB;aACF,CAAe,CAAC;QACnB,CAAC,CAAC;QAEF,OAAO,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE;YAChC,iBAAiB,EAAE;gBACjB,KAAK,EAAE,IAAI;aACZ;YACD,YAAY,EAAE;gBACZ,KAAK,EAAE,YAAY;aACpB;YACD,SAAS,EAAE;gBACT,KAAK,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;aAC5B;SACF,CAAkC,CAAC;IACtC,CAAC;;AA9iBH,kCA+iBC;AAED,KAAK,UAAU,kBAAkB,CAC/B,OAAgB,EAChB,aAA4B,EAC5B,uBAAyC,EACzC,KAAwC;IAExC,MAAM,EACJ,cAAc,GAAG,uCAAwB,EACzC,UAAU,GAAG,MAAM,CAAC,gBAAgB,EACpC,UAAU,GAAG,MAAM,CAAC,gBAAgB,EACpC,MAAM,GAAG,kCAAmB,EAC5B,IAAI,GAAG,aAAa,GACrB,GAAG,aAAa,CAAC;IAElB,IAAI,cAAc,GAAG,cAAc,CAAC;IACpC,IAAI,WAAW,GAAG,UAAU,CAAC;IAE7B,iDAAiD;IACjD,OAAO,IAAI,EAAE;QACX,IAAI;YACF,OAAO,MAAM,uBAAuB,EAAE,CAAC;SACxC;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,YAAY,sBAAa,EAAE;gBAC9B,MAAM,CAAC,CAAC;aACT;YAED,uCAAuC;YACvC,gFAAgF;YAChF,gCAAgC;YAChC,mGAAmG;YACnG,mFAAmF;YACnF,8DAA8D;YAC9D,IACE,CAAC,YAAY,qBAAY;gBACzB,CAAC,CAAC,IAAI,IAAI,0BAAiB,CAAC,gBAAgB,EAC5C;gBACA,MAAM,CAAC,CAAC;aACT;YAED,MAAM,KAAK,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC;YAE7B,OAAO,CAAC,KAAK,CACX,iDAAiD,EACjD,IAAI,EACJ,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,OAAO,CACd,CAAC;YACF,IAAI,KAAK,CAAC,KAAK,EAAE;gBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;aAC5B;YAED,IAAI,WAAW,GAAG,CAAC,EAAE;gBACnB,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC;aACpD;iBAAM;gBACL,OAAO,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBAClC,MAAM,IAAI,sBAAa,CACrB,yBAAyB,IAAI,iBAAiB,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,EAC5E;oBACE,SAAS,EAAE,mBAAU,CAAC,QAAQ;iBAC/B,CACF,CAAC;aACH;SACF;QAED,MAAM,KAAK,CAAC,cAAc,CAAC,CAAC;QAE5B,WAAW,IAAI,CAAC,CAAC;QACjB,cAAc,GAAG,IAAI,CAAC,GAAG,CACvB,MAAM,CAAC,gBAAgB,CAAC,cAAc,CAAC,EACvC,UAAU,CACX,CAAC;KACH;AACH,CAAC"}

package/dist/embedded/api.d.ts

@@ -1,5 +1,5 @@
 /// <reference types="node" />
-import { RpcContext } from "../restate_context";
+import { Context } from "../context";
 import { RemoteContext } from "../generated/proto/services";
 import { OutgoingHttpHeaders } from "http";
 export type RestateConnectionOptions = {
@@ -22,6 +22,6 @@ export declare const connection: (address: string, opt?: RestateConnectionOption
 export declare class RestateConnection {
     private readonly remote;
     constructor(remote: RemoteContext);
-    invoke<I, O>(id: string, input: I, handler: (ctx: RpcContext, input: I) => Promise<O>, opt?: RestateInvocationOptions): Promise<O>;
+    invoke<I, O>(id: string, input: I, handler: (ctx: Context, input: I) => Promise<O>, opt?: RestateInvocationOptions): Promise<O>;
 }
 //# sourceMappingURL=api.d.ts.map

package/dist/embedded/api.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../src/embedded/api.ts"],"names":[],"mappings":";AAWA,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAIhD,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAE5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,MAAM,CAAC;AAE3C,MAAM,MAAM,wBAAwB,GAAG;IACrC;;OAEG;IACH,OAAO,EAAE,mBAAmB,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC;;;;;;OAMG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,eAAO,MAAM,UAAU,YACZ,MAAM,QACT,wBAAwB,KAC7B,iBAC2D,CAAC;AAE/D,qBAAa,iBAAiB;IAChB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAN,MAAM,EAAE,aAAa;IAE3C,MAAM,CAAC,CAAC,EAAE,CAAC,EAChB,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,CAAC,EACR,OAAO,EAAE,CAAC,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,EAClD,GAAG,CAAC,EAAE,wBAAwB,GAC7B,OAAO,CAAC,CAAC,CAAC;CAKd"}
+{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../src/embedded/api.ts"],"names":[],"mappings":";AAWA,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAIrC,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAE5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,MAAM,CAAC;AAE3C,MAAM,MAAM,wBAAwB,GAAG;IACrC;;OAEG;IACH,OAAO,EAAE,mBAAmB,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC;;;;;;OAMG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,eAAO,MAAM,UAAU,YACZ,MAAM,QACT,wBAAwB,KAC7B,iBAC2D,CAAC;AAE/D,qBAAa,iBAAiB;IAChB,OAAO,CAAC,QAAQ,CAAC,MAAM;gBAAN,MAAM,EAAE,aAAa;IAE3C,MAAM,CAAC,CAAC,EAAE,CAAC,EAChB,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,CAAC,EACR,OAAO,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,EAC/C,GAAG,CAAC,EAAE,wBAAwB,GAC7B,OAAO,CAAC,CAAC,CAAC;CAKd"}

package/dist/embedded/api.js

@@ -0,0 +1,35 @@
+"use strict";
+/*
+ * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH
+ *
+ * This file is part of the Restate SDK for Node.js/TypeScript,
+ * which is released under the MIT license.
+ *
+ * You can find a copy of the license in file LICENSE in the root
+ * directory of this repository or package, or at
+ * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RestateConnection = exports.connection = void 0;
+const invocation_1 = require("./invocation");
+const handler_1 = require("./handler");
+const crypto_1 = __importDefault(require("crypto"));
+const http2_remote_1 = require("./http2_remote");
+const connection = (address, opt) => new RestateConnection((0, http2_remote_1.bufConnectRemoteContext)(address, opt));
+exports.connection = connection;
+class RestateConnection {
+    remote;
+    constructor(remote) {
+        this.remote = remote;
+    }
+    invoke(id, input, handler, opt) {
+        const method = (0, handler_1.wrapHandler)(handler);
+        const streamId = crypto_1.default.randomUUID();
+        return (0, invocation_1.doInvoke)(this.remote, id, streamId, input, method, opt);
+    }
+}
+exports.RestateConnection = RestateConnection;
+//# sourceMappingURL=api.js.map

package/dist/embedded/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../src/embedded/api.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;AAGH,6CAAwC;AACxC,uCAAwC;AACxC,oDAA4B;AAE5B,iDAAyD;AAqBlD,MAAM,UAAU,GAAG,CACxB,OAAe,EACf,GAA8B,EACX,EAAE,CACrB,IAAI,iBAAiB,CAAC,IAAA,sCAAuB,EAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;AAJlD,QAAA,UAAU,cAIwC;AAE/D,MAAa,iBAAiB;IACC;IAA7B,YAA6B,MAAqB;QAArB,WAAM,GAAN,MAAM,CAAe;IAAG,CAAC;IAE/C,MAAM,CACX,EAAU,EACV,KAAQ,EACR,OAA+C,EAC/C,GAA8B;QAE9B,MAAM,MAAM,GAAG,IAAA,qBAAW,EAAC,OAAO,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,gBAAM,CAAC,UAAU,EAAE,CAAC;QACrC,OAAO,IAAA,qBAAQ,EAAO,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;IACvE,CAAC;CACF;AAbD,8CAaC"}

package/dist/embedded/handler.d.ts

@@ -1,4 +1,4 @@
-import { RpcContext } from "../restate_context";
+import { Context } from "../context";
 import { HostedGrpcServiceMethod } from "../types/grpc";
-export declare function wrapHandler<I, O>(handler: (ctx: RpcContext, input: I) => Promise<O>): HostedGrpcServiceMethod<I, O>;
+export declare function wrapHandler<I, O>(handler: (ctx: Context, input: I) => Promise<O>): HostedGrpcServiceMethod<I, O>;
 //# sourceMappingURL=handler.d.ts.map

package/dist/embedded/handler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../src/embedded/handler.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,UAAU,EAAc,MAAM,oBAAoB,CAAC;AAE5D,OAAO,EAAqB,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAE3E,wBAAgB,WAAW,CAAC,CAAC,EAAE,CAAC,EAC9B,OAAO,EAAE,CAAC,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,GACjD,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,CAmB/B"}
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../src/embedded/handler.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,OAAO,EAAc,MAAM,YAAY,CAAC;AACjD,OAAO,EAAqB,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAE3E,wBAAgB,WAAW,CAAC,CAAC,EAAE,CAAC,EAC9B,OAAO,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,GAC9C,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,CAmB/B"}