@restatedev/restate-sdk 1.0.1 → 1.1.1

package/dist/cjs/src/endpoint/endpoint_builder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"endpoint_builder.d.ts","sourceRoot":"","sources":["../../../../src/endpoint/endpoint_builder.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EACV,iBAAiB,EACjB,uBAAuB,EACvB,kBAAkB,EACnB,MAAM,8BAA8B,CAAC;AAGtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAOxD,OAAO,KAAK,KAAK,SAAS,MAAM,uBAAuB,CAAC;AACxD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAqBxD,qBAAa,eAAe;IAC1B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqC;IAC9D,OAAO,CAAC,OAAO,CAAC,CAAW;IAE3B,IAAW,MAAM,IAAI,QAAQ,GAAG,SAAS,CAExC;IAEM,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAI7D,YAAY,CAAC,SAAS,EAAE,SAAS;IAIjC,IAAI,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,EAC7B,UAAU,EACN,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,GACvB,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,GAC7B,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC;IA4BvB,cAAc,CAAC,GAAG,IAAI,EAAE,MAAM,EAAE;IAWvC,gBAAgB,CAAC,YAAY,EAAE,SAAS,CAAC,YAAY,GAAG,SAAS,CAAC,QAAQ;IAa1E,OAAO,CAAC,oBAAoB;IAoB5B,OAAO,CAAC,0BAA0B;IAmBlC,OAAO,CAAC,2BAA2B;CAkBpC"}
+{"version":3,"file":"endpoint_builder.d.ts","sourceRoot":"","sources":["../../../../src/endpoint/endpoint_builder.ts"],"names":[],"mappings":";AAcA,OAAO,KAAK,EACV,iBAAiB,EACjB,uBAAuB,EACvB,kBAAkB,EACnB,MAAM,8BAA8B,CAAC;AAGtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAOxD,OAAO,KAAK,KAAK,SAAS,MAAM,uBAAuB,CAAC;AACxD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAEL,KAAK,MAAM,EAGZ,MAAM,cAAc,CAAC;AAoBtB,qBAAa,eAAe;IAC1B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqC;IACvD,MAAM,EAAE,MAAM,CAAiB;IAEtC;;;;OAIG;IACI,IAAI,UAAuD;IAElE,OAAO,CAAC,OAAO,CAAC,CAAW;IAE3B,IAAW,MAAM,IAAI,QAAQ,GAAG,SAAS,CAExC;IAEM,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAI7D,YAAY,CAAC,SAAS,EAAE,SAAS;IAIjC,IAAI,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,EAC7B,UAAU,EACN,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,GACvB,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,GAC7B,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC;IA4BvB,cAAc,CAAC,GAAG,IAAI,EAAE,MAAM,EAAE;IAWhC,SAAS,CAAC,SAAS,EAAE,MAAM;IAMlC,gBAAgB,CAAC,YAAY,EAAE,SAAS,CAAC,YAAY,GAAG,SAAS,CAAC,QAAQ;IAa1E,OAAO,CAAC,oBAAoB;IAoB5B,OAAO,CAAC,0BAA0B;IAmBlC,OAAO,CAAC,2BAA2B;CAkBpC"}

package/dist/cjs/src/endpoint/endpoint_builder.js

@@ -14,6 +14,7 @@ exports.EndpointBuilder = void 0;
 const rpc_js_1 = require("../types/rpc.js");
 const components_js_1 = require("../types/components.js");
 const v1_js_1 = require("./request_signing/v1.js");
+const logger_js_1 = require("../logger.js");
 function isServiceDefinition(m) {
     return m && m.service !== undefined;
 }
@@ -25,6 +26,13 @@ function isWorkflowDefinition(m) {
 }
 class EndpointBuilder {
     services = new Map();
+    logger = logger_js_1.defaultLogger;
+    /**
+     * This is a simple console without contextual info.
+     *
+     * This should be used only in cases where no contextual info is available.
+     */
+    rlog = (0, logger_js_1.createRestateConsole)(this.logger, logger_js_1.LogSource.SYSTEM);
     _keySet;
     get keySet() {
         return this._keySet;
@@ -70,6 +78,11 @@ class EndpointBuilder {
         (0, v1_js_1.parseKeySetV1)(keys).forEach((buffer, key) => this._keySet?.set(key, buffer));
         return this;
     }
+    setLogger(newLogger) {
+        this.logger = newLogger;
+        this.rlog = (0, logger_js_1.createRestateConsole)(this.logger, logger_js_1.LogSource.SYSTEM);
+        return this;
+    }
     computeDiscovery(protocolMode) {
         const services = [...this.services.values()].map((c) => c.discovery());
         const endpoint = {

package/dist/cjs/src/endpoint/endpoint_builder.js.map

@@ -1 +1 @@
-{"version":3,"file":"endpoint_builder.js","sourceRoot":"","sources":["../../../../src/endpoint/endpoint_builder.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAWH,4CAAiD;AAEjD,0DAIgC;AAIhC,mDAAwD;AAExD,SAAS,mBAAmB,CAC1B,CAAsB;IAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,KAAK,SAAS,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CACzB,CAAsB;IAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC;AACrC,CAAC;AAED,SAAS,oBAAoB,CAC3B,CAAsB;IAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC;AACvC,CAAC;AAED,MAAa,eAAe;IACT,QAAQ,GAA2B,IAAI,GAAG,EAAE,CAAC;IACtD,OAAO,CAAY;IAE3B,IAAW,MAAM;QACf,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEM,eAAe,CAAC,aAAqB;QAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC;IAEM,YAAY,CAAC,SAAoB;QACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC;IAEM,IAAI,CACT,UAG4B;QAE5B,IAAI,mBAAmB,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC;YACrC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,kCAAkC,CAAC,CAAC;YAC1D,CAAC;YACD,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3C,CAAC;aAAM,IAAI,kBAAkB,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1C,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;YACpC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,SAAS,CAAC,iCAAiC,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,CAAC,0BAA0B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5C,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC;YACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,SAAS,CAAC,mCAAmC,CAAC,CAAC;YAC3D,CAAC;YACD,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACnD,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,SAAS,CACjB,sEAAsE,CACvE,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,cAAc,CAAC,GAAG,IAAc;QACrC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,IAAA,qBAAa,EAAC,IAAI,CAAC,CAAC;YACnC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAA,qBAAa,EAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAC1C,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAC/B,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,YAAoC;QACnD,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;QAEvE,MAAM,QAAQ,GAAuB;YACnC,YAAY;YACZ,kBAAkB,EAAE,CAAC;YACrB,kBAAkB,EAAE,CAAC;YACrB,QAAQ;SACT,CAAC;QAEF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,oBAAoB,CAAC,IAAY,EAAE,MAAW;QACpD,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,gCAAgB,CAAC,IAAI,CAAC,CAAC;QAE7C,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAC3C,MAA8B,CAC/B,EAAE,CAAC;YACF,MAAM,OAAO,GAAG,uBAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,4BAA4B,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAEO,0BAA0B,CAAC,IAAY,EAAE,MAAW;QAC1D,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,sCAAsB,CAAC,IAAI,CAAC,CAAC;QAEnD,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAC3C,MAA8B,CAC/B,EAAE,CAAC;YACF,MAAM,OAAO,GAAG,uBAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,4BAA4B,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAEO,2BAA2B,CAAC,IAAY,EAAE,QAAa;QAC7D,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,iCAAiB,CAAC,IAAI,CAAC,CAAC;QAE9C,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAC3C,QAAgC,CACjC,EAAE,CAAC;YACF,MAAM,OAAO,GAAG,uBAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,4BAA4B,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC/B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;CACF;AAjID,0CAiIC"}
+{"version":3,"file":"endpoint_builder.js","sourceRoot":"","sources":["../../../../src/endpoint/endpoint_builder.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAWH,4CAAiD;AAEjD,0DAIgC;AAIhC,mDAAwD;AACxD,4CAKsB;AAEtB,SAAS,mBAAmB,CAC1B,CAAsB;IAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,KAAK,SAAS,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CACzB,CAAsB;IAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC;AACrC,CAAC;AAED,SAAS,oBAAoB,CAC3B,CAAsB;IAEtB,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC;AACvC,CAAC;AAED,MAAa,eAAe;IACT,QAAQ,GAA2B,IAAI,GAAG,EAAE,CAAC;IACvD,MAAM,GAAW,yBAAa,CAAC;IAEtC;;;;OAIG;IACI,IAAI,GAAG,IAAA,gCAAoB,EAAC,IAAI,CAAC,MAAM,EAAE,qBAAS,CAAC,MAAM,CAAC,CAAC;IAE1D,OAAO,CAAY;IAE3B,IAAW,MAAM;QACf,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEM,eAAe,CAAC,aAAqB;QAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC;IAEM,YAAY,CAAC,SAAoB;QACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC;IAEM,IAAI,CACT,UAG4B;QAE5B,IAAI,mBAAmB,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC;YACrC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,kCAAkC,CAAC,CAAC;YAC1D,CAAC;YACD,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3C,CAAC;aAAM,IAAI,kBAAkB,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1C,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC;YACpC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,SAAS,CAAC,iCAAiC,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,CAAC,0BAA0B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5C,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC;YACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,SAAS,CAAC,mCAAmC,CAAC,CAAC;YAC3D,CAAC;YACD,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACnD,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,SAAS,CACjB,sEAAsE,CACvE,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,cAAc,CAAC,GAAG,IAAc;QACrC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,IAAA,qBAAa,EAAC,IAAI,CAAC,CAAC;YACnC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAA,qBAAa,EAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAC1C,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAC/B,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,SAAS,CAAC,SAAiB;QAChC,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,IAAA,gCAAoB,EAAC,IAAI,CAAC,MAAM,EAAE,qBAAS,CAAC,MAAM,CAAC,CAAC;QAChE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,YAAoC;QACnD,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;QAEvE,MAAM,QAAQ,GAAuB;YACnC,YAAY;YACZ,kBAAkB,EAAE,CAAC;YACrB,kBAAkB,EAAE,CAAC;YACrB,QAAQ;SACT,CAAC;QAEF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,oBAAoB,CAAC,IAAY,EAAE,MAAW;QACpD,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,gCAAgB,CAAC,IAAI,CAAC,CAAC;QAE7C,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAC3C,MAA8B,CAC/B,EAAE,CAAC;YACF,MAAM,OAAO,GAAG,uBAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,4BAA4B,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAEO,0BAA0B,CAAC,IAAY,EAAE,MAAW;QAC1D,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,sCAAsB,CAAC,IAAI,CAAC,CAAC;QAEnD,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAC3C,MAA8B,CAC/B,EAAE,CAAC;YACF,MAAM,OAAO,GAAG,uBAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,4BAA4B,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAEO,2BAA2B,CAAC,IAAY,EAAE,QAAa;QAC7D,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,iCAAiB,CAAC,IAAI,CAAC,CAAC;QAE9C,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAC3C,QAAgC,CACjC,EAAE,CAAC;YACF,MAAM,OAAO,GAAG,uBAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,4BAA4B,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC/B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;CACF;AAhJD,0CAgJC"}

package/dist/cjs/src/endpoint/fetch_endpoint.d.ts

@@ -3,12 +3,14 @@ import type { ServiceDefinition, VirtualObjectDefinition, WorkflowDefinition } f
 import type { Component } from "../types/components.js";
 import type { KeySetV1 } from "./request_signing/v1.js";
 import type { RestateEndpointBase, ServiceBundle } from "../endpoint.js";
+import { ProtocolMode } from "../types/discovery.js";
+import type { Logger } from "../logger.js";
 /**
  * Generic Fetch encapsulates all the Restate services served by this endpoint.
  *
  *
  * @example
- * A typical endpoint served would look like this:
+ * A typical request-response handler would look like this:
  * ```
  * import * as restate from "@restatedev/restate-sdk/fetch";
  *
@@ -16,13 +18,26 @@ import type { RestateEndpointBase, ServiceBundle } from "../endpoint.js";
  *   .endpoint()
  *   .bind(myService)
  *   .handler();
+ * @example
+ * A typical bidirectional handler (works with http2 and some http1.1 servers) would look like this:
+ * ```
+ * import * as restate from "@restatedev/restate-sdk/fetch";
+ *
+ * export default restate
+ *   .endpoint()
+ *   .bidirectional()
+ *   .bind(myService)
+ *   .handler();
  */
 export interface FetchEndpoint extends RestateEndpointBase<FetchEndpoint> {
     handler(): {
         fetch: (request: Request) => Promise<Response>;
     };
+    bidirectional(set?: boolean): FetchEndpoint;
 }
 export declare class FetchEndpointImpl implements FetchEndpoint {
+    private protocolMode;
+    constructor(protocolMode: ProtocolMode);
     private builder;
     get keySet(): KeySetV1 | undefined;
     componentByName(componentName: string): Component | undefined;
@@ -30,6 +45,8 @@ export declare class FetchEndpointImpl implements FetchEndpoint {
     bindBundle(services: ServiceBundle): FetchEndpoint;
     bind<P extends string, M>(definition: ServiceDefinition<P, M> | VirtualObjectDefinition<P, M> | WorkflowDefinition<P, M>): FetchEndpoint;
     withIdentityV1(...keys: string[]): FetchEndpoint;
+    setLogger(newLogger: Logger): FetchEndpoint;
+    bidirectional(set?: boolean): FetchEndpoint;
     handler(): {
         fetch: (request: Request) => Promise<Response>;
     };

package/dist/cjs/src/endpoint/fetch_endpoint.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fetch_endpoint.d.ts","sourceRoot":"","sources":["../../../../src/endpoint/fetch_endpoint.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EACV,iBAAiB,EACjB,uBAAuB,EACvB,kBAAkB,EACnB,MAAM,8BAA8B,CAAC;AAEtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAExD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,KAAK,EAEV,mBAAmB,EACnB,aAAa,EACd,MAAM,gBAAgB,CAAC;AAIxB;;;;;;;;;;;;;GAaG;AACH,MAAM,WAAW,aAAc,SAAQ,mBAAmB,CAAC,aAAa,CAAC;IACvE,OAAO,IAAI;QAAE,KAAK,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAA;KAAE,CAAC;CAC/D;AAED,qBAAa,iBAAkB,YAAW,aAAa;IACrD,OAAO,CAAC,OAAO,CAA0C;IAEzD,IAAW,MAAM,IAAI,QAAQ,GAAG,SAAS,CAExC;IAEM,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAI7D,YAAY,CAAC,SAAS,EAAE,SAAS;IAIxC,UAAU,CAAC,QAAQ,EAAE,aAAa,GAAG,aAAa;IAK3C,IAAI,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,EAC7B,UAAU,EACN,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,GACvB,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,GAC7B,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,GAC3B,aAAa;IAKT,cAAc,CAAC,GAAG,IAAI,EAAE,MAAM,EAAE,GAAG,aAAa;IAKvD,OAAO,IAAI;QACT,KAAK,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;KAChD;CAIF"}
+{"version":3,"file":"fetch_endpoint.d.ts","sourceRoot":"","sources":["../../../../src/endpoint/fetch_endpoint.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EACV,iBAAiB,EACjB,uBAAuB,EACvB,kBAAkB,EACnB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,KAAK,EAEV,mBAAmB,EACnB,aAAa,EACd,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAE3C;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,WAAW,aAAc,SAAQ,mBAAmB,CAAC,aAAa,CAAC;IACvE,OAAO,IAAI;QAAE,KAAK,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAA;KAAE,CAAC;IAC9D,aAAa,CAAC,GAAG,CAAC,EAAE,OAAO,GAAG,aAAa,CAAC;CAC7C;AAED,qBAAa,iBAAkB,YAAW,aAAa;IACzC,OAAO,CAAC,YAAY;gBAAZ,YAAY,EAAE,YAAY;IAC9C,OAAO,CAAC,OAAO,CAA0C;IAEzD,IAAW,MAAM,IAAI,QAAQ,GAAG,SAAS,CAExC;IAEM,eAAe,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAI7D,YAAY,CAAC,SAAS,EAAE,SAAS;IAIxC,UAAU,CAAC,QAAQ,EAAE,aAAa,GAAG,aAAa;IAK3C,IAAI,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,EAC7B,UAAU,EACN,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,GACvB,uBAAuB,CAAC,CAAC,EAAE,CAAC,CAAC,GAC7B,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,GAC3B,aAAa;IAKT,cAAc,CAAC,GAAG,IAAI,EAAE,MAAM,EAAE,GAAG,aAAa;IAKhD,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa;IAK3C,aAAa,CAAC,GAAG,GAAE,OAAc,GAAG,aAAa;IAOxD,OAAO,IAAI;QACT,KAAK,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;KAChD;CAIF"}

package/dist/cjs/src/endpoint/fetch_endpoint.js

@@ -14,7 +14,12 @@ exports.FetchEndpointImpl = void 0;
 const endpoint_builder_js_1 = require("./endpoint_builder.js");
 const generic_js_1 = require("./handlers/generic.js");
 const fetch_js_1 = require("./handlers/fetch.js");
+const discovery_js_1 = require("../types/discovery.js");
 class FetchEndpointImpl {
+    protocolMode;
+    constructor(protocolMode) {
+        this.protocolMode = protocolMode;
+    }
     builder = new endpoint_builder_js_1.EndpointBuilder();
     get keySet() {
         return this.builder.keySet;
@@ -37,8 +42,18 @@ class FetchEndpointImpl {
         this.builder.withIdentityV1(...keys);
         return this;
     }
+    setLogger(newLogger) {
+        this.builder.setLogger(newLogger);
+        return this;
+    }
+    bidirectional(set = true) {
+        this.protocolMode = set
+            ? discovery_js_1.ProtocolMode.BIDI_STREAM
+            : discovery_js_1.ProtocolMode.REQUEST_RESPONSE;
+        return this;
+    }
     handler() {
-        const genericHandler = new generic_js_1.GenericHandler(this.builder);
+        const genericHandler = new generic_js_1.GenericHandler(this.builder, this.protocolMode);
         return (0, fetch_js_1.fetcher)(genericHandler);
     }
 }

package/dist/cjs/src/endpoint/fetch_endpoint.js.map

@@ -1 +1 @@
-{"version":3,"file":"fetch_endpoint.js","sourceRoot":"","sources":["../../../../src/endpoint/fetch_endpoint.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAWH,+DAAwD;AAMxD,sDAAuD;AACvD,kDAA8C;AAoB9C,MAAa,iBAAiB;IACpB,OAAO,GAAoB,IAAI,qCAAe,EAAE,CAAC;IAEzD,IAAW,MAAM;QACf,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;IAC7B,CAAC;IAEM,eAAe,CAAC,aAAqB;QAC1C,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC;IAEM,YAAY,CAAC,SAAoB;QACtC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED,UAAU,CAAC,QAAuB;QAChC,QAAQ,CAAC,gBAAgB,CAAC,IAAkC,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,IAAI,CACT,UAG4B;QAE5B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,cAAc,CAAC,GAAG,IAAc;QACrC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QAGL,MAAM,cAAc,GAAG,IAAI,2BAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,OAAO,IAAA,kBAAO,EAAC,cAAc,CAAC,CAAC;IACjC,CAAC;CACF;AAzCD,8CAyCC"}
+{"version":3,"file":"fetch_endpoint.js","sourceRoot":"","sources":["../../../../src/endpoint/fetch_endpoint.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AASH,+DAAwD;AAMxD,sDAAuD;AACvD,kDAA8C;AAC9C,wDAAqD;AAgCrD,MAAa,iBAAiB;IACR;IAApB,YAAoB,YAA0B;QAA1B,iBAAY,GAAZ,YAAY,CAAc;IAAG,CAAC;IAC1C,OAAO,GAAoB,IAAI,qCAAe,EAAE,CAAC;IAEzD,IAAW,MAAM;QACf,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;IAC7B,CAAC;IAEM,eAAe,CAAC,aAAqB;QAC1C,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC;IAEM,YAAY,CAAC,SAAoB;QACtC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED,UAAU,CAAC,QAAuB;QAChC,QAAQ,CAAC,gBAAgB,CAAC,IAAkC,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,IAAI,CACT,UAG4B;QAE5B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,cAAc,CAAC,GAAG,IAAc;QACrC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,SAAS,CAAC,SAAiB;QAChC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,aAAa,CAAC,MAAe,IAAI;QACtC,IAAI,CAAC,YAAY,GAAG,GAAG;YACrB,CAAC,CAAC,2BAAY,CAAC,WAAW;YAC1B,CAAC,CAAC,2BAAY,CAAC,gBAAgB,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QAGL,MAAM,cAAc,GAAG,IAAI,2BAAc,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QAC3E,OAAO,IAAA,kBAAO,EAAC,cAAc,CAAC,CAAC;IACjC,CAAC;CACF;AAtDD,8CAsDC"}

package/dist/cjs/src/endpoint/handlers/fetch.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fetch.d.ts","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/fetch.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,cAAc,CAAC;AAEnE,wBAAgB,OAAO,CAAC,OAAO,EAAE,cAAc;mBAEtB,OAAO,KAAG,QAAQ,QAAQ,CAAC;EAmBnD"}
+{"version":3,"file":"fetch.d.ts","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/fetch.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,cAAc,CAAC;AAEnE,wBAAgB,OAAO,CAAC,OAAO,EAAE,cAAc;mBAEtB,OAAO,KAAG,QAAQ,QAAQ,CAAC;EAkBnD"}

package/dist/cjs/src/endpoint/handlers/fetch.js

@@ -15,12 +15,11 @@ function fetcher(handler) {
     return {
         fetch: async (event) => {
             const url = event.url;
-            const body = new Uint8Array(await event.arrayBuffer());
             const headers = Object.fromEntries(event.headers.entries());
             const request = {
                 url,
                 headers,
-                body,
+                body: event.body,
             };
             const resp = await handler.handle(request);
             return new Response(resp.body, {

package/dist/cjs/src/endpoint/handlers/fetch.js.map

@@ -1 +1 @@
-{"version":3,"file":"fetch.js","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/fetch.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAIH,SAAgB,OAAO,CAAC,OAAuB;IAC7C,OAAO;QACL,KAAK,EAAE,KAAK,EAAE,KAAc,EAAqB,EAAE;YACjD,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;YACtB,MAAM,IAAI,GAAe,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAE5D,MAAM,OAAO,GAAmB;gBAC9B,GAAG;gBACH,OAAO;gBACP,IAAI;aACL,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE3C,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE;gBAC7B,MAAM,EAAE,IAAI,CAAC,UAAU;gBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AArBD,0BAqBC"}
+{"version":3,"file":"fetch.js","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/fetch.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAIH,SAAgB,OAAO,CAAC,OAAuB;IAC7C,OAAO;QACL,KAAK,EAAE,KAAK,EAAE,KAAc,EAAqB,EAAE;YACjD,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;YACtB,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAE5D,MAAM,OAAO,GAAmB;gBAC9B,GAAG;gBACH,OAAO;gBACP,IAAI,EAAE,KAAK,CAAC,IAAI;aACjB,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE3C,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE;gBAC7B,MAAM,EAAE,IAAI,CAAC,UAAU;gBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AApBD,0BAoBC"}

package/dist/cjs/src/endpoint/handlers/generic.d.ts

@@ -1,4 +1,7 @@
+/// <reference types="node" />
+import type { ProtocolMode } from "../../types/discovery.js";
 import type { EndpointBuilder } from "../endpoint_builder.js";
+import { type ReadableStream } from "node:stream/web";
 export interface Headers {
     [name: string]: string | string[] | undefined;
 }
@@ -11,21 +14,19 @@ export interface AdditionalContext {
 export interface RestateRequest {
     readonly url: string;
     readonly headers: Headers;
-    readonly body: Uint8Array;
+    readonly body: ReadableStream<Uint8Array> | null;
 }
 export interface RestateResponse {
     readonly headers: ResponseHeaders;
     readonly statusCode: number;
-    readonly body: Uint8Array;
+    readonly body: ReadableStream<Uint8Array>;
 }
 export interface RestateHandler {
     handle(request: RestateRequest, context?: AdditionalContext): Promise<RestateResponse>;
 }
 /**
  * This is an internal API to support 'fetch' like handlers.
- * This is for now supports only request-reply style of invocations
- * which are suitable for platforms like AWS Lambda, Cloudflare works,
- * and runtimes that only support HTTP/1.1 like Bun.
+ * It supports both request-reply mode and bidirectional streaming mode.
  *
  * An individual handler will have to convert the shape of the incoming request
  * to a RestateRequest, and then pass it to this handler, and eventually convert back
@@ -33,9 +34,11 @@ export interface RestateHandler {
  * Different runtimes have slightly different shapes of the incoming request, and responses.
  */
 export declare class GenericHandler implements RestateHandler {
-    private readonly endpoint;
-    constructor(endpoint: EndpointBuilder);
+    readonly endpoint: EndpointBuilder;
+    private readonly protocolMode;
+    constructor(endpoint: EndpointBuilder, protocolMode: ProtocolMode);
     handle(request: RestateRequest, context?: AdditionalContext): Promise<RestateResponse>;
+    private _handle;
     private validateConnectionSignature;
     private handleInvoke;
     private handleDiscovery;

package/dist/cjs/src/endpoint/handlers/generic.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generic.d.ts","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/generic.ts"],"names":[],"mappings":"AAkCA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAE9D,MAAM,WAAW,OAAO;IACtB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,iBAAiB;IAChC,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;CAC3B;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,eAAe,CAAC;IAClC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;CAC3B;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,CACJ,OAAO,EAAE,cAAc,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,eAAe,CAAC,CAAC;CAC7B;AAED;;;;;;;;;;GAUG;AACH,qBAAa,cAAe,YAAW,cAAc;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ;gBAAR,QAAQ,EAAE,eAAe;IAczC,MAAM,CACjB,OAAO,EAAE,cAAc,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,eAAe,CAAC;YAyDb,2BAA2B;YAqC3B,YAAY;IAmD1B,OAAO,CAAC,eAAe;IAmDvB,OAAO,CAAC,eAAe;CAUxB"}
+{"version":3,"file":"generic.d.ts","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/generic.ts"],"names":[],"mappings":";AAqBA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAO7D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EACL,KAAK,cAAc,EAGpB,MAAM,iBAAiB,CAAC;AAIzB,MAAM,WAAW,OAAO;IACtB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,iBAAiB;IAChC,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAE,cAAc,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;CAClD;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,eAAe,CAAC;IAClC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,IAAI,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;CAC3C;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,CACJ,OAAO,EAAE,cAAc,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,eAAe,CAAC,CAAC;CAC7B;AAED;;;;;;;;GAQG;AACH,qBAAa,cAAe,YAAW,cAAc;IAEjD,QAAQ,CAAC,QAAQ,EAAE,eAAe;IAClC,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADpB,QAAQ,EAAE,eAAe,EACjB,YAAY,EAAE,YAAY;IAgBhC,MAAM,CACjB,OAAO,EAAE,cAAc,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,eAAe,CAAC;YAYb,OAAO;YA4DP,2BAA2B;YAqC3B,YAAY;IA+D1B,OAAO,CAAC,eAAe;IAgDvB,OAAO,CAAC,eAAe;CAUxB"}

package/dist/cjs/src/endpoint/handlers/generic.js

@@ -11,24 +11,20 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GenericHandler = void 0;
-const logger_js_1 = require("../../logger.js");
-const lambda_connection_js_1 = require("../../connection/lambda_connection.js");
 const invocation_js_1 = require("../../invocation.js");
-const decoder_js_1 = require("../../io/decoder.js");
 const state_machine_js_1 = require("../../state_machine.js");
 const errors_js_1 = require("../../types/errors.js");
 const protocol_js_1 = require("../../types/protocol.js");
-const discovery_js_1 = require("../../types/discovery.js");
 const components_js_1 = require("../../types/components.js");
 const validate_js_1 = require("../request_signing/validate.js");
 const user_agent_js_1 = require("../../user_agent.js");
-const node_buffer_1 = require("node:buffer");
 const discovery_pb_js_1 = require("../../generated/proto/discovery_pb.js");
+const web_1 = require("node:stream/web");
+const connection_js_1 = require("../../connection/connection.js");
+const streams_js_1 = require("../../utils/streams.js");
 /**
  * This is an internal API to support 'fetch' like handlers.
- * This is for now supports only request-reply style of invocations
- * which are suitable for platforms like AWS Lambda, Cloudflare works,
- * and runtimes that only support HTTP/1.1 like Bun.
+ * It supports both request-reply mode and bidirectional streaming mode.
  *
  * An individual handler will have to convert the shape of the incoming request
  * to a RestateRequest, and then pass it to this handler, and eventually convert back
@@ -37,16 +33,29 @@ const discovery_pb_js_1 = require("../../generated/proto/discovery_pb.js");
  */
 class GenericHandler {
     endpoint;
-    constructor(endpoint) {
+    protocolMode;
+    constructor(endpoint, protocolMode) {
         this.endpoint = endpoint;
+        this.protocolMode = protocolMode;
         if (!this.endpoint.keySet) {
-            logger_js_1.rlog.warn(`Accepting requests without validating request signatures; worker access must be restricted`);
+            this.endpoint.rlog.warn(`Accepting requests without validating request signatures; handler access must be restricted`);
         }
         else {
-            logger_js_1.rlog.info(`Validating requests using signing keys [${Array.from(this.endpoint.keySet.keys())}]`);
+            this.endpoint.rlog.info(`Validating requests using signing keys [${Array.from(this.endpoint.keySet.keys())}]`);
         }
     }
+    // handle does not throw.
     async handle(request, context) {
+        try {
+            return await this._handle(request, context);
+        }
+        catch (e) {
+            const error = (0, errors_js_1.ensureError)(e);
+            this.endpoint.rlog.error("Error while handling invocation: " + (error.stack ?? error.message));
+            return this.toErrorResponse(500, error.message);
+        }
+    }
+    async _handle(request, context) {
         const path = request.url;
         const error = await this.validateConnectionSignature(path, request.headers);
         if (error !== null) {
@@ -55,7 +64,7 @@ class GenericHandler {
         const parsed = (0, components_js_1.parseUrlComponents)(path);
         if (!parsed) {
             const msg = `Invalid path: path doesn't end in /invoke/SvcName/handlerName and also not in /discover: ${path}`;
-            logger_js_1.rlog.trace(msg);
+            this.endpoint.rlog.trace(msg);
             return this.toErrorResponse(404, msg);
         }
         if (parsed === "discovery") {
@@ -64,30 +73,30 @@ class GenericHandler {
         const serviceProtocolVersionString = request.headers["content-type"];
         if (typeof serviceProtocolVersionString !== "string") {
             const errorMessage = "Missing content-type header";
-            logger_js_1.rlog.warn(errorMessage);
+            this.endpoint.rlog.warn(errorMessage);
             return this.toErrorResponse(415, errorMessage);
         }
         const serviceProtocolVersion = (0, protocol_js_1.parseServiceProtocolVersion)(serviceProtocolVersionString);
         if (!(0, protocol_js_1.isServiceProtocolVersionSupported)(serviceProtocolVersion)) {
             const errorMessage = `Unsupported service protocol version '${serviceProtocolVersionString}'`;
-            logger_js_1.rlog.warn(errorMessage);
+            this.endpoint.rlog.warn(errorMessage);
             return this.toErrorResponse(415, errorMessage);
         }
         const method = this.endpoint.componentByName(parsed.componentName);
         if (!method) {
             const msg = `No service found for URL: ${JSON.stringify(parsed)}`;
-            logger_js_1.rlog.error(msg);
+            this.endpoint.rlog.error(msg);
             return this.toErrorResponse(404, msg);
         }
         const handler = method?.handlerMatching(parsed);
         if (!handler) {
             const msg = `No service found for URL: ${JSON.stringify(parsed)}`;
-            logger_js_1.rlog.error(msg);
+            this.endpoint.rlog.error(msg);
             return this.toErrorResponse(404, msg);
         }
         if (!request.body) {
             const msg = "The incoming message body was null";
-            logger_js_1.rlog.error(msg);
+            this.endpoint.rlog.error(msg);
             return this.toErrorResponse(400, msg);
         }
         return this.handleInvoke(handler, request.body, request.headers, serviceProtocolVersion, context ?? {});
@@ -100,7 +109,7 @@ class GenericHandler {
         try {
             const validateResponse = await (0, validate_js_1.validateRequestSignature)(this.endpoint.keySet, path, headers);
             if (!validateResponse.valid) {
-                logger_js_1.rlog.error(`Rejecting request as its JWT did not validate: ${
+                this.endpoint.rlog.error(`Rejecting request as its JWT did not validate: ${
                 // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
                 validateResponse.error}`);
                 return this.toErrorResponse(401, "Unauthorized");
@@ -111,59 +120,72 @@ class GenericHandler {
         }
         catch (e) {
             const error = (0, errors_js_1.ensureError)(e);
-            logger_js_1.rlog.error("Error while attempting to validate request signature: " +
+            this.endpoint.rlog.error("Error while attempting to validate request signature: " +
                 (error.stack ?? error.message));
             return this.toErrorResponse(401, "Unauthorized");
         }
     }
     async handleInvoke(handler, body, headers, serviceProtocolVersion, context) {
+        let responseController;
+        const responseBody = new web_1.TransformStream({
+            start: (ctrl) => {
+                responseController =
+                    ctrl;
+            },
+        });
+        const connection = connection_js_1.RestateConnection.from(this.endpoint.rlog, headers, {
+            readable: body,
+            writable: responseBody.writable,
+        });
+        // step 1: collect all journal events
+        const journalBuilder = new invocation_js_1.InvocationBuilder(handler);
+        connection.pipeToConsumer(journalBuilder);
         try {
-            // build the previous journal from the events
-            let decodedEntries = (0, decoder_js_1.decodeMessagesBuffer)(node_buffer_1.Buffer.from(body));
-            const journalBuilder = new invocation_js_1.InvocationBuilder(handler);
-            decodedEntries.forEach((e) => journalBuilder.handleMessage(e));
-            const alreadyCompleted = decodedEntries.find((e) => e.messageType === protocol_js_1.OUTPUT_ENTRY_MESSAGE_TYPE) !== undefined;
-            decodedEntries = null;
-            // set up and invoke the state machine
-            const connection = new lambda_connection_js_1.LambdaConnection(headers, alreadyCompleted);
-            const invocation = journalBuilder.build();
-            const stateMachine = new state_machine_js_1.StateMachine(connection, invocation, discovery_js_1.ProtocolMode.REQUEST_RESPONSE, handler.kind(), invocation.inferLoggerContext(context));
-            await stateMachine.invoke();
-            const result = await connection.getResult();
-            return {
-                headers: {
-                    "content-type": (0, protocol_js_1.serviceProtocolVersionToHeaderValue)(serviceProtocolVersion),
-                    "x-restate-server": user_agent_js_1.X_RESTATE_SERVER,
-                },
-                statusCode: 200,
-                body: result,
-            };
-        }
-        catch (e) {
-            const error = (0, errors_js_1.ensureError)(e);
-            logger_js_1.rlog.error(error.message);
-            logger_js_1.rlog.error(error.stack);
-            return this.toErrorResponse(500, error.message);
-        }
+            await journalBuilder.completion();
+        }
+        finally {
+            // ensure GC friendliness, also in case of errors
+            connection.removeCurrentConsumer();
+        }
+        // step 2: create the state machine
+        const invocation = journalBuilder.build();
+        const stateMachine = new state_machine_js_1.StateMachine(connection, invocation, handler.kind(), this.endpoint.logger, invocation.inferLoggerContext(context));
+        connection.pipeToConsumer(stateMachine);
+        // step 3: invoke the function
+        // This call would propagate errors in the state machine logic, but not errors
+        // in the application function code. Ending a function with an error as well
+        // as failign an invocation and being retried are perfectly valid actions from the
+        // SDK's perspective.
+        stateMachine
+            .invoke()
+            .catch((e) => responseController.error(e)) // in bidi case the best we can do is abort the connection
+            .finally(() => connection.removeCurrentConsumer());
+        return {
+            headers: {
+                "content-type": (0, protocol_js_1.serviceProtocolVersionToHeaderValue)(serviceProtocolVersion),
+                "x-restate-server": user_agent_js_1.X_RESTATE_SERVER,
+            },
+            statusCode: 200,
+            body: responseBody.readable,
+        };
     }
     handleDiscovery(acceptVersionsString) {
         if (typeof acceptVersionsString !== "string") {
             const errorMessage = "Missing accept header";
-            logger_js_1.rlog.warn(errorMessage);
+            this.endpoint.rlog.warn(errorMessage);
             return this.toErrorResponse(415, errorMessage);
         }
         const serviceDiscoveryProtocolVersion = (0, protocol_js_1.selectSupportedServiceDiscoveryProtocolVersion)(acceptVersionsString);
         if (serviceDiscoveryProtocolVersion ===
             discovery_pb_js_1.ServiceDiscoveryProtocolVersion.SERVICE_DISCOVERY_PROTOCOL_VERSION_UNSPECIFIED) {
             const errorMessage = `Unsupported service discovery protocol version '${acceptVersionsString}'`;
-            logger_js_1.rlog.warn(errorMessage);
+            this.endpoint.rlog.warn(errorMessage);
             return this.toErrorResponse(415, errorMessage);
         }
-        const discovery = this.endpoint.computeDiscovery(discovery_js_1.ProtocolMode.REQUEST_RESPONSE);
+        const discovery = this.endpoint.computeDiscovery(this.protocolMode);
         let body;
         if (serviceDiscoveryProtocolVersion === discovery_pb_js_1.ServiceDiscoveryProtocolVersion.V1) {
-            const discoveryJson = JSON.stringify(discovery);
-            body = node_buffer_1.Buffer.from(discoveryJson);
+            body = JSON.stringify(discovery);
         }
         else {
             // should not be reached since we check for compatibility before
@@ -175,17 +197,17 @@ class GenericHandler {
                 "x-restate-server": user_agent_js_1.X_RESTATE_SERVER,
             },
             statusCode: 200,
-            body,
+            body: (0, streams_js_1.OnceStream)(new TextEncoder().encode(body)),
         };
     }
     toErrorResponse(code, message) {
         return {
             headers: {
-                "content-type": "text/plain",
+                "content-type": "application/json",
                 "x-restate-server": user_agent_js_1.X_RESTATE_SERVER,
             },
             statusCode: code,
-            body: node_buffer_1.Buffer.from(JSON.stringify({ message })),
+            body: (0, streams_js_1.OnceStream)(new TextEncoder().encode(JSON.stringify({ message }))),
         };
     }
 }

package/dist/cjs/src/endpoint/handlers/generic.js.map

@@ -1 +1 @@
-{"version":3,"file":"generic.js","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/generic.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,+CAAuC;AACvC,gFAAyE;AACzE,uDAAwD;AACxD,oDAA2D;AAE3D,6DAAsD;AACtD,qDAAoD;AACpD,yDAOiC;AACjC,2DAAwD;AAExD,6DAA+D;AAC/D,gEAA0E;AAC1E,uDAAuD;AACvD,6CAAqC;AACrC,2EAAwF;AAmCxF;;;;;;;;;;GAUG;AACH,MAAa,cAAc;IACI;IAA7B,YAA6B,QAAyB;QAAzB,aAAQ,GAAR,QAAQ,CAAiB;QACpD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC1B,gBAAI,CAAC,IAAI,CACP,4FAA4F,CAC7F,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,gBAAI,CAAC,IAAI,CACP,2CAA2C,KAAK,CAAC,IAAI,CACnD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAC5B,GAAG,CACL,CAAC;QACJ,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CACjB,OAAuB,EACvB,OAA2B;QAE3B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC;QAEzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAC5E,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GAAG,IAAA,kCAAkB,EAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,4FAA4F,IAAI,EAAE,CAAC;YAC/G,gBAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,4BAA4B,GAAG,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACrE,IAAI,OAAO,4BAA4B,KAAK,QAAQ,EAAE,CAAC;YACrD,MAAM,YAAY,GAAG,6BAA6B,CAAC;YACnD,gBAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,sBAAsB,GAAG,IAAA,yCAA2B,EACxD,4BAA4B,CAC7B,CAAC;QACF,IAAI,CAAC,IAAA,+CAAiC,EAAC,sBAAsB,CAAC,EAAE,CAAC;YAC/D,MAAM,YAAY,GAAG,yCAAyC,4BAA4B,GAAG,CAAC;YAC9F,gBAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,6BAA6B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAClE,gBAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,6BAA6B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAClE,gBAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,GAAG,GAAG,oCAAoC,CAAC;YACjD,gBAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CACtB,OAAO,EACP,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,OAAO,EACf,sBAAsB,EACtB,OAAO,IAAI,EAAE,CACd,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,2BAA2B,CACvC,IAAY,EACZ,OAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC1B,iBAAiB;YACjB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC;YACH,MAAM,gBAAgB,GAAG,MAAM,IAAA,sCAAwB,EACrD,IAAI,CAAC,QAAQ,CAAC,MAAM,EACpB,IAAI,EACJ,OAAO,CACR,CAAC;YAEF,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;gBAC5B,gBAAI,CAAC,KAAK,CACR,kDAAkD;gBAChD,4EAA4E;gBAC5E,gBAAgB,CAAC,KACnB,EAAE,CACH,CAAC;gBACF,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YACnD,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC,CAAC,CAAC,CAAC;YAC7B,gBAAI,CAAC,KAAK,CACR,wDAAwD;gBACtD,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,CACjC,CAAC;YACF,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,OAAyB,EACzB,IAAgB,EAChB,OAAsD,EACtD,sBAA8C,EAC9C,OAA0B;QAE1B,IAAI,CAAC;YACH,6CAA6C;YAC7C,IAAI,cAAc,GAAqB,IAAA,iCAAoB,EACzD,oBAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAClB,CAAC;YACF,MAAM,cAAc,GAAG,IAAI,iCAAiB,CAAC,OAAO,CAAC,CAAC;YACtD,cAAc,CAAC,OAAO,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;YACxE,MAAM,gBAAgB,GACpB,cAAc,CAAC,IAAI,CACjB,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,uCAAyB,CAC5D,KAAK,SAAS,CAAC;YAClB,cAAc,GAAG,IAAI,CAAC;YAEtB,sCAAsC;YACtC,MAAM,UAAU,GAAG,IAAI,uCAAgB,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;YACnE,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,EAAE,CAAC;YAC1C,MAAM,YAAY,GAAG,IAAI,+BAAY,CACnC,UAAU,EACV,UAAU,EACV,2BAAY,CAAC,gBAAgB,EAC7B,OAAO,CAAC,IAAI,EAAE,EACd,UAAU,CAAC,kBAAkB,CAAC,OAAO,CAAC,CACvC,CAAC;YACF,MAAM,YAAY,CAAC,MAAM,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,EAAE,CAAC;YAE5C,OAAO;gBACL,OAAO,EAAE;oBACP,cAAc,EAAE,IAAA,iDAAmC,EACjD,sBAAsB,CACvB;oBACD,kBAAkB,EAAE,gCAAgB;iBACrC;gBACD,UAAU,EAAE,GAAG;gBACf,IAAI,EAAE,MAAM;aACb,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC,CAAC,CAAC,CAAC;YAC7B,gBAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC1B,gBAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,oBAAmD;QAEnD,IAAI,OAAO,oBAAoB,KAAK,QAAQ,EAAE,CAAC;YAC7C,MAAM,YAAY,GAAG,uBAAuB,CAAC;YAC7C,gBAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,+BAA+B,GACnC,IAAA,4DAA8C,EAAC,oBAAoB,CAAC,CAAC;QAEvE,IACE,+BAA+B;YAC/B,iDAA+B,CAAC,8CAA8C,EAC9E,CAAC;YACD,MAAM,YAAY,GAAG,mDAAmD,oBAAoB,GAAG,CAAC;YAChG,gBAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAC9C,2BAAY,CAAC,gBAAgB,CAC9B,CAAC;QAEF,IAAI,IAAI,CAAC;QAET,IACE,+BAA+B,KAAK,iDAA+B,CAAC,EAAE,EACtE,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,GAAG,oBAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,gEAAgE;YAChE,MAAM,IAAI,KAAK,CACb,mDAAmD,+BAA+B,EAAE,CACrF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP,cAAc,EAAE,IAAA,0DAA4C,EAC1D,+BAA+B,CAChC;gBACD,kBAAkB,EAAE,gCAAgB;aACrC;YACD,UAAU,EAAE,GAAG;YACf,IAAI;SACL,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,IAAY,EAAE,OAAe;QACnD,OAAO;YACL,OAAO,EAAE;gBACP,cAAc,EAAE,YAAY;gBAC5B,kBAAkB,EAAE,gCAAgB;aACrC;YACD,UAAU,EAAE,IAAI;YAChB,IAAI,EAAE,oBAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;SAC/C,CAAC;IACJ,CAAC;CACF;AAhOD,wCAgOC"}
+{"version":3,"file":"generic.js","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/generic.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,uDAAwD;AACxD,6DAAsD;AACtD,qDAAoD;AACpD,yDAMiC;AAGjC,6DAA+D;AAC/D,gEAA0E;AAC1E,uDAAuD;AACvD,2EAAwF;AAGxF,yCAIyB;AACzB,kEAAmE;AACnE,uDAAoD;AAiCpD;;;;;;;;GAQG;AACH,MAAa,cAAc;IAEd;IACQ;IAFnB,YACW,QAAyB,EACjB,YAA0B;QADlC,aAAQ,GAAR,QAAQ,CAAiB;QACjB,iBAAY,GAAZ,YAAY,CAAc;QAE3C,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC1B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CACrB,6FAA6F,CAC9F,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CACrB,2CAA2C,KAAK,CAAC,IAAI,CACnD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAC5B,GAAG,CACL,CAAC;QACJ,CAAC;IACH,CAAC;IAED,yBAAyB;IAClB,KAAK,CAAC,MAAM,CACjB,OAAuB,EACvB,OAA2B;QAE3B,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CACtB,mCAAmC,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,CACrE,CAAC;YACF,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,OAAuB,EACvB,OAA2B;QAE3B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC;QAEzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAC5E,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GAAG,IAAA,kCAAkB,EAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,4FAA4F,IAAI,EAAE,CAAC;YAC/G,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,4BAA4B,GAAG,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACrE,IAAI,OAAO,4BAA4B,KAAK,QAAQ,EAAE,CAAC;YACrD,MAAM,YAAY,GAAG,6BAA6B,CAAC;YACnD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,sBAAsB,GAAG,IAAA,yCAA2B,EACxD,4BAA4B,CAC7B,CAAC;QACF,IAAI,CAAC,IAAA,+CAAiC,EAAC,sBAAsB,CAAC,EAAE,CAAC;YAC/D,MAAM,YAAY,GAAG,yCAAyC,4BAA4B,GAAG,CAAC;YAC9F,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,6BAA6B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAClE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,6BAA6B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAClE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,GAAG,GAAG,oCAAoC,CAAC;YACjD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CACtB,OAAO,EACP,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,OAAO,EACf,sBAAsB,EACtB,OAAO,IAAI,EAAE,CACd,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,2BAA2B,CACvC,IAAY,EACZ,OAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC1B,iBAAiB;YACjB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC;YACH,MAAM,gBAAgB,GAAG,MAAM,IAAA,sCAAwB,EACrD,IAAI,CAAC,QAAQ,CAAC,MAAM,EACpB,IAAI,EACJ,OAAO,CACR,CAAC;YAEF,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;gBAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CACtB,kDAAkD;gBAChD,4EAA4E;gBAC5E,gBAAgB,CAAC,KACnB,EAAE,CACH,CAAC;gBACF,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YACnD,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,IAAA,uBAAW,EAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CACtB,wDAAwD;gBACtD,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,CACjC,CAAC;YACF,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,OAAyB,EACzB,IAAgC,EAChC,OAAsD,EACtD,sBAA8C,EAC9C,OAA0B;QAE1B,IAAI,kBAAgE,CAAC;QACrE,MAAM,YAAY,GAAG,IAAI,qBAAe,CAAa;YACnD,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE;gBACd,kBAAkB;oBAChB,IAAoD,CAAC;YACzD,CAAC;SACF,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,iCAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE;YACrE,QAAQ,EAAE,IAAI;YACd,QAAQ,EAAE,YAAY,CAAC,QAAQ;SAChC,CAAC,CAAC;QAEH,qCAAqC;QACrC,MAAM,cAAc,GAAG,IAAI,iCAAiB,CAAC,OAAO,CAAC,CAAC;QACtD,UAAU,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,UAAU,EAAE,CAAC;QACpC,CAAC;gBAAS,CAAC;YACT,iDAAiD;YACjD,UAAU,CAAC,qBAAqB,EAAE,CAAC;QACrC,CAAC;QAED,mCAAmC;QACnC,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,EAAE,CAAC;QAC1C,MAAM,YAAY,GAAG,IAAI,+BAAY,CACnC,UAAU,EACV,UAAU,EACV,OAAO,CAAC,IAAI,EAAE,EACd,IAAI,CAAC,QAAQ,CAAC,MAAM,EACpB,UAAU,CAAC,kBAAkB,CAAC,OAAO,CAAC,CACvC,CAAC;QACF,UAAU,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAExC,8BAA8B;QAE9B,8EAA8E;QAC9E,4EAA4E;QAC5E,kFAAkF;QAClF,qBAAqB;QACrB,YAAY;aACT,MAAM,EAAE;aACR,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,0DAA0D;aACpG,OAAO,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,qBAAqB,EAAE,CAAC,CAAC;QAErD,OAAO;YACL,OAAO,EAAE;gBACP,cAAc,EAAE,IAAA,iDAAmC,EACjD,sBAAsB,CACvB;gBACD,kBAAkB,EAAE,gCAAgB;aACrC;YACD,UAAU,EAAE,GAAG;YACf,IAAI,EAAE,YAAY,CAAC,QAAsC;SAC1D,CAAC;IACJ,CAAC;IAEO,eAAe,CACrB,oBAAmD;QAEnD,IAAI,OAAO,oBAAoB,KAAK,QAAQ,EAAE,CAAC;YAC7C,MAAM,YAAY,GAAG,uBAAuB,CAAC;YAC7C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,+BAA+B,GACnC,IAAA,4DAA8C,EAAC,oBAAoB,CAAC,CAAC;QAEvE,IACE,+BAA+B;YAC/B,iDAA+B,CAAC,8CAA8C,EAC9E,CAAC;YACD,MAAM,YAAY,GAAG,mDAAmD,oBAAoB,GAAG,CAAC;YAChG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEpE,IAAI,IAAI,CAAC;QAET,IACE,+BAA+B,KAAK,iDAA+B,CAAC,EAAE,EACtE,CAAC;YACD,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,gEAAgE;YAChE,MAAM,IAAI,KAAK,CACb,mDAAmD,+BAA+B,EAAE,CACrF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP,cAAc,EAAE,IAAA,0DAA4C,EAC1D,+BAA+B,CAChC;gBACD,kBAAkB,EAAE,gCAAgB;aACrC;YACD,UAAU,EAAE,GAAG;YACf,IAAI,EAAE,IAAA,uBAAU,EAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;SACjD,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,IAAY,EAAE,OAAe;QACnD,OAAO;YACL,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,kBAAkB,EAAE,gCAAgB;aACrC;YACD,UAAU,EAAE,IAAI;YAChB,IAAI,EAAE,IAAA,uBAAU,EAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;SACxE,CAAC;IACJ,CAAC;CACF;AA5PD,wCA4PC"}

package/dist/cjs/src/endpoint/handlers/lambda.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"lambda.d.ts","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/lambda.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EACV,oBAAoB,EACpB,sBAAsB,EACtB,qBAAqB,EACrB,iCAAiC,EACjC,OAAO,EACR,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,cAAc,CAAC;AAEnE,qBAAa,aAAa;IACZ,OAAO,CAAC,QAAQ,CAAC,OAAO;gBAAP,OAAO,EAAE,cAAc;IAE9C,aAAa,CACjB,KAAK,EAAE,oBAAoB,GAAG,sBAAsB,EACpD,OAAO,EAAE,OAAO,GACf,OAAO,CAAC,qBAAqB,GAAG,iCAAiC,CAAC;CA6CtE"}
+{"version":3,"file":"lambda.d.ts","sourceRoot":"","sources":["../../../../../src/endpoint/handlers/lambda.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EACV,oBAAoB,EACpB,sBAAsB,EACtB,qBAAqB,EACrB,iCAAiC,EACjC,OAAO,EACR,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,cAAc,CAAC;AAMnE,qBAAa,aAAa;IACZ,OAAO,CAAC,QAAQ,CAAC,OAAO;gBAAP,OAAO,EAAE,cAAc;IAE9C,aAAa,CACjB,KAAK,EAAE,oBAAoB,GAAG,sBAAsB,EACpD,OAAO,EAAE,OAAO,GACf,OAAO,CAAC,qBAAqB,GAAG,iCAAiC,CAAC;CAgEtE"}

package/dist/cjs/src/endpoint/handlers/lambda.js

@@ -12,6 +12,10 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LambdaHandler = void 0;
 const node_buffer_1 = require("node:buffer");
+const web_1 = require("node:stream/web");
+const streams_js_1 = require("../../utils/streams.js");
+const user_agent_js_1 = require("../../user_agent.js");
+const errors_js_1 = require("../../types/errors.js");
 class LambdaHandler {
     handler;
     constructor(handler) {
@@ -23,42 +27,55 @@ class LambdaHandler {
         //
         const path = "path" in event ? event.path : event.rawPath;
         //
-        // Convert the request body to a Uint8Array
+        // Convert the request body to a Uint8Array stream
         // Lambda functions receive the body as base64 encoded string
         //
-        let requestBody;
+        let body;
         if (!event.body) {
-            requestBody = new Uint8Array(0);
+            body = null;
         }
         else if (event.isBase64Encoded) {
-            requestBody = node_buffer_1.Buffer.from(event.body, "base64");
+            body = (0, streams_js_1.OnceStream)(node_buffer_1.Buffer.from(event.body, "base64"));
         }
         else {
-            requestBody = node_buffer_1.Buffer.from(event.body);
+            body = (0, streams_js_1.OnceStream)(new TextEncoder().encode(event.body));
         }
         const request = {
-            body: requestBody,
+            body,
             headers: event.headers,
             url: path,
         };
         const resp = await this.handler.handle(request, {
             AWSRequestId: context.awsRequestId,
         });
-        let responseBody;
-        if (!resp.body) {
-            responseBody = "";
+        const chunks = [];
+        try {
+            await resp.body.pipeTo(new web_1.WritableStream({
+                write: (chunk) => {
+                    chunks.push(chunk);
+                },
+            }));
         }
-        else if (event.isBase64Encoded) {
-            responseBody = node_buffer_1.Buffer.from(resp.body).toString("base64");
-        }
-        else {
-            responseBody = node_buffer_1.Buffer.from(resp.body).toString("utf8");
+        catch (e) {
+            // unlike in the streaming case, we can actually catch errors in the response body and form a nicer error
+            const error = (0, errors_js_1.ensureError)(e);
+            this.handler.endpoint.rlog.error("Error while collecting invocation response: " +
+                (error.stack ?? error.message));
+            return {
+                headers: {
+                    "content-type": "application/json",
+                    "x-restate-server": user_agent_js_1.X_RESTATE_SERVER,
+                },
+                statusCode: 500,
+                isBase64Encoded: false,
+                body: JSON.stringify({ message: error.message }),
+            };
         }
         return {
             headers: resp.headers,
             statusCode: resp.statusCode,
-            isBase64Encoded: event.isBase64Encoded,
-            body: responseBody,
+            isBase64Encoded: true,
+            body: node_buffer_1.Buffer.concat(chunks).toString("base64"),
         };
     }
 }