|
@@ -9,7 +9,7 @@ import (
|
|
|
|
|
|
|
|
type ModelService interface {
|
|
type ModelService interface {
|
|
|
GetList(m map[string]interface{}) ([]viewmodels.ModelInfo, error)
|
|
GetList(m map[string]interface{}) ([]viewmodels.ModelInfo, error)
|
|
|
- GetTreeData() ([]viewmodels.ModelDataInfo, error)
|
|
|
|
|
|
|
+ GetTreeData() ([]viewmodels.ModelTreeInfo, error)
|
|
|
GetPage(m map[string]interface{}) (*viewmodels.PageResult, error)
|
|
GetPage(m map[string]interface{}) (*viewmodels.PageResult, error)
|
|
|
GetById(modelId int64) (*viewmodels.ModelInfo, error)
|
|
GetById(modelId int64) (*viewmodels.ModelInfo, error)
|
|
|
Create(d *datamodels.Model) error
|
|
Create(d *datamodels.Model) error
|
|
@@ -33,7 +33,7 @@ func (s *modelService) GetList(m map[string]interface{}) ([]viewmodels.ModelInfo
|
|
|
return s.modelRepo.GetList(m)
|
|
return s.modelRepo.GetList(m)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (s *modelService) GetTreeData() ([]viewmodels.ModelDataInfo, error) {
|
|
|
|
|
|
|
+func (s *modelService) GetTreeData() ([]viewmodels.ModelTreeInfo, error) {
|
|
|
modelArr, err1 := s.modelRepo.GetAll()
|
|
modelArr, err1 := s.modelRepo.GetAll()
|
|
|
if err1 != nil {
|
|
if err1 != nil {
|
|
|
return nil, err1
|
|
return nil, err1
|
|
@@ -44,22 +44,36 @@ func (s *modelService) GetTreeData() ([]viewmodels.ModelDataInfo, error) {
|
|
|
return nil, err2
|
|
return nil, err2
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- modelDataArr := make([]viewmodels.ModelDataInfo, 0)
|
|
|
|
|
|
|
+ modelDataArr := make([]viewmodels.ModelTreeInfo, 0)
|
|
|
for _, model := range modelArr {
|
|
for _, model := range modelArr {
|
|
|
|
|
|
|
|
- modelData := viewmodels.ModelDataInfo{}
|
|
|
|
|
- modelId := model.ModelId
|
|
|
|
|
- objects := make([]viewmodels.ModelObjectInfo, 0)
|
|
|
|
|
|
|
+ modelTreeData := viewmodels.ModelTreeInfo{}
|
|
|
|
|
+ modelId := model.ModelId
|
|
|
|
|
+ children := make([]viewmodels.ModelTreeNodeInfo, 0)
|
|
|
for _, object := range objectArr {
|
|
for _, object := range objectArr {
|
|
|
if object.ModelId == modelId {
|
|
if object.ModelId == modelId {
|
|
|
- objects = append(objects, object)
|
|
|
|
|
|
|
+ treeNode := viewmodels.ModelTreeNodeInfo{
|
|
|
|
|
+ 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,
|
|
|
|
|
+ }
|
|
|
|
|
+ children = append(children, treeNode)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
- modelData.ModelId = modelId
|
|
|
|
|
- modelData.ModelTitle = model.ModelTitle
|
|
|
|
|
- modelData.ModelDesc = model.ModelDesc
|
|
|
|
|
- modelData.Objects = objects
|
|
|
|
|
- modelDataArr = append(modelDataArr, modelData)
|
|
|
|
|
|
|
+ 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
|
|
return modelDataArr, nil
|