Вот код, который я использую для вставки данных в базу данных Postgres с использованием ящика postgres
(к сожалению, отсутствует на Rust Playground):
со следующим Cargo.toml:
[package]
name = "suff-import"
version = "0.1.0"
authors = ["greg"]
[dependencies]
csv = "1"
postgres = "0.15"
uuid = "0.7"
Файл main.rs
:
extern crate csv;
extern crate postgres;
extern crate uuid;
use uuid::Uuid;
use postgres::{Connection, TlsMode};
use std::error::Error;
use std::io;
use csv::StringRecord;
struct Stuff {
stuff_id: Uuid,
created_at: String,
description: String,
is_something: bool,
}
impl Stuff {
fn from_string_record(record: StringRecord) -> Stuff {
Stuff {
stuff_id: Uuid::parse_str(record.get(0).unwrap()).unwrap(),
created_at: record.get(1).unwrap().to_string(),
description: record.get(2).unwrap().to_string(),
is_something: record.get(3).unwrap().to_string().parse::<i32>().unwrap() == 2,
}
}
}
fn save_row(dbcon: &Connection, stuff: Stuff) -> Result<(), Box<Error>> {
dbcon.execute(
"insert into public.stuff (stuff_id, created_at, description, is_something) values ($1::uuid, $2, $3, $4)",
&[&format!("{}", &stuff.stuff_id).as_str(), &stuff.created_at, &stuff.description, &stuff.is_something]
)?;
Ok(())
}
fn import() -> Result<(), Box<Error>> {
let mut reader = csv::Reader::from_reader(io::stdin());
let dbcon = Connection::connect("postgres://gregoire@10.129.198.251/gregoire", TlsMode::None).unwrap();
for result in reader.records() {
let record = result?;
println!(".");
save_row(&dbcon, Stuff::from_string_record(record))?;
}
Ok(())
}
fn main() {
if let Err(error) = import() {
println!("There were some errors: {}", error);
std::process::exit(1);
}
}
Программа компилируется, но закрывается при запуске с сообщением об ошибке:
./target/debug/suff-import <<EOF
stuff_id,created_at,description,is_something
5252fff5-d04f-4e0f-8d3e-27da489cf40c,"2019-03-15 16:39:32","This is a description",1
EOF
.
There were some errors: type conversion error: cannot convert to or from a Postgres value of type `uuid`
Я проверил превращение UUID в &str
с макросом format!
, поскольку Postgres должен неявно выполнять преобразование в UUID, но он не работает (то же сообщение об ошибке).Затем я добавил явный $1::uuid
в запрос Postgres, но проблема все еще возникает.