mirror of
https://github.com/nushell/nushell.git
synced 2025-02-18 03:21:05 +01:00
fix: now sqlite insert handles column names as expected (#12039)
- fixes #11429 - fixes #12011
This commit is contained in:
parent
ce116b5d5f
commit
93188b3eda
@ -201,29 +201,8 @@ fn insert_in_transaction(
|
|||||||
span: Span,
|
span: Span,
|
||||||
mut table: Table,
|
mut table: Table,
|
||||||
) -> Result<Value, ShellError> {
|
) -> Result<Value, ShellError> {
|
||||||
let mut stream = stream.peekable();
|
let mut stream = stream;
|
||||||
let first_val = match stream.peek() {
|
|
||||||
None => return Ok(Value::nothing(span)),
|
|
||||||
Some(val) => val.as_record()?,
|
|
||||||
};
|
|
||||||
|
|
||||||
let table_name = table.name().clone();
|
let table_name = table.name().clone();
|
||||||
let tx = table.try_init(first_val)?;
|
|
||||||
let insert_statement = format!(
|
|
||||||
"INSERT INTO [{}] VALUES ({})",
|
|
||||||
table_name,
|
|
||||||
["?"].repeat(first_val.values().len()).join(", ")
|
|
||||||
);
|
|
||||||
|
|
||||||
let mut insert_statement =
|
|
||||||
tx.prepare(&insert_statement)
|
|
||||||
.map_err(|e| ShellError::GenericError {
|
|
||||||
error: "Failed to prepare SQLite statement".into(),
|
|
||||||
msg: e.to_string(),
|
|
||||||
span: None,
|
|
||||||
help: None,
|
|
||||||
inner: Vec::new(),
|
|
||||||
})?;
|
|
||||||
|
|
||||||
// insert all the records
|
// insert all the records
|
||||||
stream.try_for_each(|stream_value| {
|
stream.try_for_each(|stream_value| {
|
||||||
@ -233,25 +212,47 @@ fn insert_in_transaction(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
insert_value(stream_value, &mut insert_statement)
|
let val = stream_value.as_record()?;
|
||||||
})?;
|
|
||||||
|
|
||||||
insert_statement
|
let tx = table.try_init(val)?;
|
||||||
.finalize()
|
let insert_statement = format!(
|
||||||
.map_err(|e| ShellError::GenericError {
|
"INSERT INTO [{}] ({}) VALUES ({})",
|
||||||
error: "Failed to finalize SQLite prepared statement".into(),
|
table_name,
|
||||||
|
val.cols.join(", "),
|
||||||
|
["?"].repeat(val.values().len()).join(", ")
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut insert_statement =
|
||||||
|
tx.prepare(&insert_statement)
|
||||||
|
.map_err(|e| ShellError::GenericError {
|
||||||
|
error: "Failed to prepare SQLite statement".into(),
|
||||||
|
msg: e.to_string(),
|
||||||
|
span: None,
|
||||||
|
help: None,
|
||||||
|
inner: Vec::new(),
|
||||||
|
})?;
|
||||||
|
|
||||||
|
let result = insert_value(stream_value, &mut insert_statement);
|
||||||
|
|
||||||
|
insert_statement
|
||||||
|
.finalize()
|
||||||
|
.map_err(|e| ShellError::GenericError {
|
||||||
|
error: "Failed to finalize SQLite prepared statement".into(),
|
||||||
|
msg: e.to_string(),
|
||||||
|
span: None,
|
||||||
|
help: None,
|
||||||
|
inner: Vec::new(),
|
||||||
|
})?;
|
||||||
|
|
||||||
|
tx.commit().map_err(|e| ShellError::GenericError {
|
||||||
|
error: "Failed to commit SQLite transaction".into(),
|
||||||
msg: e.to_string(),
|
msg: e.to_string(),
|
||||||
span: None,
|
span: None,
|
||||||
help: None,
|
help: None,
|
||||||
inner: Vec::new(),
|
inner: Vec::new(),
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
tx.commit().map_err(|e| ShellError::GenericError {
|
result
|
||||||
error: "Failed to commit SQLite transaction".into(),
|
|
||||||
msg: e.to_string(),
|
|
||||||
span: None,
|
|
||||||
help: None,
|
|
||||||
inner: Vec::new(),
|
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
Ok(Value::nothing(span))
|
Ok(Value::nothing(span))
|
||||||
|
Loading…
Reference in New Issue
Block a user