From 00c9797d91f624aa33ac5b6b6a8359a734c27b22 Mon Sep 17 00:00:00 2001 From: Jiquan Long Date: Thu, 16 Dec 2021 22:27:42 +0800 Subject: [PATCH] Add log of insert to record time spent in mq (#13527) Signed-off-by: dragondriver --- internal/proxy/task.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/internal/proxy/task.go b/internal/proxy/task.go index b71137d688..be6d33a015 100644 --- a/internal/proxy/task.go +++ b/internal/proxy/task.go @@ -985,6 +985,10 @@ func (it *insertTask) _assignSegmentID(stream msgstream.MsgStream, pack *msgstre func (it *insertTask) Execute(ctx context.Context) error { sp, ctx := trace.StartSpanFromContextWithOperationName(it.ctx, "Proxy-Insert-Execute") defer sp.Finish() + + tr := timerecord.NewTimeRecorder(fmt.Sprintf("proxy execute insert %d", it.ID())) + defer tr.Elapse("done") + collectionName := it.BaseInsertTask.CollectionName collID, err := globalMetaCache.GetCollectionID(ctx, collectionName) if err != nil { @@ -1004,6 +1008,7 @@ func (it *insertTask) Execute(ctx context.Context) error { } } it.PartitionID = partitionID + tr.Record("get collection id & partition id from cache") var tsMsg msgstream.TsMsg = &it.BaseInsertTask it.BaseMsg.Ctx = ctx @@ -1030,6 +1035,7 @@ func (it *insertTask) Execute(ctx context.Context) error { return err } } + tr.Record("get used message stream") // Assign SegmentID var pack *msgstream.MsgPack @@ -1037,6 +1043,7 @@ func (it *insertTask) Execute(ctx context.Context) error { if err != nil { return err } + tr.Record("assign segment id") err = stream.Produce(pack) if err != nil { @@ -1044,6 +1051,7 @@ func (it *insertTask) Execute(ctx context.Context) error { it.result.Status.Reason = err.Error() return err } + tr.Record("send insert request to message stream") return nil }