[Go & Cpp] Fix vector type check

Signed-off-by: sunby <[email protected]>
milvus-io · Oct 8, 2023 · 1df493f · 1df493f
1 parent c58990f
commit 1df493f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/cpp/src/storage/options.cpp b/cpp/src/storage/options.cpp
@@ -1,4 +1,5 @@
 #include "storage/options.h"
+#include <arrow/type_fwd.h>
 #include "arrow/type.h"
 #include "common/status.h"
 
@@ -30,8 +31,9 @@ Status SchemaOptions::Validate(const arrow::Schema* schema) {
     auto vector_field = schema->GetFieldByName(vector_column);
     if (!vector_field) {
       return Status::InvalidArgument("vector column is not exist");
-    } else if (vector_field->type()->id() != arrow::Type::FIXED_SIZE_BINARY) {
-      return Status::InvalidArgument("vector column is not fixed size binary");
+    } else if (vector_field->type()->id() != arrow::Type::FIXED_SIZE_BINARY &&
+               vector_field->type()->id() != arrow::Type::FIXED_SIZE_LIST) {
+      return Status::InvalidArgument("vector column is not fixed size binary or fixed size list");
     }
   } else {
     return Status::InvalidArgument("vector column is empty");

diff --git a/go/storage/schema/schema_option.go b/go/storage/schema/schema_option.go
@@ -14,7 +14,7 @@ var (
 	ErrVersionColumnNotFound = errors.New("version column not found")
 	ErrVersionColumnType     = errors.New("version column is not int64")
 	ErrVectorColumnNotFound  = errors.New("vector column not found")
-	ErrVectorColumnType      = errors.New("vector column is not fixed size binary")
+	ErrVectorColumnType      = errors.New("vector column is not fixed size binary or fixed size list")
 	ErrVectorColumnEmpty     = errors.New("vector column is empty")
 )
 
@@ -69,7 +69,7 @@ func (o *SchemaOptions) Validate(schema *arrow.Schema) error {
 		vectorField, b := schema.FieldsByName(o.VectorColumn)
 		if !b {
 			return ErrVectorColumnNotFound
-		} else if vectorField[0].Type.ID() != arrow.FIXED_SIZE_BINARY {
+		} else if vectorField[0].Type.ID() != arrow.FIXED_SIZE_BINARY && vectorField[0].Type.ID() != arrow.FIXED_SIZE_LIST {
 			return ErrVectorColumnType
 		}
 	} else {