eta_api/services/data/base_from_sci_hq.go

package data

import (
	"errors"
	"eta/eta_api/models/data_manage"
	"eta/eta_api/utils"
	"fmt"
	"time"
)

func AddSciHqClassify(classifyName, lang string, parentId int) (ok bool, err error) {
	var count int
	switch lang {
	case utils.EnLangVersion:
		count, err = data_manage.GetBaseFromSciHqClassifyEnCount(classifyName, parentId)
	default:
		count, err = data_manage.GetBaseFromSciHqClassifyCount(classifyName, parentId)
	}
	if err != nil {
		return
	}
	if count > 0 {
		return false, errors.New("分类已存在")
	}

	sort, err := data_manage.GetChildSciHqClassifyMaxSortById(parentId)
	if err != nil {
		return false, err
	}
	if parentId > 0 {
		count, err = data_manage.GetSciHqClassifyCountById(parentId)
		if err != nil {
			return false, err
		}
		if count == 0 {
			return false, errors.New("父分类不存在")
		}
		sciHqClassify := &data_manage.BaseFromSciHqClassify{
			ClassifyName:   classifyName,
			ClassifyNameEn: classifyName,
			ParentId:       parentId,
			Level:          2,
			Sort:           sort + 1,
			ModifyTime:     time.Now(),
			CreateTime:     time.Now(),
		}
		_, err = sciHqClassify.Add()
		if err != nil {
			return
		}
	} else {
		sciHqClassify := &data_manage.BaseFromSciHqClassify{
			ClassifyName:   classifyName,
			ClassifyNameEn: classifyName,
			ParentId:       0,
			Level:          1,
			Sort:           sort + 1,
			ModifyTime:     time.Now(),
			CreateTime:     time.Now(),
		}
		insertId, er := sciHqClassify.Add()
		if er != nil {
			return false, er
		}
		sciHqClassifyList := []*data_manage.BaseFromSciHqClassify{
			{
				ClassifyName:   "价格",
				ClassifyNameEn: "价格",
				ParentId:       int(insertId),
				Level:          2,
				Sort:           1,
				ModifyTime:     time.Now(),
				CreateTime:     time.Now(),
			},
			{
				ClassifyName:   "供应",
				ClassifyNameEn: "供应",
				ParentId:       int(insertId),
				Level:          2,
				Sort:           2,
				ModifyTime:     time.Now(),
				CreateTime:     time.Now(),
			},
			{
				ClassifyName:   "需求",
				ClassifyNameEn: "需求",
				ParentId:       int(insertId),
				Level:          2,
				Sort:           3,
				ModifyTime:     time.Now(),
				CreateTime:     time.Now(),
			},
			{
				ClassifyName:   "库存",
				ClassifyNameEn: "库存",
				ParentId:       int(insertId),
				Level:          2,
				Sort:           4,
				ModifyTime:     time.Now(),
				CreateTime:     time.Now(),
			},
		}
		err = data_manage.BatchAddSciHqClassify(sciHqClassifyList)
		if err != nil {
			return
		}
	}
	return true, nil
}

func DelSciHqClassify(classifyId int) (err error) {
	classify, err := data_manage.GetSciHqClassifyById(classifyId)
	if err != nil {
		return
	}
	var classifyIds []int
	classifyIds = append(classifyIds, classify.ClassifyId)
	if classify.ParentId == 0 {
		tmpClassifyIds, er := data_manage.GetChildSciHqClassifyIdsById(classify.ClassifyId)
		if er != nil {
			err = er
			return
		}
		classifyIds = append(classifyIds, tmpClassifyIds...)
	}
	// 获取分类下的所有指标
	count, err := data_manage.GetSciHqIndexCountByClassifyIds(classifyIds)
	if err != nil {
		err = errors.New("获取分类下的指标信息失败,Err:" + err.Error())
		return
	}
	if count > 0 {
		err = errors.New("该分类下有指标，不可删除")
		return
	}
	// 删除对应的分类
	err = data_manage.DeleteSciHqClassifyByClassifyId(classifyIds)
	if err != nil {
		return
	}
	return
}

// ResetSciHqIndex 指标数据清除分类接口
func ResetSciHqIndex(indexId int) (err error) {
	sciHqIndex, err := data_manage.GetSciHqIndexByIndexId(indexId)
	if err != nil {
		return
	}

	// 更新指标在未分类下的排序，永远排在未分类的最后一个
	//移动排序
	classifyId := 0
	updateCol := make([]string, 0)
	var currentSort, prevSort int
	currentSort = sciHqIndex.Sort
	//未分类的最大的sort值
	prevSort, err = data_manage.GetSciHqIndexMaxSortByClassifyId(classifyId)
	if err != nil {
		err = errors.New("获取上一个兄弟节点指标信息失败,Err:" + err.Error())
		return
	}

	//如果前面一个节点的sort值比当前大，则当前节点往下移动
	if prevSort >= currentSort {
		//往下移动
		err = data_manage.MoveDownSciIndexBySort(classifyId, prevSort, currentSort)
		if err != nil {
			err = errors.New("向下移动出错:" + err.Error())
			return
		}
		sciHqIndex.Sort = prevSort
	}

	//更新
	sciHqIndex.ModifyTime = time.Now()
	sciHqIndex.ClassifyId = classifyId
	updateCol = append(updateCol, "Sort", "ModifyTime", "ClassifyId")
	err = sciHqIndex.Update(updateCol)
	if err != nil {
		err = errors.New("移动失败,Err:" + err.Error())
		return
	}
	return
}

// 批量删除卓创红期指标
func BatchDelSciHqData(indexIds []int) (existIndex []*data_manage.BaseFromSciHqIndex, err error) {
	sciHqIndexList, err := data_manage.GetSciHqIndexListByIndexIds(indexIds)
	if err != nil {
		return
	}

	indexCodes := make([]string, 0)
	codeToIndex := make(map[string]*data_manage.BaseFromSciHqIndex)
	for _, v := range sciHqIndexList {
		indexCodes = append(indexCodes, v.IndexCode)
		codeToIndex[v.IndexCode] = v
	}

	edbInfoList, err := data_manage.GetEdbInfoListByEdbCodes(utils.DATA_SOURCE_SCI_HQ, indexCodes)
	if err != nil {
		return
	}
	for _, v := range edbInfoList {
		if index, ok := codeToIndex[v.EdbCode]; ok {
			existIndex = append(existIndex, index)
			delete(codeToIndex, v.EdbCode)
		}
	}

	// 删除对应的指标
	err = data_manage.DeleteSciHqIndexByIds(indexIds)
	if err != nil {
		return
	}
	return
}

// DelSciHqData 删除卓创红期指标
func DelSciHqData(indexId int) (err error, errMsg string) {
	baseFromSciHqIndex, err := data_manage.GetSciHqIndexByIndexId(indexId)
	if err != nil {
		errMsg = `获取数据失败`
		if err.Error() == utils.ErrNoRow() {
			errMsg = `该指标未入库`
			err = nil
		}
		return
	}

	// 获取已经加入到EDB指标库的Sci指标
	edbInfo, err := data_manage.GetEdbInfoByEdbCode(utils.DATA_SOURCE_SCI_HQ, baseFromSciHqIndex.IndexCode)
	if err != nil && err.Error() != utils.ErrNoRow() {
		errMsg = "删除失败"
		err = errors.New("获取分类下的指标信息失败,Err:" + err.Error())
		return
	}
	if edbInfo != nil {
		errMsg = "当前指标已被引用，不可删除"
		err = nil
		return
	}

	// 删除对应的指标
	err = data_manage.DeleteSciHqIndexById(indexId)
	if err != nil {
		return
	}

	return
}

// EditSciHqIndex 编辑卓创红期指标
func EditSciHqIndex(indexId, classifyId int) (baseFromSciHqIndex *data_manage.BaseFromSciHqIndex, errMsg string, err error) {
	baseFromSciHqIndex, err = data_manage.GetSciHqIndexByIndexId(indexId)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			errMsg = `该指标未入库`
			err = nil
		}
		return
	}
	classify, err := data_manage.GetSciHqClassifyById(classifyId)
	if err != nil {
		if err.Error() == utils.ErrNoRow() {
			errMsg = `该分类不存在或已删除`
			err = nil
		}
		return
	}
	if classify.Level == 1 { // 顶级分类
		errMsg = `不合法的编辑`
		err = nil
		return
	}
	maxSort, err := data_manage.GetChildSciHqClassifyMaxSortById(classifyId)
	if err != nil {
		return
	}

	baseFromSciHqIndex.ClassifyId = classifyId
	baseFromSciHqIndex.Sort = maxSort + 1
	baseFromSciHqIndex.ModifyTime = time.Now()
	err = baseFromSciHqIndex.Update([]string{"classify_id", "sort", "modify_time"})
	return
}

type SciHqIndexSource2EdbReq struct {
	EdbCode       string
	EdbName       string
	Frequency     string
	Unit          string
	ClassifyId    int
	AdminId       int
	AdminRealName string
}

// SciHqIndexSource2Edb 新增卓创红期数据从数据源到指标库
func SciHqIndexSource2Edb(req SciHqIndexSource2EdbReq, lang string) (edb *data_manage.EdbInfo, errMsg string, skip bool, err error) {
	if req.EdbCode == "" {
		err = fmt.Errorf("指标ID为空")
		return
	}
	defer func() {
		if err != nil {
			tips := fmt.Sprintf("SciHqIndexSource2Edb新增失败, Err: %s", err.Error())
			fmt.Println(tips)
			utils.FileLog.Info(tips)
		}
	}()
	source := utils.DATA_SOURCE_SCI_HQ

	// 是否新增过指标
	exist, e := data_manage.GetEdbInfoByEdbCode(source, req.EdbCode)
	if e != nil && e.Error() != utils.ErrNoRow() {
		err = fmt.Errorf("获取指标是否存在失败, err: %s", e.Error())
		return
	}
	if exist != nil {
		skip = true
		return
	}

	// 开始结束时间
	var startDate, endDate string

	// 新增指标库
	edbInfo, e, msg, _ := EdbInfoAdd(source, utils.DATA_SUB_SOURCE_EDB, req.ClassifyId, req.EdbCode, req.EdbName, req.Frequency, req.Unit, startDate, endDate, req.AdminId, req.AdminRealName, lang)
	if e != nil {
		errMsg = msg
		err = fmt.Errorf("EdbInfo: 新增指标失败, err: %s", e.Error())
		return
	}

	edb = edbInfo

	return
}

// MoveSciHqClassify 移动卓创红期分类
func MoveSciHqClassify(classifyId, parentClassifyId, prevClassifyId, nextClassifyId int) (err error, errMsg string) {
	//判断分类是否存在
	classifyInfo, err := data_manage.GetSciHqClassifyById(classifyId)
	if err != nil {
		errMsg = "移动失败"
		err = errors.New("获取分类信息失败,Err:" + err.Error())
		return
	}

	updateCol := make([]string, 0)

	//判断上级id是否一致，如果不一致的话，那么需要移动该分类层级
	if classifyInfo.ParentId != parentClassifyId && parentClassifyId != 0 {
		parentClassifyInfo, tmpErr := data_manage.GetSciHqClassifyById(parentClassifyId)
		if tmpErr != nil {
			errMsg = "移动失败"
			err = errors.New("获取上级分类信息失败,Err:" + tmpErr.Error())
			return
		}
		classifyInfo.ParentId = parentClassifyInfo.ClassifyId
		classifyInfo.Level = parentClassifyInfo.Level + 1
		updateCol = append(updateCol, "ParentId", "Level")
	}
	var currentSort, prevSort, nextSort int
	currentSort = classifyInfo.Sort

	var prevClassify *data_manage.BaseFromSciHqClassify
	var nextClassify *data_manage.BaseFromSciHqClassify
	//如果有传入 上一个兄弟节点分类id
	if prevClassifyId > 0 {
		prevClassify, err = data_manage.GetSciHqClassifyById(prevClassifyId)
		if err != nil {
			if err.Error() == utils.ErrNoRow() {
				errMsg = "移动失败"
				err = errors.New("上一个兄弟节点分类信息不存在" + err.Error())
				return
			}
			errMsg = "移动失败"
			err = errors.New("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
			return
		}
		prevSort = prevClassify.Sort
	}
	if nextClassifyId > 0 {
		//下一个兄弟节点
		nextClassify, err = data_manage.GetSciHqClassifyById(nextClassifyId)
		if err != nil {
			if err.Error() == utils.ErrNoRow() {
				errMsg = "移动失败"
				err = errors.New("下一个兄弟节点分类信息不存在" + err.Error())
				return
			}
			errMsg = "移动失败"
			err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
			return
		}
		nextSort = nextClassify.Sort
	}
	// 修改历史数据中的排序为0的情况
	if prevSort == 0 && nextSort == 0 { //目标是让当前分类处于目录的最顶部
		//更新为0排序的数据为当前最小排序
		if nextClassify != nil {
			minSort, e := data_manage.GetSciHqIndexClassifyMinSort(parentClassifyId)
			if e != nil {
				errMsg = "移动失败"
				err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + e.Error())
				return
			}
			if minSort > 1 {
				minSort -= 1
			}
			nextClassify.Sort = minSort
			err = nextClassify.Update([]string{"sort"})
			if err != nil {
				errMsg = "移动失败"
				err = errors.New("更新下一个兄弟节点分类信息失败,Err:" + err.Error())
				return
			}
			nextSort = minSort
		}
	}
	//移到两个排序值中间操作
	if prevSort >= currentSort {
		//往下移动
		err = data_manage.MoveDownSciHqIndexClassifyBySort(parentClassifyId, prevSort, currentSort)
		if err != nil {
			err = errors.New("向下移动出错:" + err.Error())
			return
		}
		classifyInfo.Sort = prevSort
	} else if nextSort <= currentSort && nextSort != 0 {
		//往上移动
		err = data_manage.MoveUpSciHqIndexClassifyBySort(parentClassifyId, nextSort, currentSort)
		if err != nil {
			err = errors.New("向上移动出错:" + err.Error())
			return
		}
		classifyInfo.Sort = nextSort
	}
	classifyInfo.ModifyTime = time.Now()
	updateCol = append(updateCol, "Sort", "ModifyTime")
	err = classifyInfo.Update(updateCol)
	if err != nil {
		errMsg = "移动失败"
		err = errors.New("修改失败,Err:" + err.Error())
		return
	}
	return
}

// MoveSciHqData 移动卓创红期指标
func MoveSciHqData(indexId, classifyId, prevIndexId, nextIndexId int) (err error, errMsg string) {
	//分类信息
	sciHqIndex, err := data_manage.GetSciHqIndexByIndexId(indexId)
	if err != nil {
		errMsg = `获取数据失败`
		if err.Error() == utils.ErrNoRow() {
			errMsg = `该指标未入库`
			err = nil
		}
		return
	}

	//判断分类是否存在
	if classifyId > 0 {
		_, err = data_manage.GetSciHqClassifyById(classifyId)
		if err != nil {
			errMsg = "移动失败"
			err = errors.New("获取分类信息失败,Err:" + err.Error())
			if err.Error() == utils.ErrNoRow() {
				errMsg = "找不到该分类"
				err = nil
			}
			return
		}
	}

	//如果改变了分类，那么移动该图表数据
	if sciHqIndex.ClassifyId != classifyId {
		tmpErr := data_manage.MoveSciHqIndex(indexId, classifyId)
		if tmpErr != nil {
			errMsg = "移动失败"
			err = errors.New("移动失败,Err:" + tmpErr.Error())
			return
		}
	}

	//移动排序
	updateCol := make([]string, 0)
	var currentSort, prevSort, nextSort int
	currentSort = sciHqIndex.Sort
	//如果有传入 上一个兄弟节点分类id
	var prevIndex *data_manage.BaseFromSciHqIndex
	var nextIndex *data_manage.BaseFromSciHqIndex
	if prevIndexId > 0 {
		prevIndex, err = data_manage.GetSciHqIndexByIndexId(prevIndexId)
		if err != nil {
			if err.Error() == utils.ErrNoRow() {
				errMsg = "移动失败"
				err = errors.New("上一个兄弟节点指标信息不存在" + err.Error())
				return
			}
			errMsg = "移动失败"
			err = errors.New("获取上一个兄弟节点指标信息失败,Err:" + err.Error())
			return
		}
		prevSort = prevIndex.Sort
	}

	if nextIndexId > 0 {
		//下一个兄弟节点
		nextIndex, err = data_manage.GetSciHqIndexByIndexId(nextIndexId)
		if err != nil {
			if err.Error() == utils.ErrNoRow() {
				errMsg = "移动失败"
				err = errors.New("下一个兄弟节点指标信息不存在" + err.Error())
				return
			}
			errMsg = "移动失败"
			err = errors.New("获取下一个兄弟节点指标信息失败,Err:" + err.Error())
			return
		}
		nextSort = nextIndex.Sort
	}

	// 修改历史数据中的排序为0的情况
	if prevSort == 0 && nextSort == 0 { //目标是让当前分类处于目录的最顶部, 但是历史数据里下个节点的sort为0的情况
		//更新为0排序的数据为当前最小排序
		if nextIndex != nil {
			minSort, e := data_manage.GetSciHqIndexMinSortByClassifyId(classifyId)
			if e != nil {
				errMsg = "移动失败"
				err = errors.New("获取下一个兄弟节点分类信息失败,Err:" + e.Error())
				return
			}
			if minSort > 1 {
				minSort -= 1
			}
			nextIndex.Sort = minSort
			err = nextIndex.Update([]string{"Sort"})
			if err != nil {
				errMsg = "移动失败"
				err = errors.New("更新下一个兄弟节点分类信息失败,Err:" + err.Error())
				return
			}
			nextSort = minSort
		}
	}
	//移到两个排序值中间操作
	if prevSort >= currentSort {
		//往下移动
		err = data_manage.MoveDownSciHqIndexBySort(classifyId, prevSort, currentSort)
		if err != nil {
			err = errors.New("向下移动出错:" + err.Error())
			return
		}
		sciHqIndex.Sort = prevSort
	} else if nextSort <= currentSort && nextSort != 0 {
		//往上移动
		err = data_manage.MoveUpSciHqIndexBySort(classifyId, nextSort, currentSort)
		if err != nil {
			err = errors.New("向上移动出错:" + err.Error())
			return
		}
		sciHqIndex.Sort = nextSort
	}

	//更新
	sciHqIndex.ModifyTime = time.Now()
	updateCol = append(updateCol, "Sort", "ModifyTime")
	err = sciHqIndex.Update(updateCol)
	if err != nil {
		errMsg = "移动失败"
		err = errors.New("修改失败,Err:" + err.Error())
		return
	}
	return
}