Files
full/module/base/mgt/internal/logic/department/fetch_tree.go

51 lines
1.4 KiB
Go

package department
import (
"bsm/full/module/base/mgt/internal/libs"
"bsm/full/module/base/mgt/internal/models"
"bsm/full/module/base/mgt/internal/types"
"git.apinb.com/bsm-sdk/core/errcode"
"git.apinb.com/bsm-sdk/core/infra"
"git.apinb.com/bsm-sdk/core/printer"
"github.com/gin-gonic/gin"
)
func FetchTree(c *gin.Context) {
var (
request = types.DptReq{}
data = make([]models.MgtDepartment, 0)
)
if err := c.ShouldBindJSON(&request); err != nil {
infra.Response.Error(c, errcode.ErrJsonUnmarshal)
return
}
if err := libs.ValidateStruct(&request); err != nil {
infra.Response.Error(c, errcode.ErrInvalidArgument)
return
}
if request.Workspace == "" {
printer.Error("workspace is required: workspace=%s", request.Workspace)
infra.Response.Error(c, errcode.ErrInvalidArgument)
return
}
appModel := models.MgtApplication{Workspace: request.Workspace}
appId, err := appModel.WorkspaceToId()
if err != nil {
printer.Error("workspace not found: workspace=%s", request.Workspace)
infra.Response.Error(c, errcode.ErrRecordNotFound)
return
}
if err := models.DBService.Model(&models.MgtDepartment{}).
Where("app_id = ?", appId).
Select("id", "identity", "name", "app_id", "parent_id").
Order("parent_id asc").Find(&data).Error; err != nil {
infra.Response.Error(c, errcode.ErrDB)
return
}
tree := buildDeptTree(data, 0)
infra.Response.Success(c, types.FetchResp{Data: tree})
}