on reprepare create new state with updated number of cursors and

registers, so that the Program insns are in sync with ProgramState

tests/integration/query_processing/test_multi_thread.rs

@@ -1,6 +1,6 @@
 use std::sync::{atomic::AtomicUsize, Arc};
 
-use turso_core::StepResult;
+use turso_core::{Statement, StepResult};
 
 use crate::common::{maybe_setup_tracing, TempDatabase};
 
@@ -234,3 +234,96 @@ fn test_schema_reprepare_write() {
         }
     }
 }
+
+fn advance(stmt: &mut Statement) -> anyhow::Result<()> {
+    stmt.step()?;
+    stmt.run_once()?;
+    Ok(())
+}
+
+/// Regression test detected by whopper
+#[test]
+fn test_interleaved_transactions() -> anyhow::Result<()> {
+    maybe_setup_tracing();
+    let tmp_db = TempDatabase::new_empty(true);
+    {
+        let bootstrap_conn = tmp_db.connect_limbo();
+        bootstrap_conn.execute("CREATE TABLE table_0 (id INTEGER,col_1 REAL,col_2 INTEGER,col_3 REAL,col_4 TEXT,col_5 REAL,col_6 TEXT)")?;
+        bootstrap_conn.execute("CREATE INDEX idx_table_0_0 ON table_0 (col_4 ASC)")?;
+        bootstrap_conn.execute("CREATE INDEX idx_table_0_1 ON table_0 (id DESC)")?;
+        bootstrap_conn.execute("CREATE INDEX idx_table_0_2 ON table_0 (col_5 DESC)")?;
+        bootstrap_conn.close()?;
+    }
+
+    let conn = [
+        tmp_db.connect_limbo(),
+        tmp_db.connect_limbo(),
+        tmp_db.connect_limbo(),
+        tmp_db.connect_limbo(),
+    ];
+
+    let mut statement2 = conn[2].prepare("BEGIN")?;
+    let mut statement0 = conn[0].prepare("BEGIN")?;
+    let mut statement1 = conn[1].prepare("BEGIN")?;
+
+    advance(&mut statement2)?;
+
+    let mut statement2 = conn[2].prepare("DELETE FROM table_0 WHERE (TRUE)")?;
+
+    advance(&mut statement0)?;
+
+    let mut statement0 = conn[0].prepare("COMMIT")?;
+
+    advance(&mut statement1)?;
+
+    let mut statement1 = conn[1].prepare("UPDATE table_0 SET col_5 = -2926216022.864461, col_1 = 1136343846.3760414, col_2 = 1260332354248861058, col_6 = 'breathtaking_wallace', col_3 = -8354252674.968108, id = -2763965266862900284 WHERE (TRUE)")?;
+
+    advance(&mut statement2)?;
+    advance(&mut statement0)?;
+    advance(&mut statement1)?;
+    advance(&mut statement2)?;
+
+    let mut statement2 = conn[2].prepare("COMMIT")?;
+
+    advance(&mut statement1)?;
+    advance(&mut statement2)?;
+
+    let mut statement2 = conn[2].prepare("BEGIN")?;
+
+    let mut statement3 = conn[3].prepare("BEGIN")?;
+
+    advance(&mut statement1)?;
+    advance(&mut statement2)?;
+
+    let mut statement2 = conn[2].prepare("INSERT INTO table_0 VALUES (3433031730186055493, -9049649117.499245, 377748201198469116, -303828055.307354, 'hardworking_pinotnoir', 3130880977.346573, 'dazzling_identity'), (4047491512698975530, -6415241771.805258, 8252804953477887816, 6468710871.6649, 'diplomatic_karamazov', 590358226.8343716, 'hilarious_strasbourg'), (2865599543545078376, 84894401.22016525, 9113810426850381627, -1136160051.7521439, 'funny_benton', 9522389352.598354, 'magnificent_french'), (-157899885850804353, -303833796.57147026, 486259919370287064, -9427424128.005714, 'considerate_diamant', -3105334243.936157, 'kind_walker'), (8502247374804763489, -2126532888.2616653, 5690470012873526939, 4011656749.2326107, 'kind_ladrido', 1381034902.7760563, 'humorous_crane'), (2487742055507017334, -5830452441.986847, 3661939929057695925, -6299976423.211256, 'adaptable_raevsky', -7871748970.666381, 'technological_bataille'), (-5348095239593101865, -998225440.4524403, 5195262288395229508, -8305444803.975374, 'upbeat_courtney', -3943473497.4281626, 'imaginative_arrigoni'), (-6470080787150464674, -5833281407.383408, 5877012236478010308, 3023123550.254177, 'fearless_cairo', -141073531.91679573, 'generous_university')")?;
+
+    advance(&mut statement3)?;
+
+    let mut statement3 = conn[3].prepare("INSERT INTO table_0 VALUES (-7468459471409934075, 8179435779.870651, -7868006515434924912, -5415470506.527203, 'affectionate_n1x', -88866295.57206345, 'agreeable_treloar'), (3000445982321368777, 3099814982.0727863, -5101787605795972474, -925278326.7265358, 'giving_individualiste', -6553332857.366568, 'brave_patrizia'), (406163996859206098, -3340292138.289094, -5058217201699339610, 2605267874.8582096, 'fabulous_burnett', -4601912326.914466, 'super_jedi'), (6398781934600428549, -6770226564.882048, -2332649333251794167, 6904161964.055864, 'shining_sergent', -4779129294.073781, 'hardworking_beggar'), (1530150677936272307, -8683321096.443897, -2211014401610293017, 2417417840.8996468, 'magnificent_datacide', -2218929107.793541, 'ravishing_nw'), (8028216547992752413, -8876487798.088352, 8974386493479719872, -6723037189.199554, 'glimmering_murray', -1973499548.0633707, 'spectacular_fitzpatrick')")?;
+
+    advance(&mut statement1)?;
+
+    let mut statement1 = conn[1].prepare(
+        "CREATE INDEX idx_table_0_persiste ON table_0 (col_2 ASC, col_6 ASC, col_3 ASC)",
+    )?;
+
+    advance(&mut statement2)?;
+    advance(&mut statement3)?;
+
+    let mut statement0 = conn[0].prepare("BEGIN")?;
+
+    advance(&mut statement1)?;
+    assert!(statement1.is_done());
+
+    let mut statement1 = conn[1].prepare("COMMIT")?;
+
+    advance(&mut statement2)?;
+    advance(&mut statement0)?;
+
+    let mut _statement0 = conn[0].prepare("UPDATE table_0 SET col_3 = 7634893024.5729065, col_6 = 'glimmering_besnard', col_1 = 9240915430.267292, col_4 = 'efficient_mekan' WHERE (TRUE)")?;
+
+    advance(&mut statement1)?;
+    advance(&mut statement2)?;
+
+    Ok(())
+}