executor: add memory track in handle cop request (pingcap#38755) (pin…

…gcap#38759) close pingcap#33893
ti-chi-bot · Jan 7, 2023 · 52ae590 · 52ae590
1 parent 4b777b4
commit 52ae590
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/executor/coprocessor.go b/executor/coprocessor.go
@@ -63,6 +63,7 @@ func (h *CoprocessorDAGHandler) HandleRequest(ctx context.Context, req *coproces
 	chk := newFirstChunk(e)
 	tps := e.base().retFieldTypes
 	var totalChunks, partChunks []tipb.Chunk
+	memTracker := h.sctx.GetSessionVars().StmtCtx.MemTracker
 	for {
 		chk.Reset()
 		err = Next(ctx, e, chk)
@@ -76,6 +77,9 @@ func (h *CoprocessorDAGHandler) HandleRequest(ctx context.Context, req *coproces
 		if err != nil {
 			return h.buildErrorResponse(err)
 		}
+		for _, ch := range partChunks {
+			memTracker.Consume(int64(ch.Size()))
+		}
 		totalChunks = append(totalChunks, partChunks...)
 	}
 	if err := e.Close(); err != nil {