1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19

use postgres::{Client, Error};

pub fn insert_record(
    client: &mut Client,
    table: String,
    columns: String,
    values: String,
) -> Result<i32, Error> {
    let query = format!(
        "INSERT INTO {} ({}) VALUES ({}) RETURNING id",
        table, columns, values
    );

    let id = match client.query_one(&query[..], &[]) {
        Ok(data) => data,
        Err(why) => panic!("{}", why),
    };
    Ok(id.get(0))
}