fix double and float data generator

Signed-off-by: ryjiang <[email protected]>
zilliztech · Nov 9, 2023 · e8afb56 · e8afb56
1 parent 940a7e1
commit e8afb56
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server/src/utils/Helper.ts b/server/src/utils/Helper.ts
@@ -8,6 +8,7 @@ export const findKeyValue = (obj: KeyValuePair[], key: string) =>
 
 export const makeDynamicBool = () => Math.random() > 0.5;
 export const makeRandomInt = () => Math.floor(Math.random() * 127);
+export const makeFloat = () => Math.random();
 
 export const genDataByType = ({ data_type, type_params }: FieldSchema) => {
   switch (data_type) {
@@ -21,6 +22,10 @@ export const genDataByType = ({ data_type, type_params }: FieldSchema) => {
       return Math.floor(Math.random() * 214748364);
     case 'Int64':
       return Math.floor(Math.random() * 214748364);
+    case 'Float':
+      return makeFloat();
+    case 'Double':
+      return makeFloat();
     case 'FloatVector':
       return Array.from({ length: (type_params as any)[0].value }).map(() =>
         Math.random()