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)
}
return ctx, r
case *milvuspb.GetFlushStateRequest:
if r.DbName == "" {
r.DbName = GetCurDBNameFromContextOrDefault(ctx)
}
return ctx, r
default:
return ctx, req
}

2
internal/proxy/database_interceptor_test.go
View File

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