Merge pull request #13 from milvus-io/fix/collection-statistic

Fix/collection statistic

README.md

@@ -37,15 +37,15 @@ Make sure that:
 1. First of all, we need to import @zilliz/milvus-sdk-node.
 
 ```javascript
-import { MilvusNode } from "@zilliz/milvus2-sdk-node";
+import { MilvusClient } from "@zilliz/milvus2-sdk-node";
 ```
 
 2. Then, we can make connection with Milvus server.
    By default Milvus runs on localhost in port 19530, so you can use default value to connect to Milvus.
 
 ```javascript
 const IP = "127.0.0.1:19530";
-const milvusClient = new MilvusNode(IP);
+const milvusClient = new MilvusClient(IP);
 ```
 
 3. After connecting, we can communicate with Milvus in the following ways.

milvus/index.ts

@@ -50,6 +50,7 @@ import { DataType, DataTypeMap, DslType } from "./types/Common";
 import { FlushReq, InsertReq } from "./types/Insert";
 import { parseFloatArrayToBytes } from "./utils/Blob";
 import { findKeyValue } from "./utils";
+import { formatKeyValueData } from "./utils/Format";
 
 const protoPath = path.resolve(__dirname, "../grpc-proto/milvus.proto");
 const schemaPath = path.resolve(__dirname, "../grpc-proto/schema.proto");
@@ -225,6 +226,9 @@ export class MilvusClient {
       "GetCollectionStatistics",
       data
     );
+
+    promise.data = formatKeyValueData(promise.stats, ["row_count"]);
+
     return promise;
   }
 
@@ -268,6 +272,7 @@ export class MilvusClient {
       "GetPartitionStatistics",
       data
     );
+    promise.data = formatKeyValueData(promise.stats, ["row_count"]);
     return promise;
   }
 

milvus/types/Response.ts

@@ -76,6 +76,7 @@ export interface DescribeCollectionResponse {
 export interface StatisticsResponse {
   status: ResStatus;
   stats: KeyValuePair[];
+  data: { [x: string]: any };
 }
 
 export interface ShowPartitionsResponse {

milvus/utils/Format.ts

@@ -0,0 +1,18 @@
+import { findKeyValue } from ".";
+import { KeyValuePair } from "../types/Common";
+
+/**
+ *  parse [{key:"row_count",value:4}] to {row_count:4}
+ * @param data key value pair array
+ * @param keys all keys in data
+ * @returns {key:value}
+ */
+export const formatKeyValueData = (data: KeyValuePair[], keys: string[]) => {
+  const result: { [x: string]: any } = {};
+  keys.forEach((k) => {
+    const value = findKeyValue(data, k);
+    result[k] = value;
+  });
+
+  return result;
+};

test/Collection.spec.ts

@@ -142,6 +142,7 @@ describe("Collection Api", () => {
     });
     expect(res.status.error_code).toEqual(ErrorCode.SUCCESS);
     expect(res.stats[0].value).toEqual("0");
+    expect(res.data.row_count).toEqual("0");
   });
 
   it("Describe Collection info", async () => {