fix: fix altercollectionfield interceptor dbname (#38662)

fix altercollectionfield interceptor dbname issue: #37436 Signed-off-by: Xianhui.Lin <[email protected]>
milvus-io · Dec 23, 2024 · d718bbb · d718bbb
1 parent 82d85e8
commit d718bbb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/internal/proxy/database_interceptor.go b/internal/proxy/database_interceptor.go
@@ -68,6 +68,11 @@ func fillDatabase(ctx context.Context, req interface{}) (context.Context, interf
 			r.DbName = GetCurDBNameFromContextOrDefault(ctx)
 		}
 		return ctx, r
+	case *milvuspb.AlterCollectionFieldRequest:
+		if r.DbName == "" {
+			r.DbName = GetCurDBNameFromContextOrDefault(ctx)
+		}
+		return ctx, r
 	case *milvuspb.CreatePartitionRequest:
 		if r.DbName == "" {
 			r.DbName = GetCurDBNameFromContextOrDefault(ctx)

diff --git a/internal/proxy/database_interceptor_test.go b/internal/proxy/database_interceptor_test.go
@@ -57,6 +57,7 @@ func TestDatabaseInterceptor(t *testing.T) {
 			&milvuspb.GetCollectionStatisticsRequest{},
 			&milvuspb.ShowCollectionsRequest{},
 			&milvuspb.AlterCollectionRequest{},
+			&milvuspb.AlterCollectionFieldRequest{},
 			&milvuspb.CreatePartitionRequest{},
 			&milvuspb.DropPartitionRequest{},
 			&milvuspb.HasPartitionRequest{},