fix: Pass index type for Scalar indexes

Related to #694 #695 Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Apr 8, 2024 · 649a6db · 649a6db
1 parent 59ff80c
commit 649a6db
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])
 }