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