Как сохранить произвольный объект JSON в sqlite с помощью Diesel - PullRequest
Новая
       42

Как сохранить произвольный объект JSON в sqlite с помощью Diesel

0 голосов
/ 07 июня 2019

У меня есть ввод JSON: 

{"key1": "val1", "key2": 1}

и я хочу сохранить его в базе данных sqlite, чтобы позже ответить на какой-либо запрос API с точно таким же значением.

Это моя миграция: 

CREATE TABLE my_table (
    id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
    arbitrary_json TEXT NOT NULL
);

Мой Cargo.toml: 

[package]
name = "diesel_playground"
version = "0.1.0"
authors = ["User <user@example.com>"]
edition = "2018"

[dependencies]
diesel = { version = "1.4" , features = ["sqlite"] }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"

Используя следующий код: 

#[macro_use]
extern crate diesel;

mod schema;

use schema::my_table;
use serde::{Deserialize, Serialize};

use diesel::prelude::*;
use diesel::sqlite::SqliteConnection;
use std::env;

pub fn establish_connection() -> SqliteConnection {
    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
    SqliteConnection::establish(&database_url)
        .expect(&format!("Error connecting to {}", database_url))
}

#[derive(Debug, Deserialize, Serialize, Queryable, Identifiable)]
#[table_name = "my_table"]
struct Item {
    id: i32,
    arbitrary_json: serde_json::Value,
}

#[derive(Debug, Deserialize, Serialize, Insertable, Queryable)]
#[table_name = "my_table"]
struct NewItem {
    arbitrary_json: serde_json::Value,
}

fn main() {
    let my_json = serde_json::json!({
    "key1": "val1",
    "key2": 1
    });

    let new_item = NewItem {
        arbitrary_json: my_json,
    };

    let conn = establish_connection();

    diesel::insert_into(my_table::table)
        .values(&new_item)
        .execute(&conn)
        .expect("Error adding new item");

    let my_item = my_table::table
        .find(1)
        .first::<Item>(&conn)
        .expect("Error selecting id 1");

    assert!(my_item.arbitrary_json == new_item.arbitrary_json);
}

Я получаю следующую ошибку: 

error[E0277]: the trait bound `serde_json::value::Value: diesel::Expression` is not satisfied
  --> src/main.rs:27:41
   |
27 | #[derive(Debug, Deserialize, Serialize, Insertable, Queryable)]
   |                                         ^^^^^^^^^^ the trait `diesel::Expression` is not implemented for `serde_json::value::Value`

Я мог бы создать модель с String представлением JSON и получить From / Into для типа ввода API, но я не хочу вставлять .into() везде в моем коде сейчас. СУХОЕ решение было бы сделать это, как я предложил в прилагаемом коде.

1 Ответ

0 голосов
/ 10 июня 2019

В своем ответе я буду хранить объект JSON в БД в строковом представлении (схема: TEXT).

Для нашего неподдерживаемого типа нам нужны следующие характеристики: ToSql, FromSql, AsExpression и FromSqlRow.

Теперь, поскольку нельзя реализовать признак для типа, поступающего из внешнего ящика, его необходимо заключить в один кортеж элемента: 

struct MyJsonType(serde_json::Value)

Теперь FromSql реализация черты: 

impl FromSql<Text, DB> for MyJsonType {
    fn from_sql(
        bytes: Option<&<diesel::sqlite::Sqlite as Backend>::RawValue>,
    ) -> deserialize::Result<Self> {
        let t = <String as FromSql<Text, DB>>::from_sql(bytes)?;
        Ok(Self(serde_json::from_str(&t)?))
    }
}

И ToSql Черта реализации: 

impl ToSql<Text, DB> for MyJsonType {
    fn to_sql<W: Write>(&self, out: &mut Output<W, DB>) -> serialize::Result {
        let s = serde_json::to_string(&self.0)?;
        <String as ToSql<Text, DB>>::to_sql(&s, out)
    }
}

Теперь остальные черты можно получить с помощью макросов: 

#[derive(AsExpression, Debug, Deserialize, Serialize, FromSqlRow)]
#[sql_type = "Text"]
struct MyJsonType(serde_json::Value);

Теперь можно использовать наш новый тип: 

#[derive(Debug, Deserialize, Serialize, Queryable, Identifiable)]
#[table_name = "my_table"]
struct Item {
    id: i32,
    arbitrary_json: MyJsonType
}
...