fix: map batch length to length in pipeline.js

stipsan · Sep 18, 2023 · 335ef03 · 335ef03
1 parent 1259f10
commit 335ef03
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/pipeline.js b/src/pipeline.js
@@ -11,6 +11,12 @@ class Pipeline {
     this._transactions = 0
 
     this.copyCommands()
+
+    Object.defineProperty(this, 'length', {
+      get () {
+        return this.batch.length;
+      },
+    });
   }
 
   copyCommands() {

diff --git a/test/integration/multi.js b/test/integration/multi.js
@@ -15,6 +15,17 @@ describe('multi', () => {
     expect(redis.batch.batch[1]).toEqual(expect.any(Function))
   })
 
+  it('should map batch length to length in pipeline',() => {
+    const redis = new Redis()
+    const pipeline = redis.pipeline()
+
+    pipeline
+      .incr('user_next')
+      .incr('post_next');
+
+    expect(pipeline.length).toBe(2);
+  })
+
   it('allows for pipelining methods', () => {
     const redis = new Redis()