mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 14:26:36 +08:00
ryan
123 lines
3.9 KiB
Go
123 lines
3.9 KiB
Go
/*
|
|
Copyright 2025 linux.do
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
"github.com/shopspring/decimal"
|
|
|
|
"github.com/linux-do/credit/internal/db"
|
|
)
|
|
|
|
// 配置键常量 - 所有系统配置的 key 定义
|
|
const (
|
|
ConfigKeyUploadAllowedExtensions = "upload_allowed_extensions" // 允许上传的文件扩展名，逗号分隔
|
|
ConfigKeySiteName = "site_name" // 站点名称
|
|
ConfigKeyPasswordLoginEnabled = "password_login_enabled" // 是否允许密码登录
|
|
ConfigKeyRegistrationEnabled = "registration_enabled" // 是否允许注册
|
|
ConfigKeyPasswordRegisterEnabled = "password_register_enabled" // 是否允许密码注册
|
|
ConfigKeyOIDCLoginEnabled = "oidc_login_enabled" // 是否允许 OIDC 登录
|
|
ConfigKeyMaxAPIKeysPerUser = "max_api_keys_per_user" // 每个用户最大 API Key 数量
|
|
)
|
|
|
|
const (
|
|
// SystemConfigRedisHashKey Redis Hash key，存储所有系统配置
|
|
SystemConfigRedisHashKey = "system:system_configs"
|
|
)
|
|
|
|
type SystemConfig struct {
|
|
Key string `json:"key" gorm:"primaryKey;size:64;not null"`
|
|
Value string `json:"value" gorm:"size:255;not null"`
|
|
Type string `json:"type" gorm:"size:32;not null;default:'system'"`
|
|
Description string `json:"description" gorm:"size:255"`
|
|
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
|
|
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
|
|
}
|
|
|
|
// GetByKey 通过 key 查询配置（带 Redis 缓存）
|
|
func (sc *SystemConfig) GetByKey(ctx context.Context, key string) error {
|
|
if err := db.HGetJSON(ctx, SystemConfigRedisHashKey, key, sc); err == nil {
|
|
return nil
|
|
} else if !errors.Is(err, redis.Nil) {
|
|
// Redis 服务错误，返回错误
|
|
return err
|
|
}
|
|
|
|
// 查数据库
|
|
if err := db.DB(ctx).Where("key = ?", key).First(sc).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// 更新 Redis Hash 缓存
|
|
_ = db.HSetJSON(ctx, SystemConfigRedisHashKey, key, sc)
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetIntByKey 通过 key 查询配置并转换为 int 类型
|
|
func GetIntByKey(ctx context.Context, key string) (int, error) {
|
|
var sc SystemConfig
|
|
if err := sc.GetByKey(ctx, key); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
value, err := strconv.Atoi(sc.Value)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("配置 %s 的值 '%s' 无法转换为整数: %w", key, sc.Value, err)
|
|
}
|
|
|
|
return value, nil
|
|
}
|
|
|
|
// GetDecimalByKey 通过 key 查询配置并转换为 decimal.Decimal 类型
|
|
// precision 指定保留的小数位数，多余的小数会被裁剪
|
|
func GetDecimalByKey(ctx context.Context, key string, precision int32) (decimal.Decimal, error) {
|
|
var sc SystemConfig
|
|
if err := sc.GetByKey(ctx, key); err != nil {
|
|
return decimal.Zero, err
|
|
}
|
|
|
|
value, err := decimal.NewFromString(sc.Value)
|
|
if err != nil {
|
|
return decimal.Zero, fmt.Errorf("配置 %s 的值 '%s' 无法转换为decimal: %w", key, sc.Value, err)
|
|
}
|
|
|
|
// 裁剪到指定小数位数
|
|
return value.Truncate(precision), nil
|
|
}
|
|
|
|
// GetBoolByKey 通过 key 查询配置并转换为 bool 类型
|
|
func GetBoolByKey(ctx context.Context, key string) (bool, error) {
|
|
var sc SystemConfig
|
|
if err := sc.GetByKey(ctx, key); err != nil {
|
|
return false, err
|
|
}
|
|
|
|
value, err := strconv.ParseBool(sc.Value)
|
|
if err != nil {
|
|
return false, fmt.Errorf("配置 %s 的值 '%s' 无法转换为布尔值: %w", key, sc.Value, err)
|
|
}
|
|
|
|
return value, nil
|
|
}
|