imphnen-backend-service/imphnen-libs/src/surrealdb/mod.rs

use crate::enviroment::ENV;
use surrealdb::engine::any;
use surrealdb::engine::local::{Db, Mem};
use surrealdb::opt::auth::Root;
use surrealdb::{Result, Surreal};

pub type SurrealWsClient = Surreal<any::Any>;
pub type SurrealMemClient = Surreal<Db>;

pub mod resource;
pub use resource::*;

pub async fn surrealdb_init_ws() -> Result<Surreal<any::Any>> {
	let env = &ENV;
	let db = any::connect(&env.surrealdb_url).await?;

	db.signin(Root {
		username: &env.surrealdb_username,
		password: &env.surrealdb_password,
	})
	.await?;
	db.use_ns(env.surrealdb_namespace.clone())
		.use_db(env.surrealdb_dbname.clone())
		.await?;
	Ok(db)
}

pub async fn surrealdb_init_mem() -> Result<SurrealMemClient> {
	let env = &ENV;
	let db = Surreal::new::<Mem>(()).await?;
	db.use_ns(&env.surrealdb_namespace)
		.use_db(&env.surrealdb_dbname)
		.await?;
	Ok(db)
}