@milaboratories/pl-client 2.16.11 → 2.16.12

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (36) hide show
  1. package/dist/core/errors.cjs +2 -0
  2. package/dist/core/errors.cjs.map +1 -1
  3. package/dist/core/errors.d.ts.map +1 -1
  4. package/dist/core/errors.js +2 -0
  5. package/dist/core/errors.js.map +1 -1
  6. package/dist/core/ll_client.cjs +21 -8
  7. package/dist/core/ll_client.cjs.map +1 -1
  8. package/dist/core/ll_client.d.ts.map +1 -1
  9. package/dist/core/ll_client.js +21 -8
  10. package/dist/core/ll_client.js.map +1 -1
  11. package/dist/core/ll_transaction.cjs +10 -0
  12. package/dist/core/ll_transaction.cjs.map +1 -1
  13. package/dist/core/ll_transaction.d.ts +1 -0
  14. package/dist/core/ll_transaction.d.ts.map +1 -1
  15. package/dist/core/ll_transaction.js +10 -0
  16. package/dist/core/ll_transaction.js.map +1 -1
  17. package/dist/core/websocket_stream.cjs +333 -0
  18. package/dist/core/websocket_stream.cjs.map +1 -0
  19. package/dist/core/websocket_stream.d.ts +60 -0
  20. package/dist/core/websocket_stream.d.ts.map +1 -0
  21. package/dist/core/websocket_stream.js +331 -0
  22. package/dist/core/websocket_stream.js.map +1 -0
  23. package/dist/helpers/retry_strategy.cjs +92 -0
  24. package/dist/helpers/retry_strategy.cjs.map +1 -0
  25. package/dist/helpers/retry_strategy.d.ts +24 -0
  26. package/dist/helpers/retry_strategy.d.ts.map +1 -0
  27. package/dist/helpers/retry_strategy.js +89 -0
  28. package/dist/helpers/retry_strategy.js.map +1 -0
  29. package/package.json +3 -3
  30. package/src/core/errors.ts +1 -0
  31. package/src/core/ll_client.ts +24 -8
  32. package/src/core/ll_transaction.test.ts +18 -0
  33. package/src/core/ll_transaction.ts +12 -0
  34. package/src/core/websocket_stream.test.ts +412 -0
  35. package/src/core/websocket_stream.ts +412 -0
  36. package/src/helpers/retry_strategy.ts +123 -0
@@ -1 +1 @@
1
- {"version":3,"file":"ll_transaction.cjs","sources":["../../src/core/ll_transaction.ts"],"sourcesContent":["import type {\n TxAPI_ClientMessage,\n TxAPI_ServerMessage,\n} from '../proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api';\nimport type { BiDiStream } from './abstract_stream';\nimport Denque from 'denque';\nimport type { Status } from '../proto-grpc/github.com/googleapis/googleapis/google/rpc/status';\nimport {\n PlErrorCodeNotFound,\n RecoverablePlError,\n rethrowMeaningfulError,\n UnrecoverablePlError,\n} from './errors';\nimport { StatefulPromise } from './StatefulPromise';\n\nexport type ClientMessageRequest = TxAPI_ClientMessage['request'];\n\nexport type ServerMessageResponse = TxAPI_ServerMessage['response'];\n\ntype TxStream = BiDiStream<TxAPI_ClientMessage, TxAPI_ServerMessage>;\n\nexport type OneOfKind<T extends { oneofKind: unknown }, Kind extends T['oneofKind']> = Extract<\n T,\n { oneofKind: Kind }\n>;\n\ninterface SingleResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: false;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>) => void;\n reject: (e: Error) => void;\n}\n\ninterface MultiResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: true;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>[]) => void;\n reject: (e: Error) => void;\n}\n\ntype AnySingleResponseHandler = SingleResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyMultiResponseHandler = MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyResponseHandler =\n | SingleResponseHandler<ServerMessageResponse['oneofKind']>\n | MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\nfunction createResponseHandler<Kind extends ServerMessageResponse['oneofKind']>(\n kind: Kind,\n expectMultiResponse: boolean,\n resolve:\n | ((v: OneOfKind<ServerMessageResponse, Kind>) => void)\n | ((v: OneOfKind<ServerMessageResponse, Kind>[]) => void),\n reject: (e: Error) => void,\n): AnyResponseHandler {\n return { kind, expectMultiResponse, resolve, reject } as AnyResponseHandler;\n}\n\nfunction isRecoverable(status: Status): boolean {\n return status.code === PlErrorCodeNotFound;\n}\n\nexport class RethrowError extends Error {\n name = 'RethrowError';\n constructor(public readonly rethrowLambda: () => never) {\n super('Rethrow error, you should never see this one.');\n }\n}\n\nexport class LLPlTransaction {\n /** Bidirectional channel through which transaction communicates with the server */\n private readonly stream: TxStream;\n\n /** Used to abort ongoing transaction stream */\n private readonly abortController = new AbortController();\n\n /** Counter of sent requests, used to calculate which future response will correspond to this request.\n * Incremented on each sent request. */\n private requestIdxCounter = 0;\n\n /** Queue from which incoming message processor takes handlers to which pass incoming messages */\n private readonly responseHandlerQueue = new Denque<AnyResponseHandler>();\n\n /** Each new resource, created by the transaction, is assigned with virtual (local) resource id, to make it possible\n * to populate its fields without awaiting actual resource id. This counter tracks those ids on client side, the\n * same way it is tracked on the server, so client can synchronously return such ids to the user. */\n private localResourceIdCounter = 0n;\n\n /** Switches to true, when this transaction closes due to normal or exceptional conditions. Prevents any new messages\n * to be sent to the stream. */\n private closed = false;\n /** Whether the outgoing stream was already closed. */\n private completed = false;\n\n /** If this transaction was terminated due to error, this is a generator to create new errors if corresponding response is required. */\n private errorFactory?: () => never;\n\n /** Timestamp when transaction was opened */\n private readonly openTimestamp = Date.now();\n\n private readonly incomingProcessorResult: Promise<(() => never) | null>;\n\n constructor(streamFactory: (abortSignal: AbortSignal) => TxStream) {\n this.stream = streamFactory(this.abortController.signal);\n\n // Starting incoming event processor\n this.incomingProcessorResult = this.incomingEventProcessor();\n }\n\n private assignErrorFactoryIfNotSet(\n errorFactory: () => never,\n reject?: (e: Error) => void,\n ): () => never {\n if (reject !== undefined) reject(new RethrowError(errorFactory));\n if (this.errorFactory) return errorFactory;\n this.errorFactory = errorFactory;\n return errorFactory;\n }\n\n private async incomingEventProcessor(): Promise<(() => never) | null> {\n /** Counter of received responses, used to check consistency of responses.\n * Increments on each received message. */\n let expectedId = -1;\n\n // defined externally to make possible to communicate any processing errors\n // to the specific request on which it happened\n let currentHandler: AnyResponseHandler | undefined = undefined;\n let responseAggregator: ServerMessageResponse[] | undefined = undefined;\n try {\n for await (const message of this.stream.responses) {\n if (currentHandler === undefined) {\n currentHandler = this.responseHandlerQueue.shift();\n\n if (currentHandler === undefined) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`orphan incoming message`);\n });\n break;\n }\n\n // allocating response aggregator array\n if (currentHandler.expectMultiResponse) responseAggregator = [];\n\n expectedId++;\n }\n\n if (message.requestId !== expectedId) {\n const errorMessage = `out of order messages, ${message.requestId} !== ${expectedId}`;\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n });\n break;\n }\n\n if (message.error !== undefined) {\n const status = message.error;\n\n if (isRecoverable(status)) {\n currentHandler.reject(\n new RethrowError(() => {\n throw new RecoverablePlError(status);\n }),\n );\n currentHandler = undefined;\n\n if (message.multiMessage !== undefined && !message.multiMessage.isLast) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error('Unexpected message sequence.');\n });\n break;\n }\n\n // We can continue to work after recoverable errors\n continue;\n } else {\n this.assignErrorFactoryIfNotSet(() => {\n throw new UnrecoverablePlError(status);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n // In case of unrecoverable errors we close the transaction\n break;\n }\n }\n\n if (\n currentHandler.kind !== message.response.oneofKind\n && message?.multiMessage?.isEmpty !== true\n ) {\n const errorMessage = `inconsistent request response types: ${currentHandler.kind} !== ${message.response.oneofKind}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n if (currentHandler.expectMultiResponse !== (message.multiMessage !== undefined)) {\n const errorMessage = `inconsistent multi state: ${currentHandler.expectMultiResponse} !== ${message.multiMessage !== undefined}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n // <- at this point we validated everything we can at this level\n\n if (message.multiMessage !== undefined) {\n if (!message.multiMessage.isEmpty) {\n if (message.multiMessage.id !== responseAggregator!.length + 1) {\n const errorMessage = `inconsistent multi id: ${message.multiMessage.id} !== ${responseAggregator!.length + 1}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n responseAggregator!.push(message.response);\n }\n\n if (message.multiMessage.isLast) {\n (currentHandler as AnyMultiResponseHandler).resolve(responseAggregator!);\n responseAggregator = undefined;\n currentHandler = undefined;\n }\n } else {\n (currentHandler as AnySingleResponseHandler).resolve(message.response);\n currentHandler = undefined;\n }\n }\n } catch (e: any) {\n return this.assignErrorFactoryIfNotSet(() => {\n rethrowMeaningfulError(e, true);\n }, currentHandler?.reject);\n } finally {\n await this.close();\n }\n return null;\n }\n\n /** Executed after termination of incoming message processor */\n private async close(): Promise<void> {\n if (this.closed) return;\n\n this.closed = true;\n\n // Rejecting all messages\n\n while (true) {\n const handler = this.responseHandlerQueue.shift();\n if (!handler) break;\n if (this.errorFactory) handler.reject(new RethrowError(this.errorFactory));\n else handler.reject(new Error('no reply'));\n }\n\n // closing outgoing stream\n await this.stream.requests.complete();\n }\n\n /** Forcefully close the transaction, terminate all connections and reject all pending requests */\n public abort(cause?: Error) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`transaction aborted`, { cause });\n });\n this.abortController.abort(cause);\n }\n\n /** Await incoming message loop termination and throw any leftover errors if it was unsuccessful */\n public async await(): Promise<void> {\n // for those who want to understand \"why?\":\n // this way there is no hanging promise that will complete with rejection\n // until await is implicitly requested, the this.incomingProcessorResult\n // always resolves with success\n\n const processingResult = await this.incomingProcessorResult;\n if (processingResult !== null) processingResult();\n }\n\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: false\n ): Promise<OneOfKind<ServerMessageResponse, Kind>>;\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: true\n ): Promise<OneOfKind<ServerMessageResponse, Kind>[]>;\n /** Generate proper client message and send it to the server, and returns a promise of future response. */\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: boolean,\n ): Promise<OneOfKind<ServerMessageResponse, Kind> | OneOfKind<ServerMessageResponse, Kind>[]> {\n if (this.errorFactory) return Promise.reject(new RethrowError(this.errorFactory));\n\n if (this.closed) return Promise.reject(new Error('Transaction already closed'));\n\n // Note: Promise synchronously executes a callback passed to a constructor\n const result = StatefulPromise.fromDeferredReject(new Promise<OneOfKind<ServerMessageResponse, Kind>>((resolve, reject) => {\n this.responseHandlerQueue.push(\n createResponseHandler(r.oneofKind, expectMultiResponse, resolve, reject),\n );\n }));\n\n // Awaiting message dispatch to catch any associated errors.\n // There is no hurry, we are not going to receive a response until message is sent.\n await this.stream.requests.send({\n requestId: this.requestIdxCounter++,\n request: r,\n });\n\n try {\n return await result;\n } catch (e: any) {\n if (e instanceof RethrowError) e.rethrowLambda();\n throw new Error('Error while waiting for response', { cause: e });\n }\n }\n\n private _completed = false;\n\n /** Safe to call multiple times */\n public async complete() {\n if (this._completed) return;\n this._completed = true;\n await this.stream.requests.complete();\n }\n}\n"],"names":["PlErrorCodeNotFound","RecoverablePlError","UnrecoverablePlError","rethrowMeaningfulError","StatefulPromise"],"mappings":";;;;;;AAgDA,SAAS,qBAAqB,CAC5B,IAAU,EACV,mBAA4B,EAC5B,OAE2D,EAC3D,MAA0B,EAAA;IAE1B,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,EAAwB;AAC7E;AAEA,SAAS,aAAa,CAAC,MAAc,EAAA;AACnC,IAAA,OAAO,MAAM,CAAC,IAAI,KAAKA,0BAAmB;AAC5C;AAEM,MAAO,YAAa,SAAQ,KAAK,CAAA;AAET,IAAA,aAAA;IAD5B,IAAI,GAAG,cAAc;AACrB,IAAA,WAAA,CAA4B,aAA0B,EAAA;QACpD,KAAK,CAAC,+CAA+C,CAAC;QAD5B,IAAA,CAAA,aAAa,GAAb,aAAa;IAEzC;AACD;MAEY,eAAe,CAAA;;AAET,IAAA,MAAM;;AAGN,IAAA,eAAe,GAAG,IAAI,eAAe,EAAE;AAExD;AACuC;IAC/B,iBAAiB,GAAG,CAAC;;AAGZ,IAAA,oBAAoB,GAAG,IAAI,MAAM,EAAsB;AAExE;;AAEoG;IAC5F,sBAAsB,GAAG,EAAE;AAEnC;AAC+B;IACvB,MAAM,GAAG,KAAK;;IAEd,SAAS,GAAG,KAAK;;AAGjB,IAAA,YAAY;;AAGH,IAAA,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE;AAE1B,IAAA,uBAAuB;AAExC,IAAA,WAAA,CAAY,aAAqD,EAAA;QAC/D,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;;AAGxD,QAAA,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,sBAAsB,EAAE;IAC9D;IAEQ,0BAA0B,CAChC,YAAyB,EACzB,MAA2B,EAAA;QAE3B,IAAI,MAAM,KAAK,SAAS;AAAE,YAAA,MAAM,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,YAAY;AAC1C,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,OAAO,YAAY;IACrB;AAEQ,IAAA,MAAM,sBAAsB,GAAA;AAClC;AAC0C;AAC1C,QAAA,IAAI,UAAU,GAAG,EAAE;;;QAInB,IAAI,cAAc,GAAmC,SAAS;QAC9D,IAAI,kBAAkB,GAAwC,SAAS;AACvE,QAAA,IAAI;YACF,WAAW,MAAM,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;AACjD,gBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,oBAAA,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AAElD,oBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAI,KAAK,CAAC,CAAA,uBAAA,CAAyB,CAAC;AAC5C,wBAAA,CAAC,CAAC;wBACF;oBACF;;oBAGA,IAAI,cAAc,CAAC,mBAAmB;wBAAE,kBAAkB,GAAG,EAAE;AAE/D,oBAAA,UAAU,EAAE;gBACd;AAEA,gBAAA,IAAI,OAAO,CAAC,SAAS,KAAK,UAAU,EAAE;oBACpC,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,SAAS,CAAA,KAAA,EAAQ,UAAU,CAAA,CAAE;AACpF,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,CAAC;oBACF;gBACF;AAEA,gBAAA,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE;AAC/B,oBAAA,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK;AAE5B,oBAAA,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE;AACzB,wBAAA,cAAc,CAAC,MAAM,CACnB,IAAI,YAAY,CAAC,MAAK;AACpB,4BAAA,MAAM,IAAIC,yBAAkB,CAAC,MAAM,CAAC;wBACtC,CAAC,CAAC,CACH;wBACD,cAAc,GAAG,SAAS;AAE1B,wBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AACtE,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC;AACjD,4BAAA,CAAC,CAAC;4BACF;wBACF;;wBAGA;oBACF;yBAAO;AACL,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAIC,2BAAoB,CAAC,MAAM,CAAC;AACxC,wBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;wBACzB,cAAc,GAAG,SAAS;;wBAG1B;oBACF;gBACF;gBAEA,IACE,cAAc,CAAC,IAAI,KAAK,OAAO,CAAC,QAAQ,CAAC;AACtC,uBAAA,OAAO,EAAE,YAAY,EAAE,OAAO,KAAK,IAAI,EAC1C;AACA,oBAAA,MAAM,YAAY,GAAG,CAAA,qCAAA,EAAwC,cAAc,CAAC,IAAI,CAAA,KAAA,EAAQ,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE;AAEpH,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;AAEA,gBAAA,IAAI,cAAc,CAAC,mBAAmB,MAAM,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,EAAE;AAC/E,oBAAA,MAAM,YAAY,GAAG,CAAA,0BAAA,EAA6B,cAAc,CAAC,mBAAmB,CAAA,KAAA,EAAQ,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AAEhI,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;;AAIA,gBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AACtC,oBAAA,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,EAAE;AACjC,wBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAC9D,4BAAA,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,YAAY,CAAC,EAAE,CAAA,KAAA,EAAQ,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAE9G,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,4BAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;4BACzB,cAAc,GAAG,SAAS;4BAE1B;wBACF;AAEA,wBAAA,kBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5C;AAEA,oBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AAC9B,wBAAA,cAA0C,CAAC,OAAO,CAAC,kBAAmB,CAAC;wBACxE,kBAAkB,GAAG,SAAS;wBAC9B,cAAc,GAAG,SAAS;oBAC5B;gBACF;qBAAO;AACJ,oBAAA,cAA2C,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBACtE,cAAc,GAAG,SAAS;gBAC5B;YACF;QACF;QAAE,OAAO,CAAM,EAAE;AACf,YAAA,OAAO,IAAI,CAAC,0BAA0B,CAAC,MAAK;AAC1C,gBAAAC,6BAAsB,CAAC,CAAC,EAAE,IAAI,CAAC;AACjC,YAAA,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC;QAC5B;gBAAU;AACR,YAAA,MAAM,IAAI,CAAC,KAAK,EAAE;QACpB;AACA,QAAA,OAAO,IAAI;IACb;;AAGQ,IAAA,MAAM,KAAK,GAAA;QACjB,IAAI,IAAI,CAAC,MAAM;YAAE;AAEjB,QAAA,IAAI,CAAC,MAAM,GAAG,IAAI;;QAIlB,OAAO,IAAI,EAAE;YACX,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AACjD,YAAA,IAAI,CAAC,OAAO;gBAAE;YACd,IAAI,IAAI,CAAC,YAAY;gBAAE,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;;gBACrE,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C;;QAGA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;;AAGO,IAAA,KAAK,CAAC,KAAa,EAAA;AACxB,QAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;YACnC,MAAM,IAAI,KAAK,CAAC,CAAA,mBAAA,CAAqB,EAAE,EAAE,KAAK,EAAE,CAAC;AACnD,QAAA,CAAC,CAAC;AACF,QAAA,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC;IACnC;;AAGO,IAAA,MAAM,KAAK,GAAA;;;;;AAMhB,QAAA,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,uBAAuB;QAC3D,IAAI,gBAAgB,KAAK,IAAI;AAAE,YAAA,gBAAgB,EAAE;IACnD;;AAWO,IAAA,MAAM,IAAI,CACf,CAAwC,EACxC,mBAA4B,EAAA;QAE5B,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEjF,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;;AAG/E,QAAA,MAAM,MAAM,GAAGC,+BAAe,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAyC,CAAC,OAAO,EAAE,MAAM,KAAI;AACxH,YAAA,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAC5B,qBAAqB,CAAC,CAAC,CAAC,SAAS,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,CAAC,CACzE;QACH,CAAC,CAAC,CAAC;;;AAIH,QAAA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;AAC9B,YAAA,SAAS,EAAE,IAAI,CAAC,iBAAiB,EAAE;AACnC,YAAA,OAAO,EAAE,CAAC;AACX,SAAA,CAAC;AAEF,QAAA,IAAI;YACF,OAAO,MAAM,MAAM;QACrB;QAAE,OAAO,CAAM,EAAE;YACf,IAAI,CAAC,YAAY,YAAY;gBAAE,CAAC,CAAC,aAAa,EAAE;YAChD,MAAM,IAAI,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACnE;IACF;IAEQ,UAAU,GAAG,KAAK;;AAGnB,IAAA,MAAM,QAAQ,GAAA;QACnB,IAAI,IAAI,CAAC,UAAU;YAAE;AACrB,QAAA,IAAI,CAAC,UAAU,GAAG,IAAI;QACtB,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;AACD;;;;;"}
1
+ {"version":3,"file":"ll_transaction.cjs","sources":["../../src/core/ll_transaction.ts"],"sourcesContent":["import type {\n TxAPI_ClientMessage,\n TxAPI_ServerMessage,\n} from '../proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api';\nimport type { BiDiStream } from './abstract_stream';\nimport Denque from 'denque';\nimport type { Status } from '../proto-grpc/github.com/googleapis/googleapis/google/rpc/status';\nimport {\n PlErrorCodeNotFound,\n RecoverablePlError,\n rethrowMeaningfulError,\n UnrecoverablePlError,\n} from './errors';\nimport { StatefulPromise } from './StatefulPromise';\n\nexport type ClientMessageRequest = TxAPI_ClientMessage['request'];\n\nexport type ServerMessageResponse = TxAPI_ServerMessage['response'];\n\ntype TxStream = BiDiStream<TxAPI_ClientMessage, TxAPI_ServerMessage>;\n\nexport type OneOfKind<T extends { oneofKind: unknown }, Kind extends T['oneofKind']> = Extract<\n T,\n { oneofKind: Kind }\n>;\n\ninterface SingleResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: false;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>) => void;\n reject: (e: Error) => void;\n}\n\ninterface MultiResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: true;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>[]) => void;\n reject: (e: Error) => void;\n}\n\ntype AnySingleResponseHandler = SingleResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyMultiResponseHandler = MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyResponseHandler =\n | SingleResponseHandler<ServerMessageResponse['oneofKind']>\n | MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\nfunction createResponseHandler<Kind extends ServerMessageResponse['oneofKind']>(\n kind: Kind,\n expectMultiResponse: boolean,\n resolve:\n | ((v: OneOfKind<ServerMessageResponse, Kind>) => void)\n | ((v: OneOfKind<ServerMessageResponse, Kind>[]) => void),\n reject: (e: Error) => void,\n): AnyResponseHandler {\n return { kind, expectMultiResponse, resolve, reject } as AnyResponseHandler;\n}\n\nfunction isRecoverable(status: Status): boolean {\n return status.code === PlErrorCodeNotFound;\n}\n\nexport class RethrowError extends Error {\n name = 'RethrowError';\n constructor(public readonly rethrowLambda: () => never) {\n super('Rethrow error, you should never see this one.');\n }\n}\n\nexport class LLPlTransaction {\n /** Bidirectional channel through which transaction communicates with the server */\n private readonly stream: TxStream;\n\n /** Used to abort ongoing transaction stream */\n private readonly abortController = new AbortController();\n\n /** Counter of sent requests, used to calculate which future response will correspond to this request.\n * Incremented on each sent request. */\n private requestIdxCounter = 0;\n\n /** Queue from which incoming message processor takes handlers to which pass incoming messages */\n private readonly responseHandlerQueue = new Denque<AnyResponseHandler>();\n\n /** Each new resource, created by the transaction, is assigned with virtual (local) resource id, to make it possible\n * to populate its fields without awaiting actual resource id. This counter tracks those ids on client side, the\n * same way it is tracked on the server, so client can synchronously return such ids to the user. */\n private localResourceIdCounter = 0n;\n\n /** Switches to true, when this transaction closes due to normal or exceptional conditions. Prevents any new messages\n * to be sent to the stream. */\n private closed = false;\n /** Whether the outgoing stream was already closed. */\n private completed = false;\n\n /** If this transaction was terminated due to error, this is a generator to create new errors if corresponding response is required. */\n private errorFactory?: () => never;\n\n /** Timestamp when transaction was opened */\n private readonly openTimestamp = Date.now();\n\n private readonly incomingProcessorResult: Promise<(() => never) | null>;\n\n constructor(streamFactory: (abortSignal: AbortSignal) => TxStream) {\n this.stream = streamFactory(this.abortController.signal);\n\n // Starting incoming event processor\n this.incomingProcessorResult = this.incomingEventProcessor();\n }\n\n private assignErrorFactoryIfNotSet(\n errorFactory: () => never,\n reject?: (e: Error) => void,\n ): () => never {\n if (reject !== undefined) reject(new RethrowError(errorFactory));\n if (this.errorFactory) return errorFactory;\n this.errorFactory = errorFactory;\n return errorFactory;\n }\n\n private async incomingEventProcessor(): Promise<(() => never) | null> {\n /** Counter of received responses, used to check consistency of responses.\n * Increments on each received message. */\n let expectedId = -1;\n\n // defined externally to make possible to communicate any processing errors\n // to the specific request on which it happened\n let currentHandler: AnyResponseHandler | undefined = undefined;\n let responseAggregator: ServerMessageResponse[] | undefined = undefined;\n try {\n for await (const message of this.stream.responses) {\n if (currentHandler === undefined) {\n currentHandler = this.responseHandlerQueue.shift();\n\n if (currentHandler === undefined) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`orphan incoming message`);\n });\n break;\n }\n\n // allocating response aggregator array\n if (currentHandler.expectMultiResponse) responseAggregator = [];\n\n expectedId++;\n }\n\n if (message.requestId !== expectedId) {\n const errorMessage = `out of order messages, ${message.requestId} !== ${expectedId}`;\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n });\n break;\n }\n\n if (message.error !== undefined) {\n const status = message.error;\n\n if (isRecoverable(status)) {\n currentHandler.reject(\n new RethrowError(() => {\n throw new RecoverablePlError(status);\n }),\n );\n currentHandler = undefined;\n\n if (message.multiMessage !== undefined && !message.multiMessage.isLast) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error('Unexpected message sequence.');\n });\n break;\n }\n\n // We can continue to work after recoverable errors\n continue;\n } else {\n this.assignErrorFactoryIfNotSet(() => {\n throw new UnrecoverablePlError(status);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n // In case of unrecoverable errors we close the transaction\n break;\n }\n }\n\n if (\n currentHandler.kind !== message.response.oneofKind\n && message?.multiMessage?.isEmpty !== true\n ) {\n const errorMessage = `inconsistent request response types: ${currentHandler.kind} !== ${message.response.oneofKind}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n if (currentHandler.expectMultiResponse !== (message.multiMessage !== undefined)) {\n const errorMessage = `inconsistent multi state: ${currentHandler.expectMultiResponse} !== ${message.multiMessage !== undefined}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n // <- at this point we validated everything we can at this level\n\n if (message.multiMessage !== undefined) {\n if (!message.multiMessage.isEmpty) {\n if (message.multiMessage.id !== responseAggregator!.length + 1) {\n const errorMessage = `inconsistent multi id: ${message.multiMessage.id} !== ${responseAggregator!.length + 1}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n responseAggregator!.push(message.response);\n }\n\n if (message.multiMessage.isLast) {\n (currentHandler as AnyMultiResponseHandler).resolve(responseAggregator!);\n responseAggregator = undefined;\n currentHandler = undefined;\n }\n } else {\n (currentHandler as AnySingleResponseHandler).resolve(message.response);\n currentHandler = undefined;\n }\n\n // After receiving a terminal response (txCommit or txDiscard), we proactively close the client stream.\n // This ensures consistent behavior between the gRPC and WebSocket transports,\n // since the server closes the connection automatically upon transaction completion in both cases.\n if (this.isTerminalResponse(message) && this.responseHandlerQueue.length === 0) {\n await this.stream.requests.complete();\n }\n }\n } catch (e: any) {\n return this.assignErrorFactoryIfNotSet(() => {\n rethrowMeaningfulError(e, true);\n }, currentHandler?.reject);\n } finally {\n await this.close();\n }\n return null;\n }\n\n /** Executed after termination of incoming message processor */\n private async close(): Promise<void> {\n if (this.closed) return;\n\n this.closed = true;\n\n // Rejecting all messages\n\n while (true) {\n const handler = this.responseHandlerQueue.shift();\n if (!handler) break;\n if (this.errorFactory) handler.reject(new RethrowError(this.errorFactory));\n else handler.reject(new Error('no reply'));\n }\n\n // closing outgoing stream\n await this.stream.requests.complete();\n }\n\n /** Forcefully close the transaction, terminate all connections and reject all pending requests */\n public abort(cause?: Error) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`transaction aborted`, { cause });\n });\n this.abortController.abort(cause);\n }\n\n /** Await incoming message loop termination and throw any leftover errors if it was unsuccessful */\n public async await(): Promise<void> {\n // for those who want to understand \"why?\":\n // this way there is no hanging promise that will complete with rejection\n // until await is implicitly requested, the this.incomingProcessorResult\n // always resolves with success\n\n const processingResult = await this.incomingProcessorResult;\n if (processingResult !== null) processingResult();\n }\n\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: false\n ): Promise<OneOfKind<ServerMessageResponse, Kind>>;\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: true\n ): Promise<OneOfKind<ServerMessageResponse, Kind>[]>;\n /** Generate proper client message and send it to the server, and returns a promise of future response. */\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: boolean,\n ): Promise<OneOfKind<ServerMessageResponse, Kind> | OneOfKind<ServerMessageResponse, Kind>[]> {\n if (this.errorFactory) return Promise.reject(new RethrowError(this.errorFactory));\n\n if (this.closed) return Promise.reject(new Error('Transaction already closed'));\n\n // Note: Promise synchronously executes a callback passed to a constructor\n const result = StatefulPromise.fromDeferredReject(new Promise<OneOfKind<ServerMessageResponse, Kind>>((resolve, reject) => {\n this.responseHandlerQueue.push(\n createResponseHandler(r.oneofKind, expectMultiResponse, resolve, reject),\n );\n }));\n\n // Awaiting message dispatch to catch any associated errors.\n // There is no hurry, we are not going to receive a response until message is sent.\n await this.stream.requests.send({\n requestId: this.requestIdxCounter++,\n request: r,\n });\n\n try {\n return await result;\n } catch (e: any) {\n if (e instanceof RethrowError) e.rethrowLambda();\n throw new Error('Error while waiting for response', { cause: e });\n }\n }\n\n private _completed = false;\n\n /** Safe to call multiple times */\n public async complete() {\n if (this._completed) return;\n this._completed = true;\n await this.stream.requests.complete();\n }\n\n private isTerminalResponse(message: TxAPI_ServerMessage): boolean {\n const kind = message.response.oneofKind;\n return kind === 'txCommit' || kind === 'txDiscard';\n }\n}\n"],"names":["PlErrorCodeNotFound","RecoverablePlError","UnrecoverablePlError","rethrowMeaningfulError","StatefulPromise"],"mappings":";;;;;;AAgDA,SAAS,qBAAqB,CAC5B,IAAU,EACV,mBAA4B,EAC5B,OAE2D,EAC3D,MAA0B,EAAA;IAE1B,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,EAAwB;AAC7E;AAEA,SAAS,aAAa,CAAC,MAAc,EAAA;AACnC,IAAA,OAAO,MAAM,CAAC,IAAI,KAAKA,0BAAmB;AAC5C;AAEM,MAAO,YAAa,SAAQ,KAAK,CAAA;AAET,IAAA,aAAA;IAD5B,IAAI,GAAG,cAAc;AACrB,IAAA,WAAA,CAA4B,aAA0B,EAAA;QACpD,KAAK,CAAC,+CAA+C,CAAC;QAD5B,IAAA,CAAA,aAAa,GAAb,aAAa;IAEzC;AACD;MAEY,eAAe,CAAA;;AAET,IAAA,MAAM;;AAGN,IAAA,eAAe,GAAG,IAAI,eAAe,EAAE;AAExD;AACuC;IAC/B,iBAAiB,GAAG,CAAC;;AAGZ,IAAA,oBAAoB,GAAG,IAAI,MAAM,EAAsB;AAExE;;AAEoG;IAC5F,sBAAsB,GAAG,EAAE;AAEnC;AAC+B;IACvB,MAAM,GAAG,KAAK;;IAEd,SAAS,GAAG,KAAK;;AAGjB,IAAA,YAAY;;AAGH,IAAA,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE;AAE1B,IAAA,uBAAuB;AAExC,IAAA,WAAA,CAAY,aAAqD,EAAA;QAC/D,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;;AAGxD,QAAA,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,sBAAsB,EAAE;IAC9D;IAEQ,0BAA0B,CAChC,YAAyB,EACzB,MAA2B,EAAA;QAE3B,IAAI,MAAM,KAAK,SAAS;AAAE,YAAA,MAAM,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,YAAY;AAC1C,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,OAAO,YAAY;IACrB;AAEQ,IAAA,MAAM,sBAAsB,GAAA;AAClC;AAC0C;AAC1C,QAAA,IAAI,UAAU,GAAG,EAAE;;;QAInB,IAAI,cAAc,GAAmC,SAAS;QAC9D,IAAI,kBAAkB,GAAwC,SAAS;AACvE,QAAA,IAAI;YACF,WAAW,MAAM,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;AACjD,gBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,oBAAA,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AAElD,oBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAI,KAAK,CAAC,CAAA,uBAAA,CAAyB,CAAC;AAC5C,wBAAA,CAAC,CAAC;wBACF;oBACF;;oBAGA,IAAI,cAAc,CAAC,mBAAmB;wBAAE,kBAAkB,GAAG,EAAE;AAE/D,oBAAA,UAAU,EAAE;gBACd;AAEA,gBAAA,IAAI,OAAO,CAAC,SAAS,KAAK,UAAU,EAAE;oBACpC,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,SAAS,CAAA,KAAA,EAAQ,UAAU,CAAA,CAAE;AACpF,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,CAAC;oBACF;gBACF;AAEA,gBAAA,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE;AAC/B,oBAAA,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK;AAE5B,oBAAA,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE;AACzB,wBAAA,cAAc,CAAC,MAAM,CACnB,IAAI,YAAY,CAAC,MAAK;AACpB,4BAAA,MAAM,IAAIC,yBAAkB,CAAC,MAAM,CAAC;wBACtC,CAAC,CAAC,CACH;wBACD,cAAc,GAAG,SAAS;AAE1B,wBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AACtE,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC;AACjD,4BAAA,CAAC,CAAC;4BACF;wBACF;;wBAGA;oBACF;yBAAO;AACL,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAIC,2BAAoB,CAAC,MAAM,CAAC;AACxC,wBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;wBACzB,cAAc,GAAG,SAAS;;wBAG1B;oBACF;gBACF;gBAEA,IACE,cAAc,CAAC,IAAI,KAAK,OAAO,CAAC,QAAQ,CAAC;AACtC,uBAAA,OAAO,EAAE,YAAY,EAAE,OAAO,KAAK,IAAI,EAC1C;AACA,oBAAA,MAAM,YAAY,GAAG,CAAA,qCAAA,EAAwC,cAAc,CAAC,IAAI,CAAA,KAAA,EAAQ,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE;AAEpH,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;AAEA,gBAAA,IAAI,cAAc,CAAC,mBAAmB,MAAM,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,EAAE;AAC/E,oBAAA,MAAM,YAAY,GAAG,CAAA,0BAAA,EAA6B,cAAc,CAAC,mBAAmB,CAAA,KAAA,EAAQ,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AAEhI,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;;AAIA,gBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AACtC,oBAAA,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,EAAE;AACjC,wBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAC9D,4BAAA,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,YAAY,CAAC,EAAE,CAAA,KAAA,EAAQ,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAE9G,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,4BAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;4BACzB,cAAc,GAAG,SAAS;4BAE1B;wBACF;AAEA,wBAAA,kBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5C;AAEA,oBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AAC9B,wBAAA,cAA0C,CAAC,OAAO,CAAC,kBAAmB,CAAC;wBACxE,kBAAkB,GAAG,SAAS;wBAC9B,cAAc,GAAG,SAAS;oBAC5B;gBACF;qBAAO;AACJ,oBAAA,cAA2C,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBACtE,cAAc,GAAG,SAAS;gBAC5B;;;;AAKA,gBAAA,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC9E,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;gBACvC;YACF;QACF;QAAE,OAAO,CAAM,EAAE;AACf,YAAA,OAAO,IAAI,CAAC,0BAA0B,CAAC,MAAK;AAC1C,gBAAAC,6BAAsB,CAAC,CAAC,EAAE,IAAI,CAAC;AACjC,YAAA,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC;QAC5B;gBAAU;AACR,YAAA,MAAM,IAAI,CAAC,KAAK,EAAE;QACpB;AACA,QAAA,OAAO,IAAI;IACb;;AAGQ,IAAA,MAAM,KAAK,GAAA;QACjB,IAAI,IAAI,CAAC,MAAM;YAAE;AAEjB,QAAA,IAAI,CAAC,MAAM,GAAG,IAAI;;QAIlB,OAAO,IAAI,EAAE;YACX,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AACjD,YAAA,IAAI,CAAC,OAAO;gBAAE;YACd,IAAI,IAAI,CAAC,YAAY;gBAAE,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;;gBACrE,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C;;QAGA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;;AAGO,IAAA,KAAK,CAAC,KAAa,EAAA;AACxB,QAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;YACnC,MAAM,IAAI,KAAK,CAAC,CAAA,mBAAA,CAAqB,EAAE,EAAE,KAAK,EAAE,CAAC;AACnD,QAAA,CAAC,CAAC;AACF,QAAA,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC;IACnC;;AAGO,IAAA,MAAM,KAAK,GAAA;;;;;AAMhB,QAAA,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,uBAAuB;QAC3D,IAAI,gBAAgB,KAAK,IAAI;AAAE,YAAA,gBAAgB,EAAE;IACnD;;AAWO,IAAA,MAAM,IAAI,CACf,CAAwC,EACxC,mBAA4B,EAAA;QAE5B,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEjF,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;;AAG/E,QAAA,MAAM,MAAM,GAAGC,+BAAe,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAyC,CAAC,OAAO,EAAE,MAAM,KAAI;AACxH,YAAA,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAC5B,qBAAqB,CAAC,CAAC,CAAC,SAAS,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,CAAC,CACzE;QACH,CAAC,CAAC,CAAC;;;AAIH,QAAA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;AAC9B,YAAA,SAAS,EAAE,IAAI,CAAC,iBAAiB,EAAE;AACnC,YAAA,OAAO,EAAE,CAAC;AACX,SAAA,CAAC;AAEF,QAAA,IAAI;YACF,OAAO,MAAM,MAAM;QACrB;QAAE,OAAO,CAAM,EAAE;YACf,IAAI,CAAC,YAAY,YAAY;gBAAE,CAAC,CAAC,aAAa,EAAE;YAChD,MAAM,IAAI,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACnE;IACF;IAEQ,UAAU,GAAG,KAAK;;AAGnB,IAAA,MAAM,QAAQ,GAAA;QACnB,IAAI,IAAI,CAAC,UAAU;YAAE;AACrB,QAAA,IAAI,CAAC,UAAU,GAAG,IAAI;QACtB,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;AAEQ,IAAA,kBAAkB,CAAC,OAA4B,EAAA;AACrD,QAAA,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,CAAC,SAAS;AACvC,QAAA,OAAO,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,WAAW;IACpD;AACD;;;;;"}
@@ -51,6 +51,7 @@ export declare class LLPlTransaction {
51
51
  private _completed;
52
52
  /** Safe to call multiple times */
53
53
  complete(): Promise<void>;
54
+ private isTerminalResponse;
54
55
  }
55
56
  export {};
56
57
  //# sourceMappingURL=ll_transaction.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"ll_transaction.d.ts","sourceRoot":"","sources":["../../src/core/ll_transaction.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,mBAAmB,EACnB,mBAAmB,EACpB,MAAM,+DAA+D,CAAC;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAWpD,MAAM,MAAM,oBAAoB,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;AAElE,MAAM,MAAM,qBAAqB,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;AAEpE,KAAK,QAAQ,GAAG,UAAU,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAC;AAErE,MAAM,MAAM,SAAS,CAAC,CAAC,SAAS;IAAE,SAAS,EAAE,OAAO,CAAA;CAAE,EAAE,IAAI,SAAS,CAAC,CAAC,WAAW,CAAC,IAAI,OAAO,CAC5F,CAAC,EACD;IAAE,SAAS,EAAE,IAAI,CAAA;CAAE,CACpB,CAAC;AAuCF,qBAAa,YAAa,SAAQ,KAAK;aAET,aAAa,EAAE,MAAM,KAAK;IADtD,IAAI,SAAkB;gBACM,aAAa,EAAE,MAAM,KAAK;CAGvD;AAED,qBAAa,eAAe;IAC1B,mFAAmF;IACnF,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAW;IAElC,+CAA+C;IAC/C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAyB;IAEzD;2CACuC;IACvC,OAAO,CAAC,iBAAiB,CAAK;IAE9B,iGAAiG;IACjG,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoC;IAEzE;;wGAEoG;IACpG,OAAO,CAAC,sBAAsB,CAAM;IAEpC;mCAC+B;IAC/B,OAAO,CAAC,MAAM,CAAS;IACvB,sDAAsD;IACtD,OAAO,CAAC,SAAS,CAAS;IAE1B,uIAAuI;IACvI,OAAO,CAAC,YAAY,CAAC,CAAc;IAEnC,4CAA4C;IAC5C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAc;IAE5C,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAgC;gBAE5D,aAAa,EAAE,CAAC,WAAW,EAAE,WAAW,KAAK,QAAQ;IAOjE,OAAO,CAAC,0BAA0B;YAUpB,sBAAsB;IAiIpC,+DAA+D;YACjD,KAAK;IAkBnB,kGAAkG;IAC3F,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK;IAO1B,mGAAmG;IACtF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAUtB,IAAI,CAAC,IAAI,SAAS,oBAAoB,CAAC,WAAW,CAAC,EAC9D,CAAC,EAAE,SAAS,CAAC,oBAAoB,EAAE,IAAI,CAAC,EACxC,mBAAmB,EAAE,KAAK,GACzB,OAAO,CAAC,SAAS,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;IACrC,IAAI,CAAC,IAAI,SAAS,oBAAoB,CAAC,WAAW,CAAC,EAC9D,CAAC,EAAE,SAAS,CAAC,oBAAoB,EAAE,IAAI,CAAC,EACxC,mBAAmB,EAAE,IAAI,GACxB,OAAO,CAAC,SAAS,CAAC,qBAAqB,EAAE,IAAI,CAAC,EAAE,CAAC;IAgCpD,OAAO,CAAC,UAAU,CAAS;IAE3B,kCAAkC;IACrB,QAAQ;CAKtB"}
1
+ {"version":3,"file":"ll_transaction.d.ts","sourceRoot":"","sources":["../../src/core/ll_transaction.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,mBAAmB,EACnB,mBAAmB,EACpB,MAAM,+DAA+D,CAAC;AACvE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAWpD,MAAM,MAAM,oBAAoB,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;AAElE,MAAM,MAAM,qBAAqB,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;AAEpE,KAAK,QAAQ,GAAG,UAAU,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAC;AAErE,MAAM,MAAM,SAAS,CAAC,CAAC,SAAS;IAAE,SAAS,EAAE,OAAO,CAAA;CAAE,EAAE,IAAI,SAAS,CAAC,CAAC,WAAW,CAAC,IAAI,OAAO,CAC5F,CAAC,EACD;IAAE,SAAS,EAAE,IAAI,CAAA;CAAE,CACpB,CAAC;AAuCF,qBAAa,YAAa,SAAQ,KAAK;aAET,aAAa,EAAE,MAAM,KAAK;IADtD,IAAI,SAAkB;gBACM,aAAa,EAAE,MAAM,KAAK;CAGvD;AAED,qBAAa,eAAe;IAC1B,mFAAmF;IACnF,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAW;IAElC,+CAA+C;IAC/C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAyB;IAEzD;2CACuC;IACvC,OAAO,CAAC,iBAAiB,CAAK;IAE9B,iGAAiG;IACjG,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoC;IAEzE;;wGAEoG;IACpG,OAAO,CAAC,sBAAsB,CAAM;IAEpC;mCAC+B;IAC/B,OAAO,CAAC,MAAM,CAAS;IACvB,sDAAsD;IACtD,OAAO,CAAC,SAAS,CAAS;IAE1B,uIAAuI;IACvI,OAAO,CAAC,YAAY,CAAC,CAAc;IAEnC,4CAA4C;IAC5C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAc;IAE5C,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAgC;gBAE5D,aAAa,EAAE,CAAC,WAAW,EAAE,WAAW,KAAK,QAAQ;IAOjE,OAAO,CAAC,0BAA0B;YAUpB,sBAAsB;IAwIpC,+DAA+D;YACjD,KAAK;IAkBnB,kGAAkG;IAC3F,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK;IAO1B,mGAAmG;IACtF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAUtB,IAAI,CAAC,IAAI,SAAS,oBAAoB,CAAC,WAAW,CAAC,EAC9D,CAAC,EAAE,SAAS,CAAC,oBAAoB,EAAE,IAAI,CAAC,EACxC,mBAAmB,EAAE,KAAK,GACzB,OAAO,CAAC,SAAS,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;IACrC,IAAI,CAAC,IAAI,SAAS,oBAAoB,CAAC,WAAW,CAAC,EAC9D,CAAC,EAAE,SAAS,CAAC,oBAAoB,EAAE,IAAI,CAAC,EACxC,mBAAmB,EAAE,IAAI,GACxB,OAAO,CAAC,SAAS,CAAC,qBAAqB,EAAE,IAAI,CAAC,EAAE,CAAC;IAgCpD,OAAO,CAAC,UAAU,CAAS;IAE3B,kCAAkC;IACrB,QAAQ;IAMrB,OAAO,CAAC,kBAAkB;CAI3B"}
@@ -148,6 +148,12 @@ class LLPlTransaction {
148
148
  currentHandler.resolve(message.response);
149
149
  currentHandler = undefined;
150
150
  }
151
+ // After receiving a terminal response (txCommit or txDiscard), we proactively close the client stream.
152
+ // This ensures consistent behavior between the gRPC and WebSocket transports,
153
+ // since the server closes the connection automatically upon transaction completion in both cases.
154
+ if (this.isTerminalResponse(message) && this.responseHandlerQueue.length === 0) {
155
+ await this.stream.requests.complete();
156
+ }
151
157
  }
152
158
  }
153
159
  catch (e) {
@@ -228,6 +234,10 @@ class LLPlTransaction {
228
234
  this._completed = true;
229
235
  await this.stream.requests.complete();
230
236
  }
237
+ isTerminalResponse(message) {
238
+ const kind = message.response.oneofKind;
239
+ return kind === 'txCommit' || kind === 'txDiscard';
240
+ }
231
241
  }
232
242
 
233
243
  export { LLPlTransaction, RethrowError };
@@ -1 +1 @@
1
- {"version":3,"file":"ll_transaction.js","sources":["../../src/core/ll_transaction.ts"],"sourcesContent":["import type {\n TxAPI_ClientMessage,\n TxAPI_ServerMessage,\n} from '../proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api';\nimport type { BiDiStream } from './abstract_stream';\nimport Denque from 'denque';\nimport type { Status } from '../proto-grpc/github.com/googleapis/googleapis/google/rpc/status';\nimport {\n PlErrorCodeNotFound,\n RecoverablePlError,\n rethrowMeaningfulError,\n UnrecoverablePlError,\n} from './errors';\nimport { StatefulPromise } from './StatefulPromise';\n\nexport type ClientMessageRequest = TxAPI_ClientMessage['request'];\n\nexport type ServerMessageResponse = TxAPI_ServerMessage['response'];\n\ntype TxStream = BiDiStream<TxAPI_ClientMessage, TxAPI_ServerMessage>;\n\nexport type OneOfKind<T extends { oneofKind: unknown }, Kind extends T['oneofKind']> = Extract<\n T,\n { oneofKind: Kind }\n>;\n\ninterface SingleResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: false;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>) => void;\n reject: (e: Error) => void;\n}\n\ninterface MultiResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: true;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>[]) => void;\n reject: (e: Error) => void;\n}\n\ntype AnySingleResponseHandler = SingleResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyMultiResponseHandler = MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyResponseHandler =\n | SingleResponseHandler<ServerMessageResponse['oneofKind']>\n | MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\nfunction createResponseHandler<Kind extends ServerMessageResponse['oneofKind']>(\n kind: Kind,\n expectMultiResponse: boolean,\n resolve:\n | ((v: OneOfKind<ServerMessageResponse, Kind>) => void)\n | ((v: OneOfKind<ServerMessageResponse, Kind>[]) => void),\n reject: (e: Error) => void,\n): AnyResponseHandler {\n return { kind, expectMultiResponse, resolve, reject } as AnyResponseHandler;\n}\n\nfunction isRecoverable(status: Status): boolean {\n return status.code === PlErrorCodeNotFound;\n}\n\nexport class RethrowError extends Error {\n name = 'RethrowError';\n constructor(public readonly rethrowLambda: () => never) {\n super('Rethrow error, you should never see this one.');\n }\n}\n\nexport class LLPlTransaction {\n /** Bidirectional channel through which transaction communicates with the server */\n private readonly stream: TxStream;\n\n /** Used to abort ongoing transaction stream */\n private readonly abortController = new AbortController();\n\n /** Counter of sent requests, used to calculate which future response will correspond to this request.\n * Incremented on each sent request. */\n private requestIdxCounter = 0;\n\n /** Queue from which incoming message processor takes handlers to which pass incoming messages */\n private readonly responseHandlerQueue = new Denque<AnyResponseHandler>();\n\n /** Each new resource, created by the transaction, is assigned with virtual (local) resource id, to make it possible\n * to populate its fields without awaiting actual resource id. This counter tracks those ids on client side, the\n * same way it is tracked on the server, so client can synchronously return such ids to the user. */\n private localResourceIdCounter = 0n;\n\n /** Switches to true, when this transaction closes due to normal or exceptional conditions. Prevents any new messages\n * to be sent to the stream. */\n private closed = false;\n /** Whether the outgoing stream was already closed. */\n private completed = false;\n\n /** If this transaction was terminated due to error, this is a generator to create new errors if corresponding response is required. */\n private errorFactory?: () => never;\n\n /** Timestamp when transaction was opened */\n private readonly openTimestamp = Date.now();\n\n private readonly incomingProcessorResult: Promise<(() => never) | null>;\n\n constructor(streamFactory: (abortSignal: AbortSignal) => TxStream) {\n this.stream = streamFactory(this.abortController.signal);\n\n // Starting incoming event processor\n this.incomingProcessorResult = this.incomingEventProcessor();\n }\n\n private assignErrorFactoryIfNotSet(\n errorFactory: () => never,\n reject?: (e: Error) => void,\n ): () => never {\n if (reject !== undefined) reject(new RethrowError(errorFactory));\n if (this.errorFactory) return errorFactory;\n this.errorFactory = errorFactory;\n return errorFactory;\n }\n\n private async incomingEventProcessor(): Promise<(() => never) | null> {\n /** Counter of received responses, used to check consistency of responses.\n * Increments on each received message. */\n let expectedId = -1;\n\n // defined externally to make possible to communicate any processing errors\n // to the specific request on which it happened\n let currentHandler: AnyResponseHandler | undefined = undefined;\n let responseAggregator: ServerMessageResponse[] | undefined = undefined;\n try {\n for await (const message of this.stream.responses) {\n if (currentHandler === undefined) {\n currentHandler = this.responseHandlerQueue.shift();\n\n if (currentHandler === undefined) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`orphan incoming message`);\n });\n break;\n }\n\n // allocating response aggregator array\n if (currentHandler.expectMultiResponse) responseAggregator = [];\n\n expectedId++;\n }\n\n if (message.requestId !== expectedId) {\n const errorMessage = `out of order messages, ${message.requestId} !== ${expectedId}`;\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n });\n break;\n }\n\n if (message.error !== undefined) {\n const status = message.error;\n\n if (isRecoverable(status)) {\n currentHandler.reject(\n new RethrowError(() => {\n throw new RecoverablePlError(status);\n }),\n );\n currentHandler = undefined;\n\n if (message.multiMessage !== undefined && !message.multiMessage.isLast) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error('Unexpected message sequence.');\n });\n break;\n }\n\n // We can continue to work after recoverable errors\n continue;\n } else {\n this.assignErrorFactoryIfNotSet(() => {\n throw new UnrecoverablePlError(status);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n // In case of unrecoverable errors we close the transaction\n break;\n }\n }\n\n if (\n currentHandler.kind !== message.response.oneofKind\n && message?.multiMessage?.isEmpty !== true\n ) {\n const errorMessage = `inconsistent request response types: ${currentHandler.kind} !== ${message.response.oneofKind}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n if (currentHandler.expectMultiResponse !== (message.multiMessage !== undefined)) {\n const errorMessage = `inconsistent multi state: ${currentHandler.expectMultiResponse} !== ${message.multiMessage !== undefined}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n // <- at this point we validated everything we can at this level\n\n if (message.multiMessage !== undefined) {\n if (!message.multiMessage.isEmpty) {\n if (message.multiMessage.id !== responseAggregator!.length + 1) {\n const errorMessage = `inconsistent multi id: ${message.multiMessage.id} !== ${responseAggregator!.length + 1}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n responseAggregator!.push(message.response);\n }\n\n if (message.multiMessage.isLast) {\n (currentHandler as AnyMultiResponseHandler).resolve(responseAggregator!);\n responseAggregator = undefined;\n currentHandler = undefined;\n }\n } else {\n (currentHandler as AnySingleResponseHandler).resolve(message.response);\n currentHandler = undefined;\n }\n }\n } catch (e: any) {\n return this.assignErrorFactoryIfNotSet(() => {\n rethrowMeaningfulError(e, true);\n }, currentHandler?.reject);\n } finally {\n await this.close();\n }\n return null;\n }\n\n /** Executed after termination of incoming message processor */\n private async close(): Promise<void> {\n if (this.closed) return;\n\n this.closed = true;\n\n // Rejecting all messages\n\n while (true) {\n const handler = this.responseHandlerQueue.shift();\n if (!handler) break;\n if (this.errorFactory) handler.reject(new RethrowError(this.errorFactory));\n else handler.reject(new Error('no reply'));\n }\n\n // closing outgoing stream\n await this.stream.requests.complete();\n }\n\n /** Forcefully close the transaction, terminate all connections and reject all pending requests */\n public abort(cause?: Error) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`transaction aborted`, { cause });\n });\n this.abortController.abort(cause);\n }\n\n /** Await incoming message loop termination and throw any leftover errors if it was unsuccessful */\n public async await(): Promise<void> {\n // for those who want to understand \"why?\":\n // this way there is no hanging promise that will complete with rejection\n // until await is implicitly requested, the this.incomingProcessorResult\n // always resolves with success\n\n const processingResult = await this.incomingProcessorResult;\n if (processingResult !== null) processingResult();\n }\n\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: false\n ): Promise<OneOfKind<ServerMessageResponse, Kind>>;\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: true\n ): Promise<OneOfKind<ServerMessageResponse, Kind>[]>;\n /** Generate proper client message and send it to the server, and returns a promise of future response. */\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: boolean,\n ): Promise<OneOfKind<ServerMessageResponse, Kind> | OneOfKind<ServerMessageResponse, Kind>[]> {\n if (this.errorFactory) return Promise.reject(new RethrowError(this.errorFactory));\n\n if (this.closed) return Promise.reject(new Error('Transaction already closed'));\n\n // Note: Promise synchronously executes a callback passed to a constructor\n const result = StatefulPromise.fromDeferredReject(new Promise<OneOfKind<ServerMessageResponse, Kind>>((resolve, reject) => {\n this.responseHandlerQueue.push(\n createResponseHandler(r.oneofKind, expectMultiResponse, resolve, reject),\n );\n }));\n\n // Awaiting message dispatch to catch any associated errors.\n // There is no hurry, we are not going to receive a response until message is sent.\n await this.stream.requests.send({\n requestId: this.requestIdxCounter++,\n request: r,\n });\n\n try {\n return await result;\n } catch (e: any) {\n if (e instanceof RethrowError) e.rethrowLambda();\n throw new Error('Error while waiting for response', { cause: e });\n }\n }\n\n private _completed = false;\n\n /** Safe to call multiple times */\n public async complete() {\n if (this._completed) return;\n this._completed = true;\n await this.stream.requests.complete();\n }\n}\n"],"names":[],"mappings":";;;;AAgDA,SAAS,qBAAqB,CAC5B,IAAU,EACV,mBAA4B,EAC5B,OAE2D,EAC3D,MAA0B,EAAA;IAE1B,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,EAAwB;AAC7E;AAEA,SAAS,aAAa,CAAC,MAAc,EAAA;AACnC,IAAA,OAAO,MAAM,CAAC,IAAI,KAAK,mBAAmB;AAC5C;AAEM,MAAO,YAAa,SAAQ,KAAK,CAAA;AAET,IAAA,aAAA;IAD5B,IAAI,GAAG,cAAc;AACrB,IAAA,WAAA,CAA4B,aAA0B,EAAA;QACpD,KAAK,CAAC,+CAA+C,CAAC;QAD5B,IAAA,CAAA,aAAa,GAAb,aAAa;IAEzC;AACD;MAEY,eAAe,CAAA;;AAET,IAAA,MAAM;;AAGN,IAAA,eAAe,GAAG,IAAI,eAAe,EAAE;AAExD;AACuC;IAC/B,iBAAiB,GAAG,CAAC;;AAGZ,IAAA,oBAAoB,GAAG,IAAI,MAAM,EAAsB;AAExE;;AAEoG;IAC5F,sBAAsB,GAAG,EAAE;AAEnC;AAC+B;IACvB,MAAM,GAAG,KAAK;;IAEd,SAAS,GAAG,KAAK;;AAGjB,IAAA,YAAY;;AAGH,IAAA,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE;AAE1B,IAAA,uBAAuB;AAExC,IAAA,WAAA,CAAY,aAAqD,EAAA;QAC/D,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;;AAGxD,QAAA,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,sBAAsB,EAAE;IAC9D;IAEQ,0BAA0B,CAChC,YAAyB,EACzB,MAA2B,EAAA;QAE3B,IAAI,MAAM,KAAK,SAAS;AAAE,YAAA,MAAM,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,YAAY;AAC1C,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,OAAO,YAAY;IACrB;AAEQ,IAAA,MAAM,sBAAsB,GAAA;AAClC;AAC0C;AAC1C,QAAA,IAAI,UAAU,GAAG,EAAE;;;QAInB,IAAI,cAAc,GAAmC,SAAS;QAC9D,IAAI,kBAAkB,GAAwC,SAAS;AACvE,QAAA,IAAI;YACF,WAAW,MAAM,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;AACjD,gBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,oBAAA,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AAElD,oBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAI,KAAK,CAAC,CAAA,uBAAA,CAAyB,CAAC;AAC5C,wBAAA,CAAC,CAAC;wBACF;oBACF;;oBAGA,IAAI,cAAc,CAAC,mBAAmB;wBAAE,kBAAkB,GAAG,EAAE;AAE/D,oBAAA,UAAU,EAAE;gBACd;AAEA,gBAAA,IAAI,OAAO,CAAC,SAAS,KAAK,UAAU,EAAE;oBACpC,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,SAAS,CAAA,KAAA,EAAQ,UAAU,CAAA,CAAE;AACpF,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,CAAC;oBACF;gBACF;AAEA,gBAAA,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE;AAC/B,oBAAA,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK;AAE5B,oBAAA,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE;AACzB,wBAAA,cAAc,CAAC,MAAM,CACnB,IAAI,YAAY,CAAC,MAAK;AACpB,4BAAA,MAAM,IAAI,kBAAkB,CAAC,MAAM,CAAC;wBACtC,CAAC,CAAC,CACH;wBACD,cAAc,GAAG,SAAS;AAE1B,wBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AACtE,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC;AACjD,4BAAA,CAAC,CAAC;4BACF;wBACF;;wBAGA;oBACF;yBAAO;AACL,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAI,oBAAoB,CAAC,MAAM,CAAC;AACxC,wBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;wBACzB,cAAc,GAAG,SAAS;;wBAG1B;oBACF;gBACF;gBAEA,IACE,cAAc,CAAC,IAAI,KAAK,OAAO,CAAC,QAAQ,CAAC;AACtC,uBAAA,OAAO,EAAE,YAAY,EAAE,OAAO,KAAK,IAAI,EAC1C;AACA,oBAAA,MAAM,YAAY,GAAG,CAAA,qCAAA,EAAwC,cAAc,CAAC,IAAI,CAAA,KAAA,EAAQ,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE;AAEpH,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;AAEA,gBAAA,IAAI,cAAc,CAAC,mBAAmB,MAAM,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,EAAE;AAC/E,oBAAA,MAAM,YAAY,GAAG,CAAA,0BAAA,EAA6B,cAAc,CAAC,mBAAmB,CAAA,KAAA,EAAQ,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AAEhI,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;;AAIA,gBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AACtC,oBAAA,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,EAAE;AACjC,wBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAC9D,4BAAA,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,YAAY,CAAC,EAAE,CAAA,KAAA,EAAQ,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAE9G,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,4BAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;4BACzB,cAAc,GAAG,SAAS;4BAE1B;wBACF;AAEA,wBAAA,kBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5C;AAEA,oBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AAC9B,wBAAA,cAA0C,CAAC,OAAO,CAAC,kBAAmB,CAAC;wBACxE,kBAAkB,GAAG,SAAS;wBAC9B,cAAc,GAAG,SAAS;oBAC5B;gBACF;qBAAO;AACJ,oBAAA,cAA2C,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBACtE,cAAc,GAAG,SAAS;gBAC5B;YACF;QACF;QAAE,OAAO,CAAM,EAAE;AACf,YAAA,OAAO,IAAI,CAAC,0BAA0B,CAAC,MAAK;AAC1C,gBAAA,sBAAsB,CAAC,CAAC,EAAE,IAAI,CAAC;AACjC,YAAA,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC;QAC5B;gBAAU;AACR,YAAA,MAAM,IAAI,CAAC,KAAK,EAAE;QACpB;AACA,QAAA,OAAO,IAAI;IACb;;AAGQ,IAAA,MAAM,KAAK,GAAA;QACjB,IAAI,IAAI,CAAC,MAAM;YAAE;AAEjB,QAAA,IAAI,CAAC,MAAM,GAAG,IAAI;;QAIlB,OAAO,IAAI,EAAE;YACX,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AACjD,YAAA,IAAI,CAAC,OAAO;gBAAE;YACd,IAAI,IAAI,CAAC,YAAY;gBAAE,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;;gBACrE,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C;;QAGA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;;AAGO,IAAA,KAAK,CAAC,KAAa,EAAA;AACxB,QAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;YACnC,MAAM,IAAI,KAAK,CAAC,CAAA,mBAAA,CAAqB,EAAE,EAAE,KAAK,EAAE,CAAC;AACnD,QAAA,CAAC,CAAC;AACF,QAAA,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC;IACnC;;AAGO,IAAA,MAAM,KAAK,GAAA;;;;;AAMhB,QAAA,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,uBAAuB;QAC3D,IAAI,gBAAgB,KAAK,IAAI;AAAE,YAAA,gBAAgB,EAAE;IACnD;;AAWO,IAAA,MAAM,IAAI,CACf,CAAwC,EACxC,mBAA4B,EAAA;QAE5B,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEjF,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;;AAG/E,QAAA,MAAM,MAAM,GAAG,eAAe,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAyC,CAAC,OAAO,EAAE,MAAM,KAAI;AACxH,YAAA,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAC5B,qBAAqB,CAAC,CAAC,CAAC,SAAS,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,CAAC,CACzE;QACH,CAAC,CAAC,CAAC;;;AAIH,QAAA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;AAC9B,YAAA,SAAS,EAAE,IAAI,CAAC,iBAAiB,EAAE;AACnC,YAAA,OAAO,EAAE,CAAC;AACX,SAAA,CAAC;AAEF,QAAA,IAAI;YACF,OAAO,MAAM,MAAM;QACrB;QAAE,OAAO,CAAM,EAAE;YACf,IAAI,CAAC,YAAY,YAAY;gBAAE,CAAC,CAAC,aAAa,EAAE;YAChD,MAAM,IAAI,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACnE;IACF;IAEQ,UAAU,GAAG,KAAK;;AAGnB,IAAA,MAAM,QAAQ,GAAA;QACnB,IAAI,IAAI,CAAC,UAAU;YAAE;AACrB,QAAA,IAAI,CAAC,UAAU,GAAG,IAAI;QACtB,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;AACD;;;;"}
1
+ {"version":3,"file":"ll_transaction.js","sources":["../../src/core/ll_transaction.ts"],"sourcesContent":["import type {\n TxAPI_ClientMessage,\n TxAPI_ServerMessage,\n} from '../proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api';\nimport type { BiDiStream } from './abstract_stream';\nimport Denque from 'denque';\nimport type { Status } from '../proto-grpc/github.com/googleapis/googleapis/google/rpc/status';\nimport {\n PlErrorCodeNotFound,\n RecoverablePlError,\n rethrowMeaningfulError,\n UnrecoverablePlError,\n} from './errors';\nimport { StatefulPromise } from './StatefulPromise';\n\nexport type ClientMessageRequest = TxAPI_ClientMessage['request'];\n\nexport type ServerMessageResponse = TxAPI_ServerMessage['response'];\n\ntype TxStream = BiDiStream<TxAPI_ClientMessage, TxAPI_ServerMessage>;\n\nexport type OneOfKind<T extends { oneofKind: unknown }, Kind extends T['oneofKind']> = Extract<\n T,\n { oneofKind: Kind }\n>;\n\ninterface SingleResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: false;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>) => void;\n reject: (e: Error) => void;\n}\n\ninterface MultiResponseHandler<Kind extends ServerMessageResponse['oneofKind']> {\n kind: Kind;\n expectMultiResponse: true;\n resolve: (v: OneOfKind<ServerMessageResponse, Kind>[]) => void;\n reject: (e: Error) => void;\n}\n\ntype AnySingleResponseHandler = SingleResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyMultiResponseHandler = MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\ntype AnyResponseHandler =\n | SingleResponseHandler<ServerMessageResponse['oneofKind']>\n | MultiResponseHandler<ServerMessageResponse['oneofKind']>;\n\nfunction createResponseHandler<Kind extends ServerMessageResponse['oneofKind']>(\n kind: Kind,\n expectMultiResponse: boolean,\n resolve:\n | ((v: OneOfKind<ServerMessageResponse, Kind>) => void)\n | ((v: OneOfKind<ServerMessageResponse, Kind>[]) => void),\n reject: (e: Error) => void,\n): AnyResponseHandler {\n return { kind, expectMultiResponse, resolve, reject } as AnyResponseHandler;\n}\n\nfunction isRecoverable(status: Status): boolean {\n return status.code === PlErrorCodeNotFound;\n}\n\nexport class RethrowError extends Error {\n name = 'RethrowError';\n constructor(public readonly rethrowLambda: () => never) {\n super('Rethrow error, you should never see this one.');\n }\n}\n\nexport class LLPlTransaction {\n /** Bidirectional channel through which transaction communicates with the server */\n private readonly stream: TxStream;\n\n /** Used to abort ongoing transaction stream */\n private readonly abortController = new AbortController();\n\n /** Counter of sent requests, used to calculate which future response will correspond to this request.\n * Incremented on each sent request. */\n private requestIdxCounter = 0;\n\n /** Queue from which incoming message processor takes handlers to which pass incoming messages */\n private readonly responseHandlerQueue = new Denque<AnyResponseHandler>();\n\n /** Each new resource, created by the transaction, is assigned with virtual (local) resource id, to make it possible\n * to populate its fields without awaiting actual resource id. This counter tracks those ids on client side, the\n * same way it is tracked on the server, so client can synchronously return such ids to the user. */\n private localResourceIdCounter = 0n;\n\n /** Switches to true, when this transaction closes due to normal or exceptional conditions. Prevents any new messages\n * to be sent to the stream. */\n private closed = false;\n /** Whether the outgoing stream was already closed. */\n private completed = false;\n\n /** If this transaction was terminated due to error, this is a generator to create new errors if corresponding response is required. */\n private errorFactory?: () => never;\n\n /** Timestamp when transaction was opened */\n private readonly openTimestamp = Date.now();\n\n private readonly incomingProcessorResult: Promise<(() => never) | null>;\n\n constructor(streamFactory: (abortSignal: AbortSignal) => TxStream) {\n this.stream = streamFactory(this.abortController.signal);\n\n // Starting incoming event processor\n this.incomingProcessorResult = this.incomingEventProcessor();\n }\n\n private assignErrorFactoryIfNotSet(\n errorFactory: () => never,\n reject?: (e: Error) => void,\n ): () => never {\n if (reject !== undefined) reject(new RethrowError(errorFactory));\n if (this.errorFactory) return errorFactory;\n this.errorFactory = errorFactory;\n return errorFactory;\n }\n\n private async incomingEventProcessor(): Promise<(() => never) | null> {\n /** Counter of received responses, used to check consistency of responses.\n * Increments on each received message. */\n let expectedId = -1;\n\n // defined externally to make possible to communicate any processing errors\n // to the specific request on which it happened\n let currentHandler: AnyResponseHandler | undefined = undefined;\n let responseAggregator: ServerMessageResponse[] | undefined = undefined;\n try {\n for await (const message of this.stream.responses) {\n if (currentHandler === undefined) {\n currentHandler = this.responseHandlerQueue.shift();\n\n if (currentHandler === undefined) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`orphan incoming message`);\n });\n break;\n }\n\n // allocating response aggregator array\n if (currentHandler.expectMultiResponse) responseAggregator = [];\n\n expectedId++;\n }\n\n if (message.requestId !== expectedId) {\n const errorMessage = `out of order messages, ${message.requestId} !== ${expectedId}`;\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n });\n break;\n }\n\n if (message.error !== undefined) {\n const status = message.error;\n\n if (isRecoverable(status)) {\n currentHandler.reject(\n new RethrowError(() => {\n throw new RecoverablePlError(status);\n }),\n );\n currentHandler = undefined;\n\n if (message.multiMessage !== undefined && !message.multiMessage.isLast) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error('Unexpected message sequence.');\n });\n break;\n }\n\n // We can continue to work after recoverable errors\n continue;\n } else {\n this.assignErrorFactoryIfNotSet(() => {\n throw new UnrecoverablePlError(status);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n // In case of unrecoverable errors we close the transaction\n break;\n }\n }\n\n if (\n currentHandler.kind !== message.response.oneofKind\n && message?.multiMessage?.isEmpty !== true\n ) {\n const errorMessage = `inconsistent request response types: ${currentHandler.kind} !== ${message.response.oneofKind}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n if (currentHandler.expectMultiResponse !== (message.multiMessage !== undefined)) {\n const errorMessage = `inconsistent multi state: ${currentHandler.expectMultiResponse} !== ${message.multiMessage !== undefined}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n // <- at this point we validated everything we can at this level\n\n if (message.multiMessage !== undefined) {\n if (!message.multiMessage.isEmpty) {\n if (message.multiMessage.id !== responseAggregator!.length + 1) {\n const errorMessage = `inconsistent multi id: ${message.multiMessage.id} !== ${responseAggregator!.length + 1}`;\n\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(errorMessage);\n }, currentHandler.reject);\n currentHandler = undefined;\n\n break;\n }\n\n responseAggregator!.push(message.response);\n }\n\n if (message.multiMessage.isLast) {\n (currentHandler as AnyMultiResponseHandler).resolve(responseAggregator!);\n responseAggregator = undefined;\n currentHandler = undefined;\n }\n } else {\n (currentHandler as AnySingleResponseHandler).resolve(message.response);\n currentHandler = undefined;\n }\n\n // After receiving a terminal response (txCommit or txDiscard), we proactively close the client stream.\n // This ensures consistent behavior between the gRPC and WebSocket transports,\n // since the server closes the connection automatically upon transaction completion in both cases.\n if (this.isTerminalResponse(message) && this.responseHandlerQueue.length === 0) {\n await this.stream.requests.complete();\n }\n }\n } catch (e: any) {\n return this.assignErrorFactoryIfNotSet(() => {\n rethrowMeaningfulError(e, true);\n }, currentHandler?.reject);\n } finally {\n await this.close();\n }\n return null;\n }\n\n /** Executed after termination of incoming message processor */\n private async close(): Promise<void> {\n if (this.closed) return;\n\n this.closed = true;\n\n // Rejecting all messages\n\n while (true) {\n const handler = this.responseHandlerQueue.shift();\n if (!handler) break;\n if (this.errorFactory) handler.reject(new RethrowError(this.errorFactory));\n else handler.reject(new Error('no reply'));\n }\n\n // closing outgoing stream\n await this.stream.requests.complete();\n }\n\n /** Forcefully close the transaction, terminate all connections and reject all pending requests */\n public abort(cause?: Error) {\n this.assignErrorFactoryIfNotSet(() => {\n throw new Error(`transaction aborted`, { cause });\n });\n this.abortController.abort(cause);\n }\n\n /** Await incoming message loop termination and throw any leftover errors if it was unsuccessful */\n public async await(): Promise<void> {\n // for those who want to understand \"why?\":\n // this way there is no hanging promise that will complete with rejection\n // until await is implicitly requested, the this.incomingProcessorResult\n // always resolves with success\n\n const processingResult = await this.incomingProcessorResult;\n if (processingResult !== null) processingResult();\n }\n\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: false\n ): Promise<OneOfKind<ServerMessageResponse, Kind>>;\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: true\n ): Promise<OneOfKind<ServerMessageResponse, Kind>[]>;\n /** Generate proper client message and send it to the server, and returns a promise of future response. */\n public async send<Kind extends ClientMessageRequest['oneofKind']>(\n r: OneOfKind<ClientMessageRequest, Kind>,\n expectMultiResponse: boolean,\n ): Promise<OneOfKind<ServerMessageResponse, Kind> | OneOfKind<ServerMessageResponse, Kind>[]> {\n if (this.errorFactory) return Promise.reject(new RethrowError(this.errorFactory));\n\n if (this.closed) return Promise.reject(new Error('Transaction already closed'));\n\n // Note: Promise synchronously executes a callback passed to a constructor\n const result = StatefulPromise.fromDeferredReject(new Promise<OneOfKind<ServerMessageResponse, Kind>>((resolve, reject) => {\n this.responseHandlerQueue.push(\n createResponseHandler(r.oneofKind, expectMultiResponse, resolve, reject),\n );\n }));\n\n // Awaiting message dispatch to catch any associated errors.\n // There is no hurry, we are not going to receive a response until message is sent.\n await this.stream.requests.send({\n requestId: this.requestIdxCounter++,\n request: r,\n });\n\n try {\n return await result;\n } catch (e: any) {\n if (e instanceof RethrowError) e.rethrowLambda();\n throw new Error('Error while waiting for response', { cause: e });\n }\n }\n\n private _completed = false;\n\n /** Safe to call multiple times */\n public async complete() {\n if (this._completed) return;\n this._completed = true;\n await this.stream.requests.complete();\n }\n\n private isTerminalResponse(message: TxAPI_ServerMessage): boolean {\n const kind = message.response.oneofKind;\n return kind === 'txCommit' || kind === 'txDiscard';\n }\n}\n"],"names":[],"mappings":";;;;AAgDA,SAAS,qBAAqB,CAC5B,IAAU,EACV,mBAA4B,EAC5B,OAE2D,EAC3D,MAA0B,EAAA;IAE1B,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,EAAwB;AAC7E;AAEA,SAAS,aAAa,CAAC,MAAc,EAAA;AACnC,IAAA,OAAO,MAAM,CAAC,IAAI,KAAK,mBAAmB;AAC5C;AAEM,MAAO,YAAa,SAAQ,KAAK,CAAA;AAET,IAAA,aAAA;IAD5B,IAAI,GAAG,cAAc;AACrB,IAAA,WAAA,CAA4B,aAA0B,EAAA;QACpD,KAAK,CAAC,+CAA+C,CAAC;QAD5B,IAAA,CAAA,aAAa,GAAb,aAAa;IAEzC;AACD;MAEY,eAAe,CAAA;;AAET,IAAA,MAAM;;AAGN,IAAA,eAAe,GAAG,IAAI,eAAe,EAAE;AAExD;AACuC;IAC/B,iBAAiB,GAAG,CAAC;;AAGZ,IAAA,oBAAoB,GAAG,IAAI,MAAM,EAAsB;AAExE;;AAEoG;IAC5F,sBAAsB,GAAG,EAAE;AAEnC;AAC+B;IACvB,MAAM,GAAG,KAAK;;IAEd,SAAS,GAAG,KAAK;;AAGjB,IAAA,YAAY;;AAGH,IAAA,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE;AAE1B,IAAA,uBAAuB;AAExC,IAAA,WAAA,CAAY,aAAqD,EAAA;QAC/D,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;;AAGxD,QAAA,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,sBAAsB,EAAE;IAC9D;IAEQ,0BAA0B,CAChC,YAAyB,EACzB,MAA2B,EAAA;QAE3B,IAAI,MAAM,KAAK,SAAS;AAAE,YAAA,MAAM,CAAC,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;QAChE,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,YAAY;AAC1C,QAAA,IAAI,CAAC,YAAY,GAAG,YAAY;AAChC,QAAA,OAAO,YAAY;IACrB;AAEQ,IAAA,MAAM,sBAAsB,GAAA;AAClC;AAC0C;AAC1C,QAAA,IAAI,UAAU,GAAG,EAAE;;;QAInB,IAAI,cAAc,GAAmC,SAAS;QAC9D,IAAI,kBAAkB,GAAwC,SAAS;AACvE,QAAA,IAAI;YACF,WAAW,MAAM,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;AACjD,gBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,oBAAA,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AAElD,oBAAA,IAAI,cAAc,KAAK,SAAS,EAAE;AAChC,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAI,KAAK,CAAC,CAAA,uBAAA,CAAyB,CAAC;AAC5C,wBAAA,CAAC,CAAC;wBACF;oBACF;;oBAGA,IAAI,cAAc,CAAC,mBAAmB;wBAAE,kBAAkB,GAAG,EAAE;AAE/D,oBAAA,UAAU,EAAE;gBACd;AAEA,gBAAA,IAAI,OAAO,CAAC,SAAS,KAAK,UAAU,EAAE;oBACpC,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,SAAS,CAAA,KAAA,EAAQ,UAAU,CAAA,CAAE;AACpF,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,CAAC;oBACF;gBACF;AAEA,gBAAA,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE;AAC/B,oBAAA,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK;AAE5B,oBAAA,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE;AACzB,wBAAA,cAAc,CAAC,MAAM,CACnB,IAAI,YAAY,CAAC,MAAK;AACpB,4BAAA,MAAM,IAAI,kBAAkB,CAAC,MAAM,CAAC;wBACtC,CAAC,CAAC,CACH;wBACD,cAAc,GAAG,SAAS;AAE1B,wBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AACtE,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC;AACjD,4BAAA,CAAC,CAAC;4BACF;wBACF;;wBAGA;oBACF;yBAAO;AACL,wBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,4BAAA,MAAM,IAAI,oBAAoB,CAAC,MAAM,CAAC;AACxC,wBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;wBACzB,cAAc,GAAG,SAAS;;wBAG1B;oBACF;gBACF;gBAEA,IACE,cAAc,CAAC,IAAI,KAAK,OAAO,CAAC,QAAQ,CAAC;AACtC,uBAAA,OAAO,EAAE,YAAY,EAAE,OAAO,KAAK,IAAI,EAC1C;AACA,oBAAA,MAAM,YAAY,GAAG,CAAA,qCAAA,EAAwC,cAAc,CAAC,IAAI,CAAA,KAAA,EAAQ,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE;AAEpH,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;AAEA,gBAAA,IAAI,cAAc,CAAC,mBAAmB,MAAM,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,EAAE;AAC/E,oBAAA,MAAM,YAAY,GAAG,CAAA,0BAAA,EAA6B,cAAc,CAAC,mBAAmB,CAAA,KAAA,EAAQ,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AAEhI,oBAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,wBAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,oBAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;oBACzB,cAAc,GAAG,SAAS;oBAE1B;gBACF;;AAIA,gBAAA,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE;AACtC,oBAAA,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,EAAE;AACjC,wBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAC9D,4BAAA,MAAM,YAAY,GAAG,CAAA,uBAAA,EAA0B,OAAO,CAAC,YAAY,CAAC,EAAE,CAAA,KAAA,EAAQ,kBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE;AAE9G,4BAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;AACnC,gCAAA,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC;AAC/B,4BAAA,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC;4BACzB,cAAc,GAAG,SAAS;4BAE1B;wBACF;AAEA,wBAAA,kBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;oBAC5C;AAEA,oBAAA,IAAI,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE;AAC9B,wBAAA,cAA0C,CAAC,OAAO,CAAC,kBAAmB,CAAC;wBACxE,kBAAkB,GAAG,SAAS;wBAC9B,cAAc,GAAG,SAAS;oBAC5B;gBACF;qBAAO;AACJ,oBAAA,cAA2C,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;oBACtE,cAAc,GAAG,SAAS;gBAC5B;;;;AAKA,gBAAA,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC9E,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;gBACvC;YACF;QACF;QAAE,OAAO,CAAM,EAAE;AACf,YAAA,OAAO,IAAI,CAAC,0BAA0B,CAAC,MAAK;AAC1C,gBAAA,sBAAsB,CAAC,CAAC,EAAE,IAAI,CAAC;AACjC,YAAA,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC;QAC5B;gBAAU;AACR,YAAA,MAAM,IAAI,CAAC,KAAK,EAAE;QACpB;AACA,QAAA,OAAO,IAAI;IACb;;AAGQ,IAAA,MAAM,KAAK,GAAA;QACjB,IAAI,IAAI,CAAC,MAAM;YAAE;AAEjB,QAAA,IAAI,CAAC,MAAM,GAAG,IAAI;;QAIlB,OAAO,IAAI,EAAE;YACX,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE;AACjD,YAAA,IAAI,CAAC,OAAO;gBAAE;YACd,IAAI,IAAI,CAAC,YAAY;gBAAE,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;;gBACrE,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C;;QAGA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;;AAGO,IAAA,KAAK,CAAC,KAAa,EAAA;AACxB,QAAA,IAAI,CAAC,0BAA0B,CAAC,MAAK;YACnC,MAAM,IAAI,KAAK,CAAC,CAAA,mBAAA,CAAqB,EAAE,EAAE,KAAK,EAAE,CAAC;AACnD,QAAA,CAAC,CAAC;AACF,QAAA,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC;IACnC;;AAGO,IAAA,MAAM,KAAK,GAAA;;;;;AAMhB,QAAA,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,uBAAuB;QAC3D,IAAI,gBAAgB,KAAK,IAAI;AAAE,YAAA,gBAAgB,EAAE;IACnD;;AAWO,IAAA,MAAM,IAAI,CACf,CAAwC,EACxC,mBAA4B,EAAA;QAE5B,IAAI,IAAI,CAAC,YAAY;AAAE,YAAA,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEjF,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;;AAG/E,QAAA,MAAM,MAAM,GAAG,eAAe,CAAC,kBAAkB,CAAC,IAAI,OAAO,CAAyC,CAAC,OAAO,EAAE,MAAM,KAAI;AACxH,YAAA,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAC5B,qBAAqB,CAAC,CAAC,CAAC,SAAS,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,CAAC,CACzE;QACH,CAAC,CAAC,CAAC;;;AAIH,QAAA,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;AAC9B,YAAA,SAAS,EAAE,IAAI,CAAC,iBAAiB,EAAE;AACnC,YAAA,OAAO,EAAE,CAAC;AACX,SAAA,CAAC;AAEF,QAAA,IAAI;YACF,OAAO,MAAM,MAAM;QACrB;QAAE,OAAO,CAAM,EAAE;YACf,IAAI,CAAC,YAAY,YAAY;gBAAE,CAAC,CAAC,aAAa,EAAE;YAChD,MAAM,IAAI,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACnE;IACF;IAEQ,UAAU,GAAG,KAAK;;AAGnB,IAAA,MAAM,QAAQ,GAAA;QACnB,IAAI,IAAI,CAAC,UAAU;YAAE;AACrB,QAAA,IAAI,CAAC,UAAU,GAAG,IAAI;QACtB,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE;IACvC;AAEQ,IAAA,kBAAkB,CAAC,OAA4B,EAAA;AACrD,QAAA,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,CAAC,SAAS;AACvC,QAAA,OAAO,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,WAAW;IACpD;AACD;;;;"}
@@ -0,0 +1,333 @@
1
+ 'use strict';
2
+
3
+ var undici = require('undici');
4
+ var api = require('../proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api.cjs');
5
+ var Denque = require('denque');
6
+ var retry_strategy = require('../helpers/retry_strategy.cjs');
7
+
8
+ /**
9
+ * WebSocket-based bidirectional stream implementation for LLTransaction.
10
+ * Implements BiDiStream interface which is compatible with DuplexStreamingCall.
11
+ */
12
+ class WebSocketBiDiStream {
13
+ // Connection
14
+ ws = null;
15
+ connectionState = 'disconnected';
16
+ url;
17
+ jwtToken;
18
+ abortSignal;
19
+ reconnection;
20
+ // Send management
21
+ sendQueue = new Denque();
22
+ sendCompleted = false;
23
+ // Response management
24
+ responseQueue = new Denque();
25
+ responseResolvers = [];
26
+ // Error tracking
27
+ connectionError = null;
28
+ // === Public API ===
29
+ requests = {
30
+ send: async (message) => {
31
+ this.validateSendState();
32
+ return this.enqueueSend(message);
33
+ },
34
+ complete: async () => {
35
+ if (this.sendCompleted)
36
+ return;
37
+ this.sendCompleted = true;
38
+ await this.drainSendQueue();
39
+ this.closeConnection();
40
+ },
41
+ };
42
+ responses = {
43
+ [Symbol.asyncIterator]: () => this.createResponseIterator(),
44
+ };
45
+ constructor(url, abortSignal, jwtToken, retryConfig = {}) {
46
+ this.url = url;
47
+ this.jwtToken = jwtToken;
48
+ this.abortSignal = abortSignal;
49
+ this.reconnection = new retry_strategy.RetryStrategy(retryConfig, {
50
+ onRetry: () => { void this.connect(); },
51
+ onMaxAttemptsReached: (error) => this.handleError(error),
52
+ });
53
+ if (abortSignal.aborted) {
54
+ this.connectionState = 'closed';
55
+ return;
56
+ }
57
+ this.attachAbortSignalHandler();
58
+ void this.connect();
59
+ }
60
+ // === Connection Lifecycle ===
61
+ connect() {
62
+ if (this.isConnectingOrConnected() || this.abortSignal.aborted)
63
+ return;
64
+ this.connectionState = 'connecting';
65
+ this.connectionError = null;
66
+ try {
67
+ this.ws = this.createWebSocket();
68
+ this.attachWebSocketHandlers();
69
+ }
70
+ catch (error) {
71
+ this.connectionError = this.toError(error);
72
+ this.connectionState = 'disconnected';
73
+ this.reconnection.schedule();
74
+ }
75
+ }
76
+ createWebSocket() {
77
+ const options = this.jwtToken
78
+ ? { headers: { authorization: `Bearer ${this.jwtToken}` } }
79
+ : undefined;
80
+ const ws = new undici.WebSocket(this.url, options);
81
+ if (ws) {
82
+ ws.binaryType = 'arraybuffer';
83
+ }
84
+ return ws;
85
+ }
86
+ attachWebSocketHandlers() {
87
+ if (!this.ws)
88
+ return;
89
+ this.ws.addEventListener('open', () => this.onOpen());
90
+ this.ws.addEventListener('message', (event) => this.onMessage(event.data));
91
+ this.ws.addEventListener('error', (error) => this.onError(error));
92
+ this.ws.addEventListener('close', () => this.onClose());
93
+ }
94
+ attachAbortSignalHandler() {
95
+ this.abortSignal.addEventListener('abort', () => this.close());
96
+ }
97
+ onOpen() {
98
+ this.connectionState = 'connected';
99
+ this.reconnection.reset();
100
+ void this.processSendQueue();
101
+ }
102
+ onClose() {
103
+ this.ws = null;
104
+ if (this.isClosed() || this.abortSignal.aborted)
105
+ return;
106
+ if (this.sendCompleted) {
107
+ this.finalizeStream();
108
+ }
109
+ else {
110
+ this.connectionState = 'disconnected';
111
+ this.reconnection.schedule();
112
+ }
113
+ }
114
+ onError(error) {
115
+ this.handleError(this.toError(error));
116
+ }
117
+ onMessage(data) {
118
+ try {
119
+ const message = this.parseMessage(data);
120
+ this.deliverResponse(message);
121
+ }
122
+ catch (error) {
123
+ this.handleError(this.toError(error));
124
+ }
125
+ }
126
+ closeConnection() {
127
+ if (this.ws?.readyState === undici.WebSocket.OPEN) {
128
+ this.ws.close();
129
+ }
130
+ }
131
+ close() {
132
+ if (this.isClosed())
133
+ return;
134
+ this.connectionState = 'closed';
135
+ this.reconnection.cancel();
136
+ this.closeWebSocket();
137
+ this.rejectAllPendingOperations();
138
+ }
139
+ closeWebSocket() {
140
+ if (!this.ws)
141
+ return;
142
+ try {
143
+ this.ws.close();
144
+ }
145
+ catch {
146
+ // Suppress close errors
147
+ }
148
+ this.ws = null;
149
+ }
150
+ finalizeStream() {
151
+ this.connectionState = 'closed';
152
+ this.resolveAllPendingResponses();
153
+ }
154
+ resolveAllPendingResponses() {
155
+ while (this.responseResolvers.length > 0) {
156
+ const resolver = this.responseResolvers.shift();
157
+ resolver.resolve({ value: undefined, done: true });
158
+ }
159
+ }
160
+ parseMessage(data) {
161
+ if (data instanceof ArrayBuffer) {
162
+ return api.TxAPI_ServerMessage.fromBinary(new Uint8Array(data));
163
+ }
164
+ throw new Error(`Unsupported message format: ${typeof data}`);
165
+ }
166
+ // === Send Queue Management ===
167
+ validateSendState() {
168
+ if (this.sendCompleted) {
169
+ throw new Error('Cannot send: stream already completed');
170
+ }
171
+ if (this.abortSignal.aborted) {
172
+ throw new Error('Cannot send: stream aborted');
173
+ }
174
+ }
175
+ enqueueSend(message) {
176
+ return new Promise((resolve, reject) => {
177
+ this.sendQueue.push({ message, resolve, reject });
178
+ void this.processSendQueue();
179
+ });
180
+ }
181
+ processSendQueue() {
182
+ if (!this.canSendMessages())
183
+ return;
184
+ while (this.sendQueue.length > 0) {
185
+ const queued = this.sendQueue.shift();
186
+ this.sendQueuedMessage(queued);
187
+ }
188
+ }
189
+ canSendMessages() {
190
+ return this.connectionState === 'connected' && this.ws !== null;
191
+ }
192
+ sendQueuedMessage(queued) {
193
+ try {
194
+ const ws = this.ws;
195
+ if (!ws) {
196
+ throw new Error('WebSocket is not connected');
197
+ }
198
+ // Check if WebSocket is in a valid state for sending
199
+ if (ws.readyState !== undici.WebSocket.OPEN) {
200
+ throw new Error(`WebSocket is not open (readyState: ${ws.readyState})`);
201
+ }
202
+ const binary = api.TxAPI_ClientMessage.toBinary(queued.message);
203
+ ws.send(binary);
204
+ queued.resolve();
205
+ }
206
+ catch (error) {
207
+ queued.reject(this.toError(error));
208
+ }
209
+ }
210
+ async drainSendQueue() {
211
+ const POLL_INTERVAL_MS = 10;
212
+ while (this.sendQueue.length > 0) {
213
+ await this.waitForCondition(() => this.sendQueue.length === 0, POLL_INTERVAL_MS);
214
+ }
215
+ }
216
+ waitForCondition(condition, intervalMs) {
217
+ return new Promise((resolve, reject) => {
218
+ if (this.abortSignal.aborted) {
219
+ return reject(this.toError(this.abortSignal.reason) ?? new Error('Stream aborted'));
220
+ }
221
+ let timeoutId;
222
+ const onAbort = () => {
223
+ clearTimeout(timeoutId);
224
+ reject(this.toError(this.abortSignal.reason) ?? new Error('Stream aborted'));
225
+ };
226
+ this.abortSignal.addEventListener('abort', onAbort, { once: true });
227
+ const check = () => {
228
+ if (condition() || this.isStreamEnded()) {
229
+ this.abortSignal.removeEventListener('abort', onAbort);
230
+ resolve();
231
+ }
232
+ else {
233
+ timeoutId = setTimeout(check, intervalMs);
234
+ }
235
+ };
236
+ check();
237
+ });
238
+ }
239
+ // === Response Delivery ===
240
+ deliverResponse(message) {
241
+ if (this.responseResolvers.length > 0) {
242
+ const resolver = this.responseResolvers.shift();
243
+ resolver.resolve({ value: message, done: false });
244
+ }
245
+ else {
246
+ this.responseQueue.push(message);
247
+ }
248
+ }
249
+ async *createResponseIterator() {
250
+ while (true) {
251
+ const result = await this.nextResponse();
252
+ if (result.done)
253
+ break;
254
+ yield result.value;
255
+ }
256
+ }
257
+ nextResponse() {
258
+ return new Promise((resolve, reject) => {
259
+ // Fast path: message already available
260
+ if (this.responseQueue.length > 0) {
261
+ const message = this.responseQueue.shift();
262
+ resolve({ value: message, done: false });
263
+ return;
264
+ }
265
+ // Stream ended
266
+ if (this.isStreamEnded()) {
267
+ if (this.connectionError) {
268
+ reject(this.connectionError);
269
+ }
270
+ else {
271
+ resolve({ value: undefined, done: true });
272
+ }
273
+ return;
274
+ }
275
+ // Wait for next message
276
+ this.responseResolvers.push({ resolve, reject });
277
+ });
278
+ }
279
+ // === Error Handling ===
280
+ handleError(error) {
281
+ if (this.isClosed())
282
+ return;
283
+ this.connectionState = 'closed';
284
+ this.connectionError = error;
285
+ this.reconnection.cancel();
286
+ this.closeWebSocket();
287
+ this.rejectAllPendingOperations(error);
288
+ }
289
+ rejectAllPendingOperations(error) {
290
+ const err = error ?? this.createStreamClosedError();
291
+ this.rejectAllSendOperations(err);
292
+ this.rejectAllResponseResolvers(err);
293
+ }
294
+ rejectAllSendOperations(error) {
295
+ while (this.sendQueue.length > 0) {
296
+ const queued = this.sendQueue.shift();
297
+ queued.reject(error);
298
+ }
299
+ }
300
+ rejectAllResponseResolvers(error) {
301
+ while (this.responseResolvers.length > 0) {
302
+ const resolver = this.responseResolvers.shift();
303
+ resolver.reject(error);
304
+ }
305
+ }
306
+ createStreamClosedError() {
307
+ if (this.abortSignal.aborted) {
308
+ const reason = this.abortSignal.reason;
309
+ if (reason instanceof Error) {
310
+ return reason;
311
+ }
312
+ return new Error('Stream aborted', { cause: reason });
313
+ }
314
+ return new Error('Stream closed');
315
+ }
316
+ // === State Checks ===
317
+ isConnectingOrConnected() {
318
+ return this.connectionState === 'connecting'
319
+ || this.connectionState === 'connected';
320
+ }
321
+ isClosed() {
322
+ return this.connectionState === 'closed';
323
+ }
324
+ isStreamEnded() {
325
+ return this.isClosed() || this.abortSignal.aborted;
326
+ }
327
+ toError(error) {
328
+ return error instanceof Error ? error : new Error(String(error));
329
+ }
330
+ }
331
+
332
+ exports.WebSocketBiDiStream = WebSocketBiDiStream;
333
+ //# sourceMappingURL=websocket_stream.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"websocket_stream.cjs","sources":["../../src/core/websocket_stream.ts"],"sourcesContent":["import { WebSocket } from 'undici';\nimport {\n TxAPI_ClientMessage as ClientMessageType,\n TxAPI_ServerMessage as ServerMessageType,\n} from '../proto-grpc/github.com/milaboratory/pl/plapi/plapiproto/api';\nimport type { BiDiStream } from './abstract_stream';\nimport Denque from 'denque';\nimport type { RetryConfig } from '../helpers/retry_strategy';\nimport { RetryStrategy } from '../helpers/retry_strategy';\n\ntype ConnectionState = 'disconnected' | 'connecting' | 'connected' | 'closing' | 'closed';\n\ninterface QueuedMessage {\n message: ClientMessageType;\n resolve: () => void;\n reject: (error: Error) => void;\n}\n\ninterface ResponseResolver {\n resolve: (value: IteratorResult<ServerMessageType>) => void;\n reject: (error: Error) => void;\n}\n\n/**\n * WebSocket-based bidirectional stream implementation for LLTransaction.\n * Implements BiDiStream interface which is compatible with DuplexStreamingCall.\n */\nexport class WebSocketBiDiStream implements BiDiStream<ClientMessageType, ServerMessageType> {\n // Connection\n private ws: WebSocket | null = null;\n private connectionState: ConnectionState = 'disconnected';\n private readonly url: string;\n private readonly jwtToken?: string;\n private readonly abortSignal: AbortSignal;\n private readonly reconnection: RetryStrategy;\n\n // Send management\n private readonly sendQueue = new Denque<QueuedMessage>();\n private sendCompleted = false;\n\n // Response management\n private readonly responseQueue = new Denque<ServerMessageType>();\n private responseResolvers: ResponseResolver[] = [];\n\n // Error tracking\n private connectionError: Error | null = null;\n\n // === Public API ===\n\n public readonly requests = {\n send: async (message: ClientMessageType): Promise<void> => {\n this.validateSendState();\n return this.enqueueSend(message);\n },\n\n complete: async (): Promise<void> => {\n if (this.sendCompleted) return;\n\n this.sendCompleted = true;\n await this.drainSendQueue();\n this.closeConnection();\n },\n };\n\n public readonly responses: AsyncIterable<ServerMessageType> = {\n [Symbol.asyncIterator]: () => this.createResponseIterator(),\n };\n\n constructor(\n url: string,\n abortSignal: AbortSignal,\n jwtToken?: string,\n retryConfig: Partial<RetryConfig> = {},\n ) {\n this.url = url;\n this.jwtToken = jwtToken;\n this.abortSignal = abortSignal;\n\n this.reconnection = new RetryStrategy(retryConfig, {\n onRetry: () => { void this.connect(); },\n onMaxAttemptsReached: (error) => this.handleError(error),\n });\n\n if (abortSignal.aborted) {\n this.connectionState = 'closed';\n return;\n }\n\n this.attachAbortSignalHandler();\n void this.connect();\n }\n\n // === Connection Lifecycle ===\n\n private connect(): void {\n if (this.isConnectingOrConnected() || this.abortSignal.aborted) return;\n\n this.connectionState = 'connecting';\n this.connectionError = null;\n\n try {\n this.ws = this.createWebSocket();\n this.attachWebSocketHandlers();\n } catch (error) {\n this.connectionError = this.toError(error);\n this.connectionState = 'disconnected';\n this.reconnection.schedule();\n }\n }\n\n private createWebSocket(): WebSocket {\n const options = this.jwtToken\n ? { headers: { authorization: `Bearer ${this.jwtToken}` } }\n : undefined;\n\n const ws = new (WebSocket as any)(this.url, options);\n if (ws) {\n ws.binaryType = 'arraybuffer';\n }\n return ws;\n }\n\n private attachWebSocketHandlers(): void {\n if (!this.ws) return;\n\n this.ws.addEventListener('open', () => this.onOpen());\n this.ws.addEventListener('message', (event) => this.onMessage(event.data));\n this.ws.addEventListener('error', (error) => this.onError(error));\n this.ws.addEventListener('close', () => this.onClose());\n }\n\n private attachAbortSignalHandler(): void {\n this.abortSignal.addEventListener('abort', () => this.close());\n }\n\n private onOpen(): void {\n this.connectionState = 'connected';\n this.reconnection.reset();\n void this.processSendQueue();\n }\n\n private onClose(): void {\n this.ws = null;\n\n if (this.isClosed() || this.abortSignal.aborted) return;\n\n if (this.sendCompleted) {\n this.finalizeStream();\n } else {\n this.connectionState = 'disconnected';\n this.reconnection.schedule();\n }\n }\n\n private onError(error: unknown): void {\n this.handleError(this.toError(error));\n }\n\n private onMessage(data: unknown): void {\n try {\n const message = this.parseMessage(data);\n this.deliverResponse(message);\n } catch (error) {\n this.handleError(this.toError(error));\n }\n }\n\n private closeConnection(): void {\n if (this.ws?.readyState === WebSocket.OPEN) {\n this.ws.close();\n }\n }\n\n private close(): void {\n if (this.isClosed()) return;\n\n this.connectionState = 'closed';\n this.reconnection.cancel();\n this.closeWebSocket();\n this.rejectAllPendingOperations();\n }\n\n private closeWebSocket(): void {\n if (!this.ws) return;\n\n try {\n this.ws.close();\n } catch {\n // Suppress close errors\n }\n\n this.ws = null;\n }\n\n private finalizeStream(): void {\n this.connectionState = 'closed';\n this.resolveAllPendingResponses();\n }\n\n private resolveAllPendingResponses(): void {\n while (this.responseResolvers.length > 0) {\n const resolver = this.responseResolvers.shift()!;\n resolver.resolve({ value: undefined as any, done: true });\n }\n }\n\n private parseMessage(data: unknown): ServerMessageType {\n if (data instanceof ArrayBuffer) {\n return ServerMessageType.fromBinary(new Uint8Array(data));\n }\n\n throw new Error(`Unsupported message format: ${typeof data}`);\n }\n\n // === Send Queue Management ===\n\n private validateSendState(): void {\n if (this.sendCompleted) {\n throw new Error('Cannot send: stream already completed');\n }\n\n if (this.abortSignal.aborted) {\n throw new Error('Cannot send: stream aborted');\n }\n }\n\n private enqueueSend(message: ClientMessageType): Promise<void> {\n return new Promise<void>((resolve, reject) => {\n this.sendQueue.push({ message, resolve, reject });\n void this.processSendQueue();\n });\n }\n\n private processSendQueue(): void {\n if (!this.canSendMessages()) return;\n\n while (this.sendQueue.length > 0) {\n const queued = this.sendQueue.shift()!;\n this.sendQueuedMessage(queued);\n }\n }\n\n private canSendMessages(): boolean {\n return this.connectionState === 'connected' && this.ws !== null;\n }\n\n private sendQueuedMessage(queued: QueuedMessage): void {\n try {\n const ws = this.ws;\n if (!ws) {\n throw new Error('WebSocket is not connected');\n }\n\n // Check if WebSocket is in a valid state for sending\n if (ws.readyState !== WebSocket.OPEN) {\n throw new Error(`WebSocket is not open (readyState: ${ws.readyState})`);\n }\n\n const binary = ClientMessageType.toBinary(queued.message);\n ws.send(binary);\n queued.resolve();\n } catch (error) {\n queued.reject(this.toError(error));\n }\n }\n\n private async drainSendQueue(): Promise<void> {\n const POLL_INTERVAL_MS = 10;\n\n while (this.sendQueue.length > 0) {\n await this.waitForCondition(\n () => this.sendQueue.length === 0,\n POLL_INTERVAL_MS,\n );\n }\n }\n\n private waitForCondition(\n condition: () => boolean,\n intervalMs: number,\n ): Promise<void> {\n return new Promise<void>((resolve, reject) => {\n if (this.abortSignal.aborted) {\n return reject(this.toError(this.abortSignal.reason) ?? new Error('Stream aborted'));\n }\n\n let timeoutId: ReturnType<typeof setTimeout>;\n const onAbort = () => {\n clearTimeout(timeoutId);\n reject(this.toError(this.abortSignal.reason) ?? new Error('Stream aborted'));\n };\n\n this.abortSignal.addEventListener('abort', onAbort, { once: true });\n\n const check = () => {\n if (condition() || this.isStreamEnded()) {\n this.abortSignal.removeEventListener('abort', onAbort);\n resolve();\n } else {\n timeoutId = setTimeout(check, intervalMs);\n }\n };\n\n check();\n });\n }\n\n // === Response Delivery ===\n\n private deliverResponse(message: ServerMessageType): void {\n if (this.responseResolvers.length > 0) {\n const resolver = this.responseResolvers.shift()!;\n resolver.resolve({ value: message, done: false });\n } else {\n this.responseQueue.push(message);\n }\n }\n\n private async *createResponseIterator(): AsyncIterator<ServerMessageType> {\n while (true) {\n const result = await this.nextResponse();\n\n if (result.done) break;\n\n yield result.value;\n }\n }\n\n private nextResponse(): Promise<IteratorResult<ServerMessageType>> {\n return new Promise<IteratorResult<ServerMessageType>>((resolve, reject) => {\n // Fast path: message already available\n if (this.responseQueue.length > 0) {\n const message = this.responseQueue.shift()!;\n resolve({ value: message, done: false });\n return;\n }\n\n // Stream ended\n if (this.isStreamEnded()) {\n if (this.connectionError) {\n reject(this.connectionError);\n } else {\n resolve({ value: undefined as any, done: true });\n }\n return;\n }\n\n // Wait for next message\n this.responseResolvers.push({ resolve, reject });\n });\n }\n\n // === Error Handling ===\n private handleError(error: Error): void {\n if (this.isClosed()) return;\n\n this.connectionState = 'closed';\n this.connectionError = error;\n this.reconnection.cancel();\n this.closeWebSocket();\n this.rejectAllPendingOperations(error);\n }\n\n private rejectAllPendingOperations(error?: Error): void {\n const err = error ?? this.createStreamClosedError();\n this.rejectAllSendOperations(err);\n this.rejectAllResponseResolvers(err);\n }\n\n private rejectAllSendOperations(error: Error): void {\n while (this.sendQueue.length > 0) {\n const queued = this.sendQueue.shift()!;\n queued.reject(error);\n }\n }\n\n private rejectAllResponseResolvers(error: Error): void {\n while (this.responseResolvers.length > 0) {\n const resolver = this.responseResolvers.shift()!;\n resolver.reject(error);\n }\n }\n\n private createStreamClosedError(): Error {\n if (this.abortSignal.aborted) {\n const reason = this.abortSignal.reason;\n if (reason instanceof Error) {\n return reason;\n }\n return new Error('Stream aborted', { cause: reason });\n }\n return new Error('Stream closed');\n }\n // === State Checks ===\n\n private isConnectingOrConnected(): boolean {\n return this.connectionState === 'connecting'\n || this.connectionState === 'connected';\n }\n\n private isClosed(): boolean {\n return this.connectionState === 'closed';\n }\n\n private isStreamEnded(): boolean {\n return this.isClosed() || this.abortSignal.aborted;\n }\n\n private toError(error: unknown): Error {\n return error instanceof Error ? error : new Error(String(error));\n }\n}\n"],"names":["RetryStrategy","WebSocket","ServerMessageType","ClientMessageType"],"mappings":";;;;;;;AAuBA;;;AAGG;MACU,mBAAmB,CAAA;;IAEtB,EAAE,GAAqB,IAAI;IAC3B,eAAe,GAAoB,cAAc;AACxC,IAAA,GAAG;AACH,IAAA,QAAQ;AACR,IAAA,WAAW;AACX,IAAA,YAAY;;AAGZ,IAAA,SAAS,GAAG,IAAI,MAAM,EAAiB;IAChD,aAAa,GAAG,KAAK;;AAGZ,IAAA,aAAa,GAAG,IAAI,MAAM,EAAqB;IACxD,iBAAiB,GAAuB,EAAE;;IAG1C,eAAe,GAAiB,IAAI;;AAI5B,IAAA,QAAQ,GAAG;AACzB,QAAA,IAAI,EAAE,OAAO,OAA0B,KAAmB;YACxD,IAAI,CAAC,iBAAiB,EAAE;AACxB,YAAA,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QAClC,CAAC;QAED,QAAQ,EAAE,YAA0B;YAClC,IAAI,IAAI,CAAC,aAAa;gBAAE;AAExB,YAAA,IAAI,CAAC,aAAa,GAAG,IAAI;AACzB,YAAA,MAAM,IAAI,CAAC,cAAc,EAAE;YAC3B,IAAI,CAAC,eAAe,EAAE;QACxB,CAAC;KACF;AAEe,IAAA,SAAS,GAAqC;QAC5D,CAAC,MAAM,CAAC,aAAa,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE;KAC5D;AAED,IAAA,WAAA,CACE,GAAW,EACX,WAAwB,EACxB,QAAiB,EACjB,cAAoC,EAAE,EAAA;AAEtC,QAAA,IAAI,CAAC,GAAG,GAAG,GAAG;AACd,QAAA,IAAI,CAAC,QAAQ,GAAG,QAAQ;AACxB,QAAA,IAAI,CAAC,WAAW,GAAG,WAAW;AAE9B,QAAA,IAAI,CAAC,YAAY,GAAG,IAAIA,4BAAa,CAAC,WAAW,EAAE;YACjD,OAAO,EAAE,MAAK,EAAG,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACvC,oBAAoB,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC;AACzD,SAAA,CAAC;AAEF,QAAA,IAAI,WAAW,CAAC,OAAO,EAAE;AACvB,YAAA,IAAI,CAAC,eAAe,GAAG,QAAQ;YAC/B;QACF;QAEA,IAAI,CAAC,wBAAwB,EAAE;AAC/B,QAAA,KAAK,IAAI,CAAC,OAAO,EAAE;IACrB;;IAIQ,OAAO,GAAA;QACb,IAAI,IAAI,CAAC,uBAAuB,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO;YAAE;AAEhE,QAAA,IAAI,CAAC,eAAe,GAAG,YAAY;AACnC,QAAA,IAAI,CAAC,eAAe,GAAG,IAAI;AAE3B,QAAA,IAAI;AACF,YAAA,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,eAAe,EAAE;YAChC,IAAI,CAAC,uBAAuB,EAAE;QAChC;QAAE,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;AAC1C,YAAA,IAAI,CAAC,eAAe,GAAG,cAAc;AACrC,YAAA,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;QAC9B;IACF;IAEQ,eAAe,GAAA;AACrB,QAAA,MAAM,OAAO,GAAG,IAAI,CAAC;AACnB,cAAE,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,CAAA,OAAA,EAAU,IAAI,CAAC,QAAQ,CAAA,CAAE,EAAE;cACvD,SAAS;QAEb,MAAM,EAAE,GAAG,IAAKC,gBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC;QACpD,IAAI,EAAE,EAAE;AACN,YAAA,EAAE,CAAC,UAAU,GAAG,aAAa;QAC/B;AACA,QAAA,OAAO,EAAE;IACX;IAEQ,uBAAuB,GAAA;QAC7B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE;AAEd,QAAA,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QACrD,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AAC1E,QAAA,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AACjE,QAAA,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;IACzD;IAEQ,wBAAwB,GAAA;AAC9B,QAAA,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IAChE;IAEQ,MAAM,GAAA;AACZ,QAAA,IAAI,CAAC,eAAe,GAAG,WAAW;AAClC,QAAA,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE;AACzB,QAAA,KAAK,IAAI,CAAC,gBAAgB,EAAE;IAC9B;IAEQ,OAAO,GAAA;AACb,QAAA,IAAI,CAAC,EAAE,GAAG,IAAI;QAEd,IAAI,IAAI,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO;YAAE;AAEjD,QAAA,IAAI,IAAI,CAAC,aAAa,EAAE;YACtB,IAAI,CAAC,cAAc,EAAE;QACvB;aAAO;AACL,YAAA,IAAI,CAAC,eAAe,GAAG,cAAc;AACrC,YAAA,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;QAC9B;IACF;AAEQ,IAAA,OAAO,CAAC,KAAc,EAAA;QAC5B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACvC;AAEQ,IAAA,SAAS,CAAC,IAAa,EAAA;AAC7B,QAAA,IAAI;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;AACvC,YAAA,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;QAC/B;QAAE,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACvC;IACF;IAEQ,eAAe,GAAA;QACrB,IAAI,IAAI,CAAC,EAAE,EAAE,UAAU,KAAKA,gBAAS,CAAC,IAAI,EAAE;AAC1C,YAAA,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE;QACjB;IACF;IAEQ,KAAK,GAAA;QACX,IAAI,IAAI,CAAC,QAAQ,EAAE;YAAE;AAErB,QAAA,IAAI,CAAC,eAAe,GAAG,QAAQ;AAC/B,QAAA,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QAC1B,IAAI,CAAC,cAAc,EAAE;QACrB,IAAI,CAAC,0BAA0B,EAAE;IACnC;IAEQ,cAAc,GAAA;QACpB,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE;AAEd,QAAA,IAAI;AACF,YAAA,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE;QACjB;AAAE,QAAA,MAAM;;QAER;AAEA,QAAA,IAAI,CAAC,EAAE,GAAG,IAAI;IAChB;IAEQ,cAAc,GAAA;AACpB,QAAA,IAAI,CAAC,eAAe,GAAG,QAAQ;QAC/B,IAAI,CAAC,0BAA0B,EAAE;IACnC;IAEQ,0BAA0B,GAAA;QAChC,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAG;AAChD,YAAA,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC3D;IACF;AAEQ,IAAA,YAAY,CAAC,IAAa,EAAA;AAChC,QAAA,IAAI,IAAI,YAAY,WAAW,EAAE;YAC/B,OAAOC,uBAAiB,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;QAC3D;QAEA,MAAM,IAAI,KAAK,CAAC,CAAA,4BAAA,EAA+B,OAAO,IAAI,CAAA,CAAE,CAAC;IAC/D;;IAIQ,iBAAiB,GAAA;AACvB,QAAA,IAAI,IAAI,CAAC,aAAa,EAAE;AACtB,YAAA,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC;QAC1D;AAEA,QAAA,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;AAC5B,YAAA,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC;QAChD;IACF;AAEQ,IAAA,WAAW,CAAC,OAA0B,EAAA;QAC5C,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,KAAI;AAC3C,YAAA,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AACjD,YAAA,KAAK,IAAI,CAAC,gBAAgB,EAAE;AAC9B,QAAA,CAAC,CAAC;IACJ;IAEQ,gBAAgB,GAAA;AACtB,QAAA,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;YAAE;QAE7B,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAG;AACtC,YAAA,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC;QAChC;IACF;IAEQ,eAAe,GAAA;QACrB,OAAO,IAAI,CAAC,eAAe,KAAK,WAAW,IAAI,IAAI,CAAC,EAAE,KAAK,IAAI;IACjE;AAEQ,IAAA,iBAAiB,CAAC,MAAqB,EAAA;AAC7C,QAAA,IAAI;AACF,YAAA,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE;YAClB,IAAI,CAAC,EAAE,EAAE;AACP,gBAAA,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC;YAC/C;;YAGA,IAAI,EAAE,CAAC,UAAU,KAAKD,gBAAS,CAAC,IAAI,EAAE;gBACpC,MAAM,IAAI,KAAK,CAAC,CAAA,mCAAA,EAAsC,EAAE,CAAC,UAAU,CAAA,CAAA,CAAG,CAAC;YACzE;YAEA,MAAM,MAAM,GAAGE,uBAAiB,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC;AACzD,YAAA,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACf,MAAM,CAAC,OAAO,EAAE;QAClB;QAAE,OAAO,KAAK,EAAE;YACd,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACpC;IACF;AAEQ,IAAA,MAAM,cAAc,GAAA;QAC1B,MAAM,gBAAgB,GAAG,EAAE;QAE3B,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AAChC,YAAA,MAAM,IAAI,CAAC,gBAAgB,CACzB,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EACjC,gBAAgB,CACjB;QACH;IACF;IAEQ,gBAAgB,CACtB,SAAwB,EACxB,UAAkB,EAAA;QAElB,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,KAAI;AAC3C,YAAA,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;AAC5B,gBAAA,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACrF;AAEA,YAAA,IAAI,SAAwC;YAC5C,MAAM,OAAO,GAAG,MAAK;gBACnB,YAAY,CAAC,SAAS,CAAC;AACvB,gBAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;AAC9E,YAAA,CAAC;AAED,YAAA,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAEnE,MAAM,KAAK,GAAG,MAAK;gBACjB,IAAI,SAAS,EAAE,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;oBACvC,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC;AACtD,oBAAA,OAAO,EAAE;gBACX;qBAAO;AACL,oBAAA,SAAS,GAAG,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC;gBAC3C;AACF,YAAA,CAAC;AAED,YAAA,KAAK,EAAE;AACT,QAAA,CAAC,CAAC;IACJ;;AAIQ,IAAA,eAAe,CAAC,OAA0B,EAAA;QAChD,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAG;AAChD,YAAA,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACnD;aAAO;AACL,YAAA,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC;QAClC;IACF;IAEQ,OAAO,sBAAsB,GAAA;QACnC,OAAO,IAAI,EAAE;AACX,YAAA,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE;YAExC,IAAI,MAAM,CAAC,IAAI;gBAAE;YAEjB,MAAM,MAAM,CAAC,KAAK;QACpB;IACF;IAEQ,YAAY,GAAA;QAClB,OAAO,IAAI,OAAO,CAAoC,CAAC,OAAO,EAAE,MAAM,KAAI;;YAExE,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;gBACjC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAG;gBAC3C,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACxC;YACF;;AAGA,YAAA,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;AACxB,gBAAA,IAAI,IAAI,CAAC,eAAe,EAAE;AACxB,oBAAA,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;gBAC9B;qBAAO;oBACL,OAAO,CAAC,EAAE,KAAK,EAAE,SAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAClD;gBACA;YACF;;YAGA,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAClD,QAAA,CAAC,CAAC;IACJ;;AAGQ,IAAA,WAAW,CAAC,KAAY,EAAA;QAC9B,IAAI,IAAI,CAAC,QAAQ,EAAE;YAAE;AAErB,QAAA,IAAI,CAAC,eAAe,GAAG,QAAQ;AAC/B,QAAA,IAAI,CAAC,eAAe,GAAG,KAAK;AAC5B,QAAA,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;QAC1B,IAAI,CAAC,cAAc,EAAE;AACrB,QAAA,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC;IACxC;AAEQ,IAAA,0BAA0B,CAAC,KAAa,EAAA;QAC9C,MAAM,GAAG,GAAG,KAAK,IAAI,IAAI,CAAC,uBAAuB,EAAE;AACnD,QAAA,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC;AACjC,QAAA,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC;IACtC;AAEQ,IAAA,uBAAuB,CAAC,KAAY,EAAA;QAC1C,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAG;AACtC,YAAA,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;QACtB;IACF;AAEQ,IAAA,0BAA0B,CAAC,KAAY,EAAA;QAC7C,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAG;AAChD,YAAA,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC;QACxB;IACF;IAEQ,uBAAuB,GAAA;AAC7B,QAAA,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE;AAC5B,YAAA,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM;AACtC,YAAA,IAAI,MAAM,YAAY,KAAK,EAAE;AAC3B,gBAAA,OAAO,MAAM;YACf;YACA,OAAO,IAAI,KAAK,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QACvD;AACA,QAAA,OAAO,IAAI,KAAK,CAAC,eAAe,CAAC;IACnC;;IAGQ,uBAAuB,GAAA;AAC7B,QAAA,OAAO,IAAI,CAAC,eAAe,KAAK;AAC3B,eAAA,IAAI,CAAC,eAAe,KAAK,WAAW;IAC3C;IAEQ,QAAQ,GAAA;AACd,QAAA,OAAO,IAAI,CAAC,eAAe,KAAK,QAAQ;IAC1C;IAEQ,aAAa,GAAA;QACnB,OAAO,IAAI,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO;IACpD;AAEQ,IAAA,OAAO,CAAC,KAAc,EAAA;AAC5B,QAAA,OAAO,KAAK,YAAY,KAAK,GAAG,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAClE;AACD;;;;"}