From f2b14b43de3711c67352a37f1108c5d4d6691fb8 Mon Sep 17 00:00:00 2001 From: congqixia Date: Fri, 14 Jul 2023 17:34:31 +0800 Subject: [PATCH] Add log field for local worker call (#25594) Signed-off-by: Congqi Xia --- internal/querynodev2/local_worker.go | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/internal/querynodev2/local_worker.go b/internal/querynodev2/local_worker.go index 569343912c..6f565a7876 100644 --- a/internal/querynodev2/local_worker.go +++ b/internal/querynodev2/local_worker.go @@ -43,7 +43,13 @@ func NewLocalWorker(node *QueryNode) *LocalWorker { } func (w *LocalWorker) LoadSegments(ctx context.Context, req *querypb.LoadSegmentsRequest) error { - log := log.Ctx(ctx) + log := log.Ctx(ctx).With( + zap.Int64("collectionID", req.GetCollectionID()), + zap.Int64s("segmentIDs", lo.Map(req.GetInfos(), func(info *querypb.SegmentLoadInfo, _ int) int64 { + return info.GetSegmentID() + })), + zap.String("loadScope", req.GetLoadScope().String()), + ) log.Info("start to load segments...") loaded, err := w.node.loader.Load(ctx, req.GetCollectionID(), @@ -63,7 +69,11 @@ func (w *LocalWorker) LoadSegments(ctx context.Context, req *querypb.LoadSegment } func (w *LocalWorker) ReleaseSegments(ctx context.Context, req *querypb.ReleaseSegmentsRequest) error { - log := log.Ctx(ctx) + log := log.Ctx(ctx).With( + zap.Int64("collectionID", req.GetCollectionID()), + zap.Int64s("segmentIDs", req.GetSegmentIDs()), + zap.String("scope", req.GetScope().String()), + ) log.Info("start to release segments") for _, id := range req.GetSegmentIDs() { w.node.manager.Segment.Remove(id, req.GetScope()) @@ -72,7 +82,10 @@ func (w *LocalWorker) ReleaseSegments(ctx context.Context, req *querypb.ReleaseS } func (w *LocalWorker) Delete(ctx context.Context, req *querypb.DeleteRequest) error { - log := log.Ctx(ctx) + log := log.Ctx(ctx).With( + zap.Int64("collectionID", req.GetCollectionId()), + zap.Int64("segmentID", req.GetSegmentId()), + ) log.Info("start to process segment delete") status, err := w.node.Delete(ctx, req) if err != nil {