milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix flush slowly (#27123) 

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
pull/27140/head
yihao.dai 2023-09-15 17:45:21 +08:00 committed by GitHub
parent 361bb34e33
commit 7a59de1fb1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
internal/proxy/database_interceptor.go
View File

@ -238,6 +238,11 @@ func fillDatabase(ctx context.Context, req interface{}) (context.Context, interf
r.DbName = GetCurDBNameFromContextOrDefault(ctx) r.DbName = GetCurDBNameFromContextOrDefault(ctx)
} }
return ctx, r return ctx, r
case *milvuspb.GetFlushStateRequest:
if r.DbName == "" {
r.DbName = GetCurDBNameFromContextOrDefault(ctx)
}
return ctx, r
default: default:
return ctx, req return ctx, req
} }

2
internal/proxy/database_interceptor_test.go
View File

@ -74,6 +74,7 @@ func TestDatabaseInterceptor(t *testing.T) {
&milvuspb.DeleteRequest{}, &milvuspb.DeleteRequest{},
&milvuspb.SearchRequest{}, &milvuspb.SearchRequest{},
&milvuspb.FlushRequest{}, &milvuspb.FlushRequest{},
&milvuspb.GetFlushStateRequest{},
&milvuspb.QueryRequest{}, &milvuspb.QueryRequest{},
&milvuspb.CreateAliasRequest{}, &milvuspb.CreateAliasRequest{},
&milvuspb.DropAliasRequest{}, &milvuspb.DropAliasRequest{},
@ -113,7 +114,6 @@ func TestDatabaseInterceptor(t *testing.T) {
&milvuspb.GetCompactionStateRequest{}, &milvuspb.GetCompactionStateRequest{},
&milvuspb.ManualCompactionRequest{}, &milvuspb.ManualCompactionRequest{},
&milvuspb.GetCompactionPlansRequest{}, &milvuspb.GetCompactionPlansRequest{},
&milvuspb.GetFlushStateRequest{},
&milvuspb.GetFlushAllStateRequest{}, &milvuspb.GetFlushAllStateRequest{},
&milvuspb.GetImportStateRequest{}, &milvuspb.GetImportStateRequest{},
} }