diff --git a/internal/proxy/database_interceptor.go b/internal/proxy/database_interceptor.go index 5373cf65c9..6adb0bcd40 100644 --- a/internal/proxy/database_interceptor.go +++ b/internal/proxy/database_interceptor.go @@ -28,6 +28,11 @@ func fillDatabase(ctx context.Context, req interface{}) (context.Context, interf r.DbName = GetCurDBNameFromContextOrDefault(ctx) } return ctx, r + case *milvuspb.TruncateCollectionRequest: + if r.DbName == "" { + r.DbName = GetCurDBNameFromContextOrDefault(ctx) + } + return ctx, r case *milvuspb.HasCollectionRequest: if r.DbName == "" { r.DbName = GetCurDBNameFromContextOrDefault(ctx) diff --git a/internal/proxy/database_interceptor_test.go b/internal/proxy/database_interceptor_test.go index 598a7b7c48..26e64a45e1 100644 --- a/internal/proxy/database_interceptor_test.go +++ b/internal/proxy/database_interceptor_test.go @@ -49,6 +49,7 @@ func TestDatabaseInterceptor(t *testing.T) { availableReqs := []proto.Message{ &milvuspb.CreateCollectionRequest{}, &milvuspb.DropCollectionRequest{}, + &milvuspb.TruncateCollectionRequest{}, &milvuspb.HasCollectionRequest{}, &milvuspb.LoadCollectionRequest{}, &milvuspb.ReleaseCollectionRequest{},