fix ut

Signed-off-by: xufei <xufeixw@mail.ustc.edu.cn>
pingcap · Oct 13, 2023 · 869d02f · 869d02f
1 parent 1fa5652
commit 869d02f
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/util/chunk/row_container.go b/util/chunk/row_container.go
@@ -643,7 +643,7 @@ func (c *SortedRowContainer) ActionSpill() *SortAndSpillDiskAction {
 	if c.actionSpill == nil {
 		c.actionSpill = &SortAndSpillDiskAction{
 			c:                   c,
-			baseSpillDiskAction: &baseSpillDiskAction{cond: spillStatusCond{sync.NewCond(new(sync.Mutex)), notSpilled}},
+			baseSpillDiskAction: c.RowContainer.ActionSpill().baseSpillDiskAction,
 		}
 	}
 	return c.actionSpill
@@ -652,16 +652,8 @@ func (c *SortedRowContainer) ActionSpill() *SortAndSpillDiskAction {
 // ActionSpillForTest returns a SortAndSpillDiskAction for sorting and spilling over to disk for test.
 func (c *SortedRowContainer) ActionSpillForTest() *SortAndSpillDiskAction {
 	c.actionSpill = &SortAndSpillDiskAction{
-		c: c,
-		baseSpillDiskAction: &baseSpillDiskAction{
-			testSyncInputFunc: func() {
-				c.actionSpill.testWg.Add(1)
-			},
-			testSyncOutputFunc: func() {
-				c.actionSpill.testWg.Done()
-			},
-			cond: spillStatusCond{sync.NewCond(new(sync.Mutex)), notSpilled},
-		},
+		c:                   c,
+		baseSpillDiskAction: c.RowContainer.ActionSpillForTest().baseSpillDiskAction,
 	}
 	return c.actionSpill
 }