xps-server-g5/pkg/model/service/model_service.go

package service

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

type ModelService interface {
    GetList(m map[string]interface{}) ([]viewmodel.ModelInfo, error)
    GetTreeData() ([]viewmodel.ModelTreeInfo, error)
    GetPage(m map[string]interface{}) (*common.PageResult, error)
    GetById(modelId int64) (*viewmodel.ModelInfo, error)
    Create(d *datamodel.Model) error
    Update(d *datamodel.Model) error
    Delete(m map[string]interface{}) error
}

type modelService struct {
    modelRepo  *repository.ModelRepo
    objectRepo *repository.ModelObjectRepo
}

func NewModelService() ModelService {
    return &modelService {
        modelRepo:  repository.NewModelRepo(datasource.InstanceMaster()),
        objectRepo: repository.NewModelObjectRepo(datasource.InstanceMaster()),
    }
}

func (s *modelService) GetList(m map[string]interface{}) ([]viewmodel.ModelInfo, error) {
    return s.modelRepo.GetList(m)
}

func (s *modelService) GetTreeData() ([]viewmodel.ModelTreeInfo, error) {
    modelArr, err1 := s.modelRepo.GetAll()
    if err1 != nil {
        return nil, err1
    }

    objectArr, err2 := s.objectRepo.GetAll()
    if err2 != nil {
        return nil, err2
    }

    modelDataArr := make([]viewmodel.ModelTreeInfo, 0)
    for _, model := range modelArr {

        modelTreeData := viewmodel.ModelTreeInfo{}
        modelId  := model.ModelId
        children := make([]viewmodel.ModelTreeNodeInfo, 0)
        for _, object := range objectArr {
            if object.ModelId == modelId {
                treeNode := viewmodel.ModelTreeNodeInfo{
                    Label:       object.ObjectTitle,
                    Value:       object.ObjectId,
                    ObjectId:    object.ObjectId,
                    ObjectType:  object.ObjectType,
                    ObjectTitle: object.ObjectTitle,
                    ObjectCode:  object.ObjectCode,
                    IsFixed:     object.IsFixed,
                    IsLeaf:      1,
                    Comment:     object.Comment,
                }
                children = append(children, treeNode)
            }
        }
        modelTreeData.Label      = model.ModelTitle
        modelTreeData.Value      = model.ModelId
        modelTreeData.IsLeaf     = 0
        modelTreeData.ModelId    = modelId
        modelTreeData.ModelTitle = model.ModelTitle
        modelTreeData.ModelDesc  = model.ModelDesc
        modelTreeData.Children   = children
        modelDataArr = append(modelDataArr, modelTreeData)
    }

    return modelDataArr, nil
}

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

func (s *modelService) GetById(modelId int64) (*viewmodel.ModelInfo, error) {
    return s.modelRepo.GetById(modelId)
}

func (s *modelService) Create(m *datamodel.Model) error {
    m.ModelId = NewID()
    return s.modelRepo.Create(m)
}

func (s *modelService) Update(m *datamodel.Model) error {
    return s.modelRepo.Update(m)
}

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