未验证 提交 74e3cc64 编写于 作者: T Ten Thousand Leaves 提交者: GitHub

Add refresh path to proxy (#21933)

/kind improvement
Signed-off-by: NYuchen Gao <yuchen.gao@zilliz.com>
上级 d41cc0b2
...@@ -383,7 +383,8 @@ func (node *Proxy) LoadCollection(ctx context.Context, request *milvuspb.LoadCol ...@@ -383,7 +383,8 @@ func (node *Proxy) LoadCollection(ctx context.Context, request *milvuspb.LoadCol
log := log.Ctx(ctx).With( log := log.Ctx(ctx).With(
zap.String("role", typeutil.ProxyRole), zap.String("role", typeutil.ProxyRole),
zap.String("db", request.DbName), zap.String("db", request.DbName),
zap.String("collection", request.CollectionName)) zap.String("collection", request.CollectionName),
zap.Bool("refreshMode", request.Refresh))
log.Debug("LoadCollection received") log.Debug("LoadCollection received")
...@@ -1129,7 +1130,8 @@ func (node *Proxy) LoadPartitions(ctx context.Context, request *milvuspb.LoadPar ...@@ -1129,7 +1130,8 @@ func (node *Proxy) LoadPartitions(ctx context.Context, request *milvuspb.LoadPar
zap.String("role", typeutil.ProxyRole), zap.String("role", typeutil.ProxyRole),
zap.String("db", request.DbName), zap.String("db", request.DbName),
zap.String("collection", request.CollectionName), zap.String("collection", request.CollectionName),
zap.Any("partitions", request.PartitionNames)) zap.Strings("partitions", request.PartitionNames),
zap.Bool("refreshMode", request.Refresh))
log.Debug(rpcReceived(method)) log.Debug(rpcReceived(method))
......
...@@ -1369,6 +1369,7 @@ func (lct *loadCollectionTask) Execute(ctx context.Context) (err error) { ...@@ -1369,6 +1369,7 @@ func (lct *loadCollectionTask) Execute(ctx context.Context) (err error) {
Schema: collSchema, Schema: collSchema,
ReplicaNumber: lct.ReplicaNumber, ReplicaNumber: lct.ReplicaNumber,
FieldIndexID: fieldIndexIDs, FieldIndexID: fieldIndexIDs,
Refresh: lct.Refresh,
} }
log.Debug("send LoadCollectionRequest to query coordinator", log.Debug("send LoadCollectionRequest to query coordinator",
zap.Any("schema", request.Schema)) zap.Any("schema", request.Schema))
...@@ -1595,6 +1596,7 @@ func (lpt *loadPartitionsTask) Execute(ctx context.Context) error { ...@@ -1595,6 +1596,7 @@ func (lpt *loadPartitionsTask) Execute(ctx context.Context) error {
Schema: collSchema, Schema: collSchema,
ReplicaNumber: lpt.ReplicaNumber, ReplicaNumber: lpt.ReplicaNumber,
FieldIndexID: fieldIndexIDs, FieldIndexID: fieldIndexIDs,
Refresh: lpt.Refresh,
} }
lpt.result, err = lpt.queryCoord.LoadPartitions(ctx, request) lpt.result, err = lpt.queryCoord.LoadPartitions(ctx, request)
return err return err
......
...@@ -213,6 +213,7 @@ func (s *Server) LoadCollection(ctx context.Context, req *querypb.LoadCollection ...@@ -213,6 +213,7 @@ func (s *Server) LoadCollection(ctx context.Context, req *querypb.LoadCollection
zap.Any("schema", req.Schema), zap.Any("schema", req.Schema),
zap.Int32("replicaNumber", req.ReplicaNumber), zap.Int32("replicaNumber", req.ReplicaNumber),
zap.Int64s("fieldIndexes", lo.Values(req.GetFieldIndexID())), zap.Int64s("fieldIndexes", lo.Values(req.GetFieldIndexID())),
zap.Bool("refreshMode", req.GetRefresh()),
) )
metrics.QueryCoordLoadCount.WithLabelValues(metrics.TotalLabel).Inc() metrics.QueryCoordLoadCount.WithLabelValues(metrics.TotalLabel).Inc()
...@@ -301,6 +302,7 @@ func (s *Server) LoadPartitions(ctx context.Context, req *querypb.LoadPartitions ...@@ -301,6 +302,7 @@ func (s *Server) LoadPartitions(ctx context.Context, req *querypb.LoadPartitions
zap.Int64("collectionID", req.GetCollectionID()), zap.Int64("collectionID", req.GetCollectionID()),
zap.Int32("replicaNumber", req.GetReplicaNumber()), zap.Int32("replicaNumber", req.GetReplicaNumber()),
zap.Strings("resourceGroups", req.GetResourceGroups()), zap.Strings("resourceGroups", req.GetResourceGroups()),
zap.Bool("refreshMode", req.GetRefresh()),
) )
log.Info("received load partitions request", log.Info("received load partitions request",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册