Stash proxy_surrealdb_example for now
This commit is contained in:
Chris Tsang
parent
955bbcbc12
commit
5539e7c5ad
@ -1,27 +0,0 @@
|
|||||||
[package]
|
|
||||||
name = "sea-orm-proxy-surrealdb-example"
|
|
||||||
version = "0.1.0"
|
|
||||||
authors = ["Langyo <langyo.china@gmail.com>"]
|
|
||||||
edition = "2021"
|
|
||||||
publish = false
|
|
||||||
|
|
||||||
[workspace]
|
|
||||||
|
|
||||||
[dependencies]
|
|
||||||
async-std = { version = "1.12", features = ["attributes", "tokio1"] }
|
|
||||||
serde_json = { version = "1" }
|
|
||||||
serde = { version = "1" }
|
|
||||||
futures = { version = "0.3" }
|
|
||||||
async-stream = { version = "0.3" }
|
|
||||||
futures-util = { version = "0.3" }
|
|
||||||
|
|
||||||
sqlparser = "0.40"
|
|
||||||
sea-orm = { path = "../../", features = [
|
|
||||||
"proxy",
|
|
||||||
"debug-print",
|
|
||||||
] }
|
|
||||||
surrealdb = { version = "1", features = ["kv-mem"] }
|
|
||||||
|
|
||||||
[dev-dependencies]
|
|
||||||
smol = { version = "1.2" }
|
|
||||||
smol-potat = { version = "1.1" }
|
|
||||||
@ -1,7 +0,0 @@
|
|||||||
# SeaORM Proxy Demo for SurrealDB
|
|
||||||
|
|
||||||
Run this demo for [SurrealDB](https://surrealdb.com/) with the following command:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
cargo run
|
|
||||||
```
|
|
||||||
@ -1 +0,0 @@
|
|||||||
pub mod post;
|
|
||||||
@ -1,17 +0,0 @@
|
|||||||
use sea_orm::entity::prelude::*;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Deserialize, Serialize)]
|
|
||||||
#[sea_orm(table_name = "posts")]
|
|
||||||
pub struct Model {
|
|
||||||
#[sea_orm(primary_key)]
|
|
||||||
pub id: String,
|
|
||||||
|
|
||||||
pub title: String,
|
|
||||||
pub text: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
||||||
pub enum Relation {}
|
|
||||||
|
|
||||||
impl ActiveModelBehavior for ActiveModel {}
|
|
||||||
@ -1,271 +0,0 @@
|
|||||||
//! Proxy connection example.
|
|
||||||
|
|
||||||
#![deny(missing_docs)]
|
|
||||||
|
|
||||||
mod entity;
|
|
||||||
|
|
||||||
use std::{
|
|
||||||
collections::BTreeMap,
|
|
||||||
sync::{Arc, Mutex},
|
|
||||||
};
|
|
||||||
|
|
||||||
use sea_orm::{
|
|
||||||
ActiveValue::Set, Database, DbBackend, DbErr, EntityTrait, ProxyDatabaseTrait, ProxyExecResult,
|
|
||||||
ProxyRow, Statement,
|
|
||||||
};
|
|
||||||
use surrealdb::{
|
|
||||||
engine::local::{Db, Mem},
|
|
||||||
Surreal,
|
|
||||||
};
|
|
||||||
|
|
||||||
use entity::post::{ActiveModel, Entity};
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
struct ProxyDb {
|
|
||||||
mem: Surreal<Db>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ProxyDatabaseTrait for ProxyDb {
|
|
||||||
fn query(&self, statement: Statement) -> Result<Vec<ProxyRow>, DbErr> {
|
|
||||||
println!("SQL query: {:?}", statement);
|
|
||||||
let mut ret = async_std::task::block_on(async {
|
|
||||||
// Surrealdb's grammar is not compatible with sea-orm's
|
|
||||||
// so we need to remove the extra clauses
|
|
||||||
// from "SELECT `from`.`col` FROM `from` WHERE `from`.`col` = xx"
|
|
||||||
// to "SELECT `col` FROM `from` WHERE `col` = xx"
|
|
||||||
|
|
||||||
use sqlparser::ast::{Expr, SelectItem, SetExpr, TableFactor};
|
|
||||||
use sqlparser::dialect::GenericDialect;
|
|
||||||
use sqlparser::parser::Parser;
|
|
||||||
|
|
||||||
let dialect = GenericDialect {};
|
|
||||||
let mut ast = Parser::parse_sql(&dialect, statement.sql.as_str()).unwrap();
|
|
||||||
match &mut ast[0] {
|
|
||||||
sqlparser::ast::Statement::Query(query) => match &mut *query.body {
|
|
||||||
SetExpr::Select(body) => {
|
|
||||||
body.projection.iter_mut().for_each(|item| {
|
|
||||||
match item {
|
|
||||||
SelectItem::UnnamedExpr(expr) => {
|
|
||||||
match expr {
|
|
||||||
Expr::CompoundIdentifier(idents) => {
|
|
||||||
// Remove the head of the identifier
|
|
||||||
// e.g. `from`.`col` -> `col`
|
|
||||||
let ident = idents.pop().unwrap();
|
|
||||||
*expr = Expr::Identifier(ident);
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
}
|
|
||||||
});
|
|
||||||
body.from.iter_mut().for_each(|item| {
|
|
||||||
match &mut item.relation {
|
|
||||||
TableFactor::Table { name, .. } => {
|
|
||||||
// Remove the head of the identifier
|
|
||||||
// e.g. `from`.`col` -> `col`
|
|
||||||
let ident = name.0.pop().unwrap();
|
|
||||||
name.0 = vec![ident];
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
},
|
|
||||||
_ => todo!(),
|
|
||||||
};
|
|
||||||
|
|
||||||
let statement = &ast[0];
|
|
||||||
let sql = statement.to_string();
|
|
||||||
println!("SQL: {}", sql);
|
|
||||||
self.mem.query(sql).await
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
// Convert the result to sea-orm's format
|
|
||||||
let ret: Vec<serde_json::Value> = ret.take(0).unwrap();
|
|
||||||
println!("SQL query result: {}", serde_json::to_string(&ret).unwrap());
|
|
||||||
let ret = ret
|
|
||||||
.iter()
|
|
||||||
.map(|row| {
|
|
||||||
let mut map = serde_json::Map::new();
|
|
||||||
for (k, v) in row.as_object().unwrap().iter() {
|
|
||||||
if k == "id" {
|
|
||||||
// Get `tb` and `id` columns from surrealdb
|
|
||||||
// and convert them to sea-orm's `id`
|
|
||||||
let tb = v.as_object().unwrap().get("tb").unwrap().to_string();
|
|
||||||
let id = v
|
|
||||||
.as_object()
|
|
||||||
.unwrap()
|
|
||||||
.get("id")
|
|
||||||
.unwrap()
|
|
||||||
.get("String")
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
// Remove the quotes
|
|
||||||
let tb = tb.to_string().replace("\"", "");
|
|
||||||
let id = id.to_string().replace("\"", "");
|
|
||||||
|
|
||||||
map.insert("id".to_owned(), format!("{}:{}", tb, id).into());
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
map.insert(k.to_owned(), v.to_owned());
|
|
||||||
}
|
|
||||||
serde_json::Value::Object(map)
|
|
||||||
})
|
|
||||||
.map(|v: serde_json::Value| {
|
|
||||||
let mut ret: BTreeMap<String, sea_orm::Value> = BTreeMap::new();
|
|
||||||
for (k, v) in v.as_object().unwrap().iter() {
|
|
||||||
ret.insert(
|
|
||||||
k.to_owned(),
|
|
||||||
match v {
|
|
||||||
serde_json::Value::Bool(b) => {
|
|
||||||
sea_orm::Value::TinyInt(if *b { Some(1) } else { Some(0) })
|
|
||||||
}
|
|
||||||
serde_json::Value::Number(n) => {
|
|
||||||
if n.is_i64() {
|
|
||||||
sea_orm::Value::BigInt(Some(n.as_i64().unwrap()))
|
|
||||||
} else if n.is_u64() {
|
|
||||||
sea_orm::Value::BigUnsigned(Some(n.as_u64().unwrap()))
|
|
||||||
} else if n.is_f64() {
|
|
||||||
sea_orm::Value::Double(Some(n.as_f64().unwrap()))
|
|
||||||
} else {
|
|
||||||
unreachable!()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
serde_json::Value::String(s) => {
|
|
||||||
sea_orm::Value::String(Some(Box::new(s.to_owned())))
|
|
||||||
}
|
|
||||||
_ => sea_orm::Value::Json(Some(Box::new(v.to_owned()))),
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
ProxyRow { values: ret }
|
|
||||||
})
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
Ok(ret)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn execute(&self, statement: Statement) -> Result<ProxyExecResult, DbErr> {
|
|
||||||
async_std::task::block_on(async {
|
|
||||||
if let Some(values) = statement.values {
|
|
||||||
// Replace all the '?' with the statement values
|
|
||||||
use sqlparser::ast::{Expr, Value};
|
|
||||||
use sqlparser::dialect::GenericDialect;
|
|
||||||
use sqlparser::parser::Parser;
|
|
||||||
|
|
||||||
let dialect = GenericDialect {};
|
|
||||||
let mut ast = Parser::parse_sql(&dialect, statement.sql.as_str()).unwrap();
|
|
||||||
match &mut ast[0] {
|
|
||||||
sqlparser::ast::Statement::Insert {
|
|
||||||
table_name,
|
|
||||||
columns,
|
|
||||||
source,
|
|
||||||
..
|
|
||||||
} => {
|
|
||||||
// Replace the table name's quote style
|
|
||||||
table_name.0[0].quote_style = Some('`');
|
|
||||||
|
|
||||||
// Replace all the column names' quote style
|
|
||||||
for item in columns.iter_mut() {
|
|
||||||
item.quote_style = Some('`');
|
|
||||||
}
|
|
||||||
|
|
||||||
// Convert the values to sea-orm's format
|
|
||||||
if let Some(obj) = source {
|
|
||||||
match &mut *obj.body {
|
|
||||||
sqlparser::ast::SetExpr::Values(obj) => {
|
|
||||||
for (mut item, val) in
|
|
||||||
obj.rows[0].iter_mut().zip(values.0.iter())
|
|
||||||
{
|
|
||||||
match &mut item {
|
|
||||||
Expr::Value(item) => {
|
|
||||||
*item = match val {
|
|
||||||
sea_orm::Value::String(val) => {
|
|
||||||
Value::SingleQuotedString(match val {
|
|
||||||
Some(val) => val.to_string(),
|
|
||||||
None => "".to_string(),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
sea_orm::Value::BigInt(val) => Value::Number(
|
|
||||||
val.unwrap_or(0).to_string(),
|
|
||||||
false,
|
|
||||||
),
|
|
||||||
_ => todo!(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => todo!(),
|
|
||||||
}
|
|
||||||
|
|
||||||
let statement = &ast[0];
|
|
||||||
let sql = statement.to_string();
|
|
||||||
println!("SQL: {}", sql);
|
|
||||||
self.mem.query(sql).await
|
|
||||||
} else {
|
|
||||||
self.mem.query(statement.sql).await
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
Ok(ProxyExecResult {
|
|
||||||
last_insert_id: 1,
|
|
||||||
rows_affected: 1,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_std::main]
|
|
||||||
async fn main() {
|
|
||||||
let mem = Surreal::new::<Mem>(()).await.unwrap();
|
|
||||||
mem.use_ns("test").use_db("post").await.unwrap();
|
|
||||||
|
|
||||||
let db = Database::connect_proxy(
|
|
||||||
DbBackend::MySql,
|
|
||||||
Arc::new(Mutex::new(Box::new(ProxyDb { mem }))),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
println!("Initialized");
|
|
||||||
|
|
||||||
let data = ActiveModel {
|
|
||||||
title: Set("Homo".to_owned()),
|
|
||||||
text: Set("いいよ、来いよ".to_owned()),
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
Entity::insert(data).exec(&db).await.unwrap();
|
|
||||||
let data = ActiveModel {
|
|
||||||
title: Set("Homo".to_owned()),
|
|
||||||
text: Set("そうだよ".to_owned()),
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
Entity::insert(data).exec(&db).await.unwrap();
|
|
||||||
let data = ActiveModel {
|
|
||||||
title: Set("Homo".to_owned()),
|
|
||||||
text: Set("悔い改めて".to_owned()),
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
Entity::insert(data).exec(&db).await.unwrap();
|
|
||||||
|
|
||||||
let list = Entity::find().all(&db).await.unwrap().to_vec();
|
|
||||||
println!("Result: {:?}", list);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
#[smol_potat::test]
|
|
||||||
async fn try_run() {
|
|
||||||
crate::main()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
x
Reference in New Issue
Block a user