admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-02-02 01:06:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add refresh path to proxy (#21931)

Browse Source 
/kind improvement

Signed-off-by: Yuchen Gao <yuchen.gao@zilliz.com>
This commit is contained in:
Ten Thousand Leaves 2023-02-03 15:09:52 +08:00 committed by GitHub
parent 14f7b7559a
commit 932ced5cd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/proxy/impl.go
View File

@ -453,7 +453,8 @@ func (node *Proxy) LoadCollection(ctx context.Context, request *milvuspb.LoadCol
zap.String("traceID", traceID),
zap.String("role", typeutil.ProxyRole),
zap.String("db", request.DbName),
zap.String("collection", request.CollectionName))
zap.String("collection", request.CollectionName),
zap.Bool("refreshMode", request.Refresh))
if err := node.sched.ddQueue.Enqueue(lct); err != nil {
log.Warn("LoadCollection failed to enqueue",
@ -1407,7 +1408,8 @@ func (node *Proxy) LoadPartitions(ctx context.Context, request *milvuspb.LoadPar
zap.String("role", typeutil.ProxyRole),
zap.String("db", request.DbName),
zap.String("collection", request.CollectionName),
zap.Any("partitions", request.PartitionNames))
zap.Strings("partitions", request.PartitionNames),
zap.Bool("refreshMode", request.Refresh))
if err := node.sched.ddQueue.Enqueue(lpt); err != nil {
log.Warn(

2
internal/proxy/task.go
View File

@ -1364,6 +1364,7 @@ func (lct *loadCollectionTask) Execute(ctx context.Context) (err error) {
Schema: collSchema,
ReplicaNumber: lct.ReplicaNumber,
FieldIndexID: fieldIndexIDs,
Refresh: lct.Refresh,
}
log.Info("send LoadCollectionRequest to query coordinator", zap.String("role", typeutil.ProxyRole),
zap.Int64("msgID", request.Base.MsgID), zap.Int64("collectionID", request.CollectionID),
@ -1586,6 +1587,7 @@ func (lpt *loadPartitionsTask) Execute(ctx context.Context) error {
Schema: collSchema,
ReplicaNumber: lpt.ReplicaNumber,
FieldIndexID: fieldIndexIDs,
Refresh: lpt.Refresh,
}
lpt.result, err = lpt.queryCoord.LoadPartitions(ctx, request)
return err

2
internal/querycoordv2/services.go
View File

@ -215,6 +215,7 @@ func (s *Server) LoadCollection(ctx context.Context, req *querypb.LoadCollection
zap.Any("schema", req.Schema),
zap.Int32("replicaNumber", req.ReplicaNumber),
zap.Int64s("fieldIndexes", lo.Values(req.GetFieldIndexID())),
zap.Bool("refreshMode", req.GetRefresh()),
)
metrics.QueryCoordLoadCount.WithLabelValues(metrics.TotalLabel).Inc()
@ -304,6 +305,7 @@ func (s *Server) LoadPartitions(ctx context.Context, req *querypb.LoadPartitions
zap.Int64("collectionID", req.GetCollectionID()),
zap.Int32("replicaNumber", req.GetReplicaNumber()),
zap.Strings("resourceGroups", req.GetResourceGroups()),
zap.Bool("refreshMode", req.GetRefresh()),
)
log.Info("received load partitions request",