package controllers import ( "errors" "fmt" "github.com/gin-gonic/gin" "gorm.io/gorm" "wms/extend/code" "wms/extend/util" "wms/models" "wms/pkg/convertx" "wms/pkg/structx" "wms/request" ) type DepartmentController struct{} // Add // @Tags 部门信息 // @Summary 添加部门信息 // @Produce application/json // @Param object body request.AddDepartment true "部门信息信息" // @Success 200 {object} util.Response "成功" // @Router /api-s/v1/organize/department [post] func (slf DepartmentController) Add(c *gin.Context) { var reqParams request.AddDepartment var params models.Department if err := c.BindJSON(&reqParams); err != nil { util.ResponseFormat(c, code.RequestParamError, "参数解析失败,数据类型错误") return } if err := structx.AssignTo(reqParams, ¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, "数据转换错误") return } if err := slf.ParamsCheck(params); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } if err := models.NewDepartmentSearch().Create(¶ms); err != nil { util.ResponseFormat(c, code.SaveFail, "插入失败") return } util.ResponseFormat(c, code.Success, "添加成功") } // Update // @Tags 部门信息 // @Summary 编辑部门信息 // @Produce application/json // @Param object body request.UpdateDepartment true "部门信息信息" // @Success 200 {object} util.Response "成功" // @Router /api-s/v1/organize/department/{id} [put] func (slf DepartmentController) Update(c *gin.Context) { idStr := c.Param("id") if idStr == "0" || idStr == "" { util.ResponseFormat(c, code.RequestParamError, "空的记录id") return } var ( reqParams request.UpdateDepartment params models.Department ) reqParams.ID = uint(convertx.Atoi(idStr)) if err := c.BindJSON(&reqParams); err != nil { util.ResponseFormat(c, code.RequestParamError, fmt.Sprintf("参数解析失败: %v"+err.Error())) return } if err := structx.AssignTo(reqParams, ¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, fmt.Sprintf("数据转换错误: %v", err.Error())) return } if err := slf.ParamsCheck(params); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } _, err := models.NewDepartmentSearch().SetID(params.ID).First() if err != nil { util.ResponseFormat(c, code.RequestParamError, "ID不存在") return } if params.ParentId != 0 { //判断有没有环(修改上级部门时错选了自己的下级部门) parentId := params.ParentId for { if parentId == params.ID { util.ResponseFormat(c, code.RequestParamError, "不能把自己或自己的下级部门当做上级部门") return } if parentId == 0 { break } parent, err := models.NewDepartmentSearch().SetID(parentId).First() if err == gorm.ErrRecordNotFound { break } else if err != nil { util.ResponseFormat(c, code.RequestParamError, "数据查询错误") return } parentId = parent.ParentId } } err = models.NewDepartmentSearch().SetID(params.ID).Save(¶ms) if err != nil { util.ResponseFormat(c, code.RequestParamError, "修改失败") return } util.ResponseFormat(c, code.UpdateSuccess, "更新成功") } func (slf DepartmentController) ParamsCheck(params models.Department) (err error) { if params.ParentId != 0 { _, err := models.NewDepartmentSearch().SetID(params.ParentId).First() if err != nil { return errors.New("上级部门不存在") } } dep, err := models.NewDepartmentSearch().SetNumber(params.Number).First() if err != gorm.ErrRecordNotFound && dep != nil && dep.ID != params.ID { return errors.New("部门编号重复") } _, err = models.NewDepartmentSearch().SetName(params.Name).First() if err != gorm.ErrRecordNotFound && dep != nil && dep.ID != params.ID { return errors.New("部门名称重复") } return nil } // List // @Tags 部门信息 // @Summary 查询部门信息列表 // @Produce application/json // @Param object query request.GetDepartmentList true "查询参数" // @Success 200 {object} util.ResponseList{data=[]models.Department} "成功" // @Router /api-s/v1/organize/department [get] func (slf DepartmentController) List(c *gin.Context) { var params request.GetDepartmentList if err := c.ShouldBindQuery(¶ms); err != nil { util.ResponseFormat(c, code.RequestParamError, err.Error()) return } treeMap := make(map[uint][]*models.Department) allMenus, err := models.NewDepartmentSearch().FindNotTotal() for _, v := range allMenus { treeMap[v.ParentId] = append(treeMap[v.ParentId], v) } menus := treeMap[0] //parentId == 0 代表1级部门(没有上级部门) for i := 0; i < len(menus); i++ { _ = getMenuChildrenList(menus[i], treeMap) } if err != nil { util.ResponseFormat(c, code.RequestParamError, "查找失败") return } util.ResponseFormatList(c, code.Success, menus, len(menus)) } func getMenuChildrenList(menu *models.Department, treeMap map[uint][]*models.Department) (err error) { menu.Children = treeMap[menu.ID] for i := 0; i < len(menu.Children); i++ { err = getMenuChildrenList(menu.Children[i], treeMap) } return err } // Delete // @Tags 部门信息 // @Summary 编辑部门信息 // @Produce application/json // @Param object body request.UpdateDepartment true "部门信息信息" // @Success 200 {object} util.Response "成功" // @Router /api-s/v1/organize/department/{id} [delete] func (slf DepartmentController) Delete(c *gin.Context) { idStr := c.Param("id") if idStr == "0" || idStr == "" { util.ResponseFormat(c, code.RequestParamError, "空的记录id") return } id := convertx.StringToUInt(idStr) _, err := models.NewDepartmentSearch().SetParentId(id).First() if err != gorm.ErrRecordNotFound { util.ResponseFormat(c, code.RequestParamError, "请先删除下级部门") return } err = models.NewDepartmentSearch().SetID(id).Delete() if err != nil { util.ResponseFormat(c, code.RequestParamError, "删除失败") return } util.ResponseFormat(c, code.UpdateSuccess, "删除成功") }