170 lines
3.3 KiB
Rust
170 lines
3.3 KiB
Rust
|
#![allow(unused_imports, dead_code)]
|
||
|
|
||
|
use serde::Deserialize;
|
||
|
use serde::Serialize;
|
||
|
use serde_json::json;
|
||
|
use std::borrow::Cow;
|
||
|
use std::ops::Bound;
|
||
|
use surrealdb::opt::auth::Database;
|
||
|
use surrealdb::opt::auth::Jwt;
|
||
|
use surrealdb::opt::auth::Namespace;
|
||
|
use surrealdb::opt::auth::Root;
|
||
|
use surrealdb::opt::auth::Scope;
|
||
|
use surrealdb::opt::PatchOp;
|
||
|
use surrealdb::sql::statements::BeginStatement;
|
||
|
use surrealdb::sql::statements::CommitStatement;
|
||
|
use surrealdb::Surreal;
|
||
|
use ulid::Ulid;
|
||
|
|
||
|
const NS: &str = "test-ns";
|
||
|
const ROOT_USER: &str = "root";
|
||
|
const ROOT_PASS: &str = "root";
|
||
|
|
||
|
#[derive(Debug, Serialize)]
|
||
|
struct Record<'a> {
|
||
|
name: &'a str,
|
||
|
}
|
||
|
|
||
|
#[derive(Debug, Deserialize)]
|
||
|
struct RecordId {
|
||
|
id: String,
|
||
|
}
|
||
|
|
||
|
#[derive(Debug, Deserialize)]
|
||
|
struct RecordName {
|
||
|
name: String,
|
||
|
}
|
||
|
|
||
|
#[derive(Debug, Deserialize, Eq, PartialEq, Ord, PartialOrd)]
|
||
|
struct RecordBuf {
|
||
|
id: String,
|
||
|
name: String,
|
||
|
}
|
||
|
|
||
|
#[derive(Debug, Serialize, Deserialize)]
|
||
|
struct AuthParams<'a> {
|
||
|
email: &'a str,
|
||
|
pass: &'a str,
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "protocol-ws")]
|
||
|
mod ws {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::remote::ws::Client;
|
||
|
use surrealdb::engines::remote::ws::Ws;
|
||
|
|
||
|
async fn new_db() -> Surreal<Client> {
|
||
|
let db = Surreal::new::<Ws>("127.0.0.1:8000").await.unwrap();
|
||
|
db.signin(Root {
|
||
|
username: ROOT_USER,
|
||
|
password: ROOT_PASS,
|
||
|
})
|
||
|
.await
|
||
|
.unwrap();
|
||
|
db
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/auth.rs");
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "protocol-http")]
|
||
|
mod http {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::remote::http::Client;
|
||
|
use surrealdb::engines::remote::http::Http;
|
||
|
|
||
|
async fn new_db() -> Surreal<Client> {
|
||
|
let db = Surreal::new::<Http>("127.0.0.1:8000").await.unwrap();
|
||
|
db.signin(Root {
|
||
|
username: ROOT_USER,
|
||
|
password: ROOT_PASS,
|
||
|
})
|
||
|
.await
|
||
|
.unwrap();
|
||
|
db
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/auth.rs");
|
||
|
include!("api/backup.rs");
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "kv-mem")]
|
||
|
mod mem {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::local::Db;
|
||
|
use surrealdb::engines::local::Mem;
|
||
|
|
||
|
async fn new_db() -> Surreal<Db> {
|
||
|
Surreal::new::<Mem>(()).await.unwrap()
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/backup.rs");
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "kv-rocksdb")]
|
||
|
mod file {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::local::Db;
|
||
|
use surrealdb::engines::local::File;
|
||
|
|
||
|
async fn new_db() -> Surreal<Db> {
|
||
|
let path = format!("/tmp/{}.db", Ulid::new());
|
||
|
Surreal::new::<File>(path.as_str()).await.unwrap()
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/backup.rs");
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "kv-tikv")]
|
||
|
mod tikv {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::local::Db;
|
||
|
use surrealdb::engines::local::TiKv;
|
||
|
|
||
|
async fn new_db() -> Surreal<Db> {
|
||
|
Surreal::new::<TiKv>("127.0.0.1:2379").await.unwrap()
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/backup.rs");
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "kv-fdb")]
|
||
|
mod fdb {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::local::Db;
|
||
|
use surrealdb::engines::local::FDb;
|
||
|
|
||
|
async fn new_db() -> Surreal<Db> {
|
||
|
Surreal::new::<FDb>("/tmp/fdb.cluster").await.unwrap()
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/backup.rs");
|
||
|
}
|
||
|
|
||
|
#[cfg(feature = "protocol-http")]
|
||
|
mod any {
|
||
|
use super::*;
|
||
|
use surrealdb::engines::any::Any;
|
||
|
|
||
|
async fn new_db() -> Surreal<Any> {
|
||
|
let db = surrealdb::engines::any::connect("http://127.0.0.1:8000").await.unwrap();
|
||
|
db.signin(Root {
|
||
|
username: ROOT_USER,
|
||
|
password: ROOT_PASS,
|
||
|
})
|
||
|
.await
|
||
|
.unwrap();
|
||
|
db
|
||
|
}
|
||
|
|
||
|
include!("api/mod.rs");
|
||
|
include!("api/auth.rs");
|
||
|
include!("api/backup.rs");
|
||
|
}
|