修复登录及验证码功能
This commit is contained in:
@@ -1,17 +1,26 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"photowall/internal/model"
|
||||
"strconv"
|
||||
"sync"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type ConfigService struct {
|
||||
db *gorm.DB
|
||||
db *gorm.DB
|
||||
cache map[string]string
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
func NewConfigService(db *gorm.DB) *ConfigService {
|
||||
return &ConfigService{db: db}
|
||||
return &ConfigService{
|
||||
db: db,
|
||||
cache: make(map[string]string),
|
||||
}
|
||||
}
|
||||
|
||||
// defaultConfigs 所有可配置项及说明
|
||||
@@ -23,8 +32,8 @@ var defaultConfigs = []model.SystemConfig{
|
||||
{Key: "audit_grace_days", Value: "30", Label: "审核补齐期限(天)", Description: "班级被打回后，超过该天数未补齐则自动删除", Category: "basic", Placeholder: "30", Sort: 4},
|
||||
|
||||
// ===== 极验行为验证 =====
|
||||
{Key: "geetest_id", Value: "", Label: "极验 ID", Description: "极验行为验证的Captcha ID，用于防机器人和暴力破解", Category: "geetest", Placeholder: "如：64a####################", DocLink: "https://www.geetest.com 注册后在应用管理获取", Sort: 1},
|
||||
{Key: "geetest_key", Value: "", Label: "极验 Key", Description: "极验行为验证的私钥，与ID配对使用", Category: "geetest", Placeholder: "如：64a####################", DocLink: "https://www.geetest.com 注册后在应用管理获取", Sort: 2},
|
||||
{Key: "geetest_id", Value: "662e68b7ce24be211fa97a53975c5d96", Label: "极验 ID", Description: "极验行为验证的Captcha ID，用于防机器人和暴力破解", Category: "geetest", Placeholder: "如：64a####################", DocLink: "https://www.geetest.com 注册后在应用管理获取", Sort: 1},
|
||||
{Key: "geetest_key", Value: "2789075bc2b25a04c0f7c5420bfa3d6f", Label: "极验 Key", Description: "极验行为验证的私钥，与ID配对使用", Category: "geetest", Placeholder: "如：64a####################", DocLink: "https://www.geetest.com 注册后在应用管理获取", Sort: 2},
|
||||
|
||||
// ===== 短信验证码 =====
|
||||
{Key: "sms_provider", Value: "aliyun", Label: "短信服务商", Description: "选择短信服务商，目前支持阿里云和腾讯云", Category: "sms", Placeholder: "aliyun 或 tencent", DocLink: "阿里云: https://dysms.console.aliyun.com | 腾讯云: https://console.cloud.tencent.com/sms", Sort: 1},
|
||||
@@ -43,17 +52,29 @@ var defaultConfigs = []model.SystemConfig{
|
||||
{Key: "max_file_size_mb", Value: "10", Label: "单文件大小上限(MB)", Description: "上传图片的最大大小限制", Category: "storage", Placeholder: "10", Sort: 2},
|
||||
}
|
||||
|
||||
// InitDefaults 初始化默认配置（不存在则创建）
|
||||
// InitDefaults 初始化默认配置（不存在则创建，已存在但为空且默认有初始值则补齐）
|
||||
func (s *ConfigService) InitDefaults() error {
|
||||
for _, c := range defaultConfigs {
|
||||
var count int64
|
||||
s.db.Model(&model.SystemConfig{}).Where("`key` = ?", c.Key).Count(&count)
|
||||
if count == 0 {
|
||||
var existing model.SystemConfig
|
||||
err := s.db.Where("`key` = ?", c.Key).First(&existing).Error
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
if err := s.db.Create(&c).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
} else if err == nil && existing.Value == "" && c.Value != "" {
|
||||
s.db.Model(&existing).Update("value", c.Value)
|
||||
}
|
||||
}
|
||||
|
||||
// 预热内存缓存
|
||||
var list []model.SystemConfig
|
||||
if err := s.db.Find(&list).Error; err == nil {
|
||||
s.mu.Lock()
|
||||
for _, item := range list {
|
||||
s.cache[item.Key] = item.Value
|
||||
}
|
||||
s.mu.Unlock()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -71,14 +92,49 @@ func (s *ConfigService) GetByCategory(category string) ([]model.SystemConfig, er
|
||||
return list, err
|
||||
}
|
||||
|
||||
// UpdateReq 更新配置请求
|
||||
// UpdateConfigReq 更新配置请求（支持 map 与 key-value 列表格式）
|
||||
type UpdateConfigReq struct {
|
||||
Configs map[string]string `json:"configs" binding:"required"`
|
||||
Configs map[string]string `json:"configs"`
|
||||
}
|
||||
|
||||
func (r *UpdateConfigReq) UnmarshalJSON(data []byte) error {
|
||||
// 格式1: {"configs": {"k1": "v1", "k2": "v2"}}
|
||||
var auxMap struct {
|
||||
Configs map[string]string `json:"configs"`
|
||||
}
|
||||
if err := json.Unmarshal(data, &auxMap); err == nil && auxMap.Configs != nil {
|
||||
r.Configs = auxMap.Configs
|
||||
return nil
|
||||
}
|
||||
|
||||
// 格式2: {"configs": [{"key": "k1", "value": "v1"}]}
|
||||
var auxList struct {
|
||||
Configs []struct {
|
||||
Key string `json:"key"`
|
||||
Value string `json:"value"`
|
||||
} `json:"configs"`
|
||||
}
|
||||
if err := json.Unmarshal(data, &auxList); err == nil && auxList.Configs != nil {
|
||||
r.Configs = make(map[string]string, len(auxList.Configs))
|
||||
for _, item := range auxList.Configs {
|
||||
r.Configs[item.Key] = item.Value
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 格式3: 直接传入 map: {"k1": "v1"}
|
||||
var directMap map[string]string
|
||||
if err := json.Unmarshal(data, &directMap); err == nil {
|
||||
r.Configs = directMap
|
||||
return nil
|
||||
}
|
||||
|
||||
return errors.New("invalid configs format")
|
||||
}
|
||||
|
||||
// Update 批量更新配置
|
||||
func (s *ConfigService) Update(req *UpdateConfigReq) error {
|
||||
return s.db.Transaction(func(tx *gorm.DB) error {
|
||||
err := s.db.Transaction(func(tx *gorm.DB) error {
|
||||
for key, value := range req.Configs {
|
||||
if err := tx.Model(&model.SystemConfig{}).Where("`key` = ?", key).Update("value", value).Error; err != nil {
|
||||
return err
|
||||
@@ -86,13 +142,66 @@ func (s *ConfigService) Update(req *UpdateConfigReq) error {
|
||||
}
|
||||
return nil
|
||||
})
|
||||
if err == nil {
|
||||
s.mu.Lock()
|
||||
for key, value := range req.Configs {
|
||||
s.cache[key] = value
|
||||
}
|
||||
s.mu.Unlock()
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// GetValue 获取单个配置值
|
||||
// GetValue 获取单个配置值（优先走内存缓存）
|
||||
func (s *ConfigService) GetValue(key string) string {
|
||||
s.mu.RLock()
|
||||
val, ok := s.cache[key]
|
||||
s.mu.RUnlock()
|
||||
if ok {
|
||||
return val
|
||||
}
|
||||
|
||||
var c model.SystemConfig
|
||||
if err := s.db.Where("`key` = ?", key).First(&c).Error; err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
s.mu.Lock()
|
||||
s.cache[key] = c.Value
|
||||
s.mu.Unlock()
|
||||
|
||||
return c.Value
|
||||
}
|
||||
|
||||
// GetInt 获取整数配置值，解析失败或不存在则返回 def
|
||||
func (s *ConfigService) GetInt(key string, def int) int {
|
||||
val := s.GetValue(key)
|
||||
if val == "" {
|
||||
return def
|
||||
}
|
||||
if n, err := strconv.Atoi(val); err == nil {
|
||||
return n
|
||||
}
|
||||
return def
|
||||
}
|
||||
|
||||
// GetGeetestConfig 获取极验验证配置
|
||||
func (s *ConfigService) GetGeetestConfig() (id, key string) {
|
||||
return s.GetValue("geetest_id"), s.GetValue("geetest_key")
|
||||
}
|
||||
|
||||
// GetSmsConfig 获取短信配置
|
||||
func (s *ConfigService) GetSmsConfig() (provider, accessKey, secretKey, signName, templateCode string) {
|
||||
return s.GetValue("sms_provider"),
|
||||
s.GetValue("sms_access_key"),
|
||||
s.GetValue("sms_secret_key"),
|
||||
s.GetValue("sms_sign_name"),
|
||||
s.GetValue("sms_template_code")
|
||||
}
|
||||
|
||||
// GetWechatConfig 获取微信登录配置
|
||||
func (s *ConfigService) GetWechatConfig() (appID, appSecret, redirectURI string) {
|
||||
return s.GetValue("wechat_app_id"),
|
||||
s.GetValue("wechat_app_secret"),
|
||||
s.GetValue("wechat_redirect_uri")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user