address comment

Signed-off-by: lance6716 <lance6716@gmail.com>
lance6716 · Nov 25, 2022 · 0121d7e · 0121d7e
1 parent 56dbede
commit 0121d7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dm/worker/subtask.go b/dm/worker/subtask.go
@@ -75,7 +75,7 @@ func createRealUnits(cfg *config.SubTaskConfig, etcdClient *clientv3.Client, wor
 
 func newLoadUnit(cfg *config.SubTaskConfig, etcdClient *clientv3.Client, workerName string) unit.Unit {
 	// tidb-lightning doesn't support column mapping currently
-	if cfg.ImportMode == config.LoadModeLoader || len(cfg.ColumnMappingRules) > 0 {
+	if !cfg.NeedUseLightning() || len(cfg.ColumnMappingRules) > 0 {
 		return loader.NewLoader(cfg, etcdClient, workerName)
 	}
 	return loader.NewLightning(cfg, etcdClient, workerName)