fix(getRange): cannot support multiple pk (#8)

Signed-off-by: Kevin Cui <bh@bugs.cc>
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

Author: BlackHole1

README.md

@@ -193,14 +193,20 @@ const result = await batchWriteRow.do({
 Query rows within a range of primary keys.
 
 ```typescript
-import { GetRange, Direction } from "alicloud-tablestore";
+import { GetRange, Direction, INF_MIN, INF_MAX } from "alicloud-tablestore";
 
 const getRange = new GetRange(client);
 const result = await getRange.do({
   tableName: "users",
   direction: Direction.FORWARD, // or Direction.BACKWARD
-  inclusiveStartPrimaryKey: createPrimaryKey("id", "100"),
-  exclusiveEndPrimaryKey: createPrimaryKey("id", "200"),
+  inclusiveStartPrimaryKey: [
+    createPrimaryKey("id", "100"),
+    createPrimaryKey("age", INF_MIN),
+  ],
+  exclusiveEndPrimaryKey: [
+    createPrimaryKey("id", "200"),
+    createPrimaryKey("age", INF_MAX),
+  ],
   limit: 100,
   maxVersions: 1,
 });

src/const.ts

@@ -22,3 +22,6 @@ export const OTS_API_NAME = {
     BatchGetRow: "BatchGetRow",
     BatchWriteRow: "BatchWriteRow",
 } as const;
+
+export const INF_MIN = {};
+export const INF_MAX = {};

src/index.ts

@@ -3,6 +3,8 @@ export {
 } from "./client";
 
 export {
+    INF_MAX,
+    INF_MIN,
     OTS_API_NAME,
 } from "./const";
 

src/operator/get-range.ts

@@ -18,8 +18,8 @@ export interface GetRangeData {
     timeRange?: TimeRange;
     maxVersions?: number;
     limit?: number;
-    inclusiveStartPrimaryKey: PlainBufferCell;
-    exclusiveEndPrimaryKey: PlainBufferCell;
+    inclusiveStartPrimaryKey: PlainBufferCell[];
+    exclusiveEndPrimaryKey: PlainBufferCell[];
     filter?: Filter;
     startColumn?: string;
     endColumn?: string;
@@ -40,11 +40,11 @@ export class GetRange {
             tableName: options.tableName,
             direction: options.direction,
             inclusiveStartPrimaryKey: Buffer.from(encodePlainBuffer([{
-                primaryKey: [fixPlainBufferCellType(options.inclusiveStartPrimaryKey)],
+                primaryKey: options.inclusiveStartPrimaryKey.map(fixPlainBufferCellType),
                 attributes: [],
             }])),
             exclusiveEndPrimaryKey: Buffer.from(encodePlainBuffer([{
-                primaryKey: [fixPlainBufferCellType(options.exclusiveEndPrimaryKey)],
+                primaryKey: options.exclusiveEndPrimaryKey.map(fixPlainBufferCellType),
                 attributes: [],
             }])),
         };

src/utils.ts

@@ -1,11 +1,19 @@
-import type { PlainBufferCell, VariantType } from "./plainbuffer";
-import { inferVariantType } from "./plainbuffer";
+import type { PlainBufferCell } from "./plainbuffer";
+import { INF_MAX, INF_MIN } from "./const";
+import { inferVariantType, VariantType } from "./plainbuffer";
 
 export function fixPlainBufferCellType(cell: PlainBufferCell): PlainBufferCell {
     if (cell.type) {
         return cell;
     }
 
+    if (cell.value === INF_MIN) {
+        return { ...cell, type: VariantType.INF_MIN };
+    }
+    if (cell.value === INF_MAX) {
+        return { ...cell, type: VariantType.INF_MAX };
+    }
+
     cell.type = inferVariantType(cell.value).type;
     return cell;
 }