eta_api/services/classify.go

package services

import (
	"errors"
	"eta_gn/eta_api/models"
	"eta_gn/eta_api/models/report_approve"
	"eta_gn/eta_api/utils"
	"fmt"
	"sort"
	"time"
)

func MoveReportClassify(req models.ClassifyMoveReq) (err error, errMsg string) {
	classifyId := req.ClassifyId
	prevClassifyId := req.PrevClassifyId
	nextClassifyId := req.NextClassifyId

	var (
		classifyInfo *models.Classify
		prevClassify *models.Classify
		nextClassify *models.Classify

		prevSort int
		nextSort int
	)

	classifyInfo, err = models.GetClassifyById(classifyId)
	if err != nil {
		if utils.IsErrNoRow(err) {
			errMsg = "分类不存在, 请刷新页面"
			err = fmt.Errorf("获取分类信息失败,Err:" + err.Error())
			return
		}
		errMsg = "移动失败"
		err = fmt.Errorf("获取分类信息失败,Err:" + err.Error())
		return
	} else if classifyInfo.Id == 0 {
		errMsg = "分类不存在, 请刷新页面"
		err = fmt.Errorf("获取分类信息失败,Err:" + err.Error())
		return
	}

	parentClassifyId := classifyInfo.ParentId
	if prevClassifyId > 0 {
		prevClassify, err = models.GetClassifyById(prevClassifyId)
		if err != nil {
			if utils.IsErrNoRow(err) {
				errMsg = "上一个分类不存在, 请刷新页面"
				err = fmt.Errorf("获取分类信息失败,Err:" + err.Error())
				return
			}
			errMsg = "移动失败"
			err = fmt.Errorf("获取上一个兄弟节点分类信息失败,Err:" + err.Error())
			return
		}
		if prevClassify.ParentId != parentClassifyId {
			errMsg = "禁止拖动到其他节点"
			err = fmt.Errorf(errMsg)
			return
		}
		prevSort = prevClassify.Sort
	}

	if nextClassifyId > 0 {
		nextClassify, err = models.GetClassifyById(nextClassifyId)
		if err != nil {
			if utils.IsErrNoRow(err) {
				errMsg = "下一个分类不存在, 请刷新页面"
				err = fmt.Errorf("获取分类信息失败,Err:" + err.Error())
				return
			}
			errMsg = "移动失败"
			err = fmt.Errorf("获取下一个兄弟节点分类信息失败,Err:" + err.Error())
			return
		}
		if nextClassify.ParentId != parentClassifyId {
			errMsg = "禁止拖动到其他节点"
			err = fmt.Errorf(errMsg)
			return
		}
		nextSort = nextClassify.Sort
	}

	err, errMsg = moveReportClassify(classifyInfo, prevClassify, nextClassify, parentClassifyId, prevSort, nextSort)
	return
}

func moveReportClassify(classifyInfo, prevClassify, nextClassify *models.Classify, parentId, prevSort, nextSort int) (err error, errMsg string) {
	ob := new(models.Classify)
	updateCol := make([]string, 0)

	if classifyInfo.ParentId != parentId {
		errMsg = "移动失败"
		err = fmt.Errorf("不支持目录层级变更")
		return
	}

	if prevSort > 0 {
		if nextSort > 0 {
			if prevSort == nextSort || prevSort == classifyInfo.Sort {
				updateSortStr := `sort + 2`

				if prevClassify != nil {
					_ = models.UpdateClassifySortByParentId(parentId, prevClassify.Id, prevClassify.Sort, updateSortStr)
				} else {
					_ = models.UpdateClassifySortByParentId(parentId, 0, prevSort, updateSortStr)
				}

			} else {
				if nextSort-prevSort == 1 {
					updateSortStr := `sort + 1`

					if prevClassify != nil {
						_ = models.UpdateClassifySortByParentId(parentId, prevClassify.Id, prevSort, updateSortStr)
					} else {
						_ = models.UpdateClassifySortByParentId(parentId, 0, prevSort, updateSortStr)
					}

				}
			}
		}

		classifyInfo.Sort = prevSort + 1
		classifyInfo.ModifyTime = time.Now()
		updateCol = append(updateCol, "Sort", "ModifyTime")
	} else if prevClassify == nil && nextClassify == nil && parentId > 0 {
		var maxSort int
		maxSort, err = ob.GetMaxSortByParentId(parentId)
		if err != nil {
			errMsg = "移动失败"
			err = fmt.Errorf("查询组内排序信息失败,Err:" + err.Error())
			return
		}
		classifyInfo.Sort = maxSort + 1 //那就是排在组内最后一位
		classifyInfo.ModifyTime = time.Now()
		updateCol = append(updateCol, "Sort", "ModifyTime")
	} else {
		firstPermission, tmpErr := ob.GetFirstClassifyByParentId(parentId)
		if tmpErr != nil && !utils.IsErrNoRow(tmpErr) {
			errMsg = "移动失败"
			err = fmt.Errorf("获取获取当前父级分类下的排序第一条的分类信息失败,Err:" + tmpErr.Error())
			return
		}

		if firstPermission != nil && firstPermission.Id > 0 && firstPermission.Sort == 0 {
			updateSortStr := ` sort + 1 `
			_ = models.UpdateClassifySortByParentId(parentId, firstPermission.Id-1, 0, updateSortStr)
		}

		classifyInfo.Sort = 0 //那就是排在第一位
		classifyInfo.ModifyTime = time.Now()
		updateCol = append(updateCol, "Sort", "ModifyTime")
	}

	if len(updateCol) > 0 {
		err = classifyInfo.UpdateClassify(updateCol)
		if err != nil {
			errMsg = "移动失败"
			err = fmt.Errorf("修改失败,Err:" + err.Error())
			return
		}
	}
	return
}

func AddReportClassify(classifyName string, parentId int, chartPermissionIdList []int) (err error, errMsg string, isSendEmail bool) {
	isSendEmail = true
	errMsg = `添加失败`
	item, err := models.GetClassifyByName(classifyName, parentId)
	if err != nil && !utils.IsErrNoRow(err) {
		errMsg = "获取分类信息失败"
		if utils.IsErrNoRow(err) {
			errMsg = "分类名称:" + classifyName + "已存在"
			isSendEmail = false
		}
		return
	}
	if item != nil && item.Id > 0 {
		errMsg = "分类名称:" + classifyName + "已存在"
		isSendEmail = false
		err = errors.New(errMsg)
		return
	}

	level := 1

	var parentClassifyItem *models.Classify
	var childClassifyCount int

	if parentId > 0 {
		parentClassifyItem, err = models.GetClassifyById(parentId)
		if err != nil {
			errMsg = "获取父级分类信息失败"
			if utils.IsErrNoRow(err) {
				errMsg = "父级分类不存在"
			}
			return
		}
		level = parentClassifyItem.Level + 1

		if level > 3 {
			errMsg = "分类层级不可超过三级"
			isSendEmail = false
			return
		}

		err, errMsg = checkClassifyApprove(parentClassifyItem)
		if err != nil {
			return
		}

		childClassifyCount, err = models.GetCountClassifyChildByParentId(parentId)
		if err != nil {
			errMsg = "获取父级分类的子分类信息失败"
			return
		}

	}

	nowTime := time.Now().Local()
	classify := new(models.Classify)

	maxSort, err := classify.GetMaxSort()
	if err != nil {
		errMsg = "操作失败"
		err = errors.New("查询品种排序失败, Err: " + err.Error())
		return
	}
	classify.ClassifyName = classifyName
	classify.ParentId = parentId
	classify.CreateTime = nowTime
	classify.ModifyTime = nowTime
	classify.Sort = maxSort + 1
	classify.Enabled = 1
	classify.ShowType = 1             //默认列表格式
	classify.ReportDetailShowType = 1 //默认列表格式
	classify.IsShow = 1
	classify.Level = level

	err = models.AddClassify(classify)
	if err != nil {
		return
	}
	if parentClassifyItem != nil {
		parentClassifyItem.HasChild = 1
		parentClassifyItem.UpdateClassify([]string{"HasChild"})

		if childClassifyCount <= 0 {
			tmpErr := inheritReportChapterType(parentId, classify.Id)
			if tmpErr != nil {
				return
			}

			go inheritReportApproveFlow(parentClassifyItem, classify)

			moveReportByAddClassify(parentClassifyItem, classify)
		}
	}

	return
}

func checkClassifyApprove(currClassify *models.Classify) (err error, errMsg string) {
	errMsg = `判断是否有审批流关联失败`
	var firstClassifyId, secondClassifyId int
	if currClassify.ParentId > 0 {
		parentClassifyItem, tmpErr := models.GetClassifyById(currClassify.ParentId)
		if tmpErr != nil {
			err = tmpErr
			errMsg = "获取父级分类信息失败"
			if utils.IsErrNoRow(tmpErr) {
				errMsg = "父级分类不存在"
			}
			return
		}
		firstClassifyId = parentClassifyItem.Id
		secondClassifyId = currClassify.Id
	} else {
		firstClassifyId = currClassify.Id
	}
	{
		flowOb := new(report_approve.ReportApproveFlow)

		existCond := fmt.Sprintf(` AND %s = ? AND %s = ? AND %s = ? AND %s = ?`, report_approve.ReportApproveFlowCols.ReportType, report_approve.ReportApproveFlowCols.ClassifyFirstId, report_approve.ReportApproveFlowCols.ClassifySecondId, report_approve.ReportApproveFlowCols.ClassifyThirdId)
		existPars := make([]interface{}, 0)
		existPars = append(existPars, report_approve.FlowReportTypeChinese, firstClassifyId, secondClassifyId, 0)
		flowItem, e := flowOb.GetItemByCondition(existCond, existPars, "")
		if e != nil {
			if !utils.IsErrNoRow(e) {
				err = errors.New("获取审批流是否已存在失败, Err: " + e.Error())
				return
			}
			err = nil
			return
		}
		if flowItem == nil {
			return
		}

		approvingOb := new(report_approve.ReportApprove)
		approvingCond := fmt.Sprintf(` AND %s = ? AND %s = ? AND %s = ?`, report_approve.ReportApproveCols.FlowId, report_approve.ReportApproveCols.FlowVersion, report_approve.ReportApproveCols.State)
		approvingPars := make([]interface{}, 0)
		approvingPars = append(approvingPars, flowItem.ReportApproveFlowId, flowItem.CurrVersion, report_approve.ReportApproveStateApproving)
		count, e := approvingOb.GetCountByCondition(approvingCond, approvingPars)
		if e != nil {
			err = errors.New("获取审批流关联进行中的审批数失败. Err: " + e.Error())
			return
		}
		if count > 0 {
			errMsg = "当前有未走完流程的报告，请走完流程后再做变更"
			err = errors.New(errMsg)
			return
		}
	}

	return
}

func moveReportByAddClassify(parentClassifyInfo, currClassifyInfo *models.Classify) (err error) {
	defer func() {
		if err != nil {
			utils.FileLog.Error(fmt.Sprint("历史报告更改分类失败，父级分类ID：", parentClassifyInfo.Id, "；当前分类ID:", currClassifyInfo.Id, "；错误信息：", err.Error()))
		}
	}()
	if currClassifyInfo.Level > 3 {
		err = errors.New("父级分类不支持三级分类以上")
		return
	}


	var condition, updateStr string
	pars := make([]interface{}, 0)
	switch currClassifyInfo.Level {
	case 3: // 当前分类是3级分类
		updateStr += ` classify_id_third = ?,classify_name_third = ?`
		condition += ` AND classify_id_second = ? `
	case 2: // 当前分类是2级分类
		updateStr += ` classify_id_second = ?,classify_name_second = ?`
		condition += ` AND classify_id_first = ? `
	default:
		err = errors.New("错误的分类层级")
		return
	}

	pars = append(pars, currClassifyInfo.Id, currClassifyInfo.ClassifyName, parentClassifyInfo.Id)

	currReportChapterTypeList, err := models.GetAllReportChapterTypeListByClassifyId(currClassifyInfo.Id)
	if err != nil {
		return
	}
	chapterTypeIdMap := make(map[int]int)
	for _, v := range currReportChapterTypeList {
		chapterTypeIdMap[v.ReportChapterTypeId] = v.InheritReportChapterTypeId
	}

	err = models.ModifyReportClassifyAndReportChapterTypeByCondition(condition, pars, updateStr, chapterTypeIdMap, parentClassifyInfo.Id, currClassifyInfo.Id, currClassifyInfo.ClassifyName)
	if err != nil {
		return
	}

	return
}

func inheritReportChapterType(parentClassifyId, currClassifyId int) (err error) {
	defer func() {
		if err != nil {
			utils.FileLog.Error(fmt.Sprint("继承父级分类下的章节类型失败，父级分类ID：", parentClassifyId, "；当前分类ID:", currClassifyId, "；错误信息：", err.Error()))
		}
	}()
	parentReportChapterTypeList, err := models.GetAllReportChapterTypeListByClassifyId(parentClassifyId)
	if err != nil {
		return
	}

	if len(parentReportChapterTypeList) <= 0 {
		return
	}

	addList := make([]*models.ReportChapterType, 0)
	for _, v := range parentReportChapterTypeList {
		addList = append(addList, &models.ReportChapterType{
			ReportChapterTypeKey:       v.ReportChapterTypeKey,
			ReportChapterTypeThumb:     v.ReportChapterTypeThumb,
			BannerUrl:                  v.BannerUrl,
			ReportChapterTypeName:      v.ReportChapterTypeName,
			Sort:                       v.Sort,
			Enabled:                    v.Enabled,
			CreatedTime:                time.Now(),
			LastUpdatedTime:            time.Now(),
			ResearchType:               v.ResearchType,
			SelectedImage:              v.SelectedImage,
			UnselectedImage:            v.UnselectedImage,
			PcSelectedImage:            v.PcSelectedImage,
			PcUnselectedImage:          v.PcUnselectedImage,
			EditImgUrl:                 v.EditImgUrl,
			TickerTitle:                v.TickerTitle,
			IsShow:                     v.IsShow,
			PauseStartTime:             v.PauseStartTime,
			PauseEndTime:               v.PauseEndTime,
			IsSet:                      v.IsSet,
			YbIconUrl:                  v.YbIconUrl,
			YbBottomIcon:               v.YbBottomIcon,
			ReportClassifyId:           currClassifyId,
			InheritReportChapterTypeId: v.ReportChapterTypeId,
		})
	}

	obj := models.ReportChapterType{}
	err = obj.MultiCreate(addList)

	go func() {
		var syncReq ChapterTypeSyncReq
		_, _ = ReportChapterTypeSync(&syncReq)
	}()

	return
}

func inheritReportApproveFlow(parentClassifyItem, currClassifyItem *models.Classify) (err error) {
	defer func() {
		if err != nil {
			utils.FileLog.Error(fmt.Sprint("继承父级分类下的审批流失败，父级分类ID：", parentClassifyItem.Id, "；当前分类ID:", currClassifyItem.Id, "；错误信息：", err.Error()))
		}
	}()

	var firstClassify, secondClassify, thirdClassify *models.Classify
	if parentClassifyItem.ParentId > 0 {
		firstClassify, err = models.GetClassifyById(parentClassifyItem.ParentId)
		if err != nil {
			return
		}

		secondClassify = parentClassifyItem
		thirdClassify = currClassifyItem
	} else {
		firstClassify = parentClassifyItem
		secondClassify = currClassifyItem
	}

	flowObj := report_approve.ReportApproveFlow{}

	existCond := fmt.Sprintf(` AND %s = ? AND %s = ? `, report_approve.ReportApproveFlowCols.ReportType, report_approve.ReportApproveFlowCols.ClassifyFirstId)
	existPars := make([]interface{}, 0)
	existPars = append(existPars, report_approve.FlowReportTypeChinese, firstClassify.Id)

	if thirdClassify != nil {
		existCond = fmt.Sprintf(`%s AND %s = ?`, existCond, report_approve.ReportApproveFlowCols.ClassifySecondId)
		existPars = append(existPars, secondClassify.Id)
	}

	parentFlow, err := flowObj.GetItemByCondition(existCond, existPars, "")
	if err != nil {
		if utils.IsErrNoRow(err) {
			err = nil
		}
		return
	}

	nodeObj := report_approve.ReportApproveNode{}
	nodeCond := fmt.Sprintf(` AND %s = ? AND %s = ?`, report_approve.ReportApproveNodeCols.ReportApproveFlowId, report_approve.ReportApproveNodeCols.CurrVersion)
	nodePars := make([]interface{}, 0)
	nodePars = append(nodePars, parentFlow.ReportApproveFlowId, parentFlow.CurrVersion)
	parentNodeList, err := nodeObj.GetItemsByCondition(nodeCond, nodePars, []string{}, "")
	if err != nil {
		return
	}

	currFlow := &report_approve.ReportApproveFlow{
		ReportApproveFlowId: 0,
		FlowName:            currClassifyItem.ClassifyName,
		ReportType:          parentFlow.ReportType,
		ClassifyFirstId:     firstClassify.Id,
		ClassifySecondId:    secondClassify.Id,
		CurrVersion: 1,
		Enabled:     1,
		CreateTime:  time.Now().Local(),
		ModifyTime:  time.Now().Local(),
	}
	if thirdClassify != nil {
		currFlow.ClassifyThirdId = thirdClassify.Id
	}

	nodeItems := make([]*report_approve.ReportApproveNode, 0)
	for _, v := range parentNodeList {
		n := &report_approve.ReportApproveNode{
			PrevNodeId:  0,
			NextNodeId:  0,
			NodeType:    v.NodeType,
			ApproveType: v.ApproveType,
			Users:       v.Users,
			CurrVersion: 1,
			CreateTime:  time.Now().Local(),
		}
		nodeItems = append(nodeItems, n)
	}

	err = flowObj.CreateFlowAndNodes(currFlow, nodeItems)
	if err != nil {
		return
	}

	parentFlow.Enabled = 0
	err = parentFlow.Update([]string{"Enabled"})

	return
}

func EditReportClassify(classifyId int, classifyName string, chartPermissionIdList []int) (err error, errMsg string, isSendEmail bool) {
	isSendEmail = true
	errMsg = `修改失败`

	item, err := models.GetClassifyById(classifyId)
	if err != nil {
		errMsg = "获取分类信息失败"
		if utils.IsErrNoRow(err) {
			errMsg = "分类不存在, 或已被删除"
			isSendEmail = false
		}
		return
	}

	existName, e := models.GetClassifyByName(classifyName, item.ParentId)
	if e != nil && !utils.IsErrNoRow(e) {
		errMsg = "获取信息失败"
		err = errors.New("获取重名分类失败, Err: " + err.Error())
		return
	}
	if existName != nil && existName.Id > 0 && existName.Id != item.Id {
		errMsg = "分类名称:" + classifyName + "已存在"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	item.ClassifyName = classifyName

	item.ModifyTime = time.Now().Local()
	cols := make([]string, 0)
	cols = append(cols, "ClassifyName", "ModifyTime")
	err = item.UpdateClassify(cols)
	if err != nil {
		return
	}
	return
}

func GetClassifyTreeRecursive(list []*models.ClassifyItem, parentId int) []*models.ClassifyItem {
	res := make([]*models.ClassifyItem, 0)
	for _, v := range list {
		if v.ParentId == parentId {
			v.Child = GetClassifyTreeRecursive(list, v.Id)
			res = append(res, v)
		}
	}
	return res
}

func GetParentClassifyListByParentIdList(parentClassifyIdList []int) (list []*models.ClassifyList, err error) {
	num := len(parentClassifyIdList)
	if num <= 0 {
		return
	}
	list, err = models.GetClassifyListByParentIdList(parentClassifyIdList)
	if err != nil {
		return
	}

	{
		currParentClassifyIdList := make([]int, 0)
		for _, v := range list {
			if v.ParentId > 0 {
				currParentClassifyIdList = append(currParentClassifyIdList, v.ParentId)
			}
		}

		if len(currParentClassifyIdList) > 0 {
			tmpList, tmpErr := GetParentClassifyListByParentIdList(currParentClassifyIdList)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			list = append(tmpList, list...)
		}
	}

	return
}

func GetClassifyListTreeRecursive(list []*models.ClassifyList, parentId int) []*models.ClassifyList {
	res := make([]*models.ClassifyList, 0)
	for _, v := range list {
		if v.ParentId == parentId {
			v.Child = GetClassifyListTreeRecursive(list, v.Id)
			res = append(res, v)
		}
	}

	if len(res) <= 0 {
		res = nil
	}

	return res
}

type BySortAndCreateTime []*models.ClassifyList

func (a BySortAndCreateTime) Len() int {
	return len(a)
}

func (a BySortAndCreateTime) Swap(i, j int) {
	a[i], a[j] = a[j], a[i]
}

func (a BySortAndCreateTime) Less(i, j int) bool {
	if a[i].Sort == a[j].Sort {
		return a[i].CreateTime.Before(a[j].CreateTime)
	}
	return a[i].Sort < a[j].Sort
}

func SortClassifyListBySortAndCreateTime(classifyList []*models.ClassifyList) {
	sort.Sort(BySortAndCreateTime(classifyList))
}