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

后青春期的诗go2025-05-26 22:27
登录处理代码

在src目录新建login_handle.rs文件

目录结构如下：

project

|---src

|---params //封装参数结构体模块

|---req.rs //封装请求参数结构体

|---resp.rs //封装返回数据结构体

|---result_parse.rs //解析返回参数结构体

|---mod.rs //导出模块

|---common //公共模块

|---db.rs //构造数据库连接池

|---jwt.rs //处理token

|---mistake.rs //处理错误

|---util.rs //工具函数

|---mod.rs //导出模块

|---logic //业务逻辑模块

|---wechat.rs //处理业务

|---mod.rs //导出模块

|---login_handle.rs //登录处理函数

login_handle.rs代码如下：

rust 复制代码 
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(),
                })
            }
        }
    }

  
}
CRUD与上传文件处理代码

在src目录新建xxx_handle.rs文件

目录结构如下：

project

|---src

|---params //封装参数结构体模块

|---req.rs //封装请求参数结构体

|---resp.rs //封装返回数据结构体

|---result_parse.rs //解析返回参数结构体

|---mod.rs //导出模块

|---common //公共模块

|---db.rs //构造数据库连接池

|---jwt.rs //处理token

|---mistake.rs //处理错误

|---util.rs //工具函数

|---mod.rs //导出模块

|---logic //业务逻辑模块

|---wechat.rs //处理业务

|---mod.rs //导出模块

|---login_handle.rs //登录处理函数

|---xxx_handle.rs //CRUD与上传文件处理函数示例

xxx_handle.rs处理函数代码示例：

rust 复制代码 
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())
        }
    }  
    
}
相关推荐
AdSet聚合广告2 小时前
APP广告变现，开发者如何判断对接的广告SDK安全合规？
大数据·后端·算法·安全·uni-app
不二狗2 小时前
每日算法 -【Swift 算法】实现回文数判断！
开发语言·算法·swift
csdn_aspnet3 小时前
Java 程序求圆弧段的面积(Program to find area of a Circular Segment)
java·开发语言
野犬寒鸦4 小时前
Redis核心数据结构操作指南：字符串、哈希、列表详解
数据结构·数据库·redis·后端·缓存·哈希算法
无敌小肥0075 小时前
Springboot 整合 WebSocket 实现聊天室功能
spring boot·后端·websocket
进击的_鹏5 小时前
【C++】红黑树的实现
开发语言·c++
hdsoft_huge5 小时前
VUE npm ERR! code ERESOLVE， npm ERR! ERESOLVE could not resolve， 错误有效解决
vue.js·rust·npm
无心水5 小时前
【后端高阶面经：MongoDB篇】41、MongoDB 是怎么做到高可用的？
java·开发语言·mongodb·java面试·高可用·后端高阶面经·后端工程师的高阶面经
无心水5 小时前
【后端高阶面经：MongoDB篇】40、怎么优化MongoDB的查询性能？
java·开发语言·mongodb·java面试·后端高阶面经·后端工程师的高阶面经·java高阶面经
信息化未来5 小时前
pyｔｈｏｎ 生成复杂表格，自动分页等功能
开发语言·数据结构·python
热门推荐
01KGG转MP3工具|非KGM文件|解密音频02YOLOv8入门 | 重要性能衡量指标、训练结果评价及分析及影响mAP的因素【发论文关注的指标】03从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑04【SpeedAI科研小助手】2分钟极速解决知网维普重复率、AIGC率过高，一键全文降！文件格式不变，公式都保留的！05VMware虚拟机安装Win7专业版保姆级教程（附镜像包）06DeepSeek各版本说明与优缺点分析07头歌的数据库的第二次作业的答案08组基轨迹建模 GBTM的介绍与实现（Stata 或 R）09Coze扣子平台完整体验和实践（附国内和国际版对比）10【操作系统】Linux之网络编程(UDP)（头歌作业）