feat: add setrange command (#1134)

compat.md

@@ -136,7 +136,7 @@
 | [setbit]               | :white_check_mark: | :white_check_mark: |
 | [setex]                | :white_check_mark: | :white_check_mark: |
 | [setnx]                | :white_check_mark: | :white_check_mark: |
-| [setrange]             | :white_check_mark: |        :x:         |
+| [setrange]             | :white_check_mark: | :white_check_mark: |
 | [shutdown]             | :white_check_mark: |        :x:         |
 | [sinter]               | :white_check_mark: | :white_check_mark: |
 | [sinterstore]          | :white_check_mark: | :white_check_mark: |

src/commands/index.js

@@ -92,6 +92,7 @@ export * from './set'
 export * from './setbit'
 export * from './setex'
 export * from './setnx'
+export * from './setrange'
 export * from './sinter'
 export * from './sinterstore'
 export * from './sismember'

src/commands/setrange.js

@@ -0,0 +1,26 @@
+export function setrange(key, offset, value) {
+  const oldVal = this.data.get(key) || ''
+  const start = parseInt(offset, 10)
+  const end = parseInt(start + value.length, 10)
+
+  let newVal = ''
+  if (start > 0) {
+    newVal = oldVal.slice(0, start)
+  }
+
+  newVal += value
+
+  if (end < oldVal.length) {
+    newVal += oldVal.slice(end, oldVal.length + 1)
+  }
+
+  if (newVal.length < end) {
+    newVal = newVal.padStart(end, String.fromCharCode(0))
+  }
+
+  this.data.set(key, newVal)
+
+  return newVal.length
+}
+
+export const setrangeBuffer = setrange

test/integration/commands/setbit.js

@@ -6,11 +6,7 @@ import { runTwinSuite } from '../../../test-utils'
 runTwinSuite('setbit', command => {
   describe(command, () => {
     it('should return old bit value of key', async () => {
-      const redis = new Redis({
-        data: {
-          foo: '@',
-        },
-      })
+      const redis = new Redis()
       await redis.set('foo', '@')
 
       return redis[command]('foo', 1, 0)
@@ -38,11 +34,7 @@ runTwinSuite('setbit', command => {
     })
 
     it('should override bit value of key', async () => {
-      const redis = new Redis({
-        data: {
-          foo: 'bar',
-        },
-      })
+      const redis = new Redis()
       await redis.set('foo', 'bar')
 
       return redis[command]('foo', 3, 1)

test/integration/commands/setrange.js

@@ -0,0 +1,24 @@
+import Redis from 'ioredis'
+
+// eslint-disable-next-line import/no-relative-parent-imports
+import { runTwinSuite } from '../../../test-utils'
+
+runTwinSuite('setrange', command => {
+  describe(command, () => {
+    const redis = new Redis()
+
+    afterAll(() => {
+      redis.disconnect()
+    })
+
+    it('should return "Hello Redis"', async () => {
+      await redis.set('key1', 'Hello World')
+      expect(await redis[command]('key1', 6, 'Redis')).toBe(11)
+    })
+
+    it('should return zero padding', async () => {
+      expect(await redis[command]('key2', 6, 'Redis')).toBe(11)
+      expect(await redis.get('key2')).toBe('\x00\x00\x00\x00\x00\x00Redis')
+    })
+  })
+})