admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-07 01:28:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add log field for local worker call (#25594)

Browse Source 
Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2023-07-14 17:34:31 +08:00 committed by GitHub
parent a7e6f08183
commit f2b14b43de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
internal/querynodev2/local_worker.go
View File

@ -43,7 +43,13 @@ func NewLocalWorker(node *QueryNode) *LocalWorker {
} }
func (w *LocalWorker) LoadSegments(ctx context.Context, req *querypb.LoadSegmentsRequest) error { 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...") log.Info("start to load segments...")
loaded, err := w.node.loader.Load(ctx, loaded, err := w.node.loader.Load(ctx,
req.GetCollectionID(), 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 { 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") log.Info("start to release segments")
for _, id := range req.GetSegmentIDs() { for _, id := range req.GetSegmentIDs() {
w.node.manager.Segment.Remove(id, req.GetScope()) 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 { 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") log.Info("start to process segment delete")
status, err := w.node.Delete(ctx, req) status, err := w.node.Delete(ctx, req)
if err != nil { if err != nil {