fix: map batch length to length in pipeline (#1201)

Fixes #1046
stipsan · Jan 29, 2023 · a4e8a4a · a4e8a4a
1 parent 9213bc7
commit a4e8a4a
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
@@ -10,6 +10,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()