@effect/cluster 0.28.4 → 0.29.0

package/dist/esm/HttpShardManager.js

@@ -0,0 +1,128 @@
+import * as HttpRouter from "@effect/platform/HttpRouter";
+import * as HttpServer from "@effect/platform/HttpServer";
+import * as RpcServer from "@effect/rpc/RpcServer";
+import * as Effect from "effect/Effect";
+import { identity } from "effect/Function";
+import * as Layer from "effect/Layer";
+import { layerClientProtocolHttp, layerClientProtocolWebsocket } from "./HttpCommon.js";
+import * as MessageStorage from "./MessageStorage.js";
+import * as RunnerHealth from "./RunnerHealth.js";
+import * as Runners from "./Runners.js";
+import * as ShardManager from "./ShardManager.js";
+/**
+ * @since 1.0.0
+ * @category Http App
+ */
+export const toHttpApp = /*#__PURE__*/Effect.gen(function* () {
+  const handlers = yield* Layer.build(ShardManager.layerServerHandlers);
+  return yield* RpcServer.toHttpApp(ShardManager.Rpcs).pipe(Effect.provide(handlers));
+});
+/**
+ * @since 1.0.0
+ * @category Http App
+ */
+export const toHttpAppWebsocket = /*#__PURE__*/Effect.gen(function* () {
+  const handlers = yield* Layer.build(ShardManager.layerServerHandlers);
+  return yield* RpcServer.toHttpAppWebsocket(ShardManager.Rpcs).pipe(Effect.provide(handlers));
+});
+/**
+ * A layer for the `ShardManager` service, that does not run a server.
+ *
+ * It only provides the `Runners` rpc client.
+ *
+ * You can use this with the `toHttpApp` and `toHttpAppWebsocket` apis
+ * to run a complete `ShardManager` server.
+ *
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerNoServerHttp = options => ShardManager.layer.pipe(Layer.provide(Runners.layerRpc.pipe(Layer.provide([layerClientProtocolHttp({
+  path: options.runnerPath,
+  https: options.runnerHttps
+}), MessageStorage.layerNoop]))));
+/**
+ * A layer for the `ShardManager` service, that does not run a server.
+ *
+ * It only provides the `Runners` rpc client.
+ *
+ * You can use this with the `toHttpApp` and `toHttpAppWebsocket` apis
+ * to run a complete `ShardManager` server.
+ *
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerNoServerWebsocket = options => ShardManager.layer.pipe(Layer.provide(Runners.layerRpc.pipe(Layer.provide([layerClientProtocolWebsocket({
+  path: options.runnerPath,
+  https: options.runnerHttps
+}), MessageStorage.layerNoop]))));
+/**
+ * A HTTP layer for the `ShardManager` server, that adds a route to the provided
+ * `HttpRouter.Tag`.
+ *
+ * By default, it uses the `HttpRouter.Default` tag.
+ *
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerHttpOptions = options => {
+  const routerTag = options.routerTag ?? HttpRouter.Default;
+  return routerTag.serve().pipe(options.logAddress ? withLogAddress : identity, Layer.merge(ShardManager.layerServer), Layer.provide(RpcServer.layerProtocolHttp(options)), Layer.provideMerge(layerNoServerHttp(options)));
+};
+/**
+ * A WebSocket layer for the `ShardManager` server, that adds a route to the provided
+ * `HttpRouter.Tag`.
+ *
+ * By default, it uses the `HttpRouter.Default` tag.
+ *
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerWebsocketOptions = options => {
+  const routerTag = options.routerTag ?? HttpRouter.Default;
+  return routerTag.serve().pipe(options.logAddress ? withLogAddress : identity, Layer.merge(ShardManager.layerServer), Layer.provide(RpcServer.layerProtocolWebsocket(options)), Layer.provideMerge(layerNoServerWebsocket(options)));
+};
+const withLogAddress = layer => Layer.effectDiscard(HttpServer.addressFormattedWith(address => Effect.annotateLogs(Effect.logInfo(`Listening on: ${address}`), {
+  package: "@effect/cluster",
+  service: "ShardManager"
+}))).pipe(Layer.provideMerge(layer));
+/**
+ * A HTTP layer for the `ShardManager` server, that adds a route to the provided
+ * `HttpRouter.Tag`.
+ *
+ * By default, it uses the `HttpRouter.Default` tag.
+ *
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerHttp = /*#__PURE__*/layerHttpOptions({
+  path: "/",
+  runnerPath: "/"
+});
+/**
+ * A Websocket layer for the `ShardManager` server, that adds a route to the provided
+ * `HttpRouter.Tag`.
+ *
+ * By default, it uses the `HttpRouter.Default` tag.
+ *
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerWebsocket = /*#__PURE__*/layerWebsocketOptions({
+  path: "/",
+  runnerPath: "/"
+});
+/**
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerRunnerHealthHttp = /*#__PURE__*/Layer.provide(RunnerHealth.layerRpc, /*#__PURE__*/layerClientProtocolHttp({
+  path: "/"
+}));
+/**
+ * @since 1.0.0
+ * @category Layers
+ */
+export const layerRunnerHealthWebsocket = /*#__PURE__*/Layer.provide(RunnerHealth.layerRpc, /*#__PURE__*/layerClientProtocolWebsocket({
+  path: "/"
+}));
+//# sourceMappingURL=HttpShardManager.js.map

package/dist/esm/HttpShardManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"HttpShardManager.js","names":["HttpRouter","HttpServer","RpcServer","Effect","identity","Layer","layerClientProtocolHttp","layerClientProtocolWebsocket","MessageStorage","RunnerHealth","Runners","ShardManager","toHttpApp","gen","handlers","build","layerServerHandlers","Rpcs","pipe","provide","toHttpAppWebsocket","layerNoServerHttp","options","layer","layerRpc","path","runnerPath","https","runnerHttps","layerNoop","layerNoServerWebsocket","layerHttpOptions","routerTag","Default","serve","logAddress","withLogAddress","merge","layerServer","layerProtocolHttp","provideMerge","layerWebsocketOptions","layerProtocolWebsocket","effectDiscard","addressFormattedWith","address","annotateLogs","logInfo","package","service","layerHttp","layerWebsocket","layerRunnerHealthHttp","layerRunnerHealthWebsocket"],"sources":["../../src/HttpShardManager.ts"],"sourcesContent":[null],"mappings":"AAKA,OAAO,KAAKA,UAAU,MAAM,6BAA6B;AACzD,OAAO,KAAKC,UAAU,MAAM,6BAA6B;AAGzD,OAAO,KAAKC,SAAS,MAAM,uBAAuB;AAClD,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,SAASC,QAAQ,QAAQ,iBAAiB;AAC1C,OAAO,KAAKC,KAAK,MAAM,cAAc;AAErC,SAASC,uBAAuB,EAAEC,4BAA4B,QAAQ,iBAAiB;AACvF,OAAO,KAAKC,cAAc,MAAM,qBAAqB;AACrD,OAAO,KAAKC,YAAY,MAAM,mBAAmB;AACjD,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC,OAAO,KAAKC,YAAY,MAAM,mBAAmB;AAGjD;;;;AAIA,OAAO,MAAMC,SAAS,gBAIlBT,MAAM,CAACU,GAAG,CAAC,aAAS;EACtB,MAAMC,QAAQ,GAAG,OAAOT,KAAK,CAACU,KAAK,CAACJ,YAAY,CAACK,mBAAmB,CAAC;EACrE,OAAO,OAAOd,SAAS,CAACU,SAAS,CAACD,YAAY,CAACM,IAAI,CAAC,CAACC,IAAI,CACvDf,MAAM,CAACgB,OAAO,CAACL,QAAQ,CAAC,CACzB;AACH,CAAC,CAAC;AAEF;;;;AAIA,OAAO,MAAMM,kBAAkB,gBAI3BjB,MAAM,CAACU,GAAG,CAAC,aAAS;EACtB,MAAMC,QAAQ,GAAG,OAAOT,KAAK,CAACU,KAAK,CAACJ,YAAY,CAACK,mBAAmB,CAAC;EACrE,OAAO,OAAOd,SAAS,CAACkB,kBAAkB,CAACT,YAAY,CAACM,IAAI,CAAC,CAACC,IAAI,CAChEf,MAAM,CAACgB,OAAO,CAACL,QAAQ,CAAC,CACzB;AACH,CAAC,CAAC;AAEF;;;;;;;;;;;AAWA,OAAO,MAAMO,iBAAiB,GAC5BC,OAGC,IAWDX,YAAY,CAACY,KAAK,CAACL,IAAI,CACrBb,KAAK,CAACc,OAAO,CAACT,OAAO,CAACc,QAAQ,CAACN,IAAI,CACjCb,KAAK,CAACc,OAAO,CAAC,CACZb,uBAAuB,CAAC;EACtBmB,IAAI,EAAEH,OAAO,CAACI,UAAU;EACxBC,KAAK,EAAEL,OAAO,CAACM;CAChB,CAAC,EACFpB,cAAc,CAACqB,SAAS,CACzB,CAAC,CACH,CAAC,CACH;AAEH;;;;;;;;;;;AAWA,OAAO,MAAMC,sBAAsB,GACjCR,OAGC,IAWDX,YAAY,CAACY,KAAK,CAACL,IAAI,CACrBb,KAAK,CAACc,OAAO,CAACT,OAAO,CAACc,QAAQ,CAACN,IAAI,CACjCb,KAAK,CAACc,OAAO,CAAC,CACZZ,4BAA4B,CAAC;EAC3BkB,IAAI,EAAEH,OAAO,CAACI,UAAU;EACxBC,KAAK,EAAEL,OAAO,CAACM;CAChB,CAAC,EACFpB,cAAc,CAACqB,SAAS,CACzB,CAAC,CACH,CAAC,CACH;AAEH;;;;;;;;;AASA,OAAO,MAAME,gBAAgB,GAC3BT,OAMC,IAWC;EACF,MAAMU,SAAS,GAAGV,OAAO,CAACU,SAAS,IAAIhC,UAAU,CAACiC,OAAO;EACzD,OAAOD,SAAS,CAACE,KAAK,EAAE,CAAChB,IAAI,CAC3BI,OAAO,CAACa,UAAU,GAAGC,cAAc,GAAGhC,QAAQ,EAC9CC,KAAK,CAACgC,KAAK,CAAC1B,YAAY,CAAC2B,WAAW,CAAC,EACrCjC,KAAK,CAACc,OAAO,CAACjB,SAAS,CAACqC,iBAAiB,CAACjB,OAAO,CAAC,CAAC,EACnDjB,KAAK,CAACmC,YAAY,CAACnB,iBAAiB,CAACC,OAAO,CAAC,CAAC,CAC/C;AACH,CAAC;AAED;;;;;;;;;AASA,OAAO,MAAMmB,qBAAqB,GAChCnB,OAMC,IAWC;EACF,MAAMU,SAAS,GAAGV,OAAO,CAACU,SAAS,IAAIhC,UAAU,CAACiC,OAAO;EACzD,OAAOD,SAAS,CAACE,KAAK,EAAE,CAAChB,IAAI,CAC3BI,OAAO,CAACa,UAAU,GAAGC,cAAc,GAAGhC,QAAQ,EAC9CC,KAAK,CAACgC,KAAK,CAAC1B,YAAY,CAAC2B,WAAW,CAAC,EACrCjC,KAAK,CAACc,OAAO,CAACjB,SAAS,CAACwC,sBAAsB,CAACpB,OAAO,CAAC,CAAC,EACxDjB,KAAK,CAACmC,YAAY,CAACV,sBAAsB,CAACR,OAAO,CAAC,CAAC,CACpD;AACH,CAAC;AAED,MAAMc,cAAc,GAAab,KAA2B,IAC1DlB,KAAK,CAACsC,aAAa,CACjB1C,UAAU,CAAC2C,oBAAoB,CAAEC,OAAO,IACtC1C,MAAM,CAAC2C,YAAY,CAAC3C,MAAM,CAAC4C,OAAO,CAAC,iBAAiBF,OAAO,EAAE,CAAC,EAAE;EAC9DG,OAAO,EAAE,iBAAiB;EAC1BC,OAAO,EAAE;CACV,CAAC,CACH,CACF,CAAC/B,IAAI,CAACb,KAAK,CAACmC,YAAY,CAACjB,KAAK,CAAC,CAAC;AAEnC;;;;;;;;;AASA,OAAO,MAAM2B,SAAS,gBAUlBnB,gBAAgB,CAAC;EAAEN,IAAI,EAAE,GAAG;EAAEC,UAAU,EAAE;AAAG,CAAE,CAAC;AAEpD;;;;;;;;;AASA,OAAO,MAAMyB,cAAc,gBAUvBV,qBAAqB,CAAC;EAAEhB,IAAI,EAAE,GAAG;EAAEC,UAAU,EAAE;AAAG,CAAE,CAAC;AAEzD;;;;AAIA,OAAO,MAAM0B,qBAAqB,gBAI9B/C,KAAK,CAACc,OAAO,CAACV,YAAY,CAACe,QAAQ,eAAElB,uBAAuB,CAAC;EAAEmB,IAAI,EAAE;AAAG,CAAE,CAAC,CAAC;AAEhF;;;;AAIA,OAAO,MAAM4B,0BAA0B,gBAInChD,KAAK,CAACc,OAAO,CAACV,YAAY,CAACe,QAAQ,eAAEjB,4BAA4B,CAAC;EAAEkB,IAAI,EAAE;AAAG,CAAE,CAAC,CAAC","ignoreList":[]}

package/dist/esm/MachineId.js

@@ -0,0 +1,17 @@
+/**
+ * @since 1.0.0
+ */
+import * as Schema from "effect/Schema";
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const MachineId = /*#__PURE__*/Schema.Int.pipe( /*#__PURE__*/Schema.brand("MachineId"), /*#__PURE__*/Schema.annotations({
+  pretty: () => machineId => `MachineId(${machineId})`
+}));
+/**
+ * @since 1.0.0
+ * @category Constructors
+ */
+export const make = shardId => MachineId.make(shardId);
+//# sourceMappingURL=MachineId.js.map

package/dist/esm/MachineId.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MachineId.js","names":["Schema","MachineId","Int","pipe","brand","annotations","pretty","machineId","make","shardId"],"sources":["../../src/MachineId.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,MAAM,MAAM,eAAe;AAEvC;;;;AAIA,OAAO,MAAMC,SAAS,gBAAGD,MAAM,CAACE,GAAG,CAACC,IAAI,eACtCH,MAAM,CAACI,KAAK,CAAC,WAAW,CAAC,eACzBJ,MAAM,CAACK,WAAW,CAAC;EACjBC,MAAM,EAAEA,CAAA,KAAOC,SAAS,IAAK,aAAaA,SAAS;CACpD,CAAC,CACH;AAQD;;;;AAIA,OAAO,MAAMC,IAAI,GAAIC,OAAe,IAAgBR,SAAS,CAACO,IAAI,CAACC,OAAO,CAAC","ignoreList":[]}

package/dist/esm/Message.js

@@ -1,33 +1,104 @@
-import * as internal from "./internal/message.js";
+import { FiberRef } from "effect";
+import * as Data from "effect/Data";
+import * as Effect from "effect/Effect";
+import * as Option from "effect/Option";
+import * as Schema from "effect/Schema";
+import { MalformedMessage } from "./ClusterError.js";
+import * as Envelope from "./Envelope.js";
 /**
  * @since 1.0.0
- * @category schemas
+ * @category incoming
  */
-export const TaggedMessage = internal.TaggedMessage_;
+export const incomingLocalFromOutgoing = self => {
+  if (self._tag === "OutgoingEnvelope") {
+    return new IncomingEnvelope({
+      envelope: self.envelope
+    });
+  }
+  return new IncomingRequestLocal({
+    envelope: self.envelope,
+    respond: self.respond,
+    lastSentReply: Option.none()
+  });
+};
 /**
  * @since 1.0.0
- * @category utils
+ * @category incoming
  */
-export const isMessageWithResult = internal.isMessageWithResult;
+export class IncomingRequest extends /*#__PURE__*/Data.TaggedClass("IncomingRequest") {}
 /**
- * Extracts the exit schema from a Message. This schema will be used to encode the remote exit of the Message processor.
- *
  * @since 1.0.0
- * @category utils
+ * @category outgoing
  */
-export const exitSchema = internal.exitSchema;
+export class IncomingRequestLocal extends /*#__PURE__*/Data.TaggedClass("IncomingRequestLocal") {}
 /**
- * Extracts the failure schema from a Message. This schema will be used to encode remote failures of the Message processor.
- *
  * @since 1.0.0
- * @category utils
+ * @category incoming
  */
-export const failureSchema = internal.failureSchema;
+export class IncomingEnvelope extends /*#__PURE__*/Data.TaggedClass("IncomingEnvelope") {}
 /**
- * Extracts the success schema from a Message. This schema will be used to encode the remote success of the Message processor.
- *
  * @since 1.0.0
- * @category utils
+ * @category outgoing
  */
-export const successSchema = internal.successSchema;
+export class OutgoingRequest extends /*#__PURE__*/Data.TaggedClass("OutgoingRequest") {
+  /**
+   * @since 1.0.0
+   */
+  encodedCache;
+}
+/**
+ * @since 1.0.0
+ * @category outgoing
+ */
+export class OutgoingEnvelope extends /*#__PURE__*/Data.TaggedClass("OutgoingEnvelope") {}
+/**
+ * @since 1.0.0
+ * @category serialization / deserialization
+ */
+export const serialize = message => {
+  if (message._tag !== "OutgoingRequest") {
+    return Effect.succeed(message.envelope);
+  }
+  return Effect.suspend(() => message.encodedCache ? Effect.succeed(message.encodedCache) : serializeRequest(message));
+};
+/**
+ * @since 1.0.0
+ * @category serialization / deserialization
+ */
+export const serializeEnvelope = message => Effect.flatMap(serialize(message), envelope => MalformedMessage.refail(Schema.encode(Envelope.PartialEncoded)(envelope)));
+/**
+ * @since 1.0.0
+ * @category serialization / deserialization
+ */
+export const serializeRequest = self => {
+  const rpc = self.rpc;
+  return Schema.encode(rpc.payloadSchema)(self.envelope.payload).pipe(Effect.locally(FiberRef.currentContext, self.context), MalformedMessage.refail, Effect.map(payload => ({
+    ...self.envelope,
+    payload
+  })));
+};
+/**
+ * @since 1.0.0
+ * @category serialization / deserialization
+ */
+export const deserializeLocal = (self, encoded) => {
+  if (encoded._tag !== "Request") {
+    return Effect.succeed(new IncomingEnvelope({
+      envelope: encoded
+    }));
+  } else if (self._tag !== "OutgoingRequest") {
+    return Effect.fail(new MalformedMessage({
+      cause: new Error("Can only deserialize a Request with an OutgoingRequest message")
+    }));
+  }
+  const rpc = self.rpc;
+  return Schema.decode(rpc.payloadSchema)(encoded.payload).pipe(Effect.locally(FiberRef.currentContext, self.context), MalformedMessage.refail, Effect.map(payload => new IncomingRequestLocal({
+    envelope: Envelope.makeRequest({
+      ...encoded,
+      payload
+    }),
+    lastSentReply: Option.none(),
+    respond: self.respond
+  })));
+};
 //# sourceMappingURL=Message.js.map

package/dist/esm/Message.js.map

@@ -1 +1 @@
-{"version":3,"file":"Message.js","names":["internal","TaggedMessage","TaggedMessage_","isMessageWithResult","exitSchema","failureSchema","successSchema"],"sources":["../../src/Message.ts"],"sourcesContent":[null],"mappings":"AAOA,OAAO,KAAKA,QAAQ,MAAM,uBAAuB;AAoFjD;;;;AAIA,OAAO,MAAMC,aAAa,GAAGD,QAAQ,CAACE,cAAc;AAEpD;;;;AAIA,OAAO,MAAMC,mBAAmB,GAC9BH,QAAQ,CAACG,mBAAmB;AAE9B;;;;;;AAMA,OAAO,MAAMC,UAAU,GAEwBJ,QAAQ,CAACI,UAAU;AAElE;;;;;;AAMA,OAAO,MAAMC,aAAa,GAEsBL,QAAQ,CAACK,aAAa;AAEtE;;;;;;AAMA,OAAO,MAAMC,aAAa,GAEwBN,QAAQ,CAACM,aAAa","ignoreList":[]}
+{"version":3,"file":"Message.js","names":["FiberRef","Data","Effect","Option","Schema","MalformedMessage","Envelope","incomingLocalFromOutgoing","self","_tag","IncomingEnvelope","envelope","IncomingRequestLocal","respond","lastSentReply","none","IncomingRequest","TaggedClass","OutgoingRequest","encodedCache","OutgoingEnvelope","serialize","message","succeed","suspend","serializeRequest","serializeEnvelope","flatMap","refail","encode","PartialEncoded","rpc","payloadSchema","payload","pipe","locally","currentContext","context","map","deserializeLocal","encoded","fail","cause","Error","decode","makeRequest"],"sources":["../../src/Message.ts"],"sourcesContent":[null],"mappings":"AAIA,SAASA,QAAQ,QAAQ,QAAQ;AAEjC,OAAO,KAAKC,IAAI,MAAM,aAAa;AACnC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAEvC,SAASC,gBAAgB,QAAQ,mBAAmB;AACpD,OAAO,KAAKC,QAAQ,MAAM,eAAe;AAezC;;;;AAIA,OAAO,MAAMC,yBAAyB,GAAuBC,IAAiB,IAAsB;EAClG,IAAIA,IAAI,CAACC,IAAI,KAAK,kBAAkB,EAAE;IACpC,OAAO,IAAIC,gBAAgB,CAAC;MAAEC,QAAQ,EAAEH,IAAI,CAACG;IAAQ,CAAE,CAAC;EAC1D;EACA,OAAO,IAAIC,oBAAoB,CAAC;IAC9BD,QAAQ,EAAEH,IAAI,CAACG,QAAQ;IACvBE,OAAO,EAAEL,IAAI,CAACK,OAAO;IACrBC,aAAa,EAAEX,MAAM,CAACY,IAAI;GAC3B,CAAC;AACJ,CAAC;AAED;;;;AAIA,OAAM,MAAOC,eAAmC,sBAAQf,IAAI,CAACgB,WAAW,CAAC,iBAAiB,CAIxF;AAEF;;;;AAIA,OAAM,MAAOL,oBAAwC,sBAAQX,IAAI,CAACgB,WAAW,CAAC,sBAAsB,CAIlG;AAEF;;;;AAIA,OAAM,MAAOP,gBAAiB,sBAAQT,IAAI,CAACgB,WAAW,CAAC,kBAAkB,CAGvE;AAQF;;;;AAIA,OAAM,MAAOC,eAAmC,sBAAQjB,IAAI,CAACgB,WAAW,CAAC,iBAAiB,CAMxF;EACA;;;EAGOE,YAAY;;AAGrB;;;;AAIA,OAAM,MAAOC,gBAAiB,sBAAQnB,IAAI,CAACgB,WAAW,CAAC,kBAAkB,CAGvE;AAEF;;;;AAIA,OAAO,MAAMI,SAAS,GACpBC,OAAsB,IAC+C;EACrE,IAAIA,OAAO,CAACb,IAAI,KAAK,iBAAiB,EAAE;IACtC,OAAOP,MAAM,CAACqB,OAAO,CAACD,OAAO,CAACX,QAAQ,CAAC;EACzC;EACA,OAAOT,MAAM,CAACsB,OAAO,CAAC,MACpBF,OAAO,CAACH,YAAY,GAChBjB,MAAM,CAACqB,OAAO,CAACD,OAAO,CAACH,YAAY,CAAC,GACpCM,gBAAgB,CAACH,OAAO,CAAC,CAC9B;AACH,CAAC;AAED;;;;AAIA,OAAO,MAAMI,iBAAiB,GAC5BJ,OAAsB,IAEtBpB,MAAM,CAACyB,OAAO,CACZN,SAAS,CAACC,OAAO,CAAC,EACjBX,QAAQ,IAAKN,gBAAgB,CAACuB,MAAM,CAACxB,MAAM,CAACyB,MAAM,CAACvB,QAAQ,CAACwB,cAAc,CAAC,CAACnB,QAAQ,CAAC,CAAC,CACxF;AAEH;;;;AAIA,OAAO,MAAMc,gBAAgB,GAC3BjB,IAA0B,IAC0C;EACpE,MAAMuB,GAAG,GAAGvB,IAAI,CAACuB,GAA8B;EAC/C,OAAO3B,MAAM,CAACyB,MAAM,CAACE,GAAG,CAACC,aAAa,CAAC,CAACxB,IAAI,CAACG,QAAQ,CAACsB,OAAO,CAAC,CAACC,IAAI,CACjEhC,MAAM,CAACiC,OAAO,CAACnC,QAAQ,CAACoC,cAAc,EAAE5B,IAAI,CAAC6B,OAAO,CAAC,EACrDhC,gBAAgB,CAACuB,MAAM,EACvB1B,MAAM,CAACoC,GAAG,CAAEL,OAAO,KAAM;IACvB,GAAGzB,IAAI,CAACG,QAAQ;IAChBsB;GACD,CAAC,CAAC,CACuE;AAC9E,CAAC;AAED;;;;AAIA,OAAO,MAAMM,gBAAgB,GAAGA,CAC9B/B,IAAmB,EACnBgC,OAAyC,KAIvC;EACF,IAAIA,OAAO,CAAC/B,IAAI,KAAK,SAAS,EAAE;IAC9B,OAAOP,MAAM,CAACqB,OAAO,CAAC,IAAIb,gBAAgB,CAAC;MAAEC,QAAQ,EAAE6B;IAAO,CAAE,CAAC,CAAC;EACpE,CAAC,MAAM,IAAIhC,IAAI,CAACC,IAAI,KAAK,iBAAiB,EAAE;IAC1C,OAAOP,MAAM,CAACuC,IAAI,CAChB,IAAIpC,gBAAgB,CAAC;MAAEqC,KAAK,EAAE,IAAIC,KAAK,CAAC,gEAAgE;IAAC,CAAE,CAAC,CAC7G;EACH;EACA,MAAMZ,GAAG,GAAGvB,IAAI,CAACuB,GAA8B;EAC/C,OAAO3B,MAAM,CAACwC,MAAM,CAACb,GAAG,CAACC,aAAa,CAAC,CAACQ,OAAO,CAACP,OAAO,CAAC,CAACC,IAAI,CAC3DhC,MAAM,CAACiC,OAAO,CAACnC,QAAQ,CAACoC,cAAc,EAAE5B,IAAI,CAAC6B,OAAO,CAAC,EACrDhC,gBAAgB,CAACuB,MAAM,EACvB1B,MAAM,CAACoC,GAAG,CAAEL,OAAO,IACjB,IAAIrB,oBAAoB,CAAC;IACvBD,QAAQ,EAAEL,QAAQ,CAACuC,WAAW,CAAC;MAC7B,GAAGL,OAAO;MACVP;KACM,CAAC;IACTnB,aAAa,EAAEX,MAAM,CAACY,IAAI,EAAE;IAC5BF,OAAO,EAAEL,IAAI,CAACK;GACf,CAAC,CACH,CAC4D;AACjE,CAAC","ignoreList":[]}

package/dist/esm/MessageStorage.js

@@ -0,0 +1,345 @@
+import * as Arr from "effect/Array";
+import * as Context from "effect/Context";
+import * as Data from "effect/Data";
+import * as Effect from "effect/Effect";
+import * as Exit from "effect/Exit";
+import * as FiberRef from "effect/FiberRef";
+import { globalValue } from "effect/GlobalValue";
+import * as Iterable from "effect/Iterable";
+import * as Layer from "effect/Layer";
+import * as Option from "effect/Option";
+import * as Schema from "effect/Schema";
+import { MalformedMessage } from "./ClusterError.js";
+import * as DeliverAt from "./DeliverAt.js";
+import * as Envelope from "./Envelope.js";
+import * as Message from "./Message.js";
+import * as Reply from "./Reply.js";
+import * as Snowflake from "./Snowflake.js";
+/**
+ * @since 1.0.0
+ * @category context
+ */
+export class MessageStorage extends /*#__PURE__*/Context.Tag("@effect/cluster/MessageStorage")() {}
+/**
+ * @since 1.0.0
+ * @category SaveResult
+ */
+export const SaveResult = /*#__PURE__*/Data.taggedEnum();
+/**
+ * @since 1.0.0
+ * @category SaveResult
+ */
+export const SaveResultEncoded = /*#__PURE__*/Data.taggedEnum();
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const make = storage => Effect.sync(() => {
+  const replyHandlers = new Map();
+  return MessageStorage.of({
+    ...storage,
+    registerReplyHandler: message => Effect.sync(() => {
+      replyHandlers.set(message.envelope.requestId, message.respond);
+    }),
+    saveReply(reply) {
+      return Effect.flatMap(storage.saveReply(reply), () => {
+        const handler = replyHandlers.get(reply.reply.requestId);
+        if (!handler) {
+          return Effect.void;
+        } else if (reply.reply._tag === "WithExit") {
+          replyHandlers.delete(reply.reply.requestId);
+        }
+        return handler(reply.reply);
+      });
+    }
+  });
+});
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const makeEncoded = /*#__PURE__*/Effect.fnUntraced(function* (encoded) {
+  const snowflakeGen = yield* Snowflake.Generator;
+  const clock = yield* Effect.clock;
+  const storage = yield* make({
+    saveRequest: message => Message.serializeEnvelope(message).pipe(Effect.flatMap(envelope => encoded.saveEnvelope({
+      envelope,
+      primaryKey: Envelope.primaryKey(message.envelope),
+      deliverAt: DeliverAt.toMillis(message.envelope.payload)
+    })), Effect.flatMap(result => {
+      if (result._tag === "Success" || result.lastReceivedReply._tag === "None") {
+        return Effect.succeed(result);
+      }
+      const duplicate = result;
+      const schema = Reply.Reply(message.rpc);
+      return Schema.decode(schema)(result.lastReceivedReply.value).pipe(Effect.locally(FiberRef.currentContext, message.context), MalformedMessage.refail, Effect.map(reply => SaveResult.Duplicate({
+        originalId: duplicate.originalId,
+        lastReceivedReply: Option.some(reply)
+      })));
+    })),
+    saveEnvelope: message => Message.serializeEnvelope(message).pipe(Effect.flatMap(envelope => encoded.saveEnvelope({
+      envelope,
+      primaryKey: null,
+      deliverAt: null
+    })), Effect.asVoid),
+    saveReply: reply => Effect.flatMap(Reply.serialize(reply), encoded.saveReply),
+    repliesFor: Effect.fnUntraced(function* (messages) {
+      const requestIds = Arr.empty();
+      const map = new Map();
+      for (const message of messages) {
+        const id = String(message.envelope.requestId);
+        requestIds.push(id);
+        map.set(id, message);
+      }
+      if (requestIds.length === 0) return [];
+      const encodedReplies = yield* encoded.repliesFor(requestIds);
+      return yield* decodeReplies(map, encodedReplies);
+    }),
+    unprocessedMessages: shardIds => {
+      const shards = Array.from(shardIds);
+      if (shards.length === 0) return Effect.succeed([]);
+      return Effect.flatMap(Effect.suspend(() => encoded.unprocessedMessages(shards, clock.unsafeCurrentTimeMillis())), decodeMessages);
+    },
+    unprocessedMessagesById(messageIds) {
+      const ids = Array.from(messageIds);
+      if (ids.length === 0) return Effect.succeed([]);
+      return Effect.flatMap(Effect.suspend(() => encoded.unprocessedMessagesById(ids, clock.unsafeCurrentTimeMillis())), decodeMessages);
+    },
+    resetAddress: address => encoded.resetAddress(address)
+  });
+  const decodeMessages = envelopes => {
+    const messages = [];
+    let index = 0;
+    // if we have a malformed message, we should not return it and update
+    // the storage with a defect
+    const decodeMessage = Effect.catchAll(Effect.suspend(() => {
+      const envelope = envelopes[index];
+      if (!envelope) return Effect.succeed(undefined);
+      return decodeEnvelopeWithReply(envelope);
+    }), error => {
+      const envelope = envelopes[index];
+      return storage.saveReply(Reply.ReplyWithContext.fromDefect({
+        id: snowflakeGen.unsafeNext(),
+        requestId: Snowflake.Snowflake(envelope.envelope.requestId),
+        defect: error.toString()
+      })).pipe(Effect.forkDaemon, Effect.asVoid);
+    });
+    return Effect.as(Effect.whileLoop({
+      while: () => index < envelopes.length,
+      body: () => decodeMessage,
+      step: message => {
+        const envelope = envelopes[index++];
+        if (!message) return;
+        messages.push(message.envelope._tag === "Request" ? new Message.IncomingRequest({
+          envelope: message.envelope,
+          lastSentReply: envelope.lastSentReply,
+          respond: storage.saveReply
+        }) : new Message.IncomingEnvelope({
+          envelope: message.envelope
+        }));
+      }
+    }), messages);
+  };
+  const decodeReplies = (messages, encodedReplies) => {
+    const replies = [];
+    const ignoredRequests = new Set();
+    let index = 0;
+    const decodeReply = Effect.catchAll(Effect.suspend(() => {
+      const reply = encodedReplies[index];
+      if (ignoredRequests.has(reply.requestId)) return Effect.void;
+      const message = messages.get(reply.requestId);
+      if (!message) return Effect.void;
+      const schema = Reply.Reply(message.rpc);
+      return Schema.decode(schema)(reply).pipe(Effect.locally(FiberRef.currentContext, message.context));
+    }), error => {
+      const reply = encodedReplies[index];
+      ignoredRequests.add(reply.requestId);
+      return Effect.succeed(new Reply.WithExit({
+        id: snowflakeGen.unsafeNext(),
+        requestId: Snowflake.Snowflake(reply.requestId),
+        exit: Exit.die(error)
+      }));
+    });
+    return Effect.as(Effect.whileLoop({
+      while: () => index < encodedReplies.length,
+      body: () => decodeReply,
+      step: reply => {
+        index++;
+        if (reply) replies.push(reply);
+      }
+    }), replies);
+  };
+  return storage;
+});
+/**
+ * @since 1.0.0
+ * @category Constructors
+ */
+export const noop = /*#__PURE__*/globalValue("@effect/cluster/MessageStorage/noop", () => Effect.runSync(make({
+  saveRequest: () => Effect.succeed(SaveResult.Success()),
+  saveEnvelope: () => Effect.void,
+  saveReply: () => Effect.void,
+  repliesFor: () => Effect.succeed([]),
+  unprocessedMessages: () => Effect.succeed([]),
+  unprocessedMessagesById: () => Effect.succeed([]),
+  resetAddress: () => Effect.void
+})));
+/**
+ * @since 1.0.0
+ * @category Memory
+ */
+export class MemoryDriver extends /*#__PURE__*/Effect.Service()("@effect/cluster/MessageStorage/MemoryDriver", {
+  dependencies: [Snowflake.layerGenerator],
+  effect: /*#__PURE__*/Effect.gen(function* () {
+    const requests = new Map();
+    const requestsByPrimaryKey = new Map();
+    const unprocessed = new Set();
+    const replyIds = new Set();
+    const journal = [];
+    const cursors = new WeakMap();
+    const unprocessedWith = predicate => {
+      const messages = [];
+      for (const envelope of unprocessed) {
+        if (!predicate(envelope)) {
+          continue;
+        }
+        if (envelope._tag === "Request") {
+          const entry = requests.get(envelope.requestId);
+          messages.push({
+            envelope,
+            lastSentReply: Option.fromNullable(entry?.replies[entry.replies.length - 1])
+          });
+        } else {
+          messages.push({
+            envelope,
+            lastSentReply: Option.none()
+          });
+        }
+      }
+      return messages;
+    };
+    const replyLatch = yield* Effect.makeLatch();
+    const encoded = {
+      saveEnvelope: ({
+        envelope,
+        primaryKey
+      }) => Effect.sync(() => {
+        const existing = primaryKey ? requestsByPrimaryKey.get(primaryKey) : envelope._tag === "Request" && requests.get(envelope.requestId);
+        if (existing) {
+          return SaveResultEncoded.Duplicate({
+            originalId: Snowflake.Snowflake(existing.envelope.requestId),
+            lastReceivedReply: existing.lastReceivedChunk
+          });
+        }
+        if (envelope._tag === "Request") {
+          const entry = {
+            envelope,
+            replies: [],
+            lastReceivedChunk: Option.none()
+          };
+          requests.set(envelope.requestId, entry);
+          if (primaryKey) {
+            requestsByPrimaryKey.set(primaryKey, entry);
+          }
+          unprocessed.add(envelope);
+        } else if (envelope._tag === "AckChunk") {
+          const entry = requests.get(envelope.requestId);
+          if (entry) {
+            entry.lastReceivedChunk = Arr.findFirst(entry.replies, r => r._tag === "Chunk" && r.id === envelope.replyId).pipe(Option.orElse(() => entry.lastReceivedChunk));
+          }
+        }
+        journal.push(envelope);
+        return SaveResultEncoded.Success();
+      }),
+      saveReply: reply => Effect.sync(() => {
+        const entry = requests.get(reply.requestId);
+        if (!entry || replyIds.has(reply.id)) return;
+        if (reply._tag === "WithExit") {
+          unprocessed.delete(entry.envelope);
+        }
+        entry.replies.push(reply);
+        replyIds.add(reply.id);
+        replyLatch.unsafeOpen();
+      }),
+      repliesFor: requestIds => Effect.sync(() => {
+        const replies = Arr.empty();
+        for (const requestId of requestIds) {
+          const request = requests.get(requestId);
+          if (!request) continue;else if (Option.isNone(request.lastReceivedChunk)) {
+            // eslint-disable-next-line no-restricted-syntax
+            replies.push(...request.replies);
+            continue;
+          }
+          const sequence = request.lastReceivedChunk.value.sequence;
+          for (const reply of request.replies) {
+            if (reply._tag === "Chunk" && reply.sequence <= sequence) {
+              continue;
+            }
+            replies.push(reply);
+          }
+        }
+        return replies;
+      }),
+      unprocessedMessages: shardIds => Effect.sync(() => {
+        if (unprocessed.size === 0) return [];
+        const messages = Arr.empty();
+        let index = journal.indexOf(Iterable.unsafeHead(unprocessed));
+        for (; index < journal.length; index++) {
+          const envelope = journal[index];
+          if (!shardIds.includes(envelope.address.shardId)) {
+            continue;
+          }
+          if (envelope._tag === "Request") {
+            const entry = requests.get(envelope.requestId);
+            messages.push({
+              envelope,
+              lastSentReply: Arr.last(entry.replies)
+            });
+          } else {
+            messages.push({
+              envelope,
+              lastSentReply: Option.none()
+            });
+          }
+        }
+        return messages;
+      }),
+      unprocessedMessagesById: ids => Effect.sync(() => {
+        const envelopeIds = new Set();
+        for (const id of ids) {
+          envelopeIds.add(String(id));
+        }
+        return unprocessedWith(envelope => envelopeIds.has(envelope.requestId));
+      }),
+      resetAddress: () => Effect.void
+    };
+    const storage = yield* makeEncoded(encoded);
+    return {
+      storage,
+      encoded,
+      requests,
+      requestsByPrimaryKey,
+      unprocessed,
+      replyIds,
+      journal,
+      cursors
+    };
+  })
+}) {}
+/**
+ * @since 1.0.0
+ * @category layers
+ */
+export const layerNoop = /*#__PURE__*/Layer.succeed(MessageStorage, noop);
+/**
+ * @since 1.0.0
+ * @category layers
+ */
+export const layerMemory = /*#__PURE__*/Layer.effect(MessageStorage, Effect.map(MemoryDriver, _ => _.storage)).pipe( /*#__PURE__*/Layer.provideMerge(MemoryDriver.Default));
+// --- internal ---
+const EnvelopeWithReply = /*#__PURE__*/Schema.Struct({
+  envelope: Envelope.PartialEncoded,
+  lastSentReply: /*#__PURE__*/Schema.OptionFromSelf(Reply.Encoded)
+});
+const decodeEnvelopeWithReply = /*#__PURE__*/Schema.decode(EnvelopeWithReply);
+//# sourceMappingURL=MessageStorage.js.map