PersonalWebApplication/Server/src/state.rs

//! 共享应用状态。
//!
//! 采样任务（写）与 HTTP handler（读）通过一个被 `RwLock` 保护的
//! `Stats` 快照通信。`AppState` 可廉价 `clone`（内部是 `Arc`）。

use std::sync::Arc;

use tokio::sync::RwLock;

use crate::auth::{self, DevGate, UserStore};
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>,
    /// 照片墙图片仓库（按用户隔离的磁盘目录）。
    pub photos: PhotoStore,
}

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)),
            weather: Arc::new(WeatherService::from_env()),
            photos: PhotoStore::from_env(),
        }
    }
}