@restatedev/restate-sdk 0.7.3-worker → 0.8.1

package/dist/io/decoder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"decoder.d.ts","sourceRoot":"","sources":["../../src/io/decoder.ts"],"names":[],"mappings":";;AAsBA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAU,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA4EjD,wBAAgB,aAAa,IAAI,MAAM,CAAC,SAAS,CAiBhD;AAMD,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,EAAE,CAS7D;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,EAAE,CAsB9D"}
+{"version":3,"file":"decoder.d.ts","sourceRoot":"","sources":["../../src/io/decoder.ts"],"names":[],"mappings":";;AAsBA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAU,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAUjD,eAAO,MAAM,0BAA0B,IAAI,CAAC;AAgF5C,wBAAgB,aAAa,IAAI,MAAM,CAAC,SAAS,CAiBhD;AAMD,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,EAAE,CAS7D;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,EAAE,CAsB9D"}

package/dist/io/decoder.js

@@ -0,0 +1,140 @@
+"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.decodeMessagesBuffer = exports.decodeLambdaBody = exports.streamDecoder = exports.SUPPORTED_PROTOCOL_VERSION = void 0;
+// This is a NodeJs stream transformer. It is used to convert a chunked stream of bytes to
+// a stream of JavaScript objects of the form { header: .. , message: ..} where:
+// * header has some information about the frame like, the message type, and some flags.
+// * message is the Protobuf decoded message.
+//
+// To use this one would need to do the following:
+//
+// let decodedStream = stream.pipe(streamDecoder());
+//
+// at this point the decodedStream is a high level stream of objects {header, message}
+const stream_1 = __importDefault(require("stream"));
+const protocol_1 = require("../types/protocol");
+const types_1 = require("../types/types");
+const assert_1 = __importDefault(require("assert"));
+const errors_1 = require("../types/errors");
+const WAITING_FOR_HEADER = 0;
+const WAITING_FOR_BODY = 1;
+exports.SUPPORTED_PROTOCOL_VERSION = 1;
+function initalDecoderState(buf) {
+    return {
+        state: WAITING_FOR_HEADER,
+        header: undefined,
+        buf,
+    };
+}
+function appendBufferToDecoder(state, chunk) {
+    state.buf = Buffer.concat([state.buf, chunk]);
+}
+function decodeMessages(decoderState, out) {
+    let buf = decoderState.buf;
+    while (buf.length > 0 || decoderState.state === WAITING_FOR_BODY) {
+        switch (decoderState.state) {
+            case WAITING_FOR_HEADER: {
+                (0, assert_1.default)(decoderState.header === undefined);
+                if (buf.length < 8) {
+                    decoderState.buf = buf;
+                    return decoderState;
+                }
+                const h = buf.readBigUInt64BE();
+                buf = buf.subarray(8);
+                const materializedHeader = types_1.Header.fromU64be(h);
+                decoderState.header = materializedHeader;
+                decoderState.state = WAITING_FOR_BODY;
+                // Check protocol version
+                if (materializedHeader.protocolVersion !== undefined &&
+                    materializedHeader.protocolVersion !== exports.SUPPORTED_PROTOCOL_VERSION) {
+                    throw new Error(`Unsupported protocol version ${materializedHeader.protocolVersion}, only version ${exports.SUPPORTED_PROTOCOL_VERSION} is supported`);
+                }
+                break;
+            }
+            case WAITING_FOR_BODY: {
+                const header = decoderState.header;
+                (0, assert_1.default)(header !== undefined);
+                if (buf.length < header.frameLength) {
+                    decoderState.buf = buf;
+                    return decoderState;
+                }
+                const frame = buf.subarray(0, header.frameLength);
+                buf = buf.subarray(header.frameLength);
+                decoderState.state = WAITING_FOR_HEADER;
+                decoderState.header = undefined;
+                const pbType = protocol_1.PROTOBUF_MESSAGE_BY_TYPE.get(header.messageType);
+                if (pbType === undefined) {
+                    throw new Error("Got unknown message type " + header.messageType);
+                }
+                else {
+                    const message = pbType.decode(frame);
+                    out.push(new types_1.Message(header.messageType, message, header.completedFlag, header.protocolVersion, header.requiresAckFlag));
+                }
+                break;
+            }
+        }
+    }
+    decoderState.buf = buf;
+    return decoderState;
+}
+function streamDecoder() {
+    let decoderState = initalDecoderState(Buffer.alloc(0));
+    return new stream_1.default.Transform({
+        writableObjectMode: true,
+        objectMode: true,
+        transform(chunk, _encoding, cb) {
+            try {
+                appendBufferToDecoder(decoderState, chunk);
+                decoderState = decodeMessages(decoderState, this);
+                cb();
+            }
+            catch (e) {
+                cb((0, errors_1.ensureError)(e), null);
+            }
+        },
+    });
+}
+exports.streamDecoder = streamDecoder;
+// Decodes messages from Lambda requests to an array of headers + protocol messages
+const base64regex = /^([0-9a-zA-Z+/]{4})*(([0-9a-zA-Z+/]{2}==)|([0-9a-zA-Z+/]{3}=))?$/;
+function decodeLambdaBody(msgBase64) {
+    if (!base64regex.test(msgBase64)) {
+        throw new Error("Cannot parse the lambda request body, body was not valid base64 encoded: " +
+            msgBase64);
+    }
+    const buffer = Buffer.from(msgBase64, "base64");
+    return decodeMessagesBuffer(buffer);
+}
+exports.decodeLambdaBody = decodeLambdaBody;
+function decodeMessagesBuffer(buffer) {
+    const decodedEntries = [];
+    let finalState;
+    try {
+        finalState = decodeMessages(initalDecoderState(buffer), decodedEntries);
+    }
+    catch (e) {
+        const err = (0, errors_1.ensureError)(e);
+        throw new Error("Cannot parse the lambda request body, message was not a valid sequence of Restate messages: " +
+            err.message, { cause: err });
+    }
+    if (finalState.buf.length > 0) {
+        throw new Error("Cannot parse the request body: Trailing data (incomplete message) in request body: " +
+            finalState.buf.toString("hex"));
+    }
+    return decodedEntries;
+}
+exports.decodeMessagesBuffer = decodeMessagesBuffer;
+//# sourceMappingURL=decoder.js.map

package/dist/io/decoder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"decoder.js","sourceRoot":"","sources":["../../src/io/decoder.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;AAEH,0FAA0F;AAC1F,gFAAgF;AAChF,wFAAwF;AACxF,6CAA6C;AAC7C,EAAE;AACF,kDAAkD;AAClD,EAAE;AACF,oDAAoD;AACpD,EAAE;AACF,sFAAsF;AAEtF,oDAA4B;AAC5B,gDAA6D;AAC7D,0CAAiD;AACjD,oDAA4B;AAC5B,4CAA8C;AAK9C,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAC7B,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAEd,QAAA,0BAA0B,GAAG,CAAC,CAAC;AAE5C,SAAS,kBAAkB,CAAC,GAAW;IACrC,OAAO;QACL,KAAK,EAAE,kBAAkB;QACzB,MAAM,EAAE,SAAS;QACjB,GAAG;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAmB,EAAE,KAAa;IAC/D,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,cAAc,CAAC,YAA0B,EAAE,GAAW;IAC7D,IAAI,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC;IAE3B,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,KAAK,KAAK,gBAAgB,EAAE;QAChE,QAAQ,YAAY,CAAC,KAAK,EAAE;YAC1B,KAAK,kBAAkB,CAAC,CAAC;gBACvB,IAAA,gBAAM,EAAC,YAAY,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;gBAC1C,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE;oBAClB,YAAY,CAAC,GAAG,GAAG,GAAG,CAAC;oBACvB,OAAO,YAAY,CAAC;iBACrB;gBACD,MAAM,CAAC,GAAG,GAAG,CAAC,eAAe,EAAE,CAAC;gBAChC,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;gBACtB,MAAM,kBAAkB,GAAG,cAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC/C,YAAY,CAAC,MAAM,GAAG,kBAAkB,CAAC;gBACzC,YAAY,CAAC,KAAK,GAAG,gBAAgB,CAAC;gBAEtC,yBAAyB;gBACzB,IACE,kBAAkB,CAAC,eAAe,KAAK,SAAS;oBAChD,kBAAkB,CAAC,eAAe,KAAK,kCAA0B,EACjE;oBACA,MAAM,IAAI,KAAK,CACb,gCAAgC,kBAAkB,CAAC,eAAe,kBAAkB,kCAA0B,eAAe,CAC9H,CAAC;iBACH;gBAED,MAAM;aACP;YACD,KAAK,gBAAgB,CAAC,CAAC;gBACrB,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBACnC,IAAA,gBAAM,EAAC,MAAM,KAAK,SAAS,CAAC,CAAC;gBAE7B,IAAI,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,WAAW,EAAE;oBACnC,YAAY,CAAC,GAAG,GAAG,GAAG,CAAC;oBACvB,OAAO,YAAY,CAAC;iBACrB;gBACD,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;gBAClD,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gBACvC,YAAY,CAAC,KAAK,GAAG,kBAAkB,CAAC;gBACxC,YAAY,CAAC,MAAM,GAAG,SAAS,CAAC;gBAEhC,MAAM,MAAM,GAAG,mCAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gBAChE,IAAI,MAAM,KAAK,SAAS,EAAE;oBACxB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;iBACnE;qBAAM;oBACL,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACrC,GAAG,CAAC,IAAI,CACN,IAAI,eAAO,CACT,MAAM,CAAC,WAAW,EAClB,OAAO,EACP,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,eAAe,CACvB,CACF,CAAC;iBACH;gBACD,MAAM;aACP;SACF;KACF;IAED,YAAY,CAAC,GAAG,GAAG,GAAG,CAAC;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAgB,aAAa;IAC3B,IAAI,YAAY,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvD,OAAO,IAAI,gBAAM,CAAC,SAAS,CAAC;QAC1B,kBAAkB,EAAE,IAAI;QACxB,UAAU,EAAE,IAAI;QAEhB,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;YAC5B,IAAI;gBACF,qBAAqB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBAC3C,YAAY,GAAG,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;gBAClD,EAAE,EAAE,CAAC;aACN;YAAC,OAAO,CAAC,EAAE;gBACV,EAAE,CAAC,IAAA,oBAAW,EAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;aAC1B;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAjBD,sCAiBC;AAED,mFAAmF;AACnF,MAAM,WAAW,GACf,kEAAkE,CAAC;AAErE,SAAgB,gBAAgB,CAAC,SAAiB;IAChD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE;QAChC,MAAM,IAAI,KAAK,CACb,2EAA2E;YACzE,SAAS,CACZ,CAAC;KACH;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChD,OAAO,oBAAoB,CAAC,MAAM,CAAC,CAAC;AACtC,CAAC;AATD,4CASC;AAED,SAAgB,oBAAoB,CAAC,MAAc;IACjD,MAAM,cAAc,GAAc,EAAE,CAAC;IACrC,IAAI,UAAU,CAAC;IACf,IAAI;QACF,UAAU,GAAG,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,cAAc,CAAC,CAAC;KACzE;IAAC,OAAO,CAAC,EAAE;QACV,MAAM,GAAG,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,KAAK,CACb,8FAA8F;YAC5F,GAAG,CAAC,OAAO,EACb,EAAE,KAAK,EAAE,GAAG,EAAE,CACf,CAAC;KACH;IAED,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE;QAC7B,MAAM,IAAI,KAAK,CACb,qFAAqF;YACnF,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CACjC,CAAC;KACH;IAED,OAAO,cAAc,CAAC;AACxB,CAAC;AAtBD,oDAsBC"}

package/dist/io/encoder.d.ts

@@ -1,8 +1,7 @@
 /// <reference types="node" />
-/// <reference types="node" />
 import stream from "stream";
 import { Message } from "../types/types";
 export declare function streamEncoder(): stream.Transform;
 export declare function encodeMessage(msg: Message): Uint8Array;
-export declare function encodeMessages(messages: Message[]): Buffer;
+export declare function encodeMessages(messages: Message[]): Uint8Array;
 //# sourceMappingURL=encoder.d.ts.map

package/dist/io/encoder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../src/io/encoder.ts"],"names":[],"mappings":";;AAWA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAU,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGjD,wBAAgB,aAAa,IAAI,MAAM,CAAC,SAAS,CAYhD;AAED,wBAAgB,aAAa,CAAC,GAAG,EAAE,OAAO,GAAG,UAAU,CAEtD;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CA2C1D"}
+{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../src/io/encoder.ts"],"names":[],"mappings":";AAWA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAU,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGjD,wBAAgB,aAAa,IAAI,MAAM,CAAC,SAAS,CAYhD;AAED,wBAAgB,aAAa,CAAC,GAAG,EAAE,OAAO,GAAG,UAAU,CAEtD;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,UAAU,CAqC9D"}

package/dist/io/encoder.js

@@ -0,0 +1,68 @@
+"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.encodeMessages = exports.encodeMessage = exports.streamEncoder = void 0;
+const stream_1 = __importDefault(require("stream"));
+const protocol_1 = require("../types/protocol");
+const types_1 = require("../types/types");
+const minimal_1 = __importDefault(require("protobufjs/minimal"));
+function streamEncoder() {
+    return new stream_1.default.Transform({
+        writableObjectMode: true,
+        objectMode: true,
+        transform(msg, _encoding, cb) {
+            // We do not catch errors here because we want them to be handled at the Connection level,
+            // so we can close the state machine.
+            const result = encodeMessage(msg);
+            cb(null, result);
+        },
+    });
+}
+exports.streamEncoder = streamEncoder;
+function encodeMessage(msg) {
+    return encodeMessages([msg]);
+}
+exports.encodeMessage = encodeMessage;
+function encodeMessages(messages) {
+    const offsets = [];
+    const headers = [];
+    let off = 0;
+    const writer = minimal_1.default.Writer.create();
+    for (const message of messages) {
+        const pbType = protocol_1.PROTOBUF_MESSAGE_BY_TYPE.get(BigInt(message.messageType));
+        if (pbType === undefined) {
+            throw new Error("Trying to encode a message with unknown message type " +
+                message.messageType);
+        }
+        offsets.push(off);
+        writer.fixed64(0);
+        pbType.encode(message.message, writer);
+        const messageLen = writer.len - 8 - off;
+        off = writer.len;
+        const header = new types_1.Header(BigInt(message.messageType), messageLen, message.completed, message.protocolVersion, // only set for incoming start message
+        message.requiresAck);
+        const header64 = header.toU64be();
+        headers.push(header64);
+    }
+    const buffer = writer.finish();
+    for (let i = 0; i < offsets.length; i++) {
+        const offset = offsets[i];
+        const header = headers[i];
+        buffer.writeBigUInt64BE(header, offset);
+    }
+    return buffer;
+}
+exports.encodeMessages = encodeMessages;
+//# sourceMappingURL=encoder.js.map

package/dist/io/encoder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"encoder.js","sourceRoot":"","sources":["../../src/io/encoder.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;AAEH,oDAA4B;AAC5B,gDAA6D;AAC7D,0CAAiD;AACjD,iEAAqC;AAErC,SAAgB,aAAa;IAC3B,OAAO,IAAI,gBAAM,CAAC,SAAS,CAAC;QAC1B,kBAAkB,EAAE,IAAI;QACxB,UAAU,EAAE,IAAI;QAEhB,SAAS,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YAC1B,0FAA0F;YAC1F,qCAAqC;YACrC,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;YAClC,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACnB,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAZD,sCAYC;AAED,SAAgB,aAAa,CAAC,GAAY;IACxC,OAAO,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AAC/B,CAAC;AAFD,sCAEC;AAED,SAAgB,cAAc,CAAC,QAAmB;IAChD,MAAM,OAAO,GAAG,EAAE,CAAC;IACnB,MAAM,OAAO,GAAG,EAAE,CAAC;IACnB,IAAI,GAAG,GAAG,CAAC,CAAC;IAEZ,MAAM,MAAM,GAAG,iBAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;IACnC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;QAC9B,MAAM,MAAM,GAAG,mCAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;QACzE,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,MAAM,IAAI,KAAK,CACb,uDAAuD;gBACrD,OAAO,CAAC,WAAW,CACtB,CAAC;SACH;QACD,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAClB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACvC,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;QACxC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QAEjB,MAAM,MAAM,GAAG,IAAI,cAAM,CACvB,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAC3B,UAAU,EACV,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,eAAe,EAAE,sCAAsC;QAC/D,OAAO,CAAC,WAAW,CACpB,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;QAClC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;KACxB;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,EAAY,CAAC;IACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;QACvC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KACzC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AArCD,wCAqCC"}

package/dist/journal.d.ts

@@ -1,7 +1,9 @@
 import * as p from "./types/protocol";
 import { Failure } from "./generated/proto/protocol";
 import { CompletionMessage, EntryAckMessage, OutputStreamEntryMessage, SuspensionMessage } from "./types/protocol";
+import { Message } from "./types/types";
 import { Invocation } from "./invocation";
+import { CompletablePromise } from "./utils/promises";
 export declare class Journal<I, O> {
     readonly invocation: Invocation<I, O>;
     private state;
@@ -19,9 +21,18 @@ export declare class Journal<I, O> {
     private checkJournalMatch;
     getCompletableIndices(): number[];
     private transitionState;
-    private incrementUserCodeIndex;
+    incrementUserCodeIndex(): void;
+    /**
+     * Read the next replay entry
+     */
+    readNextReplayEntry(): Message | undefined;
+    /**
+     * Append journal entry. This won't increment the journal index.
+     */
+    appendJournalEntry(messageType: bigint, message: p.ProtocolMessage | Uint8Array): Promise<unknown>;
     isClosed(): boolean;
     isProcessing(): boolean;
+    isReplaying(): boolean;
     getUserCodeJournalIndex(): number;
     close(): void;
     outputMsgWasReplayed(): boolean | undefined;
@@ -30,9 +41,7 @@ export declare class Journal<I, O> {
 export declare class JournalEntry {
     readonly messageType: bigint;
     readonly message: p.ProtocolMessage | Uint8Array;
-    promise: Promise<any>;
-    resolve: (value: any) => void;
-    reject: (reason?: any) => void;
+    completablePromise: CompletablePromise<unknown>;
     constructor(messageType: bigint, message: p.ProtocolMessage | Uint8Array);
 }
 export declare enum NewExecutionState {

package/dist/journal.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"journal.d.ts","sourceRoot":"","sources":["../src/journal.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,CAAC,MAAM,kBAAkB,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAML,iBAAiB,EACjB,eAAe,EAMf,wBAAwB,EAQxB,iBAAiB,EAClB,MAAM,kBAAkB,CAAC;AAI1B,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAK1C,qBAAa,OAAO,CAAC,CAAC,EAAE,CAAC;IASX,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC;IARjD,OAAO,CAAC,KAAK,CAA+B;IAE5C,OAAO,CAAC,oBAAoB,CAAK;IAIjC,OAAO,CAAC,qBAAqB,CAAmC;gBAE3C,UAAU,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC;IAejD,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,2BAA2B;IAa5C,qBAAqB,CAC1B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU,GAEtC,OAAO,CAAC,GAAG,GAAG,SAAS,CAAC;IAgFpB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAKpC,8BAA8B,CAAC,CAAC,EAAE,iBAAiB;IA4BnD,qBAAqB,CAAC,CAAC,EAAE,eAAe;IAa/C,OAAO,CAAC,YAAY;IA6HpB,aAAa,CAAC,CAAC,EACb,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,YAAY,EAC1B,KAAK,EAAE,CAAC,GAAG,SAAS,EACpB,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,EAC7B,sBAAsB,CAAC,EAAE,OAAO;IAclC,oBAAoB,CAClB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,wBAAwB,GAAG,iBAAiB;IAiBvD,OAAO,CAAC,iBAAiB;IAmBlB,qBAAqB,IAAI,MAAM,EAAE;IAOxC,OAAO,CAAC,eAAe;IAcvB,OAAO,CAAC,sBAAsB;IAUvB,QAAQ,IAAI,OAAO;IAInB,YAAY,IAAI,OAAO;IAIvB,uBAAuB,IAAI,MAAM;IAIjC,KAAK;IAIL,oBAAoB;IAgB3B,uBAAuB;CAGxB;AAED,qBAAa,YAAY;IASrB,QAAQ,CAAC,WAAW,EAAE,MAAM;IAC5B,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU;IAR3C,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;IAEtB,OAAO,EAAG,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAE/B,MAAM,EAAG,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,IAAI,CAAC;gBAG5B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU;CAQnD;AAOD,oBAAY,iBAAiB;IAC3B,SAAS,cAAc;IACvB,UAAU,eAAe;IACzB,MAAM,WAAW;CAClB"}
+{"version":3,"file":"journal.d.ts","sourceRoot":"","sources":["../src/journal.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,CAAC,MAAM,kBAAkB,CAAC;AACtC,OAAO,EACL,OAAO,EAER,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAQL,iBAAiB,EACjB,eAAe,EAQf,wBAAwB,EAQxB,iBAAiB,EAClB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAE1C,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAItD,qBAAa,OAAO,CAAC,CAAC,EAAE,CAAC;IASX,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC;IARjD,OAAO,CAAC,KAAK,CAA+B;IAE5C,OAAO,CAAC,oBAAoB,CAAK;IAIjC,OAAO,CAAC,qBAAqB,CAAmC;gBAE3C,UAAU,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC;IAejD,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,2BAA2B;IAa5C,qBAAqB,CAC1B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU,GAEtC,OAAO,CAAC,GAAG,GAAG,SAAS,CAAC;IAgFpB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAKpC,8BAA8B,CAAC,CAAC,EAAE,iBAAiB;IAoCnD,qBAAqB,CAAC,CAAC,EAAE,eAAe;IAa/C,OAAO,CAAC,YAAY;IAyIpB,aAAa,CAAC,CAAC,EACb,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,YAAY,EAC1B,KAAK,EAAE,CAAC,GAAG,SAAS,EACpB,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,EAC7B,sBAAsB,CAAC,EAAE,OAAO;IAclC,oBAAoB,CAClB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,wBAAwB,GAAG,iBAAiB;IAmBvD,OAAO,CAAC,iBAAiB;IAmBlB,qBAAqB,IAAI,MAAM,EAAE;IAOxC,OAAO,CAAC,eAAe;IAcvB,sBAAsB;IAUtB;;OAEG;IACI,mBAAmB;IAK1B;;OAEG;IACI,kBAAkB,CACvB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU,GACtC,OAAO,CAAC,OAAO,CAAC;IAMZ,QAAQ,IAAI,OAAO;IAInB,YAAY,IAAI,OAAO;IAIvB,WAAW,IAAI,OAAO;IAItB,uBAAuB,IAAI,MAAM;IAIjC,KAAK;IAIL,oBAAoB;IAgB3B,uBAAuB;CAGxB;AAED,qBAAa,YAAY;IAIrB,QAAQ,CAAC,WAAW,EAAE,MAAM;IAC5B,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU;IAJ3C,kBAAkB,EAAE,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBAG5C,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU;CAKnD;AAOD,oBAAY,iBAAiB;IAC3B,SAAS,cAAc;IACvB,UAAU,eAAe;IACzB,MAAM,WAAW;CAClB"}