- 创建Repository接口定义(UserRepository、ProfileRepository、TextureRepository等) - 创建Repository接口实现 - 创建依赖注入容器(container.Container) - 改造Handler层使用依赖注入(AuthHandler、UserHandler、TextureHandler) - 创建新的路由注册方式(RegisterRoutesWithDI) - 提供main.go示例文件展示如何使用依赖注入 同时包含之前的安全修复: - CORS配置安全加固 - 头像URL验证安全修复 - JWT algorithm confusion漏洞修复 - Recovery中间件增强 - 敏感错误信息泄露修复 - 类型断言安全修复
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"carrotskin/internal/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// systemConfigRepositoryImpl SystemConfigRepository的实现
|
|
type systemConfigRepositoryImpl struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewSystemConfigRepository 创建SystemConfigRepository实例
|
|
func NewSystemConfigRepository(db *gorm.DB) SystemConfigRepository {
|
|
return &systemConfigRepositoryImpl{db: db}
|
|
}
|
|
|
|
func (r *systemConfigRepositoryImpl) GetByKey(key string) (*model.SystemConfig, error) {
|
|
var config model.SystemConfig
|
|
err := r.db.Where("key = ?", key).First(&config).Error
|
|
return handleNotFoundResult(&config, err)
|
|
}
|
|
|
|
func (r *systemConfigRepositoryImpl) GetPublic() ([]model.SystemConfig, error) {
|
|
var configs []model.SystemConfig
|
|
err := r.db.Where("is_public = ?", true).Find(&configs).Error
|
|
return configs, err
|
|
}
|
|
|
|
func (r *systemConfigRepositoryImpl) GetAll() ([]model.SystemConfig, error) {
|
|
var configs []model.SystemConfig
|
|
err := r.db.Find(&configs).Error
|
|
return configs, err
|
|
}
|
|
|
|
func (r *systemConfigRepositoryImpl) Update(config *model.SystemConfig) error {
|
|
return r.db.Save(config).Error
|
|
}
|
|
|
|
func (r *systemConfigRepositoryImpl) UpdateValue(key, value string) error {
|
|
return r.db.Model(&model.SystemConfig{}).Where("key = ?", key).Update("value", value).Error
|
|
}
|
|
|