使用了logger来进行后端的消息记录

This commit is contained in:
2025-09-30 15:59:15 +08:00
parent 9880124c87
commit 7fa1ae6301
6 changed files with 353 additions and 26 deletions

32
src/config.rs Normal file
View File

@@ -0,0 +1,32 @@
use std::env;
use dotenvy::dotenv;
#[derive(Debug, Clone)]
pub struct Config {
pub database_url: String,
pub jwt_secret: String,
pub wechat_appid: String,
pub wechat_secret: String,
pub server_ports: Vec<u16>,
}
impl Config {
pub fn load() -> Result<Self, String> {
// 加载.env文件
if let Err(e) = dotenv() {
eprintln!("警告: 无法加载.env文件 - {}", e);
}
Ok(Self {
database_url: env::var("DATABASE_URL")
.map_err(|_| "环境变量DATABASE_URL未设置".to_string())?,
jwt_secret: env::var("JWT_SECRET")
.map_err(|_| "环境变量JWT_SECRET未设置".to_string())?,
wechat_appid: env::var("WECHAT_APPID")
.map_err(|_| "环境变量WECHAT_APPID未设置".to_string())?,
wechat_secret: env::var("WECHAT_SECRET")
.map_err(|_| "环境变量WECHAT_SECRET未设置".to_string())?,
server_ports: vec![8080, 3000, 8000, 8888],
})
}
}

View File

@@ -1,6 +1,7 @@
use actix_web::middleware::from_fn;
use actix_web::{App, HttpResponse, HttpServer, Responder, delete, get, post, web};
use include_dir::{Dir, include_dir};
use log::{debug, error, info, warn};
use reqwest::Client;
use sqlx::postgres::PgPool;
mod auth;
@@ -11,13 +12,12 @@ use auth::{generate_token, jwt_middleware};
use db::{create_pool, insert_weather_data};
use models::{ErrorResponse, WeChatApiResponse, WeChatLoginRequest, WeatherData};
use crate::models::Claims;
use crate::models::{Claims, TokenResponse};
// 编译时嵌入整个 static 目录
static STATIC_DIR: Dir = include_dir!("$CARGO_MANIFEST_DIR/static");
static TEMPLATES_DIR: Dir = include_dir!("$CARGO_MANIFEST_DIR/templates");
// 登录的API端点
#[post("/api/login")]
async fn login(
pool: web::Data<PgPool>,
@@ -26,8 +26,12 @@ async fn login(
) -> impl Responder {
// 从环境变量获取微信小程序配置
let appid = match std::env::var("WECHAT_APPID") {
Ok(id) => id,
Ok(id) => {
debug!("成功加载微信小程序appid");
id
}
Err(_) => {
error!("环境变量中未配置WECHAT_APPID");
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "服务器配置错误缺少微信小程序appid".to_string(),
errcode: None,
@@ -37,8 +41,12 @@ async fn login(
};
let secret = match std::env::var("WECHAT_SECRET") {
Ok(secret) => secret,
Ok(secret) => {
debug!("成功加载微信小程序secret");
secret
}
Err(_) => {
error!("环境变量中未配置WECHAT_SECRET");
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "服务器配置错误缺少微信小程序secret".to_string(),
errcode: None,
@@ -52,12 +60,16 @@ async fn login(
"https://api.weixin.qq.com/sns/jscode2session?appid={}&secret={}&js_code={}&grant_type=authorization_code",
appid, secret, req.code
);
debug!("准备调用微信API: {}", url.replace(&secret, "***")); // 日志中隐藏secret
// 调用微信API
let wechat_response = match http_client.get(&url).send().await {
Ok(response) => response,
Ok(response) => {
debug!("微信API请求成功状态码: {}", response.status());
response
}
Err(e) => {
eprintln!("请求微信API失败: {}", e);
error!("微信API请求失败: {}", e);
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "请求微信服务失败".to_string(),
errcode: None,
@@ -68,9 +80,12 @@ async fn login(
// 解析微信API响应
let wechat_data: WeChatApiResponse = match wechat_response.json().await {
Ok(data) => data,
Ok(data) => {
debug!("微信API响应解析成功");
data
}
Err(e) => {
eprintln!("解析微信API响应失败: {}", e);
error!("微信API响应解析失败: {}", e);
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "解析微信响应失败".to_string(),
errcode: None,
@@ -81,6 +96,10 @@ async fn login(
// 检查微信API响应中的错误
if let Some(errcode) = wechat_data.errcode {
warn!(
"微信登录验证失败 - 错误码: {}, 错误信息: {:?}",
errcode, wechat_data.errmsg
);
return HttpResponse::BadRequest().json(ErrorResponse {
error: "微信登录失败".to_string(),
errcode: Some(errcode),
@@ -90,8 +109,14 @@ async fn login(
// 获取openid
let openid = match wechat_data.openid {
Some(id) => id,
Some(id) => {
// 日志中只显示openid的前4位和后4位保护用户隐私
let masked_openid = format!("{}***{}", &id[0..4], &id[id.len() - 4..]);
debug!("获取到用户openid: {}", masked_openid);
id
}
None => {
error!("微信API响应中未包含openid");
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "微信API未返回openid".to_string(),
errcode: None,
@@ -100,7 +125,7 @@ async fn login(
}
};
// 3. 检查用户是否已存在,不存在则创建(显式处理数据库错误)
// 检查用户是否已存在,不存在则创建
let query = r#"
INSERT INTO users (openid, name, type)
VALUES ($1, left($1, 8), 2)
@@ -114,21 +139,22 @@ async fn login(
.await
{
Ok(Some((id,))) => {
// 已存在用户直接获取ID
println!("用户已存在ID: {}", id);
info!("用户已存在用户ID: {}", id);
id
}
Ok(None) => {
// 新用户创建成功,需再次查询ID因ON CONFLICT DO NOTHING不返回值
println!("新用户已创建openid: {}", openid);
// 新用户创建成功查询ID
match sqlx::query_as::<_, (i32,)>("SELECT id FROM users WHERE openid = $1")
.bind(&openid)
.fetch_one(pool.get_ref())
.await
{
Ok((id,)) => id,
Ok((id,)) => {
info!("新用户创建成功用户ID: {}", id);
id
}
Err(e) => {
eprintln!("查询新用户ID失败: {}", e);
error!("新用户创建后查询ID失败: {}", e);
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "创建用户后查询ID失败".to_string(),
errcode: Some(500),
@@ -138,8 +164,7 @@ async fn login(
}
}
Err(e) => {
// 数据库操作失败
eprintln!("用户查询/创建失败: {}", e);
error!("用户查询/创建失败: {}", e);
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "用户信息处理失败".to_string(),
errcode: Some(500),
@@ -148,11 +173,11 @@ async fn login(
}
};
// 4. 生成JWT token(显式处理所有可能的错误)
// 4.1 获取JWT密钥
// 生成JWT token
let jwt_secret = match std::env::var("JWT_SECRET") {
Ok(secret) => secret,
Err(_) => {
error!("环境变量中未配置JWT_SECRET");
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "服务器配置错误缺少JWT_SECRET".to_string(),
errcode: Some(500),
@@ -161,11 +186,26 @@ async fn login(
}
};
// 4.2 编码生成token
let token = generate_token(user_id, &openid, 2, &jwt_secret);
// 生成token
let token = match generate_token(user_id, &openid, 2, &jwt_secret) {
Ok(t) => {
// 日志中只显示token的前8位保护安全
debug!("JWT生成成功: {}", &t[0..8]);
t
}
Err(e) => {
error!("JWT生成失败: {}", e);
return HttpResponse::InternalServerError().json(ErrorResponse {
error: "生成身份令牌失败".to_string(),
errcode: Some(500),
errmsg: Some(e.to_string()),
});
}
};
// 5. 返回成功响应包含token
HttpResponse::Ok().json(token)
info!("用户登录成功用户ID: {}", user_id);
let token_response = TokenResponse::new(token);
HttpResponse::Ok().json(token_response)
}
#[post("/api/post-weather-data")]
@@ -566,6 +606,16 @@ fn create_server_config(
#[actix_web::main]
async fn main() -> std::io::Result<()> {
if let Err(e) = dotenvy::dotenv() {
eprintln!("警告:加载.env文件失败使用系统环境变量: {}", e);
}
// 2. 打印当前 RUST_LOG 的值(关键:验证配置是否生效)
let rust_log = std::env::var("RUST_LOG").unwrap_or_else(|_| "未设置".to_string());
eprintln!("当前 RUST_LOG 级别:{}", rust_log); // 这行用 eprintln!,不受日志级别影响
env_logger::init(); // 添加这一行
info!("logger init successful");
// 初始化数据库连接池
let pool = match create_pool().await {
Ok(pool) => pool,

View File

@@ -28,6 +28,36 @@ pub struct TempTokenClaims {
pub resource_id: i32, // 允许访问的资源ID如天气数据id
}
// 登录成功后的令牌响应结构体
/// 仅包含操作状态和令牌信息
#[derive(Debug, Serialize, Clone)]
pub struct TokenResponse {
/// 操作状态true表示登录成功false表示失败
pub success: bool,
/// 登录成功后生成的JWT令牌客户端后续请求需携带此令牌
pub token: String,
}
impl TokenResponse {
/// 快速创建登录成功的令牌响应
/// - token: 生成的JWT令牌字符串
pub fn new(token: String) -> Self {
Self {
success: true, // 登录成功场景固定为true
token,
}
}
/// 创建登录失败的响应(可选,用于统一错误响应格式)
pub fn failed() -> Self {
Self {
success: false,
token: String::new(), // 失败时令牌为空
}
}
}
#[derive(Debug, Deserialize)]
pub struct WeChatLoginRequest {
pub code: String,