fix: Pass index type for Scalar indexes

Related to milvus-io#694 milvus-io#695 Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
congqixia · Apr 3, 2024 · de44939 · de44939
1 parent 6d2bfc9
commit de44939
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/entity/index_scalar.go b/entity/index_scalar.go
@@ -7,7 +7,9 @@ type indexScalar struct {
 }
 
 func (i *indexScalar) Params() map[string]string {
-	return map[string]string{}
+	return map[string]string{
+		tIndexType: string(i.baseIndex.it),
+	}
 }
 
 func NewScalarIndex() Index {

diff --git a/entity/index_scalar_test.go b/entity/index_scalar_test.go
@@ -14,4 +14,10 @@ func TestScalarIndex(t *testing.T) {
 	idxWithType := NewScalarIndexWithType(Sorted)
 
 	assert.EqualValues(t, Sorted, idxWithType.IndexType())
+	assert.EqualValues(t, Sorted, idxWithType.Params()[tIndexType])
+
+	idxWithType = NewScalarIndexWithType(Trie)
+
+	assert.EqualValues(t, Trie, idxWithType.IndexType())
+	assert.EqualValues(t, Trie, idxWithType.Params()[tIndexType])
 }