Вернуть одиночную запись, созданную в PostgreSQL с использованием Rocket и Diesel (Rust) - PullRequest
Новая
       39

Вернуть одиночную запись, созданную в PostgreSQL с использованием Rocket и Diesel (Rust)

0 голосов
/ 02 февраля 2020

Я пытаюсь создать простое веб-приложение на основе учебных пособий: один , два и три . Все они используют немного разные подходы и были написаны для более старых версий используемых библиотек, поэтому текущие API-интерфейсы Diesel и Rocket немного отличаются. Я хочу в ответ на POST-запрос вернуть созданную запись или идентификатор записи. На данный момент я могу вернуть только список записей - список одной записи (попытки вернуть одну запись, на основе Vec.first() не компилируется).

Car go .toml : 

[dependencies]
# chrono is needed with serialization/deserialisation options
chrono = { version = "0.4.10", features = ["serde"] }
diesel = { version = "1.4.3", features = ["postgres", "chrono", "network-address"] }
ipnetwork = "0.15.1"
dotenv = "0.9.0"

# Lazy static initialization
lazy_static = "1.3.0"

# Rocket Webserver
rocket = "0.4.2"
rocket_contrib = {version = "0.4.2", features = ["json", "diesel_postgres_pool"]}

# Connection pooling
r2d2 = "0.8.7"
r2d2-diesel = "1.0.0"

# Serialization/Deserialization
serde = "1.0.104"
serde_json = "1.0.44"
serde_derive = "1.0.104"

models.rs : 

#![allow(proc_macro_derive_resolution_fallback)]
use crate::schema::bs_date_forecast;

use chrono::{NaiveDate, NaiveDateTime};
use ipnetwork::IpNetwork;
use diesel::prelude::*;

#[derive(Queryable, Insertable, Serialize, Deserialize)]
#[table_name="bs_date_forecast"]
pub struct BsDateForecast {
    pub id: Option<i32>,
    pub predicted_date: NaiveDate,
    pub seer_ip: IpNetwork,
    pub created_timestamp: NaiveDateTime,  // Defined automatically at the database level
}

impl BsDateForecast {

    // Would like to return the record (BsDateForecast) instead, but don't know how
    pub fn create(forecast: BsDateForecast,
                  conn: &PgConnection) -> QueryResult<Vec<BsDateForecast>> {
        diesel::insert_into(bs_date_forecast::table)
            .values(&forecast)
            .get_results(conn)
    }

}

schema.rs : 

table! {
    bs_date_forecast (id) {
        id -> Nullable<Int4>,
        predicted_date -> Date,
        seer_ip -> Inet,
        created_timestamp -> Timestamptz,
    }
}

rout.rs : 

use diesel::{self, prelude::*};

use rocket_contrib::json::{Json, JsonValue};
use rocket::response::status;

use crate::models::{BsDateForecast};
use crate::schema;
use crate::DbConn;
use serde_json::Value;



#[post("/predictions", format = "application/json", data = "<prediction>")]
pub fn create(prediction: Json<BsDateForecast>, conn: DbConn) -> Json<Vec<BsDateForecast>> {
    let insert = BsDateForecast{id: None, ..prediction.into_inner()};
    Json(BsDateForecast::create(insert, &conn))
}

1 Ответ

2 голосов
/ 03 февраля 2020

Это так же просто, как изменить функцию, которая загружает значения из базы данных. 

impl BsDateForecast {

    // Would like to return the record (BsDateForecast) instead, but don't know how
    pub fn create(forecast: BsDateForecast,
                  conn: &PgConnection) -> QueryResult<BsDateForecast> {
        diesel::insert_into(bs_date_forecast::table)
            .values(&forecast)
            .get_result(conn)
    }

}

Подробнее см. Документацию RunQueryDsl (и другие доступные методы)

...