executor: add missing update table delta for TxnCtx

executor/batch_point_get.go

@@ -141,6 +141,13 @@ func (e *BatchPointGetExec) Next(ctx context.Context, req *chunk.Chunk) error {
 		}
 	}
 
+	if e.lock {
+		e.updateDeltaForTableID(e.tblInfo.ID)
+		for _, pid := range e.physIDs {
+			e.updateDeltaForTableID(pid)
+		}
+	}
+
 	if e.index >= len(e.values) {
 		return nil
 	}

executor/executor.go

@@ -203,6 +203,12 @@ func (e *baseExecutor) Next(ctx context.Context, req *chunk.Chunk) error {
 	return nil
 }
 
+func (e *baseExecutor) updateDeltaForTableID(id int64) {
+	txnCtx := e.ctx.GetSessionVars().TxnCtx
+	udpp := e.ctx.GetSessionVars().UseDynamicPartitionPrune()
+	txnCtx.UpdateDeltaForTable(id, id, 0, 0, nil, udpp)
+}
+
 func newBaseExecutor(ctx sessionctx.Context, schema *expression.Schema, id int, children ...Executor) baseExecutor {
 	e := baseExecutor{
 		children:     children,
@@ -943,6 +949,18 @@ func (e *SelectLockExec) Next(ctx context.Context, req *chunk.Chunk) error {
 		lockWaitTime = int64(e.Lock.WaitSec) * 1000
 	}
 
+	if len(e.tblID2Handle) > 0 {
+		for id := range e.tblID2Handle {
+			e.updateDeltaForTableID(id)
+		}
+	}
+	if len(e.partitionedTable) > 0 {
+		for _, p := range e.partitionedTable {
+			pid := p.Meta().ID
+			e.updateDeltaForTableID(pid)
+		}
+	}
+
 	return doLockKeys(ctx, e.ctx, newLockCtx(e.ctx.GetSessionVars(), lockWaitTime), e.keys...)
 }
 

executor/point_get.go

@@ -180,6 +180,9 @@ func (e *PointGetExecutor) Next(ctx context.Context, req *chunk.Chunk) error {
 	} else {
 		tblID = e.tblInfo.ID
 	}
+	if e.lock {
+		e.updateDeltaForTableID(tblID)
+	}
 	if e.idxInfo != nil {
 		if isCommonHandleRead(e.tblInfo, e.idxInfo) {
 			handleBytes, err := EncodeUniqueIndexValuesForKey(e.ctx, e.tblInfo, e.idxInfo, e.idxVals)

executor/write.go

@@ -169,6 +169,18 @@ func updateRecord(ctx context.Context, sctx sessionctx.Context, h kv.Handle, old
 		txnCtx := sctx.GetSessionVars().TxnCtx
 		if txnCtx.IsPessimistic {
 			txnCtx.AddUnchangedRowKey(unchangedRowKey)
+			// Update DeltaForTable
+			colSize := make(map[int64]int64, len(newData))
+			for id, col := range t.Cols() {
+				_, err := codec.EstimateValueSize(sc, newData[id])
+				if err != nil {
+					continue
+				}
+				// Nothing changed so size should be 0
+				colSize[col.ID] = 0
+			}
+			udpp := sctx.GetSessionVars().UseDynamicPartitionPrune()
+			txnCtx.UpdateDeltaForTable(t.Meta().ID, physicalID, 0, 0, colSize, udpp)
 		}
 		return false, nil
 	}