address comment

Signed-off-by: lance6716 <[email protected]>
pingcap · Dec 8, 2022 · 118ed01 · 118ed01
1 parent a49082f
commit 118ed01
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dm/pkg/checker/table_structure.go b/dm/pkg/checker/table_structure.go
@@ -231,7 +231,7 @@ func (c *TablesChecker) Check(ctx context.Context) *Result {
 		pool.Go(worker.handle)
 	}
 
-	dispatchTableItemWithDownstreamTable(c.tableMap, pool.PutJob)
+	dispatchTableItemWithDownstreamTable(c.tableMap, pool)
 
 	if err := pool.Wait(); err != nil {
 		markCheckError(r, err)
@@ -930,12 +930,12 @@ func dispatchTableItem(ctx context.Context, tableMap map[string][]filter.Table,
 
 func dispatchTableItemWithDownstreamTable(
 	tableMaps map[string]map[filter.Table][]filter.Table,
-	putJobFn func(*checkItem) bool,
+	pool *WorkerPool[*checkItem, []*incompatibilityOption],
 ) {
 	for sourceID, tableMap := range tableMaps {
 		for downTable, upTables := range tableMap {
 			for _, upTable := range upTables {
-				ok := putJobFn(&checkItem{
+				ok := pool.PutJob(&checkItem{
 					upstreamTable:   upTable,
 					downstreamTable: downTable,
 					sourceID:        sourceID,