refactor: config.rs去掉TOML文件加载,统一走systemd Environment
This commit is contained in:
@@ -1,11 +1,9 @@
|
||||
//! 应用配置模块
|
||||
//!
|
||||
//! 支持多环境配置加载,优先级:环境变量 > 环境配置 > 默认配置
|
||||
//! 从环境变量加载配置,不再使用 TOML 文件。
|
||||
//! systemd service 的 Environment= 是唯一的配置来源。
|
||||
|
||||
use serde::Deserialize;
|
||||
use std::path::PathBuf;
|
||||
|
||||
#[derive(Debug, Deserialize, Clone)]
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct AppConfig {
|
||||
pub database_url: String,
|
||||
pub wechat_appid: String,
|
||||
@@ -20,65 +18,27 @@ pub struct AppConfig {
|
||||
|
||||
impl AppConfig {
|
||||
pub fn load() -> Result<Self, String> {
|
||||
let env = std::env::var("APP_ENV").unwrap_or_else(|_| "development".into());
|
||||
let environment = std::env::var("APP_ENV").unwrap_or_else(|_| "development".into());
|
||||
|
||||
let manifest_dir = std::env::var("CARGO_MANIFEST_DIR")
|
||||
.map(PathBuf::from)
|
||||
.unwrap_or_else(|_| std::env::current_dir().unwrap());
|
||||
let app_config = Self {
|
||||
database_url: env_or_fail("DATABASE_URL")?,
|
||||
wechat_appid: env_or_fail("WECHAT_APPID")?,
|
||||
wechat_secret: env_or_fail("WECHAT_SECRET")?,
|
||||
jwt_secret: env_or_fail("JWT_SECRET")?,
|
||||
ssl_key_path: std::env::var("SSL_KEY_PATH").unwrap_or_default(),
|
||||
ssl_cert_path: std::env::var("SSL_CERT_PATH").unwrap_or_default(),
|
||||
rust_log: std::env::var("RUST_LOG").unwrap_or_else(|_| "info".into()),
|
||||
free_user_data_limit: std::env::var("FREE_USER_DATA_LIMIT")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(20),
|
||||
environment,
|
||||
};
|
||||
|
||||
let config_dir = manifest_dir.join("config");
|
||||
|
||||
let default_path = config_dir.join("default.toml");
|
||||
let env_path = config_dir.join(format!("{}.toml", env));
|
||||
|
||||
let mut settings = toml::Table::new();
|
||||
|
||||
// 1. 加载 default.toml
|
||||
if default_path.exists() {
|
||||
let content = std::fs::read_to_string(&default_path)
|
||||
.map_err(|e| format!("读取 default.toml 失败: {}", e))?;
|
||||
let parsed: toml::Table =
|
||||
toml::from_str(&content).map_err(|e| format!("解析 default.toml 失败: {}", e))?;
|
||||
settings.extend(parsed);
|
||||
} else {
|
||||
return Err(format!("配置文件不存在: {}", default_path.display()));
|
||||
}
|
||||
|
||||
// 2. 合并环境配置(覆盖默认值)
|
||||
if env_path.exists() {
|
||||
let content = std::fs::read_to_string(&env_path)
|
||||
.map_err(|e| format!("读取 {}.toml 失败: {}", env, e))?;
|
||||
let parsed: toml::Table =
|
||||
toml::from_str(&content).map_err(|e| format!("解析 {}.toml 失败: {}", env, e))?;
|
||||
settings.extend(parsed);
|
||||
}
|
||||
|
||||
// 3. 从环境变量加载(最高优先级)
|
||||
// 支持两种前缀:APP_*(推荐)和直接变量名(兼容 systemd EnvironmentFile)
|
||||
if let Ok(val) = std::env::var("APP_DATABASE_URL") {
|
||||
settings.insert("database_url".into(), toml::Value::String(val));
|
||||
}
|
||||
if let Ok(val) = std::env::var("APP_JWT_SECRET").or_else(|_| std::env::var("JWT_SECRET")) {
|
||||
settings.insert("jwt_secret".into(), toml::Value::String(val));
|
||||
}
|
||||
if let Ok(val) = std::env::var("APP_WECHAT_APPID").or_else(|_| std::env::var("WECHAT_APPID")) {
|
||||
settings.insert("wechat_appid".into(), toml::Value::String(val));
|
||||
}
|
||||
if let Ok(val) = std::env::var("APP_WECHAT_SECRET").or_else(|_| std::env::var("WECHAT_SECRET")) {
|
||||
settings.insert("wechat_secret".into(), toml::Value::String(val));
|
||||
}
|
||||
if let Ok(val) = std::env::var("APP_RUST_LOG") {
|
||||
settings.insert("rust_log".into(), toml::Value::String(val));
|
||||
}
|
||||
if let Ok(val) = std::env::var("APP_FREE_USER_DATA_LIMIT")
|
||||
&& let Ok(num) = val.parse::<i64>() {
|
||||
settings.insert("free_user_data_limit".into(), toml::Value::Integer(num));
|
||||
}
|
||||
|
||||
let app_config: AppConfig = toml::from_str(&settings.to_string())
|
||||
.map_err(|e| format!("配置反序列化失败: {}", e))?;
|
||||
|
||||
tracing::info!("配置加载成功,环境={}", app_config.environment);
|
||||
tracing::info!(
|
||||
"配置加载成功,环境={}",
|
||||
app_config.environment
|
||||
);
|
||||
|
||||
Ok(app_config)
|
||||
}
|
||||
@@ -95,3 +55,7 @@ impl AppConfig {
|
||||
self.environment == "production"
|
||||
}
|
||||
}
|
||||
|
||||
fn env_or_fail(key: &str) -> Result<String, String> {
|
||||
std::env::var(key).map_err(|_| format!("环境变量 {} 未设置", key))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user