[FEAT] [JS] [KV] exposed size of the bucket via KvStatus.size

FIX #377
nats-io · Sep 29, 2022 · 9636313 · 9636313
1 parent 8426af1
commit 9636313
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/nats-base-client/kv.ts b/nats-base-client/kv.ts
@@ -849,4 +849,8 @@ export class KvStatusImpl implements KvStatus {
   get streamInfo(): StreamInfo {
     return this.si;
   }
+
+  get size(): number {
+    return this.si.state.bytes;
+  }
 }
diff --git a/nats-base-client/types.ts b/nats-base-client/types.ts
@@ -2597,6 +2597,11 @@ export interface KvStatus extends KvLimits {
    * The StreamInfo backing up the KV
    */
   streamInfo: StreamInfo;
+
+  /**
+   * Size of the bucket in bytes
+   */
+  size: number;
 }
 
 export interface KvOptions extends KvLimits {

diff --git a/tests/kv_test.ts b/tests/kv_test.ts
@@ -1510,9 +1510,29 @@ Deno.test("kv - ttl is in nanos", async () => {
   const b = await js.views.kv("a", { ttl: 1000 });
   const status = await b.status();
   assertEquals(status.ttl, 1000);
+  assertEquals(status.size, 0);
 
   const jsm = await nc.jetstreamManager();
   const si = await jsm.streams.info("KV_a");
   assertEquals(si.config.max_age, nanos(1000));
   await cleanup(ns, nc);
 });
+
+Deno.test("kv - size", async () => {
+  const { ns, nc } = await setup(
+    jetstreamServerConf({}, true),
+  );
+  const js = nc.jetstream();
+
+  const b = await js.views.kv("a", { ttl: 1000 });
+  let status = await b.status();
+  assertEquals(status.size, 0);
+  assertEquals(status.size, status.streamInfo.state.bytes);
+
+  const sc = StringCodec();
+  await b.put("a", sc.encode("hello"));
+  status = await b.status();
+  assert(status.size > 0);
+  assertEquals(status.size, status.streamInfo.state.bytes);
+  await cleanup(ns, nc);
+});