62 lines
1.5 KiB
Rust
62 lines
1.5 KiB
Rust
|
mod sql;
|
||
|
|
mod config;
|
||
|
|
use rocket::{get, launch, routes, Route};
|
||
|
|
use rocket::http::{ContentType, Status};
|
||
|
|
use rocket::serde::{ Serialize};
|
||
|
|
use rocket::response::status;
|
||
|
|
use std::sync::{Arc, Mutex};
|
||
|
|
use once_cell::sync::Lazy;
|
||
|
|
use rocket::serde::json::Json;
|
||
|
|
use tokio::sync::Mutex as TokioMutex;
|
||
|
|
use tokio_postgres::types::ToSql;
|
||
|
|
|
||
|
|
// 获取数据库连接
|
||
|
|
static GLOBAL_SQL: Lazy<Arc<TokioMutex<Option<Box<dyn sql::Database >>>>>
|
||
|
|
= Lazy::new(|| Arc::new(TokioMutex::new(None)));
|
||
|
|
|
||
|
|
// 获取数据库连接
|
||
|
|
async fn initialize_sql() {
|
||
|
|
let sql_instance = sql::loading().await;
|
||
|
|
let mut lock = GLOBAL_SQL.lock().await;
|
||
|
|
*lock = sql_instance;
|
||
|
|
}
|
||
|
|
|
||
|
|
// 网站初始化
|
||
|
|
#[get("/install")]
|
||
|
|
fn install() -> status::Custom<()> {
|
||
|
|
status::Custom(Status::Ok, ())
|
||
|
|
}
|
||
|
|
// sql查询
|
||
|
|
#[derive(Serialize)]
|
||
|
|
struct SSql{
|
||
|
|
key:String,
|
||
|
|
value:String,
|
||
|
|
}
|
||
|
|
#[get("/sql")]
|
||
|
|
async fn ssql() -> status::Custom<Json<Vec<SSql>>> {
|
||
|
|
let sql_instance=GLOBAL_SQL.lock().await;
|
||
|
|
let sql =sql_instance.as_ref().unwrap();
|
||
|
|
let query = "SELECT * FROM info";
|
||
|
|
let params: &[&(dyn ToSql + Sync)] = &[];
|
||
|
|
let data = sql.query(query, params).await.expect("查询数据失败");
|
||
|
|
let mut vec = Vec::new();
|
||
|
|
for row in data {
|
||
|
|
let key=row.get(0);
|
||
|
|
let value=row.get(1);
|
||
|
|
vec.push(SSql{
|
||
|
|
key,
|
||
|
|
value,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
status::Custom(Status::Ok, Json(vec))
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#[launch]
|
||
|
|
async fn rocket() -> _ {
|
||
|
|
initialize_sql().await;
|
||
|
|
rocket::build()
|
||
|
|
.mount("/api", routes![install,ssql])
|
||
|
|
}
|