基于Rust语言的Rocket框架和Sqlx库开发WebAPI项目记录（五）

发表于 2025-7-27 06:10:53

马上注册，结交更多好友，享用更多功能，让你轻松玩转社区。

您需要登录才可以下载或查看，没有账号？立即注册

×

use crate::common::db;
use crate::params::req;
use crate::params::req::Userparam;
use crate::params::resp;
use crate::common::utils::get_uuid;
use crate::common::jwt::get_token;
use rocket::get;
use rocket_db_pools::Connection;
use rocket_db_pools::sqlx::{self,Row};
use rocket::serde::json::{Value,Json,json};
//注册
pub async fn register(mut db: Connection<db::Prodb>,regpram:Json<req::Userparam>)->Json<resp::Regresp> {
let userid=get_uuid();
let sql = sqlx::query("insert into users (uname,id) values ($1,$2)")
.bind(&regpram.uname)
.bind(&userid)
.execute(& mut **db).await;
match sql {
Ok(_result) => {
Json(
resp::Regresp { code:1,
message:"Success".to_string(),
uid:userid
})
},
Err(err) => {
Json(
resp::Regresp { code:0,
message:err.to_string(),
uid:"null".to_owned()
})
}
}
}
//获取用户id
pub async fn get_id(mut db: Connection<db::Prodb>,uname:String)->String {
let uid = sqlx::query("select id from users where uname=$1")
.bind(&uname)
.fetch_one(&mut **db)
.await.unwrap().get(0);
return uid;
}
//验证
#[get("/verify/<uname>")]
pub async fn verify(mut db: Connection<db::Prodb>,uname:String)->Value {
let count = sqlx::query("select count(*) from users where uname=$1")
.bind(&uname)
.fetch_one(&mut **db)
.await;
match count{
Ok(a)=>{ //非空行，有数据
let num:i64 =a.get(0);
if num>0{
let id=get_id(db, uname.clone()).await;
let user=Userparam{
uname:uname.clone(),
id:id
};
get_token(Json(user)).await
}else{
let userid=get_uuid();
let regparam=Userparam{
uname:uname.clone(),
id:userid
};
let reg:Json<resp::Regresp>= register(db, Json(regparam)).await;
let uid=&reg.uid;
if reg.code==1{
let user=Userparam{
uname:uname,
id:uid.to_owned()
};
get_token(Json(user)).await
}else{
json!(
resp::Resp { code:0,
message:"认证失败！".to_string(),
})
}
}
},
Err(_b)=>{ //空行，无数据
let userid=get_uuid();
let regparam=Userparam{
uname:uname.clone(),
id:userid
};
let reg:Json<resp::Regresp>= register(db, Json(regparam)).await;
let uid=&reg.uid;
if reg.code==1{
let user=Userparam{
uname:uname,
id:uid.to_owned()
};
get_token(Json(user)).await
}else{
json!(
resp::Resp { code:0,
message:"认证失败！".to_string(),
})
}
}
}
}

复制代码

use rocket::{post,get};
use rocket_db_pools::Connection;
use rocket_db_pools::sqlx::{self,Row};
use rocket::serde::json::Json;
use sqlx::postgres::PgRow;
use rocket::form::Form;
use std::collections::HashMap;
use std::env;
use crate::logic::wechat::{ocr,parse_ocr,del_img};
use crate::common::db;
use crate::params::req::{self, UploadImg};
use crate::params::resp;
use crate::common::utils::get_uuid;
use crate::common::jwt::Token;
//上传文件处理
#[post("/upload",data="<img>")]
pub async fn upload_file(mut img:Form<UploadImg<'_>>)->Result<Json<HashMap<String, String>>,std::io::Error>{
//处理图片上传逻辑
//获取项目部署路径
let exe_path=env::current_exe().unwrap();
let mut path=exe_path
.parent()
.ok_or_else(|| std::io::Error::new(std::io::ErrorKind::Other, "No parent directory found"))
.unwrap()
.to_path_buf();
//拼接路径
path.push("tmp");
path.push(img.name);
let url=path.to_str().unwrap();
//将临时文件移动到路径
img.file.persist_to(url).await.unwrap();
//图片服务器
let http_img=String::from("https://www.xxxyyy.cn/images/")+img.name;
//返回识别数据
let datas= ocr(&http_img).await ;
if let Ok(data)=datas{
let parser =parse_ocr(&data).await;
return Ok(Json(parser));
}else{
println!("OCR文本识别失败，请重新再试");
//删除临时图片
del_img(String::from(url)).await.unwrap();
let resp=[("code".to_string(),"0".to_string()),("msg".to_string(),"识别失败".to_string())];
return Ok(Json(HashMap::from(resp)));
}
}
//插入数据
#[post("/insert",data="<liverpram>")]
pub async fn insert_liver(mut db: Connection<db::Prodb>,liverpram:Json<req::Liverparam<'_>>,auth:Token)->Json<resp::Resp> {
//获取uuid
let liverid=get_uuid();
//获取从token中解析出来的用户信息
let uname=auth.uname;
let sql = sqlx::query("insert into livers (livervalue,checktime,uid,id) values ($1,$2,$3,$4)")
.bind(liverpram.livervalue)
.bind(liverpram.checktime)
.bind(uname)
.bind(liverid)
.execute(&mut **db).await;
match sql {
Ok(_result) => {
Json(
resp::Resp { code:1,
message:"Success".to_string(),
})
},
Err(err) => {
Json(
resp::Resp { code:0,
message:err.to_string(),
})
}
}
}
//删除数据
#[get("/delete/<id>")]
pub async fn delete_liver(mut db: Connection<db::Prodb>,id:String,_auth:Token)->Json<resp::Resp> {
let sql = sqlx::query("delete from livers where id=$1")
.bind(id)
.execute(&mut **db).await;
match sql {
Ok(_result) => {
Json(
resp::Resp { code:1,
message:"Success".to_string(),
})
},
Err(err) => {
Json(
resp::Resp { code:0,
message:err.to_string(),
})
}
}
}
//返回数据
//返回liver数组--倒序
#[get("/list")]
pub async fn list_liver(mut db: Connection<db::Prodb>,auth:Token) -> Json<Vec<resp::Listliver>> {
let uname=auth.uname;
let rows =sqlx::query("SELECT id,uid,livervalue,checktime FROM livers where uid=$1 order by checktime desc limit 7").bind(uname);
let list:Result<Vec<resp::Listliver>,sqlx::Error>=rows.map(|r:PgRow|
resp::Listliver{
livervalue:r.get("livervalue"),
checktime:r.get("checktime"),
uid:r.get("uid"),
id:r.get("id")
}
)
.fetch_all(&mut **db).await;
match list {
Ok(data)=>{
Json(data)
},
Err(_err)=>{
Json(Vec::new())
}
}
}
//返回AFP数组--正序
#[get("/listasc")]
pub async fn list_liver_asc(mut db: Connection<db::Prodb>,auth:Token) -> Json<Vec<resp::Listliver>> {
let uname=auth.uname;
let rows =sqlx::query("select id,uid,livervalue,checktime from (SELECT id,uid,livervalue,checktime FROM livers where uid=$1 order by checktime desc limit 7) order by checktime asc").bind(uname);
let list:Result<Vec<resp::Listliver>,sqlx::Error>=rows.map(|r:PgRow|
resp::Listliver{
livervalue:r.get("livervalue"),
checktime:r.get("checktime"),
uid:r.get("uid"),
id:r.get("id")
}
)
.fetch_all(&mut **db).await;
match list {
Ok(data)=>{
Json(data)
},
Err(_err)=>{
Json(Vec::new())
}
}
}
//返回AFP数组--倒序
#[get("/listall")]
pub async fn list_liver_all(mut db: Connection<db::Prodb>,auth:Token) -> Json<Vec<resp::Listliver>> {
let uname=auth.uname;
let rows =sqlx::query("SELECT id,uid,livervalue,checktime FROM livers where uid=$1 order by checktime desc").bind(uname);
let list:Result<Vec<resp::Listliver>,sqlx::Error>=rows.map(|r:PgRow|
resp::Listliver{
livervalue:r.get("livervalue"),
checktime:r.get("checktime"),
uid:r.get("uid"),
id:r.get("id")
}
)
.fetch_all(&mut **db).await;
match list{
Ok(data)=>{
Json(data)
},
Err(_err)=>{
Json(Vec::new())
}
}
}
//返回AFP数组--正序
#[get("/listallasc")]
pub async fn list_liver_all_asc(mut db: Connection<db::Prodb>,auth:Token) -> Json<Vec<resp::Listliver>> {
let uname=auth.uname;
let rows =sqlx::query("SELECT id,uid,livervalue,checktime FROM livers where uid=$1 order by checktime asc").bind(uname);
let list:Result<Vec<resp::Listliver>,sqlx::Error>=rows.map(|r:PgRow|
resp::Listliver{
livervalue:r.get("livervalue"),
checktime:r.get("checktime"),
uid:r.get("uid"),
id:r.get("id")
}
)
.fetch_all(&mut **db).await;
match list{
Ok(data)=>{
Json(data)
},
Err(_err)=>{
Json(Vec::new())
}
}
}

复制代码

基于Rust语言的Rocket框架和Sqlx库开发WebAPI项目记录（五）

马上注册，结交更多好友，享用更多功能，让你轻松玩转社区。

浏览过的版块

万有斥力