first commit
This commit is contained in:
sjk
42
go_backend/config/config.dev.yaml
Normal file
42
go_backend/config/config.dev.yaml
Normal file
@@ -0,0 +1,42 @@
|
||||
server:
|
||||
port: 8080
|
||||
mode: debug # debug, release, test
|
||||
|
||||
database:
|
||||
host: localhost
|
||||
port: 3306
|
||||
username: root
|
||||
password: JKjk20011115
|
||||
dbname: ai_wht
|
||||
charset: utf8mb4
|
||||
parse_time: true
|
||||
loc: Local
|
||||
max_idle_conns: 10
|
||||
max_open_conns: 100
|
||||
conn_max_lifetime: 3600
|
||||
|
||||
jwt:
|
||||
secret: dev_secret_key_change_in_production
|
||||
expire_hours: 168 # 7天
|
||||
|
||||
wechat:
|
||||
app_id: "wxa5bf062342ef754d" # 微信小程序AppID，留空则使用默认登录
|
||||
app_secret: "69d2a3ddc902b26f82f4b56a6e277f7a" # 微信小程序AppSecret
|
||||
|
||||
xhs:
|
||||
python_service_url: "http://localhost:8000" # Python服务地址
|
||||
|
||||
scheduler:
|
||||
enabled: true # 是否启用定时任务
|
||||
publish_cron: "* * * * * *" # 每1小时执行一次（开发环境测试用）
|
||||
max_concurrent: 2 # 最大并发发布数
|
||||
publish_timeout: 300 # 发布超时时间（秒）
|
||||
max_articles_per_user_per_run: 2 # 每轮每个用户最大发文数
|
||||
max_failures_per_user_per_run: 3 # 每轮每个用户最大失败次数
|
||||
max_daily_articles_per_user: 6 # 每个用户每日最大发文数（自动发布）
|
||||
max_hourly_articles_per_user: 2 # 每个用户每小时最大发文数（自动发布）
|
||||
proxy: "" # 可选：静态全局代理地址，例如 http://user:pass@ip:port
|
||||
user_agent: "" # 可选：自定义User-Agent，不填则使用默认
|
||||
proxy_pool:
|
||||
enabled: true # 开发环境启用代理池
|
||||
api_url: "http://api.tianqiip.com/getip?secret=lu29e593&num=1&type=txt&port=1&mr=1&sign=4b81a62eaed89ba802a8f34053e2c964"
|
||||
159
go_backend/config/config.go
Normal file
159
go_backend/config/config.go
Normal file
@@ -0,0 +1,159 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Server ServerConfig `mapstructure:"server"`
|
||||
Database DatabaseConfig `mapstructure:"database"`
|
||||
JWT JWTConfig `mapstructure:"jwt"`
|
||||
Wechat WechatConfig `mapstructure:"wechat"`
|
||||
XHS XHSConfig `mapstructure:"xhs"`
|
||||
Scheduler SchedulerConfig `mapstructure:"scheduler"`
|
||||
}
|
||||
|
||||
type ServerConfig struct {
|
||||
Port int `mapstructure:"port"`
|
||||
Mode string `mapstructure:"mode"`
|
||||
}
|
||||
|
||||
type DatabaseConfig struct {
|
||||
Host string `mapstructure:"host"`
|
||||
Port int `mapstructure:"port"`
|
||||
Username string `mapstructure:"username"`
|
||||
Password string `mapstructure:"password"`
|
||||
DBName string `mapstructure:"dbname"`
|
||||
Charset string `mapstructure:"charset"`
|
||||
ParseTime bool `mapstructure:"parse_time"`
|
||||
Loc string `mapstructure:"loc"`
|
||||
MaxIdleConns int `mapstructure:"max_idle_conns"`
|
||||
MaxOpenConns int `mapstructure:"max_open_conns"`
|
||||
ConnMaxLifetime int `mapstructure:"conn_max_lifetime"`
|
||||
}
|
||||
|
||||
type JWTConfig struct {
|
||||
Secret string `mapstructure:"secret"`
|
||||
ExpireHours int `mapstructure:"expire_hours"`
|
||||
}
|
||||
|
||||
type WechatConfig struct {
|
||||
AppID string `mapstructure:"app_id"`
|
||||
AppSecret string `mapstructure:"app_secret"`
|
||||
}
|
||||
|
||||
type XHSConfig struct {
|
||||
PythonServiceURL string `mapstructure:"python_service_url"`
|
||||
}
|
||||
|
||||
type SchedulerConfig struct {
|
||||
Enabled bool `mapstructure:"enabled"` // 是否启用定时任务
|
||||
PublishCron string `mapstructure:"publish_cron"` // 发布任务的Cron表达式
|
||||
MaxConcurrent int `mapstructure:"max_concurrent"` // 最大并发发布数
|
||||
PublishTimeout int `mapstructure:"publish_timeout"` // 发布超时时间（秒）
|
||||
MaxArticlesPerUserPerRun int `mapstructure:"max_articles_per_user_per_run"` // 每轮每个用户最大发文数
|
||||
MaxFailuresPerUserPerRun int `mapstructure:"max_failures_per_user_per_run"` // 每轮每个用户最大失败次数
|
||||
MaxDailyArticlesPerUser int `mapstructure:"max_daily_articles_per_user"` // 每个用户每日最大发文数
|
||||
MaxHourlyArticlesPerUser int `mapstructure:"max_hourly_articles_per_user"` // 每个用户每小时最大发文数
|
||||
Proxy string `mapstructure:"proxy"` // 全局代理地址（可选）
|
||||
UserAgent string `mapstructure:"user_agent"` // 全局User-Agent（可选）
|
||||
ProxyFetchURL string `mapstructure:"proxy_fetch_url"` // 动态获取代理的接口地址（可选）
|
||||
}
|
||||
|
||||
var AppConfig *Config
|
||||
|
||||
// LoadConfig 加载配置文件
|
||||
// 优先级: 环境变量 > 配置文件
|
||||
func LoadConfig(env string) error {
|
||||
// 1. 优先从环境变量 APP_ENV 获取环境配置
|
||||
if envFromEnv := os.Getenv("APP_ENV"); envFromEnv != "" {
|
||||
env = envFromEnv
|
||||
log.Printf("从环境变量 APP_ENV 读取环境: %s", env)
|
||||
}
|
||||
|
||||
if env == "" {
|
||||
env = "dev"
|
||||
}
|
||||
|
||||
// 2. 加载配置文件
|
||||
viper.SetConfigName(fmt.Sprintf("config.%s", env))
|
||||
viper.SetConfigType("yaml")
|
||||
viper.AddConfigPath("./config")
|
||||
viper.AddConfigPath("../config")
|
||||
|
||||
if err := viper.ReadInConfig(); err != nil {
|
||||
return fmt.Errorf("读取配置文件失败: %w", err)
|
||||
}
|
||||
|
||||
// 3. 绑定环境变量(自动读取环境变量覆盖配置)
|
||||
viper.AutomaticEnv()
|
||||
|
||||
// 绑定特定的环境变量到配置项
|
||||
bindEnvVariables()
|
||||
|
||||
AppConfig = &Config{}
|
||||
if err := viper.Unmarshal(AppConfig); err != nil {
|
||||
return fmt.Errorf("解析配置文件失败: %w", err)
|
||||
}
|
||||
|
||||
log.Printf("配置加载成功: %s 环境", env)
|
||||
log.Printf("数据库配置: %s@%s:%d/%s", AppConfig.Database.Username, AppConfig.Database.Host, AppConfig.Database.Port, AppConfig.Database.DBName)
|
||||
return nil
|
||||
}
|
||||
|
||||
// bindEnvVariables 绑定环境变量到配置项
|
||||
func bindEnvVariables() {
|
||||
// Server 配置
|
||||
viper.BindEnv("server.port", "SERVER_PORT")
|
||||
viper.BindEnv("server.mode", "SERVER_MODE")
|
||||
|
||||
// Database 配置
|
||||
viper.BindEnv("database.host", "DB_HOST")
|
||||
viper.BindEnv("database.port", "DB_PORT")
|
||||
viper.BindEnv("database.username", "DB_USERNAME")
|
||||
viper.BindEnv("database.password", "DB_PASSWORD")
|
||||
viper.BindEnv("database.dbname", "DB_NAME")
|
||||
viper.BindEnv("database.charset", "DB_CHARSET")
|
||||
|
||||
// JWT 配置
|
||||
viper.BindEnv("jwt.secret", "JWT_SECRET")
|
||||
viper.BindEnv("jwt.expire_hours", "JWT_EXPIRE_HOURS")
|
||||
|
||||
// Wechat 配置
|
||||
viper.BindEnv("wechat.app_id", "WECHAT_APP_ID")
|
||||
viper.BindEnv("wechat.app_secret", "WECHAT_APP_SECRET")
|
||||
|
||||
// XHS 配置
|
||||
viper.BindEnv("xhs.python_service_url", "XHS_PYTHON_SERVICE_URL")
|
||||
|
||||
// Scheduler 配置
|
||||
viper.BindEnv("scheduler.enabled", "SCHEDULER_ENABLED")
|
||||
viper.BindEnv("scheduler.publish_cron", "SCHEDULER_PUBLISH_CRON")
|
||||
viper.BindEnv("scheduler.max_concurrent", "SCHEDULER_MAX_CONCURRENT")
|
||||
viper.BindEnv("scheduler.publish_timeout", "SCHEDULER_PUBLISH_TIMEOUT")
|
||||
viper.BindEnv("scheduler.max_articles_per_user_per_run", "SCHEDULER_MAX_ARTICLES_PER_USER_PER_RUN")
|
||||
viper.BindEnv("scheduler.max_failures_per_user_per_run", "SCHEDULER_MAX_FAILURES_PER_USER_PER_RUN")
|
||||
viper.BindEnv("scheduler.max_daily_articles_per_user", "SCHEDULER_MAX_DAILY_ARTICLES_PER_USER")
|
||||
viper.BindEnv("scheduler.max_hourly_articles_per_user", "SCHEDULER_MAX_HOURLY_ARTICLES_PER_USER")
|
||||
viper.BindEnv("scheduler.proxy", "SCHEDULER_PROXY")
|
||||
viper.BindEnv("scheduler.user_agent", "SCHEDULER_USER_AGENT")
|
||||
viper.BindEnv("scheduler.proxy_fetch_url", "SCHEDULER_PROXY_FETCH_URL")
|
||||
}
|
||||
|
||||
// GetDSN 获取数据库连接字符串
|
||||
func (c *DatabaseConfig) GetDSN() string {
|
||||
return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=%t&loc=%s",
|
||||
c.Username,
|
||||
c.Password,
|
||||
c.Host,
|
||||
c.Port,
|
||||
c.DBName,
|
||||
c.Charset,
|
||||
c.ParseTime,
|
||||
c.Loc,
|
||||
)
|
||||
}
|
||||
42
go_backend/config/config.prod.yaml
Normal file
42
go_backend/config/config.prod.yaml
Normal file
@@ -0,0 +1,42 @@
|
||||
server:
|
||||
port: 8070
|
||||
mode: release
|
||||
|
||||
database:
|
||||
host: 8.149.233.36
|
||||
port: 3306
|
||||
username: ai_wht_write
|
||||
password: 7aK_H2yvokVumr84lLNDt8fDBp6P
|
||||
dbname: ai_wht
|
||||
charset: utf8mb4
|
||||
parse_time: true
|
||||
loc: Local
|
||||
max_idle_conns: 20
|
||||
max_open_conns: 200
|
||||
conn_max_lifetime: 3600
|
||||
|
||||
jwt:
|
||||
secret: prod_secret_key_please_change_this
|
||||
expire_hours: 168
|
||||
|
||||
wechat:
|
||||
app_id: "wxa5bf062342ef754d" # 微信小程序AppID，留空则使用默认登录
|
||||
app_secret: "69d2a3ddc902b26f82f4b56a6e277f7a" # 微信小程序AppSecret
|
||||
|
||||
xhs:
|
||||
python_service_url: "http://localhost:8000" # Python服务地址，生产环境请修改为实际地址
|
||||
|
||||
scheduler:
|
||||
enabled: true # 是否启用定时任务
|
||||
publish_cron: "0 0 */2 * * *" # 每2小时执行一次（防封号策略）
|
||||
max_concurrent: 2 # 最大并发发布数
|
||||
publish_timeout: 300 # 发布超时时间（秒）
|
||||
max_articles_per_user_per_run: 2 # 每轮每个用户最大发文数
|
||||
max_failures_per_user_per_run: 3 # 每轮每个用户最大失败次数
|
||||
max_daily_articles_per_user: 5 # 每个用户每日最大发文数（自动发布）
|
||||
max_hourly_articles_per_user: 1 # 每个用户每小时最大发文数（自动发布）
|
||||
proxy: "" # 可选：静态全局代理地址，例如 http://user:pass@ip:port
|
||||
user_agent: "" # 可选：自定义User-Agent，不填则使用默认
|
||||
proxy_pool:
|
||||
enabled: true # 生产环境启用代理池
|
||||
api_url: "http://api.tianqiip.com/getip?secret=lu29e593&num=1&type=txt&port=1&mr=1&sign=4b81a62eaed89ba802a8f34053e2c964"
|
||||
Reference in New Issue
Block a user