add task id info in explain statement for mpp

executor/mpp_gather.go

@@ -90,14 +90,14 @@ func (e *MPPGather) appendMPPDispatchReq(pf *plannercore.Fragment, tasks []*kv.M
 	return nil
 }
 
-
 func CollectPlanIDS(plan plannercore.PhysicalPlan, ids []int) []int {
 	ids = append(ids, plan.ID())
 	for _, child := range plan.Children() {
 		ids = CollectPlanIDS(child, ids)
 	}
 	return ids
 }
+
 // Open decides the task counts and locations and generate exchange operators for every plan fragment.
 // Then dispatch tasks to tiflash stores. If any task fails, it would cancel the rest tasks.
 func (e *MPPGather) Open(ctx context.Context) (err error) {

planner/core/explain.go

@@ -864,6 +864,14 @@ func (p *PhysicalExchangeSender) ExplainInfo() string {
 		fmt.Fprintf(buffer, "HashPartition")
 		fmt.Fprintf(buffer, ", Hash Cols: %s", expression.ExplainColumnList(p.HashCols))
 	}
+	fmt.Fprintf(buffer, "tasks: [")
+	for idx, task := range p.Tasks {
+		if idx != 0 {
+			fmt.Fprintf(buffer, ", ")
+		}
+		fmt.Fprintf(buffer, "%v", task.ID)
+	}
+	fmt.Fprintf(buffer, "]")
 	return buffer.String()
 }
 

store/tikv/mpp.go

@@ -74,8 +74,8 @@ func (c *MPPClient) ConstructMPPTasks(ctx context.Context, req *kv.MPPBuildTasks
 
 // mppResponse wraps mpp data packet.
 type mppResponse struct {
-	pbResp *mpp.MPPDataPacket
-	detail *CopRuntimeStats
+	pbResp   *mpp.MPPDataPacket
+	detail   *CopRuntimeStats
 	respTime time.Duration
 	respSize int64
 

util/execdetails/execdetails.go

@@ -441,9 +441,9 @@ func (crs *CopRuntimeStats) RecordOneCopTask(address string, summary *tipb.Execu
 	defer crs.Unlock()
 	crs.stats[address] = append(crs.stats[address],
 		&BasicRuntimeStats{loop: int32(*summary.NumIterations),
-			consume: int64(*summary.TimeProcessedNs),
-			rows:    int64(*summary.NumProducedRows),
-		    concurrency: maxInt32(1, int32(summary.GetConcurrency()))})
+			consume:     int64(*summary.TimeProcessedNs),
+			rows:        int64(*summary.NumProducedRows),
+			concurrency: maxInt32(1, int32(summary.GetConcurrency()))})
 }
 
 // GetActRows return total rows of CopRuntimeStats.
@@ -571,9 +571,9 @@ func (e *BasicRuntimeStats) GetActRows() int64 {
 // Clone implements the RuntimeStats interface.
 func (e *BasicRuntimeStats) Clone() RuntimeStats {
 	return &BasicRuntimeStats{
-		loop:    e.loop,
-		consume: e.consume,
-		rows:    e.rows,
+		loop:        e.loop,
+		consume:     e.consume,
+		rows:        e.rows,
 		concurrency: e.concurrency,
 	}
 }