🎨 新增ip检测配置功能
This commit is contained in:
@@ -392,3 +392,51 @@ func (u *UserService) SetTeacherExpectRate(p request.SetTeacherInfo) (err error)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (u *UserService) SetIpConfig(p user.IpCheck) (user.IpCheck, error) {
|
||||
p.ID = 1
|
||||
err := global.GVA_DB.Save(&p).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("修改IP检测配置失败", zap.Error(err))
|
||||
return p, err
|
||||
}
|
||||
return p, nil
|
||||
}
|
||||
|
||||
func (u *UserService) GetIpConfig() (ipConfig user.IpCheck, err error) {
|
||||
ipConfig.ID = 1
|
||||
err = global.GVA_DB.First(&ipConfig).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取IP检测配置失败", zap.Error(err))
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (u *UserService) SetIpStatus(status bool) (bool, error) {
|
||||
var ipConfig user.IpCheck
|
||||
ipConfig.ID = 1
|
||||
err := global.GVA_DB.First(&ipConfig).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取IP检测配置失败", zap.Error(err))
|
||||
return false, err
|
||||
}
|
||||
ipConfig.Status = status
|
||||
err = global.GVA_DB.Save(&ipConfig).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("修改IP检测状态失败", zap.Error(err))
|
||||
return false, err
|
||||
}
|
||||
return status, nil
|
||||
}
|
||||
|
||||
func (u *UserService) GetIpStatus() (status bool, err error) {
|
||||
var ipConfig user.IpCheck
|
||||
err = global.GVA_DB.Where("id = 1").First(&ipConfig).Error
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("获取IP检测配置失败", zap.Error(err))
|
||||
return
|
||||
}
|
||||
status = ipConfig.Status
|
||||
return
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user