lightning: always enable autocommit in db connection (#1114) (#1125)

pkg/lightning/restore/tidb.go

@@ -98,6 +98,8 @@ func DBFromConfig(dsn config.DBStore) (*sql.DB, error) {
 			"allow_auto_random_explicit_insert": "1",
 			// allow use _tidb_rowid in sql statement
 			"tidb_opt_write_row_id": "1",
+			// always set auto-commit to ON
+			"autocommit": "1",
 		},
 	}
 	db, err := param.Connect()
@@ -255,7 +257,10 @@ func LoadSchemaInfo(
 		}
 
 		for _, tbl := range schema.Tables {
-			tblInfo := tableMap[strings.ToLower(tbl.Name)]
+			tblInfo, ok := tableMap[strings.ToLower(tbl.Name)]
+			if !ok {
+				return nil, errors.Errorf("table '%s' schema not found", tbl.Name)
+			}
 			tableName := tblInfo.Name.String()
 			if tblInfo.State != model.StatePublic {
 				err := errors.Errorf("table [%s.%s] state is not public", schema.Name, tableName)

tests/lightning_alter_random/run.sh

@@ -18,6 +18,9 @@ set -eu
 # FIXME: auto-random is only stable on master currently.
 check_cluster_version 4 0 0 AUTO_RANDOM || exit 0
 
+# test lightning with autocommit disabled
+run_sql "SET @@global.autocommit = '0';"
+
 for backend in tidb importer local; do
     if [ "$backend" = 'local' ]; then
         check_cluster_version 4 0 0 'local backend' || continue
@@ -35,11 +38,13 @@ for backend in tidb importer local; do
     check_contains 'inc: 3'
 
     # auto random base is 4
-    run_sql "INSERT INTO alter_random.t VALUES ();"
+    run_sql "INSERT INTO alter_random.t VALUES ();commit;"
     run_sql "SELECT id & b'000001111111111111111111111111111111111111111111111111111111111' as inc FROM alter_random.t"
     if [ "$backend" = 'tidb' ]; then
       check_contains 'inc: 30002'
     else
       check_contains 'inc: 4'
     fi
 done
+
+run_sql "SET @@global.autocommit = '1';"