eta_api/services/data/data_manage_permission/data_move.go

package data_manage_permission

import (
	"encoding/json"
	"eta_gn/eta_api/models"
	"eta_gn/eta_api/models/data_manage"
	"eta_gn/eta_api/models/data_manage/data_manage_permission"
	"eta_gn/eta_api/models/data_manage/excel"
	"eta_gn/eta_api/models/system"
	"eta_gn/eta_api/utils"
	"fmt"
	"strconv"
	"strings"
	"time"

	"github.com/rdlucklib/rdluck_tools/uuid"
)

func GetEdbChartClassifyList(source, subSource int) (resp data_manage.EdbChartClassifyResp, err error) {
	switch source {
	case 1: //手工数据指标
		list, e := models.GetEdbdataClassify(0)
		if e != nil {
			err = e
			return
		}

		for _, v := range list {
			item := data_manage.EdbChartClassify{
				ClassifyId:   v.ClassifyId,
				ClassifyName: v.ClassifyName,
				ParentId:     v.ParentId,
			}
			for _, v2 := range v.Child {
				child := data_manage.EdbChartClassify{
					ClassifyId:   v2.ClassifyId,
					ClassifyName: v2.ClassifyName,
					ParentId:     v2.ParentId,
				}
				item.Child = append(item.Child, &child)
			}
			resp.List = append(resp.List, &item)
		}

	case 3, 7:
		classifyType := utils.EdbClassifyTypeBase
		if source == 7 {
			classifyType = utils.EdbClassifyTypeCalculate
		}
		classifyAll, e := data_manage.GetAllEdbClassifyByType(classifyType)
		if e != nil && !utils.IsErrNoRow(e) {
			err = e
			return
		}
		resp.List = GetEdbMenuTreeRecursive(classifyAll, 0)

	case 4:
		classifyAll, e := data_manage.GetAllEdbClassifyByType(1)
		if e != nil && !utils.IsErrNoRow(e) {
			err = e
			return
		}
		resp.List = GetEdbMenuTreeRecursive(classifyAll, 0)
	case 5: //图库
		classifyAll, e := data_manage.GetAllChartClassifyItemsBySource(1)
		if e != nil && !utils.IsErrNoRow(e) {
			err = e
			return
		}
		resp.List = GetChartMenuTreeRecursive(classifyAll, 0)
	case 6:
		classifyAll, e := excel.GetExcelClassifyBySourceOrderByLevel(subSource)
		if e != nil && !utils.IsErrNoRow(e) {
			err = e
			return
		}
		resp.List = GetExcelMenuTreeRecursive(classifyAll, 0)
	default:
		return
	}

	return
}

func GetEdbMenuTreeRecursive(list []*data_manage.EdbClassifyItems, parentId int) []*data_manage.EdbChartClassify {
	res := make([]*data_manage.EdbChartClassify, 0)
	for _, v := range list {
		t := new(data_manage.EdbChartClassify)
		t.ClassifyId = v.ClassifyId
		t.ClassifyName = v.ClassifyName
		t.ParentId = v.ParentId
		t.IsJoinPermission = v.IsJoinPermission
		if v.ParentId == parentId {
			t.Child = GetEdbMenuTreeRecursive(list, v.ClassifyId)
			res = append(res, t)
		}
	}
	return res
}

func GetChartMenuTreeRecursive(list []*data_manage.ChartClassifyItems, parentId int) []*data_manage.EdbChartClassify {
	res := make([]*data_manage.EdbChartClassify, 0)
	for _, v := range list {
		t := new(data_manage.EdbChartClassify)
		t.ClassifyId = v.ChartClassifyId
		t.ClassifyName = v.ChartClassifyName
		t.ParentId = v.ParentId
		t.IsJoinPermission = v.IsJoinPermission
		if v.ParentId == parentId {
			t.Child = GetChartMenuTreeRecursive(list, v.ChartClassifyId)
			res = append(res, t)
		}
	}
	return res
}

func GetExcelMenuTreeRecursive(list []*excel.ExcelClassifyItems, parentId int) []*data_manage.EdbChartClassify {
	res := make([]*data_manage.EdbChartClassify, 0)
	for _, v := range list {
		t := new(data_manage.EdbChartClassify)
		t.ClassifyId = v.ExcelClassifyId
		t.ClassifyName = v.ExcelClassifyName
		t.ParentId = v.ParentId
		t.IsJoinPermission = v.IsJoinPermission
		if v.ParentId == parentId {
			t.Child = GetExcelMenuTreeRecursive(list, v.ExcelClassifyId)
			res = append(res, t)
		}
	}
	return res
}

func GetMoveEdbChartList(source, subSource, userId int, keyword, classify string, startSize, pageSize int) (list []data_manage.MoveEdbChartList, total int, err error) {
	var condition string
	var pars []interface{}
	switch source {
	case 1: //手工数据指标
		if keyword != "" {
			condition += ` AND (a.SEC_NAME LIKE ? OR a.TRADE_CODE like ? ) `
			pars = utils.GetLikeKeywordPars(pars, keyword, 2)
		}
		if userId > 0 {
			condition += ` AND a.user_id = ? `
			pars = append(pars, userId)
		}

		if classify != "" {
			condition += ` AND a.classify_id IN (` + classify + `) `
		}

		total, err = models.GetEdbinfoListCount(condition, pars, "", 0)
		if err != nil {
			return
		}

		tmpList, tmpErr := models.GetEdbinfoItemList(condition, pars, startSize, pageSize, "", 0)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		if len(tmpList) > 0 {
			adminIdList := make([]int, 0)
			for _, v := range tmpList {
				adminIdList = append(adminIdList, v.UserId)
			}
			adminList, tmpErr := system.GetAdminListByIdListWithoutEnable(adminIdList)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			adminMap := make(map[int]string)
			for _, adminInfo := range adminList {
				adminMap[adminInfo.AdminId] = adminInfo.RealName
			}

			for _, v := range tmpList {
				list = append(list, data_manage.MoveEdbChartList{
					DataId:         v.TradeCode,
					Code:           v.TradeCode,
					Name:           v.SecName,
					ClassifyName:   v.ClassifyName,
					NameEn:         v.SecName,
					ClassifyNameEn: v.ClassifyName,
					CreateUserId:   v.UserId,
					CreateUserName: adminMap[v.UserId],
				})
			}
		}
	case 3, 4, utils.EdbPermissionSourceCalculate: //ETA指标库、ETA预测指标、指标加工
		if keyword != `` {
			condition += " AND (edb_code like ? OR edb_name like ? OR edb_name_en like ? OR sys_user_real_name like ? ) "
			pars = utils.GetLikeKeywordPars(pars, keyword, 4)
		}
		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}
		if classify != "" {
			condition += ` AND classify_id IN (` + classify + `) `
		}

		edbInfoType := 0 //ETA指标库
		if source == 4 { //ETA预测指标
			edbInfoType = 1
		}
		condition += ` AND edb_info_type = ? `
		pars = append(pars, edbInfoType)

		if source != 4 {
			edbType := utils.EdbTypeBase
			if source == utils.EdbPermissionSourceCalculate {
				edbType = utils.EdbTypeCalculate
			}
			condition += ` AND edb_type = ? `
			pars = append(pars, edbType)
		}

		total, err = data_manage.GetEdbInfoByConditionCount(condition, pars)
		if err != nil {
			return
		}
		tmpList, tmpErr := data_manage.GetEdbInfoListByCondition(condition, pars, startSize, pageSize, "DESC")
		if tmpErr != nil {
			err = tmpErr
			return
		}

		for _, v := range tmpList {
			list = append(list, data_manage.MoveEdbChartList{
				DataId: strconv.Itoa(v.EdbInfoId),
				Code:   v.EdbCode,
				Name:   v.EdbName,
				NameEn: v.EdbNameEn,
				CreateUserId:   v.SysUserId,
				CreateUserName: v.SysUserRealName,
			})
		}
	case 5: //图库

		condition += ` AND source = ? `
		pars = append(pars, utils.CHART_SOURCE_DEFAULT)

		if keyword != `` {
			condition += " AND (chart_name like ? or chart_name_en like ?  OR sys_user_real_name like ? ) "
			pars = utils.GetLikeKeywordPars(pars, keyword, 3)
		}
		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}
		if classify != "" {
			condition += ` AND chart_classify_id IN (` + classify + `) `
		}

		total, err = data_manage.GetChartInfoCountByCondition(condition, pars)
		if err != nil {
			return
		}
		tmpList, tmpErr := data_manage.GetChartInfoListByCondition(condition, pars, startSize, pageSize)
		if tmpErr != nil {
			err = tmpErr
			return
		}

		firstClassifyList, tmpErr := data_manage.GetChartClassifyByParentId(0, 1)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		firstClassifyNameMap := make(map[int]string)
		firstClassifyNameEnMap := make(map[int]string)
		for _, v := range firstClassifyList {
			firstClassifyNameMap[v.ChartClassifyId] = v.ChartClassifyName
			firstClassifyNameEnMap[v.ChartClassifyId] = v.ChartClassifyNameEn
		}
		secondClassList, tmpErr := data_manage.GetChartClassifyAll(1)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		secondClassifyNameMap := make(map[int]string)
		secondClassifyNameEnMap := make(map[int]string)
		for _, v := range secondClassList {
			firstName := firstClassifyNameMap[v.ParentId]
			firstNameEn := firstClassifyNameEnMap[v.ParentId]
			if firstName != `` {
				firstName = firstName + " / "
			}
			if firstNameEn != `` {
				firstNameEn = firstNameEn + " / "
			}
			secondClassifyNameMap[v.ChartClassifyId] = firstName + v.ChartClassifyName
			secondClassifyNameEnMap[v.ChartClassifyId] = firstNameEn + v.ChartClassifyNameEn
		}

		for _, v := range tmpList {
			list = append(list, data_manage.MoveEdbChartList{
				DataId:         strconv.Itoa(v.ChartInfoId),
				Name:           v.ChartName,
				ClassifyName:   secondClassifyNameMap[v.ChartClassifyId],
				NameEn:         v.ChartNameEn,
				ClassifyNameEn: secondClassifyNameEnMap[v.ChartClassifyId],
				CreateUserId:   v.SysUserId,
				CreateUserName: v.SysUserRealName,
			})
		}
	case 6: //ETA表格
		if keyword != `` {
			condition += " AND (excel_name like ?  OR sys_user_real_name like ? ) "
			pars = utils.GetLikeKeywordPars(pars, keyword, 2)
		}
		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}
		if classify != "" {
			condition += ` AND excel_classify_id IN (` + classify + `) `
		}

		condition += " AND source = ? "
		pars = append(pars, subSource)

		if subSource == utils.BALANCE_TABLE {
			condition += " AND parent_id = 0 AND balance_type = 0 "
		}

		total, err = excel.GetExcelInfoCountByCondition(condition, pars)
		if err != nil {
			return
		}
		tmpList, tmpErr := excel.GetNoContentExcelInfoListByCondition(condition, pars, startSize, pageSize)
		if tmpErr != nil {
			err = tmpErr
			return
		}

		firstClassifyList, tmpErr := excel.GetExcelClassifyByParentId(0, subSource)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		firstClassifyNameMap := make(map[int]string)
		for _, v := range firstClassifyList {
			firstClassifyNameMap[v.ExcelClassifyId] = v.ExcelClassifyName
		}
		secondClassList, tmpErr := excel.GetExcelClassifyBySource(subSource)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		secondClassifyNameMap := make(map[int]string)
		for _, v := range secondClassList {
			firstName := firstClassifyNameMap[v.ParentId]
			if firstName != `` {
				firstName = firstName + " / "
			}
			secondClassifyNameMap[v.ExcelClassifyId] = firstName + v.ExcelClassifyName
		}

		for _, v := range tmpList {
			list = append(list, data_manage.MoveEdbChartList{
				DataId:         strconv.Itoa(v.ExcelInfoId),
				Name:           v.ExcelClassifyName,
				ClassifyName:   secondClassifyNameMap[v.ExcelClassifyId],
				NameEn:         v.ExcelClassifyName,
				ClassifyNameEn: secondClassifyNameMap[v.ExcelClassifyId],
				CreateUserId:   v.SysUserId,
				CreateUserName: v.SysUserRealName,
			})
		}
	default:
		return
	}

	return
}

func MoveEdbChart(source, subSource, oldUserId, newUserId int, isSelectAll bool, dataId, noDataId []string, keyword, classify string, opUserId int, opUserName string) (err error, errMsg string) {
	adminInfo, err := system.GetSysAdminById(newUserId)
	if err != nil {
		return
	}

	if isSelectAll {
		noDataIdMap := make(map[string]string, 0)
		for _, v := range noDataId {
			noDataIdMap[v] = v
		}

		dataId = make([]string, 0)
		list, _, tmpErr := GetMoveEdbChartList(source, subSource, oldUserId, keyword, classify, 0, 100000)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		for _, v := range list {
			if _, ok := noDataIdMap[v.DataId]; !ok {
				dataId = append(dataId, v.DataId)
			}
		}
	}

	dataPermissionMoveRecordList := make([]*data_manage_permission.DataPermissionMoveRecord, 0)
	uniqueCode := uuid.NewUUID().Hex32()
	title := `资产转移`
	content := `资产转入`

	switch source {
	case 1: //手工数据指标
		content += `（手工指标数据）`
		tmpList, tmpErr := models.GetEdbinfoListByCodeListByCodeIdList(dataId)
		if tmpErr != nil {
			err = tmpErr
			return
		}

		if len(tmpList) > 0 {
			for _, v := range tmpList {

				dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
					DataPermissionMoveRecordId: 0,
					Source:                     int32(source),
					SubSource:                  int32(subSource),
					OpUniqueCode:               uniqueCode,
					DataId:                     v.TradeCode,
					DataCode:                   v.TradeCode,
					DataName:                   v.SecName,
					OriginalUserId:             int32(v.UserId),
					OriginalUserName:           v.UserName,
					NewUserId:                  int32(adminInfo.AdminId),
					NewUserName:                adminInfo.RealName,
					CreateTime:                 time.Now(),
				})
			}
			err = models.ModifyEdbinfoUserIdByCodeList(dataId, newUserId, adminInfo.RealName)

			if err == nil && len(tmpList) > 0 {
				go afterMoveManualEdb(tmpList, adminInfo, opUserId, opUserName)
			}
		}
	case 3, 4, utils.EdbPermissionSourceCalculate: //ETA指标库、ETA预测指标
		if source == 3 {
			content += `（数据查看）`
		}
		if source == 4 {
			content += `（衍生指标）`
		}
		if source == utils.EdbPermissionSourceCalculate {
			content += `（指标加工）`
		}
		tmpList, tmpErr := data_manage.GetEdbInfoListByEdbInfoId(dataId)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		if len(tmpList) > 0 {
			for _, v := range tmpList {
				dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
					DataPermissionMoveRecordId: 0,
					Source:                     int32(source),
					SubSource:                  int32(subSource),
					OpUniqueCode:               uniqueCode,
					DataId:                     fmt.Sprint(v.EdbInfoId),
					DataCode:                   v.EdbCode,
					DataName:                   v.EdbName,
					OriginalUserId:             int32(v.SysUserId),
					OriginalUserName:           v.SysUserRealName,
					NewUserId:                  int32(adminInfo.AdminId),
					NewUserName:                adminInfo.RealName,
					CreateTime:                 time.Now(),
				})
			}
			err = data_manage.ModifyEdbInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)

			{
				obj := data_manage.EdbInfoNoPermissionAdmin{}
				for _, v := range dataId {
					edbInfoId, _ := strconv.Atoi(v)
					go obj.DeleteByEdbInfoIdAndAdminId(edbInfoId, adminInfo.AdminId)
				}
			}
		}
	case 5: //图库
		content += `（图库）`
		subSource = utils.CHART_SOURCE_DEFAULT

		tmpList, tmpErr := data_manage.GetChartInfoListByChartIdList(dataId)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		if len(tmpList) > 0 {
			for _, v := range tmpList {
				dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
					DataPermissionMoveRecordId: 0,
					Source:                     int32(source),
					SubSource:                  int32(subSource),
					OpUniqueCode:               uniqueCode,
					DataId:                     fmt.Sprint(v.ChartInfoId),
					DataCode:                   fmt.Sprint(v.ChartInfoId),
					DataName:                   v.ChartName,
					OriginalUserId:             int32(v.SysUserId),
					OriginalUserName:           v.SysUserRealName,
					NewUserId:                  int32(adminInfo.AdminId),
					NewUserName:                adminInfo.RealName,
					CreateTime:                 time.Now(),
				})
			}
			err = data_manage.ModifyChartInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)
		}
	case 6: // ETA表格
		content += `（表格）`
		tmpList, tmpErr := excel.GetNoContentExcelListByExcelInfoIdList(dataId)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		var customAnalysisIds []int
		if len(tmpList) > 0 {
			for _, v := range tmpList {
				dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
					DataPermissionMoveRecordId: 0,
					Source:                     int32(source),
					SubSource:                  int32(subSource),
					OpUniqueCode:               uniqueCode,
					DataId:                     fmt.Sprint(v.ExcelInfoId),
					DataCode:                   fmt.Sprint(v.ExcelInfoId),
					DataName:                   v.ExcelName,
					OriginalUserId:             int32(v.SysUserId),
					OriginalUserName:           v.SysUserRealName,
					NewUserId:                  int32(adminInfo.AdminId),
					NewUserName:                adminInfo.RealName,
					CreateTime:                 time.Now(),
				})

				if v.Source == utils.CUSTOM_ANALYSIS_TABLE {
					customAnalysisIds = append(customAnalysisIds, v.ExcelInfoId)
				}
			}
			err = excel.ModifyExcelInfoUserIdByCodeList(dataId, adminInfo.AdminId, adminInfo.RealName)
			if err != nil {
				return
			}

			if len(customAnalysisIds) > 0 {
				err = MoveCustomAnalysisExcel2DefaultClassify(customAnalysisIds, adminInfo.AdminId, adminInfo.RealName)
			}
		}
	default:
		return
	}

	if err != nil {
		return
	}

	dataPermissionMessage := &data_manage_permission.DataPermissionMessage{
		DataPermissionMessageId: 0,
		SendUserId:              int32(opUserId),
		ReceiveUserId:           int32(adminInfo.AdminId),
		Content:                 title,
		Remark:                  content,
		OpType:                  1,
		Source:                  int32(source),
		SubSource:               int32(subSource),
		OpUniqueCode:            uniqueCode,
		IsRead:                  0,
		CreateTime:              time.Now(),
		ModifyTime:              time.Now(),
	}
	tmpErr := data_manage_permission.AddDataPermissionMoveRecord(dataPermissionMoveRecordList, dataPermissionMessage)
	if tmpErr != nil {
		recordListJsonBytes, tmpErr2 := json.Marshal(dataPermissionMoveRecordList)
		if tmpErr2 != nil {
			utils.FileLog.Error(fmt.Sprintf("添加资产转移的记录失败:%s;list转json失败：;操作记录：%+v;消息记录：%+v", tmpErr.Error(), dataPermissionMoveRecordList, dataPermissionMessage))
			return
		}
		messageJsonBytes, tmpErr2 := json.Marshal(dataPermissionMessage)
		if tmpErr2 != nil {
			utils.FileLog.Error(fmt.Sprintf("添加资产转移的记录失败:%s;message转json失败：;操作记录：%+v;消息记录：%+v", tmpErr.Error(), dataPermissionMoveRecordList, dataPermissionMessage))
			return
		}
		utils.FileLog.Error(fmt.Sprintf("添加资产转移的记录失败:%s;操作记录：%s;消息记录：%s", tmpErr.Error(), string(recordListJsonBytes), string(messageJsonBytes)))
	}

	return
}

func MoveAllEdbChart(sourceList, oldUserIdList []int, userId, opUserId int) (err error, errMsg string) {
	adminInfo, err := system.GetSysAdminById(userId)
	if err != nil {
		return
	}

	dataPermissionMoveRecordList := make([]*data_manage_permission.DataPermissionMoveRecord, 0)
	uniqueCode := uuid.NewUUID().Hex32()
	title := `资产转移`
	content := `资产转入`
	sourceStrList := make([]string, 0)

	var isMoveManual, isMoveMysteelChemical, isMoveEdb, isMovePredictEdb, isMoveChart, isMoveExcel bool
	var customAnalysisIds []int

	for _, source := range sourceList {
		switch source {
		case 1: //手工数据指标
			sourceStrList = append(sourceStrList, "手工指标数据")
			tmpList, tmpErr := models.GetEdbinfoListByCodeListByUserId(oldUserIdList)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			if len(tmpList) > 0 {
				for _, v := range tmpList {
					dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
						DataPermissionMoveRecordId: 0,
						Source:                     int32(source),
						SubSource:                  0,
						OpUniqueCode:               uniqueCode,
						DataId:                     v.TradeCode,
						DataCode:                   v.TradeCode,
						DataName:                   v.SecName,
						OriginalUserId:             int32(v.UserId),
						OriginalUserName:           v.UserName,
						NewUserId:                  int32(adminInfo.AdminId),
						NewUserName:                adminInfo.RealName,
						CreateTime:                 time.Now(),
					})
				}
				isMoveManual = true
			}

		case 3: //ETA指标库
			sourceStrList = append(sourceStrList, "ETA指标库")
			edbInfoType := 0
			tmpList, tmpErr := data_manage.GetEdbInfoListByUserId(oldUserIdList, edbInfoType)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			if len(tmpList) > 0 {
				for _, v := range tmpList {
					dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
						DataPermissionMoveRecordId: 0,
						Source:                     int32(source),
						SubSource:                  int32(edbInfoType),
						OpUniqueCode:               uniqueCode,
						DataId:                     fmt.Sprint(v.EdbInfoId),
						DataCode:                   v.EdbCode,
						DataName:                   v.EdbName,
						OriginalUserId:             int32(v.SysUserId),
						OriginalUserName:           v.SysUserRealName,
						NewUserId:                  int32(adminInfo.AdminId),
						NewUserName:                adminInfo.RealName,
						CreateTime:                 time.Now(),
					})
				}
				isMoveEdb = true
			}

		case 4: //ETA预测指标
			sourceStrList = append(sourceStrList, "ETA预测指标")

			edbInfoType := 1
			tmpList, tmpErr := data_manage.GetEdbInfoListByUserId(oldUserIdList, edbInfoType)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			if len(tmpList) > 0 {
				for _, v := range tmpList {
					dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
						DataPermissionMoveRecordId: 0,
						Source:                     int32(source),
						SubSource:                  int32(edbInfoType),
						OpUniqueCode:               uniqueCode,
						DataId:                     fmt.Sprint(v.EdbInfoId),
						DataCode:                   v.EdbCode,
						DataName:                   v.EdbName,
						OriginalUserId:             int32(v.SysUserId),
						OriginalUserName:           v.SysUserRealName,
						NewUserId:                  int32(adminInfo.AdminId),
						NewUserName:                adminInfo.RealName,
						CreateTime:                 time.Now(),
					})
				}
				isMovePredictEdb = true
			}
		case 5: //ETA图库
			sourceStrList = append(sourceStrList, "ETA图库")
			subSource := utils.CHART_SOURCE_DEFAULT

			tmpList, tmpErr := data_manage.GetChartInfoListByUserId(oldUserIdList, subSource)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			if len(tmpList) > 0 {
				for _, v := range tmpList {
					dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
						DataPermissionMoveRecordId: 0,
						Source:                     int32(source),
						SubSource:                  int32(subSource),
						OpUniqueCode:               uniqueCode,
						DataId:                     fmt.Sprint(v.ChartInfoId),
						DataCode:                   fmt.Sprint(v.ChartInfoId),
						DataName:                   v.ChartName,
						OriginalUserId:             int32(v.SysUserId),
						OriginalUserName:           v.SysUserRealName,
						NewUserId:                  int32(adminInfo.AdminId),
						NewUserName:                adminInfo.RealName,
						CreateTime:                 time.Now(),
					})
				}
				isMoveChart = true
			}
		case 6: // ETA表格
			sourceStrList = append(sourceStrList, "ETA表格")
			tmpList, tmpErr := excel.GetNoContentExcelListByUserId(oldUserIdList)
			if tmpErr != nil {
				err = tmpErr
				return
			}
			if len(tmpList) > 0 {
				for _, v := range tmpList {
					dataPermissionMoveRecordList = append(dataPermissionMoveRecordList, &data_manage_permission.DataPermissionMoveRecord{
						DataPermissionMoveRecordId: 0,
						Source:                     int32(source),
						SubSource:                  int32(v.Source),
						OpUniqueCode:               uniqueCode,
						DataId:                     fmt.Sprint(v.ExcelInfoId),
						DataCode:                   fmt.Sprint(v.ExcelInfoId),
						DataName:                   v.ExcelName,
						OriginalUserId:             int32(v.SysUserId),
						OriginalUserName:           v.SysUserRealName,
						NewUserId:                  int32(adminInfo.AdminId),
						NewUserName:                adminInfo.RealName,
						CreateTime:                 time.Now(),
					})

					if v.Source == utils.CUSTOM_ANALYSIS_TABLE {
						customAnalysisIds = append(customAnalysisIds, v.ExcelInfoId)
					}
				}
				isMoveExcel = true
			}
		default:
		}
	}

	err = data_manage_permission.ModifyDataUserIdByOldUserId(oldUserIdList, adminInfo.AdminId, adminInfo.RealName, isMoveManual, isMoveMysteelChemical, isMoveEdb, isMovePredictEdb, isMoveChart, isMoveExcel)
	if err != nil {
		return
	}

	if len(customAnalysisIds) > 0 {
		err = MoveCustomAnalysisExcel2DefaultClassify(customAnalysisIds, adminInfo.AdminId, adminInfo.RealName)
		if err != nil {
			err = fmt.Errorf("MoveCustomAnalysisExcel2DefaultClassify, err: %v", err)
			return
		}
	}

	content += fmt.Sprintf("（%s）", strings.Join(sourceStrList, ","))
	dataPermissionMessage := &data_manage_permission.DataPermissionMessage{
		DataPermissionMessageId: 0,
		SendUserId:              int32(opUserId),
		ReceiveUserId:           int32(adminInfo.AdminId),
		Content:                 title,
		Remark:                  content,
		OpType:                  2,
		OpUniqueCode:            uniqueCode,
		IsRead:                  0,
		CreateTime:              time.Now(),
		ModifyTime:              time.Now(),
	}
	tmpErr := data_manage_permission.AddDataPermissionMoveRecord(dataPermissionMoveRecordList, dataPermissionMessage)
	if tmpErr != nil {
		recordListJsonBytes, tmpErr2 := json.Marshal(dataPermissionMoveRecordList)
		if tmpErr2 != nil {
			utils.FileLog.Error(fmt.Sprintf("添加资产转移的记录失败:%s;list转json失败：;操作记录：%+v;消息记录：%+v", tmpErr.Error(), dataPermissionMoveRecordList, dataPermissionMessage))
			return
		}
		messageJsonBytes, tmpErr2 := json.Marshal(dataPermissionMessage)
		if tmpErr2 != nil {
			utils.FileLog.Error(fmt.Sprintf("添加资产转移的记录失败:%s;message转json失败：;操作记录：%+v;消息记录：%+v", tmpErr.Error(), dataPermissionMoveRecordList, dataPermissionMessage))
			return
		}
		utils.FileLog.Error(fmt.Sprintf("添加资产转移的记录失败:%s;操作记录：%s;消息记录：%s", tmpErr.Error(), string(recordListJsonBytes), string(messageJsonBytes)))
	}

	return
}

func GetMoveEdbChartCount(userId, countType int) (sourceMap map[int]int, err error) {
	sourceMap = make(map[int]int)
	{
		var condition string
		var pars []interface{}

		if userId > 0 {
			condition += ` AND a.user_id = ? `
			pars = append(pars, userId)
		}

		total, tmpErr := models.GetEdbinfoListCount(condition, pars, "", 0)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		sourceMap[1] = total
	}

	{
		var condition string
		var pars []interface{}
		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}
		condition += ` AND edb_info_type = ? AND edb_type = ? `
		pars = append(pars, 0, utils.EdbTypeBase)

		total, tmpErr := data_manage.GetEdbInfoByConditionCount(condition, pars)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		sourceMap[3] = total
	}

	{
		var condition string
		var pars []interface{}
		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}

		condition += ` AND edb_info_type = ? `
		pars = append(pars, 1)

		total, tmpErr := data_manage.GetEdbInfoByConditionCount(condition, pars)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		sourceMap[4] = total
	}

	{
		var condition string
		var pars []interface{}

		condition += ` AND source = ? `
		pars = append(pars, utils.CHART_SOURCE_DEFAULT)

		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}

		total, tmpErr := data_manage.GetChartInfoCountByCondition(condition, pars)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		sourceMap[5] = total
	}

	{
		var condition string
		var pars []interface{}

		condition += ` AND is_delete = 0 `

		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}

		switch countType {
		case 2:
			condition += ` AND source in (2,3,5) `
		default:
			condition += ` AND source in (2,3,4,5) `
		}
		condition += " AND (parent_id = 0 AND balance_type = 0)"

		total, tmpErr := excel.GetExcelInfoCountByCondition(condition, pars)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		sourceMap[6] = total
	}

	{
		var condition string
		var pars []interface{}
		if userId > 0 {
			condition += ` AND sys_user_id = ? `
			pars = append(pars, userId)
		}
		condition += ` AND edb_info_type = ? AND edb_type = ? `
		pars = append(pars, 0, utils.EdbTypeCalculate)

		total, tmpErr := data_manage.GetEdbInfoByConditionCount(condition, pars)
		if tmpErr != nil {
			err = tmpErr
			return
		}
		sourceMap[7] = total
	}

	return
}

func afterMoveManualEdb(tmpList []*models.Edbinfo, adminInfo *system.Admin, opUserId int, opUserName string) {
	recordList := make([]*models.EdbinfoOpRecord, 0)
	for _, tmpManualInfo := range tmpList {
		remark := fmt.Sprintf("数据资产由%s转移给%s", tmpManualInfo.UserName, adminInfo.RealName)
		recordList = append(recordList, &models.EdbinfoOpRecord{
			TradeCode:  tmpManualInfo.TradeCode,
			Remark:     remark,
			UserId:     opUserId,
			UserName:   opUserName,
			CreateTime: time.Now(),
		})
	}
	obj := models.EdbinfoOpRecord{}
	_ = obj.MulCreate(recordList)
}

func MoveCustomAnalysisExcel2DefaultClassify(excelIds []int, adminId int, realName string) (err error) {
	if len(excelIds) == 0 {
		return
	}
	var moveClassifyId int
	cond := ` AND excel_classify_name = ? AND sys_user_id = ?`
	pars := make([]interface{}, 0)
	pars = append(pars, "未分类", adminId)
	classify, e := excel.GetExcelClassifyByCondition(cond, pars)
	if e != nil && !utils.IsErrNoRow(e) {
		err = fmt.Errorf("获取默认分类失败, %v", e)
		return
	}
	if e == nil && classify.ExcelClassifyId > 0 {
		moveClassifyId = classify.ExcelClassifyId
	}

	if classify == nil || classify.ExcelClassifyId <= 0 {
		classifyNew := &excel.ExcelClassify{
			ExcelClassifyName: "未分类",
			Source:            utils.CUSTOM_ANALYSIS_TABLE,
			SysUserId:         adminId,
			SysUserRealName:   realName,
			Level:             1,
			UniqueCode:        utils.MD5(utils.EXCEL_DATA_PREFIX + "_" + strconv.FormatInt(time.Now().UnixNano(), 10)),
			CreateTime:        time.Now(),
			ModifyTime:        time.Now(),
		}
		if _, e = excel.AddExcelClassify(classifyNew); e != nil {
			err = fmt.Errorf("新增默认分类失败, %v", e)
			return
		}
		moveClassifyId = classifyNew.ExcelClassifyId
	}
	if moveClassifyId <= 0 {
		err = fmt.Errorf("移动分类ID有误")
		return
	}

	if e = excel.UpdateExcelInfoClassifyIdByIds(moveClassifyId, excelIds); e != nil {
		err = fmt.Errorf("移动多表格分类失败, %v", e)
		return
	}
	return
}