refactor🎨 移除设置log函数

This commit is contained in:
wenjianzhang
2021-05-16 21:47:49 +08:00
parent 33275573f9
commit b2345039e4
4 changed files with 16 additions and 20 deletions
+1 -3
View File
@@ -114,7 +114,6 @@ func (e SysMenu) GetSysMenu(c *gin.Context) {
func (e SysMenu) InsertSysMenu(c *gin.Context) { func (e SysMenu) InsertSysMenu(c *gin.Context) {
control := new(dto.SysMenuControl) control := new(dto.SysMenuControl)
err := e.MakeContext(c). err := e.MakeContext(c).
MakeLogger().
MakeOrm(). MakeOrm().
Errors Errors
if err != nil { if err != nil {
@@ -158,7 +157,6 @@ func (e SysMenu) InsertSysMenu(c *gin.Context) {
func (e SysMenu) UpdateSysMenu(c *gin.Context) { func (e SysMenu) UpdateSysMenu(c *gin.Context) {
control := new(dto.SysMenuControl) control := new(dto.SysMenuControl)
err := e.MakeContext(c). err := e.MakeContext(c).
MakeLogger().
MakeOrm(). MakeOrm().
Errors Errors
if err != nil { if err != nil {
@@ -335,4 +333,4 @@ func (e SysMenu) GetMenuTreeSelect(c *gin.Context) {
"menus": result, "menus": result,
"checkedKeys": menuIds, "checkedKeys": menuIds,
}, "获取成功") }, "获取成功")
} }
+4 -2
View File
@@ -92,7 +92,9 @@ func (s *SysMenuControl) Bind(ctx *gin.Context) error {
// Generate 结构体数据转化 从 Control 至 model 对应的模型 // Generate 结构体数据转化 从 Control 至 model 对应的模型
func (s *SysMenuControl) Generate(model *models.SysMenu) { func (s *SysMenuControl) Generate(model *models.SysMenu) {
model.MenuId = s.MenuId if s.MenuId != 0 {
model.MenuId = s.MenuId
}
model.MenuName = s.MenuName model.MenuName = s.MenuName
model.Title = s.Title model.Title = s.Title
model.Icon = s.Icon model.Icon = s.Icon
@@ -168,4 +170,4 @@ type MenuRole struct {
type SelectRole struct { type SelectRole struct {
RoleId int `uri:"roleId"` RoleId int `uri:"roleId"`
} }
+6 -4
View File
@@ -81,9 +81,11 @@ func (e *SysMenu) GetSysMenu(d *dto.SysMenuById, model *models.SysMenu) error {
e.Log.Errorf("db error:%s", err) e.Log.Errorf("db error:%s", err)
return err return err
} }
apis := make([]int, 0)
for _, v := range model.SysApi { for _, v := range model.SysApi {
model.Apis = append(model.Apis, v.Id) apis = append(model.Apis, v.Id)
} }
model.Apis = apis
return nil return nil
} }
@@ -121,7 +123,7 @@ func (e *SysMenu) initPaths(menu *models.SysMenu) error {
// UpdateSysMenu 修改SysMenu对象 // UpdateSysMenu 修改SysMenu对象
func (e *SysMenu) UpdateSysMenu(c *dto.SysMenuControl) *SysMenu { func (e *SysMenu) UpdateSysMenu(c *dto.SysMenuControl) *SysMenu {
var err error var err error
model := models.SysMenu{} var model = models.SysMenu{}
e.Orm.First(&model, c.GetId()) e.Orm.First(&model, c.GetId())
c.Generate(&model) c.Generate(&model)
db := e.Orm.Session(&gorm.Session{FullSaveAssociations: true}).Debug().Save(&model) db := e.Orm.Session(&gorm.Session{FullSaveAssociations: true}).Debug().Save(&model)
@@ -131,7 +133,7 @@ func (e *SysMenu) UpdateSysMenu(c *dto.SysMenuControl) *SysMenu {
return e return e
} }
if db.RowsAffected == 0 { if db.RowsAffected == 0 {
e.AddError(errors.New("无权更新该数据")) _ = e.AddError(errors.New("无权更新该数据"))
return e return e
} }
return e return e
@@ -320,4 +322,4 @@ func (e *SysMenu) getByRoleName(roleName string) (Menus []models.SysMenu, err er
return return
} }
return return
} }
+5 -11
View File
@@ -21,20 +21,20 @@ type Api struct {
Errors error Errors error
} }
func (e *Api) AddError(err error) error { func (e *Api) AddError(err error) {
if e.Errors == nil { if e.Errors == nil {
e.Errors = err e.Errors = err
} else if err != nil { } else if err != nil {
e.Logger.Error(err) e.Logger.Error(err)
e.Errors = fmt.Errorf("%v; %w", e.Error, err) e.Errors = fmt.Errorf("%v; %w", e.Error, err)
} }
return e.Errors
} }
// MakeContext 设置http上下文 // MakeContext 设置http上下文
func (e *Api) MakeContext(c *gin.Context) *Api { func (e *Api) MakeContext(c *gin.Context) *Api {
fmt.Println(&c) fmt.Println(&c)
e.Context = c e.Context = c
e.Logger = api.GetRequestLogger(c)
fmt.Println(&e.Context) fmt.Println(&e.Context)
return e return e
} }
@@ -91,25 +91,19 @@ func (e Api) GetOrm() (*gorm.DB, error) {
return db, nil return db, nil
} }
// MakeLogger 设置上下文提供的日志
func (e *Api) MakeLogger() *Api {
e.Logger = api.GetRequestLogger(e.Context)
return e
}
// MakeOrm 设置Orm DB // MakeOrm 设置Orm DB
func (e *Api) MakeOrm() *Api { func (e *Api) MakeOrm() *Api {
var err error var err error
if e.Logger == nil { if e.Logger == nil {
err = errors.New("at MakeOrm logger is nil") err = errors.New("at MakeOrm logger is nil")
//e.Logger.Error(http.StatusInternalServerError, err, "at MakeOrm logger is nil") //e.Logger.Error(http.StatusInternalServerError, err, "at MakeOrm logger is nil")
_ = e.AddError(err) e.AddError(err)
return e return e
} }
db, err := pkg.GetOrm(e.Context) db, err := pkg.GetOrm(e.Context)
if err != nil { if err != nil {
e.Logger.Error(http.StatusInternalServerError, err, "数据库连接获取失败") e.Logger.Error(http.StatusInternalServerError, err, "数据库连接获取失败")
_ = e.AddError(err) e.AddError(err)
} }
e.Orm = db e.Orm = db
return e return e
@@ -133,4 +127,4 @@ func (e Api) PageOK(result interface{}, count int, pageIndex int, pageSize int,
// Custom 兼容函数 // Custom 兼容函数
func (e Api) Custom(data gin.H) { func (e Api) Custom(data gin.H) {
response.Custum(e.Context, data) response.Custum(e.Context, data)
} }