lightning: fix in-correctly check for session variable

br/pkg/lightning/lightning.go

@@ -75,6 +75,9 @@ type Lightning struct {
 }
 
 func initEnv(cfg *config.GlobalConfig) error {
+	if cfg.App.Config.File == "" {
+		return nil
+	}
 	return log.InitLogger(&cfg.App.Config, cfg.TiDB.LogLevel)
 }
 

br/pkg/lightning/restore/table_restore.go

@@ -192,6 +192,7 @@ func createColumnPermutation(columns []string, ignoreColumns []string, tableInfo
 func (tr *TableRestore) restoreEngines(pCtx context.Context, rc *Controller, cp *checkpoints.TableCheckpoint) error {
 	indexEngineCp := cp.Engines[indexEngineID]
 	if indexEngineCp == nil {
+		tr.logger.Error("fail to restoreEngines because indexengine is nil")
 		return errors.Errorf("table %v index engine checkpoint not found", tr.tableName)
 	}
 	// If there is an index engine only, it indicates no data needs to restore.

br/pkg/lightning/restore/tidb.go

@@ -118,13 +118,20 @@ func DBFromConfig(ctx context.Context, dsn config.DBStore) (*sql.DB, error) {
 		}
 	}
 
+	if dsn.Vars != nil {
+		for k, v := range dsn.Vars {
+			vars[k] = v
+		}
+	}
 	for k, v := range vars {
 		q := fmt.Sprintf("SET SESSION %s = %s;", k, v)
+		log.L().Info("Set session", zap.String("query", q))
 		if _, err1 := db.ExecContext(ctx, q); err1 != nil {
 			log.L().Warn("set session variable failed, will skip this query", zap.String("query", q),
 				zap.Error(err1))
 			delete(vars, k)
 		}
+		log.L().Info("Set session End")
 	}
 	_ = db.Close()