insert_mut

turbosql-impl/src/insert.rs

@@ -27,6 +27,17 @@ pub(super) fn insert(table: &Table) -> proc_macro2::TokenStream {
 			})
 		}
 
+		fn insert_mut(&mut self) -> Result<i64, ::turbosql::Error> {
+			assert!(self.rowid.is_none());
+			::turbosql::__TURBOSQL_DB.with(|db| {
+				let db = db.borrow_mut();
+				let mut stmt = db.prepare_cached(#sql)?;
+				let result = stmt.insert(&[#( #columns ),*] as &[&dyn ::turbosql::ToSql])?;
+				self.rowid = Some(result);
+				Ok(result)
+			})
+		}
+
 		fn insert_batch<T: AsRef<#table>>(rows: &[T]) -> Result<(), ::turbosql::Error> {
 			for row in rows {
 				row.as_ref().insert()?;

turbosql-impl/src/lib.rs

@@ -942,7 +942,7 @@ fn create(table: &Table, minitable: &MiniTable) {
 	if old_toml_str.replace("\r\n", "\n") != new_toml_str {
 		#[cfg(not(feature = "test"))]
 		if std::env::var("CI").is_ok() || std::env::var("TURBOSQL_LOCKED_MODE").is_ok() {
-			abort_call_site!("Change in `{}` detected with CI or TURBOSQL_LOCKED_MODE environment variable set. Make sure your `migrations.toml` file is up-to-date.", migrations_toml_path_lossy);
+			abort_call_site!("Change in `{}` detected with CI or TURBOSQL_LOCKED_MODE environment variable set. Make sure your `migrations.toml` file is committed and up-to-date.", migrations_toml_path_lossy);
 		};
 		fs::write(&migrations_toml_path, new_toml_str)
 			.unwrap_or_else(|e| abort_call_site!("Unable to write {}: {:?}", migrations_toml_path_lossy, e));

turbosql/tests/integration_test.rs

@@ -74,12 +74,17 @@ fn integration_test() {
 		..Default::default()
 	};
 
+	let mut row2 = row.clone();
+
 	assert_eq!(row.insert().unwrap(), 1);
 	assert_eq!(row.insert().unwrap(), 2);
 	row.rowid = Some(1);
 	row.field_u8 = Some(84);
 	assert_eq!(row.update().unwrap(), 1);
 
+	row2.insert_mut().unwrap();
+	assert_eq!(row2.rowid, Some(3));
+
 	assert_eq!(select!(i64 "1").unwrap(), 1);
 	// assert_eq!(select!(Vec<i64> "1").unwrap(), vec![1]);
 	// assert_eq!(select!(Option<i64> "1").unwrap(), Some(1));