Attempt to pass request object to createStoreWritableStream

HeartLab · Feb 22, 2024 · 4192ec9 · 4192ec9
1 parent e092f83
commit 4192ec9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -1427,7 +1427,10 @@ declare class Scp extends Network {
    * The default implementation creates a memory Writable stream that for, big instances,
    * could cause out of memory situations.
    */
-  createStoreWritableStream(acceptedPresentationContext: PresentationContext): Writable;
+  createStoreWritableStream(
+    acceptedPresentationContext: PresentationContext,
+    request: CStoreRequest
+  ): Writable;
 
   /**
    * Allows the caller to create a Dataset from the Writable stream used to

diff --git a/src/Network.js b/src/Network.js
@@ -534,7 +534,10 @@ class Network extends AsyncEventEmitter {
           if (this.dimse.getCommandFieldType() === CommandFieldType.CStoreRequest) {
             if (!this.dimseStoreStream) {
               this.dimseStream = undefined;
-              this.dimseStoreStream = this.createStoreWritableStream(presentationContext);
+              this.dimseStoreStream = this.createStoreWritableStream(
+                presentationContext,
+                this.dimse
+              );
             }
           } else {
             if (!this.dimseStream) {

diff --git a/src/Server.js b/src/Server.js
@@ -106,10 +106,11 @@ class Scp extends Network {
    * could cause out of memory situations.
    * @method
    * @param {PresentationContext} acceptedPresentationContext - The accepted presentation context.
+   * @param {CStoreRequest} request - C-STORE request.
    * @returns {Writable} The created store writable stream.
    */
   // eslint-disable-next-line no-unused-vars
-  createStoreWritableStream(acceptedPresentationContext) {
+  createStoreWritableStream(acceptedPresentationContext, request) {
     return super.createStoreWritableStream();
   }