cc
0e6c774ed6
feat(auth): 新增 wall_users 表，照片墙与开发者登录分离 登录先查 ccuser（开发者），再查 wall_users（照片墙用户）。 wall_users 初始用户：cc / UserTest。 Co-Authored-By: Claude <noreply@anthropic.com> @
52 lines
1.9 KiB
Rust
52 lines
1.9 KiB
Rust
//! 共享应用状态。
|
||
//!
|
||
//! 采样任务（写）与 HTTP handler（读）通过一个被 `RwLock` 保护的
|
||
//! `Stats` 快照通信。`AppState` 可廉价 `clone`（内部是 `Arc`）。
|
||
|
||
use std::sync::Arc;
|
||
|
||
use tokio::sync::RwLock;
|
||
|
||
use crate::auth::{self, DevGate, UserStore, WallUserStore};
|
||
use crate::folders::{FolderRegistry, FolderStore};
|
||
use crate::monitor::Stats;
|
||
use crate::photos::PhotoStore;
|
||
use crate::weather::WeatherService;
|
||
|
||
#[derive(Clone)]
|
||
pub struct AppState {
|
||
/// 最新一次采样快照。
|
||
pub snapshot: Arc<RwLock<Stats>>,
|
||
/// 用户校验后端（MySQL 或内置兜底）：内部已是连接池/只读 map，`Arc` 共享即可。
|
||
pub users: Arc<UserStore>,
|
||
/// 开发者入口门：合法入口 hash 集合，只读共享。
|
||
pub gate: Arc<DevGate>,
|
||
/// 天气/定位服务（自带每日限流 + 缓存）。
|
||
pub weather: Arc<WeatherService>,
|
||
/// 照片墙用户校验后端（wall_users 表）。
|
||
pub wall_users: Arc<WallUserStore>,
|
||
/// 照片墙图片仓库（按用户隔离的磁盘目录）。
|
||
pub photos: PhotoStore,
|
||
/// 特殊文件夹注册表（白名单）。
|
||
pub folders: FolderRegistry,
|
||
/// 特殊文件夹图片仓库（按文件夹隔离的磁盘目录）。
|
||
pub folder_photos: FolderStore,
|
||
}
|
||
|
||
impl AppState {
|
||
pub fn new() -> Self {
|
||
// 一个连接池给登录校验与入口门共用（克隆是廉价的、共享底层连接）。
|
||
let pool = auth::db_pool_from_env();
|
||
Self {
|
||
snapshot: Arc::new(RwLock::new(Stats::default())),
|
||
users: Arc::new(UserStore::new(pool.clone())),
|
||
gate: Arc::new(DevGate::new(pool.clone())),
|
||
weather: Arc::new(WeatherService::from_env()),
|
||
wall_users: Arc::new(WallUserStore::new(pool.clone())),
|
||
photos: PhotoStore::from_env(),
|
||
folders: FolderRegistry::from_env(),
|
||
folder_photos: FolderStore::from_env(),
|
||
}
|
||
}
|
||
}
|