package controller import ( "github.com/gin-gonic/gin" "hepa-calc-admin-api/api/dao" "hepa-calc-admin-api/api/dto" "hepa-calc-admin-api/api/model" "hepa-calc-admin-api/api/requests" "hepa-calc-admin-api/api/responses" "hepa-calc-admin-api/global" "hepa-calc-admin-api/utils" "strconv" ) type BaseClass struct{} // GetBaseClassPage 获取基础分类列表-分页 func (b *BaseClass) GetBaseClassPage(c *gin.Context) { baseClassRequest := requests.BaseClassRequest{} req := baseClassRequest.GetBaseClassPage if err := c.ShouldBindJSON(&req); err != nil { responses.FailWithMessage(err.Error(), c) return } // 参数验证 if err := global.Validate.Struct(req); err != nil { responses.FailWithMessage(utils.Translate(err), c) return } if req.Page == 0 { req.Page = 1 } if req.PageSize == 0 { req.PageSize = 20 } // 获取分类数据 baseClassDao := dao.BaseClassDao{} baseClass, total, err := baseClassDao.GetBaseClassPageSearch(req, req.Page, req.PageSize) if err != nil { responses.FailWithMessage(err.Error(), c) return } // 处理返回值 g := dto.GetBaseClassListDto(baseClass) result := make(map[string]interface{}) result["page"] = req.Page result["page_size"] = req.PageSize result["total"] = total result["data"] = g responses.OkWithData(result, c) } // GetBaseClassList 获取基础分类列表 func (b *BaseClass) GetBaseClassList(c *gin.Context) { baseClassRequest := requests.BaseClassRequest{} req := baseClassRequest.GetBaseClassList if err := c.ShouldBindJSON(&req); err != nil { responses.FailWithMessage(err.Error(), c) return } // 参数验证 if err := global.Validate.Struct(req); err != nil { responses.FailWithMessage(utils.Translate(err), c) return } // 获取分类数据 baseClassDao := dao.BaseClassDao{} baseClass, err := baseClassDao.GetBaseClassListSearch(req) if err != nil { responses.FailWithMessage(err.Error(), c) return } // 处理返回值 g := dto.GetBaseClassListDto(baseClass) if err != nil { responses.FailWithMessage(err.Error(), c) return } responses.OkWithData(g, c) } // PutBaseClassStatus 操作基础分类状态 func (b *BaseClass) PutBaseClassStatus(c *gin.Context) { baseClassRequest := requests.BaseClassRequest{} req := baseClassRequest.PutBaseClassStatus if err := c.ShouldBind(&req); err != nil { responses.FailWithMessage(err.Error(), c) return } // 参数验证 if err := global.Validate.Struct(req); err != nil { responses.FailWithMessage(utils.Translate(err), c) return } id := c.Param("class_id") if id == "" { responses.FailWithMessage("缺少参数", c) return } // 将 id 转换为 int64 类型 classId, err := strconv.ParseInt(id, 10, 64) if err != nil { responses.Fail(c) return } // 获取基础分类数据 baseClassDao := dao.BaseClassDao{} baseClass, err := baseClassDao.GetBaseClassById(classId) if err != nil { responses.FailWithMessage("分类异常", c) return } // 检测状态 if baseClass.ClassStatus == req.ClassStatus { responses.Ok(c) return } // 开始事务 tx := global.Db.Begin() defer func() { if r := recover(); r != nil { tx.Rollback() } }() baseClassData := make(map[string]interface{}) baseClassData["class_status"] = req.ClassStatus err = baseClassDao.EditBaseClassById(tx, classId, baseClassData) if err != nil { tx.Rollback() responses.FailWithMessage("操作失败", c) return } tx.Commit() responses.Ok(c) } // GetBaseClass 获取基础分类详情 func (b *BaseClass) GetBaseClass(c *gin.Context) { id := c.Param("class_id") if id == "" { responses.FailWithMessage("缺少参数", c) return } // 将 id 转换为 int64 类型 classId, err := strconv.ParseInt(id, 10, 64) if err != nil { responses.Fail(c) return } // 获取基础分类数据 baseClassDao := dao.BaseClassDao{} baseClass, err := baseClassDao.GetBaseClassById(classId) if err != nil { responses.FailWithMessage("分类异常", c) return } // 处理返回值 g := dto.GetBaseClassDto(baseClass) responses.OkWithData(g, c) } // PutBaseClass 修改基础分类 func (b *BaseClass) PutBaseClass(c *gin.Context) { baseClassRequest := requests.BaseClassRequest{} req := baseClassRequest.PutBaseClass if err := c.ShouldBind(&req); err != nil { responses.FailWithMessage(err.Error(), c) return } // 参数验证 if err := global.Validate.Struct(req); err != nil { responses.FailWithMessage(utils.Translate(err), c) return } id := c.Param("class_id") if id == "" { responses.FailWithMessage("缺少参数", c) return } // 将 id 转换为 int64 类型 classId, err := strconv.ParseInt(id, 10, 64) if err != nil { responses.Fail(c) return } // 获取基础分类数据 baseClassDao := dao.BaseClassDao{} baseClass, err := baseClassDao.GetBaseClassById(classId) if err != nil { responses.FailWithMessage("分类异常", c) return } // 修改值 baseClassData := make(map[string]interface{}) // 分类名称 if req.ClassName != baseClass.ClassName { baseClassData["class_name"] = req.ClassName } // 图标地址 classIcon := utils.RemoveOssDomain(req.ClassIcon) if classIcon != baseClass.ClassIcon { baseClassData["class_icon"] = classIcon } // 分类简介 if req.ClassBrief != baseClass.ClassBrief { baseClassData["class_brief"] = req.ClassBrief } // 排序值 if req.Sort != baseClass.Sort { baseClassData["sort"] = req.Sort } if len(baseClassData) > 0 { // 开始事务 tx := global.Db.Begin() defer func() { if r := recover(); r != nil { tx.Rollback() } }() err = baseClassDao.EditBaseClassById(tx, classId, baseClassData) if err != nil { tx.Rollback() responses.FailWithMessage("操作失败", c) return } tx.Commit() } responses.Ok(c) } // AddBaseClass 新增基础分类 func (b *BaseClass) AddBaseClass(c *gin.Context) { baseClassRequest := requests.BaseClassRequest{} req := baseClassRequest.AddBaseClass if err := c.ShouldBind(&req); err != nil { responses.FailWithMessage(err.Error(), c) return } // 参数验证 if err := global.Validate.Struct(req); err != nil { responses.FailWithMessage(utils.Translate(err), c) return } // 开始事务 tx := global.Db.Begin() defer func() { if r := recover(); r != nil { tx.Rollback() } }() baseClass := &model.BaseClass{ ClassName: req.ClassName, ClassStatus: 1, ClassIcon: utils.RemoveOssDomain(req.ClassIcon), ClassBrief: req.ClassBrief, Sort: req.Sort, } baseClassDao := dao.BaseClassDao{} baseClass, err := baseClassDao.AddBaseClass(tx, baseClass) if err != nil { tx.Rollback() responses.FailWithMessage(err.Error(), c) return } tx.Commit() responses.Ok(c) }