xps-server-g5/pkg/biz/service/biz_service.go

package service

import (
    "xps/cmd/datasource"
    "xps/pkg/biz/datamodel"
    "xps/pkg/biz/repository"
    "xps/pkg/biz/viewmodel"
    "xps/pkg/common"
)

type BizService interface {
    GetList(m map[string]interface{}) ([]viewmodel.BizInfo, error)
    GetTreeData() ([]viewmodel.BizTreeInfo, error)
    GetPage(m map[string]interface{}) (*common.PageResult, error)
    GetById(bizId int64) (*viewmodel.BizInfo, error)
    Create(d *datamodel.Biz) error
    Update(d *datamodel.Biz) error
    Delete(m map[string]interface{}) error
}

type bizService struct {
    bizRepo       *repository.BizRepo
    bizObjectRepo *repository.BizObjectRepo
}

func NewBizService() BizService {
    return &bizService{
        bizRepo:       repository.NewBizRepo(datasource.InstanceMaster()),
        bizObjectRepo: repository.NewBizObjectRepo(datasource.InstanceMaster()),
    }
}

func (s *bizService) GetTreeData() ([]viewmodel.BizTreeInfo, error) {
    bizArr, err1 := s.bizRepo.GetAll()
    if err1 != nil {
        return nil, err1
    }

    bizObjectArr, err2 := s.bizObjectRepo.GetAll()
    if err2 != nil {
        return nil, err2
    }

    bizDataArr := make([]viewmodel.BizTreeInfo, 0)
    for _, biz := range bizArr {

        bizData := viewmodel.BizTreeInfo{}
        bizId := biz.BizId
        treeNodes := make([]viewmodel.BizTreeNodeInfo, 0)
        for _, object := range bizObjectArr {
            if object.BizId == bizId {
                treeNode := viewmodel.BizTreeNodeInfo{
                    Label:       object.ObjectTitle,
                    Value:       object.ObjectId,
                    IsLeaf:      1,
                    ObjectId:    object.ObjectId,
                    ObjectType:  object.ObjectType,
                    ObjectTitle: object.ObjectTitle,
                    ObjectCode:  object.ObjectCode,
                    IsFixed:     object.IsFixed,
                    Comment:     object.Comment,
                }

                treeNodes = append(treeNodes, treeNode)
            }
        }
        bizData.Label = biz.BizTitle
        bizData.Value = bizId
        bizData.IsLeaf = 0
        bizData.BizId = bizId
        bizData.BizTitle = biz.BizTitle
        bizData.BizDesc = biz.BizDesc
        bizData.Children = treeNodes
        bizDataArr = append(bizDataArr, bizData)
    }

    return bizDataArr, nil
}

func (s *bizService) GetList(m map[string]interface{}) ([]viewmodel.BizInfo, error) {
    return s.bizRepo.GetList(m)
}

func (s *bizService) GetPage(m map[string]interface{}) (*common.PageResult, error) {
    return s.bizRepo.GetPage(m)
}

func (s *bizService) GetById(bizId int64) (*viewmodel.BizInfo, error) {
    return s.bizRepo.GetById(bizId)
}

func (s *bizService) Create(m *datamodel.Biz) error {
    m.BizId = NewID()
    return s.bizRepo.Create(m)
}

func (s *bizService) Update(m *datamodel.Biz) error {
    return s.bizRepo.Update(m)
}

func (s *bizService) Delete(m map[string]interface{}) error {
    return s.bizRepo.Delete(m)
}