feat: add json mset command (#1197)

* feat: add json mset command

* fix: remove unused import

Author: ogzhanolguncu

biome.json

@@ -10,9 +10,6 @@
       "performance": {
         "noDelete": "off"
       },
-      "nursery": {
-        "useAwait": "error"
-      },
       "complexity": {
         "noBannedTypes": "off"
       },

pkg/commands/json_mset.test.ts

@@ -0,0 +1,58 @@
+import { afterAll, expect, test } from "bun:test";
+import { keygen, newHttpClient } from "../test-utils";
+
+import { JsonGetCommand } from "./json_get";
+import { JsonMSetCommand } from "./json_mset";
+
+const client = newHttpClient();
+
+const { newKey, cleanup } = keygen();
+afterAll(cleanup);
+
+test("add a new value", async () => {
+  const key1 = newKey();
+  const key2 = newKey();
+  const key3 = newKey();
+  const res1 = await new JsonMSetCommand([
+    { key: key1, path: "$", value: { key: "value" } },
+    { key: key2, path: "$", value: { key: "value" } },
+    { key: key3, path: "$", value: { key: "value" } },
+  ]).exec(client);
+  expect(res1).toEqual("OK");
+});
+
+test("replace an existing value", async () => {
+  const key = newKey();
+  const res1 = await new JsonMSetCommand([
+    { key, path: "$", value: { a: 2 } },
+  ]).exec(client);
+  expect(res1).toEqual("OK");
+  const res2 = await new JsonMSetCommand([{ key, path: "$.a", value: 3 }]).exec(
+    client
+  );
+  expect(res2).toEqual("OK");
+  const res3 = await new JsonGetCommand([key, "$"]).exec(client);
+  expect(res3).toEqual([{ a: 3 }]);
+});
+
+test("update multi-paths", async () => {
+  const key = newKey();
+  const data = {
+    f1: { a: 1 },
+    f2: { a: 2 },
+  };
+  const res1 = await new JsonMSetCommand([
+    { key, path: "$", value: data },
+  ]).exec(client);
+  expect(res1).toEqual("OK");
+  const res2 = await new JsonMSetCommand([
+    { key, path: "$..a", value: 3 },
+  ]).exec(client);
+  expect(res2).toEqual("OK");
+  const res3 = await new JsonGetCommand<any[]>([key, "$"]).exec(client);
+
+  expect(res3).not.toBeNull();
+  expect(res3!.length).toEqual(1);
+  expect(res3![0]?.f1?.a).toEqual(3);
+  expect(res3![0]?.f2?.a).toEqual(3);
+});

pkg/commands/json_mset.ts

@@ -0,0 +1,25 @@
+import { Command, CommandOptions } from "./command";
+
+/**
+ * @see https://redis.io/commands/json.mset
+ */
+export class JsonMSetCommand<
+  TData extends
+    | number
+    | string
+    | boolean
+    | Record<string, unknown>
+    | (number | string | boolean | Record<string, unknown>)[]
+> extends Command<"OK" | null, "OK" | null> {
+  constructor(
+    cmd: { key: string; path: string; value: TData }[],
+    opts?: CommandOptions<"OK" | null, "OK" | null>
+  ) {
+    const command: unknown[] = ["JSON.MSET"];
+
+    for (const c of cmd) {
+      command.push(c.key, c.path, c.value);
+    }
+    super(command, opts);
+  }
+}

pkg/commands/mod.ts

@@ -58,6 +58,7 @@ export * from "./json_del";
 export * from "./json_forget";
 export * from "./json_get";
 export * from "./json_mget";
+export * from "./json_mset";
 export * from "./json_numincrby";
 export * from "./json_nummultby";
 export * from "./json_objkeys";