@restatedev/restate-sdk 0.7.3-worker → 0.8.1

package/dist/journal.js

@@ -0,0 +1,405 @@
+"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 __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NewExecutionState = exports.JournalEntry = exports.Journal = void 0;
+const p = __importStar(require("./types/protocol"));
+const protocol_1 = require("./generated/proto/protocol");
+const protocol_2 = require("./types/protocol");
+const utils_1 = require("./utils/utils");
+const types_1 = require("./types/types");
+const errors_1 = require("./types/errors");
+const promises_1 = require("./utils/promises");
+const RESOLVED = Promise.resolve(undefined);
+class Journal {
+    invocation;
+    state = NewExecutionState.REPLAYING;
+    userCodeJournalIndex = 0;
+    // Journal entries waiting for arrival of runtime completion
+    // 0 = root promise of the method invocation
+    pendingJournalEntries = new Map();
+    constructor(invocation) {
+        this.invocation = invocation;
+        const inputMessage = invocation.replayEntries.get(0);
+        if (!inputMessage ||
+            inputMessage.messageType !== protocol_2.POLL_INPUT_STREAM_ENTRY_MESSAGE_TYPE) {
+            throw errors_1.RetryableError.protocolViolation("First message of replay entries needs to be PollInputStreamMessage");
+        }
+        this.handleInputMessage(inputMessage.message);
+    }
+    handleInputMessage(m) {
+        if (this.invocation.nbEntriesToReplay === 1) {
+            this.transitionState(NewExecutionState.PROCESSING);
+        }
+        const rootEntry = new JournalEntry(p.POLL_INPUT_STREAM_ENTRY_MESSAGE_TYPE, m);
+        this.pendingJournalEntries.set(0, rootEntry);
+    }
+    handleUserSideMessage(messageType, message
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ) {
+        this.incrementUserCodeIndex();
+        switch (this.state) {
+            case NewExecutionState.REPLAYING: {
+                const replayEntry = this.invocation.replayEntries.get(this.userCodeJournalIndex);
+                if (replayEntry === undefined) {
+                    throw errors_1.RetryableError.internal(`Illegal state: no replay message was received for the entry at journal index ${this.userCodeJournalIndex}`);
+                }
+                const journalEntry = new JournalEntry(messageType, message);
+                this.handleReplay(this.userCodeJournalIndex, replayEntry, journalEntry);
+                return journalEntry.completablePromise.promise;
+            }
+            case NewExecutionState.PROCESSING: {
+                switch (messageType) {
+                    case p.SUSPENSION_MESSAGE_TYPE:
+                    case p.OUTPUT_STREAM_ENTRY_MESSAGE_TYPE: {
+                        this.handleClosingMessage(messageType, message);
+                        return RESOLVED;
+                    }
+                    case p.SET_STATE_ENTRY_MESSAGE_TYPE:
+                    case p.CLEAR_STATE_ENTRY_MESSAGE_TYPE:
+                    case p.CLEAR_ALL_STATE_ENTRY_MESSAGE_TYPE:
+                    case p.COMPLETE_AWAKEABLE_ENTRY_MESSAGE_TYPE:
+                    case p.BACKGROUND_INVOKE_ENTRY_MESSAGE_TYPE: {
+                        // Do not need completion
+                        return RESOLVED;
+                    }
+                    case p.GET_STATE_ENTRY_MESSAGE_TYPE: {
+                        const getStateMsg = message;
+                        if (getStateMsg.value !== undefined ||
+                            getStateMsg.empty !== undefined) {
+                            // State was eagerly filled by the local state store
+                            return Promise.resolve(getStateMsg.value || getStateMsg.empty);
+                        }
+                        else {
+                            // Need to retrieve state by going to the runtime.
+                            return this.appendJournalEntry(messageType, message);
+                        }
+                    }
+                    case p.GET_STATE_KEYS_ENTRY_MESSAGE_TYPE: {
+                        const getStateMsg = message;
+                        if (getStateMsg.value !== undefined) {
+                            // State was eagerly filled by the local state store
+                            return Promise.resolve(getStateMsg.value);
+                        }
+                        else {
+                            // Need to retrieve state by going to the runtime.
+                            return this.appendJournalEntry(messageType, message);
+                        }
+                    }
+                    default: {
+                        return this.appendJournalEntry(messageType, message);
+                    }
+                }
+            }
+            case NewExecutionState.CLOSED: {
+                // We cannot do anything anymore because an output was already sent back
+                // This should actually never happen because the state is only transitioned to closed if the root promise is resolved/rejected
+                // So no more user messages can come in...
+                // - Print warning log and continue...
+                //TODO received user-side message but state machine is closed
+                return RESOLVED;
+            }
+            default: {
+                throw errors_1.RetryableError.protocolViolation("Did not receive input message before other messages.");
+            }
+        }
+    }
+    isUnResolved(index) {
+        const journalEntry = this.pendingJournalEntries.get(index);
+        return journalEntry !== undefined;
+    }
+    handleRuntimeCompletionMessage(m) {
+        // Get message at that entryIndex in pendingJournalEntries
+        const journalEntry = this.pendingJournalEntries.get(m.entryIndex);
+        if (journalEntry === undefined) {
+            //TODO received completion message but there is no pending promise for that index
+            return;
+        }
+        if (m.value !== undefined) {
+            if (journalEntry.messageType === protocol_2.GET_STATE_KEYS_ENTRY_MESSAGE_TYPE) {
+                // In case of get state keys we expect the parsed message
+                journalEntry.completablePromise.resolve(protocol_1.GetStateKeysEntryMessage_StateKeys.decode(m.value));
+                this.pendingJournalEntries.delete(m.entryIndex);
+            }
+            else {
+                journalEntry.completablePromise.resolve(m.value);
+                this.pendingJournalEntries.delete(m.entryIndex);
+            }
+        }
+        else if (m.failure !== undefined) {
+            // we do all completions with Terminal Errors, because failures triggered by those exceptions
+            // when the bubble up would otherwise lead to re-tries, deterministic replay, re-throwing, and
+            // thus an infinite loop that keeps replay-ing but never makes progress
+            // these failures here consequently need to cause terminal failures, unless caught and handled
+            // by the handler code
+            journalEntry.completablePromise.reject((0, errors_1.failureToError)(m.failure, true));
+            this.pendingJournalEntries.delete(m.entryIndex);
+        }
+        else if (m.empty !== undefined) {
+            journalEntry.completablePromise.resolve(m.empty);
+            this.pendingJournalEntries.delete(m.entryIndex);
+        }
+        else {
+            //TODO completion message without a value/failure/empty
+        }
+    }
+    handleEntryAckMessage(m) {
+        // Get message at that entryIndex in pendingJournalEntries
+        const journalEntry = this.pendingJournalEntries.get(m.entryIndex);
+        if (journalEntry === undefined) {
+            return;
+        }
+        // Just needs an ack
+        journalEntry.completablePromise.resolve(undefined);
+        this.pendingJournalEntries.delete(m.entryIndex);
+    }
+    handleReplay(journalIndex, replayMessage, journalEntry) {
+        // Do the journal mismatch check
+        const match = this.checkJournalMatch(replayMessage.messageType, replayMessage.message, journalEntry.messageType, journalEntry.message);
+        // Journal mismatch check failedf
+        if (!match) {
+            throw errors_1.RetryableError.journalMismatch(journalIndex, replayMessage, journalEntry);
+        }
+        // If journal mismatch check passed
+        /*
+        - Else if the runtime replay message contains a completion
+            - If the completion is a value
+                - Return the resolved user code promise with the value
+            - Else if the completion is a failure
+                - Return the rejected user code promise with the failure as Error
+            - Else if the completion is an Empty message
+                - Return the resolved user code promise with the Empty message
+            - Remove the journal entry
+        - Else the replayed message was uncompleted
+            - Create the user code promise
+            - Add message to the pendingJournalEntries
+            - Return the user code promise
+         */
+        switch (journalEntry.messageType) {
+            case protocol_2.SUSPENSION_MESSAGE_TYPE:
+            case protocol_2.OUTPUT_STREAM_ENTRY_MESSAGE_TYPE: {
+                this.handleClosingMessage(journalEntry.messageType, journalEntry.message);
+                break;
+            }
+            case protocol_2.GET_STATE_ENTRY_MESSAGE_TYPE: {
+                const getStateMsg = replayMessage.message;
+                this.resolveResult(journalIndex, journalEntry, getStateMsg.value || getStateMsg.empty, getStateMsg.failure);
+                break;
+            }
+            case protocol_2.GET_STATE_KEYS_ENTRY_MESSAGE_TYPE: {
+                const getStateMsg = replayMessage.message;
+                this.resolveResult(journalIndex, journalEntry, getStateMsg.value, getStateMsg.failure);
+                break;
+            }
+            case protocol_2.INVOKE_ENTRY_MESSAGE_TYPE: {
+                const invokeMsg = replayMessage.message;
+                this.resolveResult(journalIndex, journalEntry, invokeMsg.value, invokeMsg.failure);
+                break;
+            }
+            case protocol_2.SLEEP_ENTRY_MESSAGE_TYPE: {
+                const sleepMsg = replayMessage.message;
+                this.resolveResult(journalIndex, journalEntry, sleepMsg.empty, sleepMsg.failure);
+                break;
+            }
+            case protocol_2.AWAKEABLE_ENTRY_MESSAGE_TYPE: {
+                const awakeableMsg = replayMessage.message;
+                this.resolveResult(journalIndex, journalEntry, awakeableMsg.value, awakeableMsg.failure);
+                break;
+            }
+            case protocol_2.SIDE_EFFECT_ENTRY_MESSAGE_TYPE: {
+                const sideEffectMsg = replayMessage.message;
+                if (sideEffectMsg.value !== undefined) {
+                    this.resolveResult(journalIndex, journalEntry, (0, utils_1.jsonDeserialize)(sideEffectMsg.value.toString()));
+                }
+                else if (sideEffectMsg.failure !== undefined) {
+                    this.resolveResult(journalIndex, journalEntry, undefined, sideEffectMsg.failure.failure, sideEffectMsg.failure.terminal);
+                }
+                else {
+                    // A side effect can have a void return type
+                    // If it was replayed, then it is acked, so we should resolve it.
+                    journalEntry.completablePromise.resolve(undefined);
+                    this.pendingJournalEntries.delete(journalIndex);
+                }
+                break;
+            }
+            case protocol_2.SET_STATE_ENTRY_MESSAGE_TYPE:
+            case protocol_2.CLEAR_STATE_ENTRY_MESSAGE_TYPE:
+            case protocol_2.CLEAR_ALL_STATE_ENTRY_MESSAGE_TYPE:
+            case protocol_2.COMPLETE_AWAKEABLE_ENTRY_MESSAGE_TYPE:
+            case protocol_2.BACKGROUND_INVOKE_ENTRY_MESSAGE_TYPE:
+            case protocol_2.COMBINATOR_ENTRY_MESSAGE: {
+                // Do not need a completion. So if the match has passed then the entry can be deleted.
+                journalEntry.completablePromise.resolve(undefined);
+                this.pendingJournalEntries.delete(journalIndex);
+                break;
+            }
+            default: {
+                // TODO received replay message of unknown type
+            }
+        }
+    }
+    resolveResult(journalIndex, journalEntry, value, failure, failureWouldBeTerminal) {
+        if (value !== undefined) {
+            journalEntry.completablePromise.resolve(value);
+            this.pendingJournalEntries.delete(journalIndex);
+        }
+        else if (failure !== undefined) {
+            const error = (0, errors_1.failureToError)(failure, failureWouldBeTerminal ?? true);
+            journalEntry.completablePromise.reject(error);
+            this.pendingJournalEntries.delete(journalIndex);
+        }
+        else {
+            this.pendingJournalEntries.set(journalIndex, journalEntry);
+        }
+    }
+    handleClosingMessage(messageType, message) {
+        this.transitionState(NewExecutionState.CLOSED);
+        const rootJournalEntry = this.pendingJournalEntries.get(0);
+        if (rootJournalEntry === undefined) {
+            // We have no other option than to throw an error here
+            // Because without the root promise we cannot resolve the method or continue
+            throw errors_1.RetryableError.internal("Illegal state: No root journal entry found to resolve with output stream message");
+        }
+        this.pendingJournalEntries.delete(0);
+        rootJournalEntry.completablePromise.resolve(new types_1.Message(messageType, message));
+    }
+    checkJournalMatch(runtimeMsgType, runtimeMsg, userCodeMsgType, userCodeMsg) {
+        if (runtimeMsgType === userCodeMsgType) {
+            const equalityFct = utils_1.equalityCheckers.get(runtimeMsgType);
+            if (equalityFct === undefined) {
+                // TODO no equality function was defined for the message type
+                return true;
+            }
+            return equalityFct(runtimeMsg, userCodeMsg);
+        }
+        else {
+            return false;
+        }
+    }
+    // To get the indices that need to be completed with suspension
+    getCompletableIndices() {
+        // return all entries except for the root entry
+        return [...this.pendingJournalEntries.entries()]
+            .filter((el) => el[0] !== 0)
+            .map((el) => el[0]);
+    }
+    transitionState(newExecState) {
+        // If the state is already closed then you cannot transition anymore
+        if (this.state === NewExecutionState.CLOSED &&
+            newExecState !== NewExecutionState.CLOSED) {
+            // do nothing
+            return;
+        }
+        else {
+            this.state = newExecState;
+            return;
+        }
+    }
+    incrementUserCodeIndex() {
+        this.userCodeJournalIndex++;
+        if (this.userCodeJournalIndex === this.invocation.nbEntriesToReplay &&
+            this.state === NewExecutionState.REPLAYING) {
+            this.transitionState(NewExecutionState.PROCESSING);
+        }
+    }
+    /**
+     * Read the next replay entry
+     */
+    readNextReplayEntry() {
+        this.incrementUserCodeIndex();
+        return this.invocation.replayEntries.get(this.userCodeJournalIndex);
+    }
+    /**
+     * Append journal entry. This won't increment the journal index.
+     */
+    appendJournalEntry(messageType, message) {
+        const journalEntry = new JournalEntry(messageType, message);
+        this.pendingJournalEntries.set(this.userCodeJournalIndex, journalEntry);
+        return journalEntry.completablePromise.promise;
+    }
+    isClosed() {
+        return this.state === NewExecutionState.CLOSED;
+    }
+    isProcessing() {
+        return this.state === NewExecutionState.PROCESSING;
+    }
+    isReplaying() {
+        return this.state === NewExecutionState.REPLAYING;
+    }
+    getUserCodeJournalIndex() {
+        return this.userCodeJournalIndex;
+    }
+    close() {
+        this.transitionState(NewExecutionState.CLOSED);
+    }
+    outputMsgWasReplayed() {
+        // Check if the last message of the replay entries is an output message
+        const lastEntry = this.invocation.replayEntries.get(this.invocation.nbEntriesToReplay - 1);
+        return (lastEntry && lastEntry.messageType === protocol_2.OUTPUT_STREAM_ENTRY_MESSAGE_TYPE);
+    }
+    // We use this for side effects.
+    // The restate context needs to know if the user-defined fct needs to be executed or not.
+    // It needs to know this before it can craft the message and call this.stateMachine.handleUserSideMessage(...)
+    // so before the index got incremented and the state got transitioned.
+    // So we cannot use isReplaying().
+    // So we need to check in the journal if the next entry (= our side effect) will be replayed or not.
+    nextEntryWillBeReplayed() {
+        return this.userCodeJournalIndex + 1 < this.invocation.nbEntriesToReplay;
+    }
+}
+exports.Journal = Journal;
+class JournalEntry {
+    messageType;
+    message;
+    completablePromise;
+    constructor(messageType, message) {
+        this.messageType = messageType;
+        this.message = message;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        this.completablePromise = new promises_1.CompletablePromise();
+    }
+}
+exports.JournalEntry = JournalEntry;
+// "WAITING_FOR_START" before receiving start message
+// "WAITING_FOR_REPLAY" when waiting for all replay entries to arrive from the runtime
+// "REPLAYING" when receiving input stream message
+// "PROCESSING" when both sides have finished replaying
+// "CLOSED" when input stream connection channel gets closed
+var NewExecutionState;
+(function (NewExecutionState) {
+    NewExecutionState["REPLAYING"] = "REPLAYING";
+    NewExecutionState["PROCESSING"] = "PROCESSING";
+    NewExecutionState["CLOSED"] = "CLOSED";
+})(NewExecutionState = exports.NewExecutionState || (exports.NewExecutionState = {}));
+//# sourceMappingURL=journal.js.map

package/dist/journal.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"journal.js","sourceRoot":"","sources":["../src/journal.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,oDAAsC;AACtC,yDAGoC;AACpC,+CA0B0B;AAC1B,yCAAkE;AAClE,yCAAwC;AAGxC,2CAAgE;AAChE,+CAAsD;AAEtD,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAE5C,MAAa,OAAO;IASG;IARb,KAAK,GAAG,iBAAiB,CAAC,SAAS,CAAC;IAEpC,oBAAoB,GAAG,CAAC,CAAC;IAEjC,4DAA4D;IAC5D,4CAA4C;IACpC,qBAAqB,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEhE,YAAqB,UAA4B;QAA5B,eAAU,GAAV,UAAU,CAAkB;QAC/C,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACrD,IACE,CAAC,YAAY;YACb,YAAY,CAAC,WAAW,KAAK,+CAAoC,EACjE;YACA,MAAM,uBAAc,CAAC,iBAAiB,CACpC,oEAAoE,CACrE,CAAC;SACH;QACD,IAAI,CAAC,kBAAkB,CACrB,YAAY,CAAC,OAAsC,CACpD,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,CAAgC;QACjD,IAAI,IAAI,CAAC,UAAU,CAAC,iBAAiB,KAAK,CAAC,EAAE;YAC3C,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;SACpD;QAED,MAAM,SAAS,GAAG,IAAI,YAAY,CAChC,CAAC,CAAC,oCAAoC,EACtC,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC;IAEM,qBAAqB,CAC1B,WAAmB,EACnB,OAAuC;IACvC,8DAA8D;;QAE9D,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAE9B,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBAChC,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,CACnD,IAAI,CAAC,oBAAoB,CAC1B,CAAC;gBACF,IAAI,WAAW,KAAK,SAAS,EAAE;oBAC7B,MAAM,uBAAc,CAAC,QAAQ,CAC3B,gFAAgF,IAAI,CAAC,oBAAoB,EAAE,CAC5G,CAAC;iBACH;gBAED,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBAC5D,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,oBAAoB,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;gBACxE,OAAO,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC;aAChD;YACD,KAAK,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBACjC,QAAQ,WAAW,EAAE;oBACnB,KAAK,CAAC,CAAC,uBAAuB,CAAC;oBAC/B,KAAK,CAAC,CAAC,gCAAgC,CAAC,CAAC;wBACvC,IAAI,CAAC,oBAAoB,CACvB,WAAW,EACX,OAA2D,CAC5D,CAAC;wBACF,OAAO,QAAQ,CAAC;qBACjB;oBACD,KAAK,CAAC,CAAC,4BAA4B,CAAC;oBACpC,KAAK,CAAC,CAAC,8BAA8B,CAAC;oBACtC,KAAK,CAAC,CAAC,kCAAkC,CAAC;oBAC1C,KAAK,CAAC,CAAC,qCAAqC,CAAC;oBAC7C,KAAK,CAAC,CAAC,oCAAoC,CAAC,CAAC;wBAC3C,yBAAyB;wBACzB,OAAO,QAAQ,CAAC;qBACjB;oBACD,KAAK,CAAC,CAAC,4BAA4B,CAAC,CAAC;wBACnC,MAAM,WAAW,GAAG,OAA+B,CAAC;wBACpD,IACE,WAAW,CAAC,KAAK,KAAK,SAAS;4BAC/B,WAAW,CAAC,KAAK,KAAK,SAAS,EAC/B;4BACA,oDAAoD;4BACpD,OAAO,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;yBAChE;6BAAM;4BACL,kDAAkD;4BAClD,OAAO,IAAI,CAAC,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;yBACtD;qBACF;oBACD,KAAK,CAAC,CAAC,iCAAiC,CAAC,CAAC;wBACxC,MAAM,WAAW,GAAG,OAAmC,CAAC;wBACxD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;4BACnC,oDAAoD;4BACpD,OAAO,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;yBAC3C;6BAAM;4BACL,kDAAkD;4BAClD,OAAO,IAAI,CAAC,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;yBACtD;qBACF;oBACD,OAAO,CAAC,CAAC;wBACP,OAAO,IAAI,CAAC,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;qBACtD;iBACF;aACF;YACD,KAAK,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC7B,wEAAwE;gBACxE,8HAA8H;gBAC9H,0CAA0C;gBAC1C,sCAAsC;gBACtC,6DAA6D;gBAC7D,OAAO,QAAQ,CAAC;aACjB;YACD,OAAO,CAAC,CAAC;gBACP,MAAM,uBAAc,CAAC,iBAAiB,CACpC,sDAAsD,CACvD,CAAC;aACH;SACF;IACH,CAAC;IAEM,YAAY,CAAC,KAAa;QAC/B,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3D,OAAO,YAAY,KAAK,SAAS,CAAC;IACpC,CAAC;IAEM,8BAA8B,CAAC,CAAoB;QACxD,0DAA0D;QAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAElE,IAAI,YAAY,KAAK,SAAS,EAAE;YAC9B,iFAAiF;YACjF,OAAO;SACR;QAED,IAAI,CAAC,CAAC,KAAK,KAAK,SAAS,EAAE;YACzB,IAAI,YAAY,CAAC,WAAW,KAAK,4CAAiC,EAAE;gBAClE,yDAAyD;gBACzD,YAAY,CAAC,kBAAkB,CAAC,OAAO,CACrC,6CAAkC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CACnD,CAAC;gBACF,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;aACjD;iBAAM;gBACL,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACjD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;aACjD;SACF;aAAM,IAAI,CAAC,CAAC,OAAO,KAAK,SAAS,EAAE;YAClC,6FAA6F;YAC7F,8FAA8F;YAC9F,uEAAuE;YACvE,8FAA8F;YAC9F,sBAAsB;YACtB,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,IAAA,uBAAc,EAAC,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACxE,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;SACjD;aAAM,IAAI,CAAC,CAAC,KAAK,KAAK,SAAS,EAAE;YAChC,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;SACjD;aAAM;YACL,uDAAuD;SACxD;IACH,CAAC;IAEM,qBAAqB,CAAC,CAAkB;QAC7C,0DAA0D;QAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAElE,IAAI,YAAY,KAAK,SAAS,EAAE;YAC9B,OAAO;SACR;QAED,oBAAoB;QACpB,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IAClD,CAAC;IAEO,YAAY,CAClB,YAAoB,EACpB,aAAsB,EACtB,YAA0B;QAE1B,gCAAgC;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAClC,aAAa,CAAC,WAAW,EACzB,aAAa,CAAC,OAAO,EACrB,YAAY,CAAC,WAAW,EACxB,YAAY,CAAC,OAAO,CACrB,CAAC;QAEF,iCAAiC;QACjC,IAAI,CAAC,KAAK,EAAE;YACV,MAAM,uBAAc,CAAC,eAAe,CAClC,YAAY,EACZ,aAAa,EACb,YAAY,CACb,CAAC;SACH;QAED,mCAAmC;QACnC;;;;;;;;;;;;;WAaG;QACH,QAAQ,YAAY,CAAC,WAAW,EAAE;YAChC,KAAK,kCAAuB,CAAC;YAC7B,KAAK,2CAAgC,CAAC,CAAC;gBACrC,IAAI,CAAC,oBAAoB,CACvB,YAAY,CAAC,WAAW,EACxB,YAAY,CAAC,OAAuD,CACrE,CAAC;gBACF,MAAM;aACP;YACD,KAAK,uCAA4B,CAAC,CAAC;gBACjC,MAAM,WAAW,GAAG,aAAa,CAAC,OAA+B,CAAC;gBAClE,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,WAAW,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,EACtC,WAAW,CAAC,OAAO,CACpB,CAAC;gBACF,MAAM;aACP;YACD,KAAK,4CAAiC,CAAC,CAAC;gBACtC,MAAM,WAAW,GAAG,aAAa,CAAC,OAAmC,CAAC;gBACtE,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,WAAW,CAAC,KAAK,EACjB,WAAW,CAAC,OAAO,CACpB,CAAC;gBACF,MAAM;aACP;YACD,KAAK,oCAAyB,CAAC,CAAC;gBAC9B,MAAM,SAAS,GAAG,aAAa,CAAC,OAA6B,CAAC;gBAC9D,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,SAAS,CAAC,KAAK,EACf,SAAS,CAAC,OAAO,CAClB,CAAC;gBACF,MAAM;aACP;YACD,KAAK,mCAAwB,CAAC,CAAC;gBAC7B,MAAM,QAAQ,GAAG,aAAa,CAAC,OAA4B,CAAC;gBAC5D,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,OAAO,CACjB,CAAC;gBACF,MAAM;aACP;YACD,KAAK,uCAA4B,CAAC,CAAC;gBACjC,MAAM,YAAY,GAAG,aAAa,CAAC,OAAgC,CAAC;gBACpE,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,YAAY,CAAC,KAAK,EAClB,YAAY,CAAC,OAAO,CACrB,CAAC;gBACF,MAAM;aACP;YACD,KAAK,yCAA8B,CAAC,CAAC;gBACnC,MAAM,aAAa,GAAG,aAAa,CAAC,OAAiC,CAAC;gBACtE,IAAI,aAAa,CAAC,KAAK,KAAK,SAAS,EAAE;oBACrC,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,IAAA,uBAAe,EAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAChD,CAAC;iBACH;qBAAM,IAAI,aAAa,CAAC,OAAO,KAAK,SAAS,EAAE;oBAC9C,IAAI,CAAC,aAAa,CAChB,YAAY,EACZ,YAAY,EACZ,SAAS,EACT,aAAa,CAAC,OAAO,CAAC,OAAO,EAC7B,aAAa,CAAC,OAAO,CAAC,QAAQ,CAC/B,CAAC;iBACH;qBAAM;oBACL,4CAA4C;oBAC5C,iEAAiE;oBACjE,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBACnD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;iBACjD;gBACD,MAAM;aACP;YACD,KAAK,uCAA4B,CAAC;YAClC,KAAK,yCAA8B,CAAC;YACpC,KAAK,6CAAkC,CAAC;YACxC,KAAK,gDAAqC,CAAC;YAC3C,KAAK,+CAAoC,CAAC;YAC1C,KAAK,mCAAwB,CAAC,CAAC;gBAC7B,sFAAsF;gBACtF,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBACnD,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;gBAChD,MAAM;aACP;YACD,OAAO,CAAC,CAAC;gBACP,+CAA+C;aAChD;SACF;IACH,CAAC;IAED,aAAa,CACX,YAAoB,EACpB,YAA0B,EAC1B,KAAoB,EACpB,OAA6B,EAC7B,sBAAgC;QAEhC,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC/C,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;SACjD;aAAM,IAAI,OAAO,KAAK,SAAS,EAAE;YAChC,MAAM,KAAK,GAAG,IAAA,uBAAc,EAAC,OAAO,EAAE,sBAAsB,IAAI,IAAI,CAAC,CAAC;YACtE,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9C,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;SACjD;aAAM;YACL,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;SAC5D;IACH,CAAC;IAED,oBAAoB,CAClB,WAAmB,EACnB,OAAqD;QAErD,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,gBAAgB,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAE3D,IAAI,gBAAgB,KAAK,SAAS,EAAE;YAClC,sDAAsD;YACtD,4EAA4E;YAC5E,MAAM,uBAAc,CAAC,QAAQ,CAC3B,kFAAkF,CACnF,CAAC;SACH;QAED,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACrC,gBAAgB,CAAC,kBAAkB,CAAC,OAAO,CACzC,IAAI,eAAO,CAAC,WAAW,EAAE,OAAO,CAAC,CAClC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,cAAsB,EACtB,UAA0C,EAC1C,eAAuB,EACvB,WAA2C;QAE3C,IAAI,cAAc,KAAK,eAAe,EAAE;YACtC,MAAM,WAAW,GAAG,wBAAgB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACzD,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,6DAA6D;gBAC7D,OAAO,IAAI,CAAC;aACb;YACD,OAAO,WAAW,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;SAC7C;aAAM;YACL,OAAO,KAAK,CAAC;SACd;IACH,CAAC;IAED,+DAA+D;IACxD,qBAAqB;QAC1B,+CAA+C;QAC/C,OAAO,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC;aAC7C,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;aAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC;IAEO,eAAe,CAAC,YAA+B;QACrD,oEAAoE;QACpE,IACE,IAAI,CAAC,KAAK,KAAK,iBAAiB,CAAC,MAAM;YACvC,YAAY,KAAK,iBAAiB,CAAC,MAAM,EACzC;YACA,aAAa;YACb,OAAO;SACR;aAAM;YACL,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;YAC1B,OAAO;SACR;IACH,CAAC;IAED,sBAAsB;QACpB,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5B,IACE,IAAI,CAAC,oBAAoB,KAAK,IAAI,CAAC,UAAU,CAAC,iBAAiB;YAC/D,IAAI,CAAC,KAAK,KAAK,iBAAiB,CAAC,SAAS,EAC1C;YACA,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;SACpD;IACH,CAAC;IAED;;OAEG;IACI,mBAAmB;QACxB,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACtE,CAAC;IAED;;OAEG;IACI,kBAAkB,CACvB,WAAmB,EACnB,OAAuC;QAEvC,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC5D,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QACxE,OAAO,YAAY,CAAC,kBAAkB,CAAC,OAAO,CAAC;IACjD,CAAC;IAEM,QAAQ;QACb,OAAO,IAAI,CAAC,KAAK,KAAK,iBAAiB,CAAC,MAAM,CAAC;IACjD,CAAC;IAEM,YAAY;QACjB,OAAO,IAAI,CAAC,KAAK,KAAK,iBAAiB,CAAC,UAAU,CAAC;IACrD,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,KAAK,KAAK,iBAAiB,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,uBAAuB;QAC5B,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC;IAEM,oBAAoB;QACzB,uEAAuE;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,CACjD,IAAI,CAAC,UAAU,CAAC,iBAAiB,GAAG,CAAC,CACtC,CAAC;QACF,OAAO,CACL,SAAS,IAAI,SAAS,CAAC,WAAW,KAAK,2CAAgC,CACxE,CAAC;IACJ,CAAC;IAED,gCAAgC;IAChC,yFAAyF;IACzF,8GAA8G;IAC9G,sEAAsE;IACtE,kCAAkC;IAClC,oGAAoG;IACpG,uBAAuB;QACrB,OAAO,IAAI,CAAC,oBAAoB,GAAG,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC;IAC3E,CAAC;CACF;AA7cD,0BA6cC;AAED,MAAa,YAAY;IAIZ;IACA;IAJJ,kBAAkB,CAA8B;IAEvD,YACW,WAAmB,EACnB,OAAuC;QADvC,gBAAW,GAAX,WAAW,CAAQ;QACnB,YAAO,GAAP,OAAO,CAAgC;QAEhD,8DAA8D;QAC9D,IAAI,CAAC,kBAAkB,GAAG,IAAI,6BAAkB,EAAO,CAAC;IAC1D,CAAC;CACF;AAVD,oCAUC;AAED,qDAAqD;AACrD,sFAAsF;AACtF,kDAAkD;AAClD,uDAAuD;AACvD,4DAA4D;AAC5D,IAAY,iBAIX;AAJD,WAAY,iBAAiB;IAC3B,4CAAuB,CAAA;IACvB,8CAAyB,CAAA;IACzB,sCAAiB,CAAA;AACnB,CAAC,EAJW,iBAAiB,GAAjB,yBAAiB,KAAjB,yBAAiB,QAI5B"}

package/dist/local_state_store.d.ts

@@ -1,13 +1,15 @@
 /// <reference types="node" />
-import { ClearStateEntryMessage, GetStateEntryMessage, SetStateEntryMessage, StartMessage_StateEntry } from "./generated/proto/protocol";
+import { ClearAllStateEntryMessage, ClearStateEntryMessage, GetStateEntryMessage, GetStateKeysEntryMessage, SetStateEntryMessage, StartMessage_StateEntry } from "./generated/proto/protocol";
 import { Empty } from "./generated/google/protobuf/empty";
 export declare class LocalStateStore {
-    readonly isPartial: boolean;
+    private isPartial;
     private state;
     constructor(isPartial: boolean, state: StartMessage_StateEntry[]);
-    get(key: string): GetStateEntryMessage;
+    tryCompleteGet(key: string, msg: GetStateEntryMessage): boolean;
+    tryCompletedGetStateKeys(msg: GetStateKeysEntryMessage): boolean;
     set<T>(key: string, value: T): SetStateEntryMessage;
     clear(key: string): ClearStateEntryMessage;
     add(key: string, result: Buffer | Empty): void;
+    clearAll(): ClearAllStateEntryMessage;
 }
 //# sourceMappingURL=local_state_store.d.ts.map

package/dist/local_state_store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"local_state_store.d.ts","sourceRoot":"","sources":["../src/local_state_store.ts"],"names":[],"mappings":";AAWA,OAAO,EACL,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,uBAAuB,EACxB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,mCAAmC,CAAC;AAG1D,qBAAa,eAAe;IAGd,QAAQ,CAAC,SAAS,EAAE,OAAO;IAFvC,OAAO,CAAC,KAAK,CAA8B;gBAEtB,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE;IAMlE,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,oBAAoB;IA4BtC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,oBAAoB;IASnD,KAAK,CAAC,GAAG,EAAE,MAAM,GAAG,sBAAsB;IAQ1C,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,KAAK,GAAG,IAAI;CAGtD"}
+{"version":3,"file":"local_state_store.d.ts","sourceRoot":"","sources":["../src/local_state_store.ts"],"names":[],"mappings":";AAWA,OAAO,EACL,yBAAyB,EACzB,sBAAsB,EACtB,oBAAoB,EACpB,wBAAwB,EAExB,oBAAoB,EACpB,uBAAuB,EACxB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,mCAAmC,CAAC;AAG1D,qBAAa,eAAe;IAGd,OAAO,CAAC,SAAS;IAF7B,OAAO,CAAC,KAAK,CAA8B;gBAEvB,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE;IAOjE,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,oBAAoB,GAAG,OAAO;IAuB/D,wBAAwB,CAAC,GAAG,EAAE,wBAAwB,GAAG,OAAO;IAWhE,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,oBAAoB;IASnD,KAAK,CAAC,GAAG,EAAE,MAAM,GAAG,sBAAsB;IAQ1C,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,KAAK,GAAG,IAAI;IAI9C,QAAQ,IAAI,yBAAyB;CAK7C"}

package/dist/local_state_store.js

@@ -0,0 +1,82 @@
+"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.LocalStateStore = void 0;
+const protocol_1 = require("./generated/proto/protocol");
+const empty_1 = require("./generated/google/protobuf/empty");
+const utils_1 = require("./utils/utils");
+class LocalStateStore {
+    isPartial;
+    state;
+    constructor(isPartial, state) {
+        this.isPartial = isPartial;
+        this.state = new Map(state.map(({ key, value }) => [key.toString(), value]));
+    }
+    // Returns true if completed
+    tryCompleteGet(key, msg) {
+        const stateEntry = this.state.get(key);
+        if (stateEntry === undefined) {
+            if (this.isPartial) {
+                // Partial eager state, so retrieve state from the runtime
+                return false;
+            }
+            else {
+                // Complete eager state, so state entry is null
+                msg.empty = empty_1.Empty.create({});
+                return true;
+            }
+        }
+        if (stateEntry instanceof Buffer) {
+            msg.value = stateEntry;
+        }
+        else {
+            // stateEntry is Empty
+            msg.empty = stateEntry;
+        }
+        return true;
+    }
+    // Returns true if completed
+    tryCompletedGetStateKeys(msg) {
+        if (this.isPartial) {
+            return false;
+        }
+        msg.value = protocol_1.GetStateKeysEntryMessage_StateKeys.create({
+            keys: Array.from(this.state.keys()).map((b) => Buffer.from(b)),
+        });
+        return true;
+    }
+    set(key, value) {
+        const bytes = Buffer.from((0, utils_1.jsonSerialize)(value));
+        this.state.set(key, bytes);
+        return protocol_1.SetStateEntryMessage.create({
+            key: Buffer.from(key, "utf8"),
+            value: bytes,
+        });
+    }
+    clear(key) {
+        this.state.set(key, empty_1.Empty.create({}));
+        return protocol_1.ClearStateEntryMessage.create({ key: Buffer.from(key) });
+    }
+    // When a GetState request does not have a local entry and we have partial state,
+    // then the request goes to the runtime.
+    // When we get the response of the runtime, we add the state to the localStateStore.
+    add(key, result) {
+        this.state.set(key, result);
+    }
+    clearAll() {
+        this.state.clear();
+        this.isPartial = false;
+        return {};
+    }
+}
+exports.LocalStateStore = LocalStateStore;
+//# sourceMappingURL=local_state_store.js.map

package/dist/local_state_store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"local_state_store.js","sourceRoot":"","sources":["../src/local_state_store.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,yDAQoC;AACpC,6DAA0D;AAC1D,yCAA8C;AAE9C,MAAa,eAAe;IAGN;IAFZ,KAAK,CAA8B;IAE3C,YAAoB,SAAkB,EAAE,KAAgC;QAApD,cAAS,GAAT,SAAS,CAAS;QACpC,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,CAClB,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC,CACvD,CAAC;IACJ,CAAC;IAED,4BAA4B;IACrB,cAAc,CAAC,GAAW,EAAE,GAAyB;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,UAAU,KAAK,SAAS,EAAE;YAC5B,IAAI,IAAI,CAAC,SAAS,EAAE;gBAClB,0DAA0D;gBAC1D,OAAO,KAAK,CAAC;aACd;iBAAM;gBACL,+CAA+C;gBAC/C,GAAG,CAAC,KAAK,GAAG,aAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC7B,OAAO,IAAI,CAAC;aACb;SACF;QAED,IAAI,UAAU,YAAY,MAAM,EAAE;YAChC,GAAG,CAAC,KAAK,GAAG,UAAU,CAAC;SACxB;aAAM;YACL,sBAAsB;YACtB,GAAG,CAAC,KAAK,GAAG,UAAU,CAAC;SACxB;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,4BAA4B;IACrB,wBAAwB,CAAC,GAA6B;QAC3D,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,OAAO,KAAK,CAAC;SACd;QAED,GAAG,CAAC,KAAK,GAAG,6CAAkC,CAAC,MAAM,CAAC;YACpD,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SAC/D,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,GAAG,CAAI,GAAW,EAAE,KAAQ;QACjC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,IAAA,qBAAa,EAAC,KAAK,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC3B,OAAO,+BAAoB,CAAC,MAAM,CAAC;YACjC,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;YAC7B,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,GAAW;QACtB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,aAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;QACtC,OAAO,iCAAsB,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,iFAAiF;IACjF,wCAAwC;IACxC,oFAAoF;IAC7E,GAAG,CAAC,GAAW,EAAE,MAAsB;QAC5C,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC9B,CAAC;IAEM,QAAQ;QACb,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,OAAO,EAAE,CAAC;IACZ,CAAC;CACF;AAtED,0CAsEC"}

package/dist/logger.d.ts

@@ -0,0 +1,19 @@
+/// <reference types="node" />
+export declare class LoggerContext {
+    readonly invocationId: string;
+    readonly additionalContext?: {
+        [name: string]: string;
+    } | undefined;
+    readonly fqMethodName: string;
+    constructor(invocationId: string, packageName: string, serviceName: string, methodName: string, additionalContext?: {
+        [name: string]: string;
+    } | undefined);
+}
+export declare function createRestateConsole(context?: LoggerContext, filter?: () => boolean): Console;
+/**
+ * This is a simple console without contextual info.
+ *
+ * This should be used only in cases where no contextual info is available.
+ */
+export declare const rlog: Console;
+//# sourceMappingURL=logger.d.ts.map

package/dist/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":";AAcA,qBAAa,aAAa;IAItB,QAAQ,CAAC,YAAY,EAAE,MAAM;IAI7B,QAAQ,CAAC,iBAAiB,CAAC;;;IAP7B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;gBAGnB,YAAY,EAAE,MAAM,EAC7B,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,EACnB,UAAU,EAAE,MAAM,EACT,iBAAiB,CAAC;;iBAA4B;CAM1D;AAeD,wBAAgB,oBAAoB,CAClC,OAAO,CAAC,EAAE,aAAa,EACvB,MAAM,CAAC,EAAE,MAAM,OAAO,GACrB,OAAO,CAmET;AAED;;;;GAIG;AACH,eAAO,MAAM,IAAI,SAAyB,CAAC"}

package/dist/logger.js

@@ -0,0 +1,90 @@
+"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.rlog = exports.createRestateConsole = exports.LoggerContext = void 0;
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/* eslint-disable no-console */
+class LoggerContext {
+    invocationId;
+    additionalContext;
+    fqMethodName;
+    constructor(invocationId, packageName, serviceName, methodName, additionalContext) {
+        this.invocationId = invocationId;
+        this.additionalContext = additionalContext;
+        this.fqMethodName = packageName
+            ? `${packageName}.${serviceName}/${methodName}`
+            : `${serviceName}/${methodName}`;
+    }
+}
+exports.LoggerContext = LoggerContext;
+function formatLogPrefix(context) {
+    if (context === undefined) {
+        return "[restate] ";
+    }
+    let prefix = `[restate] [${context.fqMethodName}][${context.invocationId}]`;
+    if (context.additionalContext !== undefined) {
+        for (const [k, v] of Object.entries(context.additionalContext)) {
+            prefix = prefix + `[${k}: ${v}]`;
+        }
+    }
+    return prefix;
+}
+function createRestateConsole(context, filter) {
+    const prefix = formatLogPrefix(context);
+    const restate_logger = Object.create(console);
+    const shouldLog = filter ?? (() => true);
+    restate_logger.log = (message, ...optionalParams) => {
+        if (!shouldLog()) {
+            return;
+        }
+        console.log(prefix + `[${new Date().toISOString()}] LOG: ` + message, ...optionalParams);
+    };
+    restate_logger.info = (message, ...optionalParams) => {
+        if (!shouldLog()) {
+            return;
+        }
+        console.info(prefix + `[${new Date().toISOString()}] INFO: ` + message, ...optionalParams);
+    };
+    restate_logger.warn = (message, ...optionalParams) => {
+        if (!shouldLog()) {
+            return;
+        }
+        console.warn(prefix + `[${new Date().toISOString()}] WARN: ` + message, ...optionalParams);
+    };
+    restate_logger.error = (message, ...optionalParams) => {
+        if (!shouldLog()) {
+            return;
+        }
+        console.error(prefix + `[${new Date().toISOString()}] ERROR: ` + message, ...optionalParams);
+    };
+    restate_logger.debug = (message, ...optionalParams) => {
+        if (!shouldLog()) {
+            return;
+        }
+        console.debug(prefix + `[${new Date().toISOString()}] DEBUG: ` + message, ...optionalParams);
+    };
+    restate_logger.trace = (message, ...optionalParams) => {
+        if (!shouldLog()) {
+            return;
+        }
+        console.trace(prefix + `[${new Date().toISOString()}] TRACE: ` + message, ...optionalParams);
+    };
+    return restate_logger;
+}
+exports.createRestateConsole = createRestateConsole;
+/**
+ * This is a simple console without contextual info.
+ *
+ * This should be used only in cases where no contextual info is available.
+ */
+exports.rlog = createRestateConsole();
+//# sourceMappingURL=logger.js.map

package/dist/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,uDAAuD;AACvD,+BAA+B;AAE/B,MAAa,aAAa;IAIb;IAIA;IAPF,YAAY,CAAS;IAE9B,YACW,YAAoB,EAC7B,WAAmB,EACnB,WAAmB,EACnB,UAAkB,EACT,iBAA8C;QAJ9C,iBAAY,GAAZ,YAAY,CAAQ;QAIpB,sBAAiB,GAAjB,iBAAiB,CAA6B;QAEvD,IAAI,CAAC,YAAY,GAAG,WAAW;YAC7B,CAAC,CAAC,GAAG,WAAW,IAAI,WAAW,IAAI,UAAU,EAAE;YAC/C,CAAC,CAAC,GAAG,WAAW,IAAI,UAAU,EAAE,CAAC;IACrC,CAAC;CACF;AAdD,sCAcC;AAED,SAAS,eAAe,CAAC,OAAuB;IAC9C,IAAI,OAAO,KAAK,SAAS,EAAE;QACzB,OAAO,YAAY,CAAC;KACrB;IACD,IAAI,MAAM,GAAG,cAAc,OAAO,CAAC,YAAY,KAAK,OAAO,CAAC,YAAY,GAAG,CAAC;IAC5E,IAAI,OAAO,CAAC,iBAAiB,KAAK,SAAS,EAAE;QAC3C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE;YAC9D,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC;SAClC;KACF;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAgB,oBAAoB,CAClC,OAAuB,EACvB,MAAsB;IAEtB,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAE9C,MAAM,SAAS,GAAkB,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAExD,cAAc,CAAC,GAAG,GAAG,CAAC,OAAa,EAAE,GAAG,cAAqB,EAAE,EAAE;QAC/D,IAAI,CAAC,SAAS,EAAE,EAAE;YAChB,OAAO;SACR;QACD,OAAO,CAAC,GAAG,CACT,MAAM,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,SAAS,GAAG,OAAO,EACxD,GAAG,cAAc,CAClB,CAAC;IACJ,CAAC,CAAC;IAEF,cAAc,CAAC,IAAI,GAAG,CAAC,OAAa,EAAE,GAAG,cAAqB,EAAE,EAAE;QAChE,IAAI,CAAC,SAAS,EAAE,EAAE;YAChB,OAAO;SACR;QACD,OAAO,CAAC,IAAI,CACV,MAAM,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,UAAU,GAAG,OAAO,EACzD,GAAG,cAAc,CAClB,CAAC;IACJ,CAAC,CAAC;IAEF,cAAc,CAAC,IAAI,GAAG,CAAC,OAAa,EAAE,GAAG,cAAqB,EAAE,EAAE;QAChE,IAAI,CAAC,SAAS,EAAE,EAAE;YAChB,OAAO;SACR;QACD,OAAO,CAAC,IAAI,CACV,MAAM,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,UAAU,GAAG,OAAO,EACzD,GAAG,cAAc,CAClB,CAAC;IACJ,CAAC,CAAC;IAEF,cAAc,CAAC,KAAK,GAAG,CAAC,OAAa,EAAE,GAAG,cAAqB,EAAE,EAAE;QACjE,IAAI,CAAC,SAAS,EAAE,EAAE;YAChB,OAAO;SACR;QACD,OAAO,CAAC,KAAK,CACX,MAAM,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,WAAW,GAAG,OAAO,EAC1D,GAAG,cAAc,CAClB,CAAC;IACJ,CAAC,CAAC;IAEF,cAAc,CAAC,KAAK,GAAG,CAAC,OAAa,EAAE,GAAG,cAAqB,EAAE,EAAE;QACjE,IAAI,CAAC,SAAS,EAAE,EAAE;YAChB,OAAO;SACR;QACD,OAAO,CAAC,KAAK,CACX,MAAM,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,WAAW,GAAG,OAAO,EAC1D,GAAG,cAAc,CAClB,CAAC;IACJ,CAAC,CAAC;IAEF,cAAc,CAAC,KAAK,GAAG,CAAC,OAAa,EAAE,GAAG,cAAqB,EAAE,EAAE;QACjE,IAAI,CAAC,SAAS,EAAE,EAAE;YAChB,OAAO;SACR;QACD,OAAO,CAAC,KAAK,CACX,MAAM,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,WAAW,GAAG,OAAO,EAC1D,GAAG,cAAc,CAClB,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,cAAc,CAAC;AACxB,CAAC;AAtED,oDAsEC;AAED;;;;GAIG;AACU,QAAA,IAAI,GAAG,oBAAoB,EAAE,CAAC"}