@restatedev/restate-sdk 0.0.0-SNAPSHOT-20231025205407

package/dist/restate_context_impl.js

@@ -0,0 +1,364 @@
+"use strict";
+/*
+ * Copyright (c) 2023 - 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.RpcContextImpl = exports.RestateGrpcContextImpl = 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 logger_1 = require("./utils/logger");
+const dynrpc_1 = require("./generated/proto/dynrpc");
+const assumpsions_1 = require("./utils/assumpsions");
+var CallContexType;
+(function (CallContexType) {
+    CallContexType[CallContexType["None"] = 0] = "None";
+    CallContexType[CallContexType["SideEffect"] = 1] = "SideEffect";
+    CallContexType[CallContexType["OneWayCall"] = 2] = "OneWayCall";
+})(CallContexType || (CallContexType = {}));
+class RestateGrpcContextImpl {
+    id;
+    serviceName;
+    stateMachine;
+    // 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();
+    constructor(id, serviceName, 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    stateMachine) {
+        this.id = id;
+        this.serviceName = serviceName;
+        this.stateMachine = stateMachine;
+    }
+    async get(name) {
+        // Check if this is a valid action
+        this.checkState("get state");
+        // Create the message and let the state machine process it
+        const msg = this.stateMachine.localStateStore.get(name);
+        const result = await this.stateMachine.handleUserCodeMessage(protocol_2.GET_STATE_ENTRY_MESSAGE_TYPE, msg);
+        // 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 (msg.value === undefined && msg.empty === undefined) {
+            this.stateMachine.localStateStore.add(name, result);
+        }
+        if (!(result instanceof Buffer)) {
+            return null;
+        }
+        return (0, utils_1.jsonDeserialize)(result.toString());
+    }
+    set(name, value) {
+        this.checkState("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");
+        const msg = this.stateMachine.localStateStore.clear(name);
+        this.stateMachine.handleUserCodeMessage(protocol_2.CLEAR_STATE_ENTRY_MESSAGE_TYPE, msg);
+    }
+    request(service, method, data) {
+        if (this.isInOneWayCall()) {
+            return this.invokeOneWay(service, method, data);
+        }
+        else {
+            return this.invoke(service, method, data);
+        }
+    }
+    async invoke(service, method, data) {
+        this.checkState("invoke");
+        const msg = protocol_1.InvokeEntryMessage.create({
+            serviceName: service,
+            methodName: method,
+            parameter: Buffer.from(data),
+        });
+        const promise = this.stateMachine.handleUserCodeMessage(protocol_2.INVOKE_ENTRY_MESSAGE_TYPE, msg);
+        return (await promise);
+    }
+    async invokeOneWay(service, method, data) {
+        const delay = this.getOneWayCallDelay();
+        const invokeTime = delay > 0 ? Date.now() + delay : 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();
+    }
+    async oneWayCall(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    call) {
+        this.checkState("oneWayCall");
+        await RestateGrpcContextImpl.callContext.run({ type: CallContexType.OneWayCall }, call);
+    }
+    async 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
+        await RestateGrpcContextImpl.callContext.run({ type: CallContexType.OneWayCall, delay: delayMillis }, call);
+    }
+    async 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 });
+        }
+        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 RestateGrpcContextImpl.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.sleep(millis);
+        return executeWithRetries(retryPolicy, executeAndLogSideEffect, sleep);
+    }
+    sleep(millis) {
+        this.checkState("sleep");
+        const msg = protocol_1.SleepEntryMessage.create({ wakeUpTime: Date.now() + millis });
+        return this.stateMachine.handleUserCodeMessage(protocol_2.SLEEP_ENTRY_MESSAGE_TYPE, msg);
+    }
+    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: Buffer.concat([this.id, encodedEntryIndex]).toString("base64url"),
+            promise: promise,
+        };
+    }
+    resolveAwakeable(id, payload) {
+        this.checkState("resolveAwakeable");
+        this.completeAwakeable(id, {
+            value: Buffer.from(JSON.stringify(payload)),
+        });
+    }
+    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));
+    }
+    isInSideEffect() {
+        const context = RestateGrpcContextImpl.callContext.getStore();
+        return context?.type === CallContexType.SideEffect;
+    }
+    isInOneWayCall() {
+        const context = RestateGrpcContextImpl.callContext.getStore();
+        return context?.type === CallContexType.OneWayCall;
+    }
+    getOneWayCallDelay() {
+        const context = RestateGrpcContextImpl.callContext.getStore();
+        return context?.delay || 0;
+    }
+    checkState(callType) {
+        const context = RestateGrpcContextImpl.callContext.getStore();
+        if (!context) {
+            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 });
+        }
+    }
+}
+exports.RestateGrpcContextImpl = RestateGrpcContextImpl;
+async function executeWithRetries(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.ErrorCodes.JOURNAL_MISMATCH) {
+                throw e;
+            }
+            const error = (0, errors_1.ensureError)(e);
+            logger_1.rlog.debug("Error while executing side effect '%s': %s - %s", name, error.name, error.message);
+            if (error.stack) {
+                logger_1.rlog.debug(error.stack);
+            }
+            if (retriesLeft > 0) {
+                logger_1.rlog.debug("Retrying in %d ms", currentDelayMs);
+            }
+            else {
+                logger_1.rlog.debug("No retries left.");
+                throw new errors_1.TerminalError(`Retries exhausted for ${name}. Last error: ${error.name}: ${error.message}`, {
+                    errorCode: errors_1.ErrorCodes.INTERNAL,
+                    cause: error,
+                });
+            }
+        }
+        await sleep(currentDelayMs);
+        retriesLeft -= 1;
+        currentDelayMs = Math.min(policy.computeNextDelay(currentDelayMs), maxDelayMs);
+    }
+}
+class RpcContextImpl {
+    ctx;
+    id;
+    serviceName;
+    constructor(ctx, id = ctx.id, serviceName = ctx.serviceName) {
+        this.ctx = ctx;
+        this.id = id;
+        this.serviceName = serviceName;
+    }
+    rpc({ path }) {
+        const clientProxy = new Proxy({}, {
+            get: (_target, prop) => {
+                const route = prop;
+                return async (...args) => {
+                    const request = (0, assumpsions_1.requestFromArgs)(args);
+                    const requestBytes = dynrpc_1.RpcRequest.encode(request).finish();
+                    const responseBytes = await this.ctx.request(path, route, requestBytes);
+                    const response = dynrpc_1.RpcResponse.decode(responseBytes);
+                    return response.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, assumpsions_1.requestFromArgs)(args);
+                    const requestBytes = dynrpc_1.RpcRequest.encode(request).finish();
+                    const sender = () => this.ctx.request(path, route, requestBytes);
+                    if (delayMillis === undefined || delayMillis === 0) {
+                        this.ctx.oneWayCall(sender);
+                    }
+                    else {
+                        this.ctx.delayedCall(sender, delayMillis);
+                    }
+                };
+            },
+        });
+        return clientProxy;
+    }
+    get(name) {
+        return this.ctx.get(name);
+    }
+    set(name, value) {
+        this.ctx.set(name, value);
+    }
+    clear(name) {
+        this.ctx.clear(name);
+    }
+    sideEffect(fn) {
+        return this.ctx.sideEffect(fn);
+    }
+    awakeable() {
+        return this.ctx.awakeable();
+    }
+    resolveAwakeable(id, payload) {
+        this.ctx.resolveAwakeable(id, payload);
+    }
+    rejectAwakeable(id, reason) {
+        this.ctx.rejectAwakeable(id, reason);
+    }
+    sleep(millis) {
+        return this.ctx.sleep(millis);
+    }
+}
+exports.RpcContextImpl = RpcContextImpl;
+//# sourceMappingURL=restate_context_impl.js.map

package/dist/restate_context_impl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"restate_context_impl.js","sourceRoot":"","sources":["../src/restate_context_impl.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAIH,yDAOoC;AACpC,+CAU0B;AAC1B,6DAAsE;AACtE,6CAAgD;AAChD,2CAOwB;AACxB,yCAA+D;AAE/D,uDAK8B;AAC9B,2CAAsC;AAEtC,qDAAmE;AACnE,qDAAsD;AAEtD,IAAK,cAIJ;AAJD,WAAK,cAAc;IACjB,mDAAI,CAAA;IACJ,+DAAU,CAAA;IACV,+DAAU,CAAA;AACZ,CAAC,EAJI,cAAc,KAAd,cAAc,QAIlB;AAOD,MAAa,sBAAsB;IAUf;IACA;IAEC;IAZnB,mFAAmF;IACnF,6CAA6C;IAC7C,2CAA2C;IAC3C,mFAAmF;IACnF,8EAA8E;IAC9E,oFAAoF;IAC5E,MAAM,CAAC,WAAW,GAAG,IAAI,+BAAiB,EAAe,CAAC;IAElE,YACkB,EAAU,EACV,WAAmB;IACnC,8DAA8D;IAC7C,YAAoC;QAHrC,OAAE,GAAF,EAAE,CAAQ;QACV,gBAAW,GAAX,WAAW,CAAQ;QAElB,iBAAY,GAAZ,YAAY,CAAwB;IACpD,CAAC;IAEG,KAAK,CAAC,GAAG,CAAI,IAAY;QAC9B,kCAAkC;QAClC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAE7B,0DAA0D;QAC1D,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC1D,uCAA4B,EAC5B,GAAG,CACJ,CAAC;QAEF,yDAAyD;QACzD,gDAAgD;QAChD,8DAA8D;QAC9D,gCAAgC;QAChC,IAAI,GAAG,CAAC,KAAK,KAAK,SAAS,IAAI,GAAG,CAAC,KAAK,KAAK,SAAS,EAAE;YACtD,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,MAAwB,CAAC,CAAC;SACvE;QAED,IAAI,CAAC,CAAC,MAAM,YAAY,MAAM,CAAC,EAAE;YAC/B,OAAO,IAAI,CAAC;SACb;QAED,OAAO,IAAA,uBAAe,EAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC5C,CAAC;IAEM,GAAG,CAAI,IAAY,EAAE,KAAQ;QAClC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7B,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;QAE/B,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,OAAO,CACZ,OAAe,EACf,MAAc,EACd,IAAgB;QAEhB,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE;YACzB,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;SACjD;aAAM;YACL,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;SAC3C;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAClB,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,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,qBAAqB,CACrD,oCAAyB,EACzB,GAAG,CACJ,CAAC;QACF,OAAO,CAAC,MAAM,OAAO,CAAe,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,OAAe,EACf,MAAc,EACd,IAAgB;QAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QAC9D,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;IAEM,KAAK,CAAC,UAAU;IACrB,8DAA8D;IAC9D,IAAwB;QAExB,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAE9B,MAAM,sBAAsB,CAAC,WAAW,CAAC,GAAG,CAC1C,EAAE,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,EACnC,IAAI,CACL,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,WAAW;IACtB,8DAA8D;IAC9D,IAAwB,EACxB,WAAoB;QAEpB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAE/B,8CAA8C;QAC9C,MAAM,sBAAsB,CAAC,WAAW,CAAC,GAAG,CAC1C,EAAE,IAAI,EAAE,cAAc,CAAC,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,EACvD,IAAI,CACL,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,UAAU,CACrB,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;QAED,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,sBAAsB,CAAC,WAAW,CAAC,GAAG,CAC7D,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,KAAK,CAAC,MAAM,CAAC,CAAC;QACrD,OAAO,kBAAkB,CAAC,WAAW,EAAE,uBAAuB,EAAE,KAAK,CAAC,CAAC;IACzE,CAAC;IAEM,KAAK,CAAC,MAAc;QACzB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAEzB,MAAM,GAAG,GAAG,4BAAiB,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAC1E,OAAO,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAC5C,mCAAwB,EACxB,GAAG,CACJ,CAAC;IACJ,CAAC;IAEM,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,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;YACrE,OAAO,EAAE,OAAO;SACjB,CAAC;IACJ,CAAC;IAEM,gBAAgB,CAAI,EAAU,EAAE,OAAU;QAC/C,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,OAAO,CAAC,CAAC;SAC5C,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;IAEO,cAAc;QACpB,MAAM,OAAO,GAAG,sBAAsB,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAC9D,OAAO,OAAO,EAAE,IAAI,KAAK,cAAc,CAAC,UAAU,CAAC;IACrD,CAAC;IAEO,cAAc;QACpB,MAAM,OAAO,GAAG,sBAAsB,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAC9D,OAAO,OAAO,EAAE,IAAI,KAAK,cAAc,CAAC,UAAU,CAAC;IACrD,CAAC;IAEO,kBAAkB;QACxB,MAAM,OAAO,GAAG,sBAAsB,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAC9D,OAAO,OAAO,EAAE,KAAK,IAAI,CAAC,CAAC;IAC7B,CAAC;IAEO,UAAU,CAAC,QAAgB;QACjC,MAAM,OAAO,GAAG,sBAAsB,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;QAC9D,IAAI,CAAC,OAAO,EAAE;YACZ,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;;AA5UH,wDA6UC;AAED,KAAK,UAAU,kBAAkB,CAC/B,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,IAAI,CAAC,YAAY,qBAAY,IAAI,CAAC,CAAC,IAAI,IAAI,mBAAU,CAAC,gBAAgB,EAAE;gBACtE,MAAM,CAAC,CAAC;aACT;YAED,MAAM,KAAK,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC;YAE7B,aAAI,CAAC,KAAK,CACR,iDAAiD,EACjD,IAAI,EACJ,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,OAAO,CACd,CAAC;YACF,IAAI,KAAK,CAAC,KAAK,EAAE;gBACf,aAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;aACzB;YAED,IAAI,WAAW,GAAG,CAAC,EAAE;gBACnB,aAAI,CAAC,KAAK,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC;aACjD;iBAAM;gBACL,aAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBAC/B,MAAM,IAAI,sBAAa,CACrB,yBAAyB,IAAI,iBAAiB,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,EAC5E;oBACE,SAAS,EAAE,mBAAU,CAAC,QAAQ;oBAC9B,KAAK,EAAE,KAAK;iBACb,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;AAED,MAAa,cAAc;IAEN;IACD;IACA;IAHlB,YACmB,GAAuB,EACxB,KAAa,GAAG,CAAC,EAAE,EACnB,cAAsB,GAAG,CAAC,WAAW;QAFpC,QAAG,GAAH,GAAG,CAAoB;QACxB,OAAE,GAAF,EAAE,CAAiB;QACnB,gBAAW,GAAX,WAAW,CAA0B;IACpD,CAAC;IAEG,GAAG,CAAI,EAAE,IAAI,EAAc;QAChC,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,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAC1C,IAAI,EACJ,KAAK,EACL,YAAY,CACb,CAAC;oBACF,MAAM,QAAQ,GAAG,oBAAW,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;oBACnD,OAAO,QAAQ,CAAC,QAAQ,CAAC;gBAC3B,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,MAAM,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;oBACjE,IAAI,WAAW,KAAK,SAAS,IAAI,WAAW,KAAK,CAAC,EAAE;wBAClD,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;qBAC7B;yBAAM;wBACL,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;qBAC3C;gBACH,CAAC,CAAC;YACJ,CAAC;SACF,CACF,CAAC;QAEF,OAAO,WAA4B,CAAC;IACtC,CAAC;IAEM,GAAG,CAAI,IAAY;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IACM,GAAG,CAAI,IAAY,EAAE,KAAQ;QAClC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IACM,KAAK,CAAC,IAAY;QACvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IACM,UAAU,CAAI,EAAoB;QACvC,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IACM,SAAS;QACd,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC;IAC9B,CAAC;IACM,gBAAgB,CAAI,EAAU,EAAE,OAAU;QAC/C,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACzC,CAAC;IACM,eAAe,CAAC,EAAU,EAAE,MAAc;QAC/C,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IACM,KAAK,CAAC,MAAc;QACzB,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF;AArFD,wCAqFC"}

package/dist/server/base_restate_server.d.ts

@@ -0,0 +1,30 @@
+import { GrpcService, GrpcServiceMethod, HostedGrpcServiceMethod, ProtoMetadata } from "../types/grpc";
+import { ProtocolMode, ServiceDiscoveryResponse } from "../generated/proto/discovery";
+import { MethodDescriptorProto as MethodDescriptorProto1 } from "ts-proto-descriptors";
+export interface ServiceOpts {
+    descriptor: ProtoMetadata;
+    service: string;
+    instance: unknown;
+}
+export declare abstract class BaseRestateServer {
+    protected readonly methods: Record<string, HostedGrpcServiceMethod<unknown, unknown>>;
+    protected readonly discovery: ServiceDiscoveryResponse;
+    protected constructor(protocolMode: ProtocolMode);
+    protected addDescriptor(descriptor: ProtoMetadata): void;
+    protected bindService({ descriptor, service, instance }: ServiceOpts): void;
+    rpcHandler(keyed: boolean, route: string, handler: Function): {
+        descriptor: MethodDescriptorProto1;
+        method: GrpcServiceMethod<unknown, unknown>;
+    };
+    stringKeyedEventHandler(keyed: boolean, route: string, handler: Function): {
+        descriptor: MethodDescriptorProto1;
+        method: GrpcServiceMethod<unknown, unknown>;
+    };
+    protected bindRpcService(name: string, router: RpcRouter, keyed: boolean): void;
+    protected methodByUrl<I, O>(url: string | undefined | null): HostedGrpcServiceMethod<I, O> | undefined;
+}
+export declare function parseService(meta: ProtoMetadata, serviceName: string, instance: any): GrpcService;
+export type RpcRouter = {
+    [key: string]: Function;
+};
+//# sourceMappingURL=base_restate_server.d.ts.map

package/dist/server/base_restate_server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base_restate_server.d.ts","sourceRoot":"","sources":["../../src/server/base_restate_server.ts"],"names":[],"mappings":"AAcA,OAAO,EACL,WAAW,EACX,iBAAiB,EACjB,uBAAuB,EACvB,aAAa,EACd,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,YAAY,EACZ,wBAAwB,EACzB,MAAM,8BAA8B,CAAC;AAOtC,OAAO,EAGL,qBAAqB,IAAI,sBAAsB,EAChD,MAAM,sBAAsB,CAAC;AAkB9B,MAAM,WAAW,WAAW;IAC1B,UAAU,EAAE,aAAa,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,CAAC;CACnB;AAED,8BAAsB,iBAAiB;IACrC,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAChC,MAAM,EACN,uBAAuB,CAAC,OAAO,EAAE,OAAO,CAAC,CAC1C,CAAM;IACP,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,wBAAwB,CAAC;IAEvD,SAAS,aAAa,YAAY,EAAE,YAAY;IAUhD,SAAS,CAAC,aAAa,CAAC,UAAU,EAAE,aAAa;IAwDjD,SAAS,CAAC,WAAW,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,WAAW;IAuBpE,UAAU,CACR,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,QAAQ,GAChB;QACD,UAAU,EAAE,sBAAsB,CAAC;QACnC,MAAM,EAAE,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;KAC7C;IA8BD,uBAAuB,CACrB,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,QAAQ,GAChB;QACD,UAAU,EAAE,sBAAsB,CAAC;QACnC,MAAM,EAAE,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;KAC7C;IA4BD,SAAS,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO;IAwDxE,SAAS,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EACxB,GAAG,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,GAC7B,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,SAAS;CAM7C;AAmDD,wBAAgB,YAAY,CAC1B,IAAI,EAAE,aAAa,EACnB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,GAAG,eAoEd;AAED,MAAM,MAAM,SAAS,GAAG;IACtB,CAAC,GAAG,EAAE,MAAM,GAAG,QAAQ,CAAC;CACzB,CAAC"}