fix: [2.5]fix altercollectionfield interceptor dbname (milvus-io#38663)

fix altercollectionfield interceptor dbname issue: milvus-io#37436 master pr: milvus-io#38662 Signed-off-by: Xianhui.Lin <[email protected]>
bigsheeper · Dec 24, 2024 · c51b458 · c51b458
1 parent bb3d993
commit c51b458
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{},