eta_api/services/data/edb_info_refresh.go

package data

import (
	"encoding/json"
	"errors"
	"eta_gn/eta_api/models/data_manage"
	"eta_gn/eta_api/models/data_manage/edb_refresh"
	"eta_gn/eta_api/models/data_manage/edb_refresh/request"
	"eta_gn/eta_api/utils"
	"fmt"
	"strings"
	"time"
)

var allDefaultEdbRefreshConfigKey = `edb_refresh_config:default:all:`

func GetAllDefaultEdbRefreshConfigListBySourceList(source, subSource int) (list []*edb_refresh.EdbRefreshDefaultConfig, err error) {
	key := getAllDefaultEdbRefreshConfigKey(source, subSource)
	if utils.Re == nil {
		if utils.Re == nil && utils.Rc.IsExist(key) {
			if data, err1 := utils.Rc.RedisBytes(key); err1 == nil {
				err = json.Unmarshal(data, &list)
				return
			}
		}
	}

	list, err = edb_refresh.GetAllListBySourceList(source, subSource)
	if err != nil {
		return
	}

	if utils.Re == nil {
		data, _ := json.Marshal(list)
		utils.Rc.Put(key, data, 2*time.Hour)
	}

	return
}

func SaveEdbRefreshDefaultConfig(source, subSource int, frequency string, list []request.RefreshConfigReq) (err error, errMsg string, isSendEmail bool) {
	isSendEmail = true
	errMsg = `保存失败`

	if source <= 0 {
		errMsg = "来源不能为空"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	if source != utils.DATA_SOURCE_YS && frequency == `` {
		errMsg = "频度不能为空"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	lenConf := len(list)
	if lenConf == 0 {
		errMsg = "至少需要一个刷新配置"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if lenConf > 5 {
		errMsg = "刷新时间设置最多不超过5个"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	tmpArr := []string{"每周", "每旬", "每月", "每季", "每半年", "每年"}
	configMap := make(map[string]string)
	for _, v := range list {
		if utils.InArrayByStr(tmpArr, v.RefreshFrequency) && v.RefreshFrequencyDay < 0 {
			errMsg = "请选择具体日期"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}

		if v.RefreshTime == "" {
			errMsg = "请选择具体时间"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}

		key := fmt.Sprint(v.RefreshFrequency, "_", v.RefreshFrequencyDay, "_", v.RefreshTime)
		if _, ok := configMap[key]; ok {
			errMsg = "刷新频率和日期不能重复"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
		configMap[key] = key
	}

	addList := make([]*edb_refresh.EdbRefreshDefaultConfig, 0)
	for _, v := range list {
		addList = append(addList, &edb_refresh.EdbRefreshDefaultConfig{
			Source:              source,
			SubSource:           subSource,
			Frequency:           frequency,
			RefreshFrequency:    v.RefreshFrequency,
			RefreshFrequencyDay: v.RefreshFrequencyDay,
			RefreshAllData:      v.RefreshAllData,
			RefreshTime:         v.RefreshTime,
			RefreshDataNum:      v.RefreshDataNum,
			ModifyTime:          time.Now(),
			CreateTime:          time.Now(),
		})
	}

	err = edb_refresh.SaveEdbRefreshDefaultConfig(source, subSource, frequency, addList)

	{
		key := getAllDefaultEdbRefreshConfigKey(source, subSource)
		if utils.Re == nil {
			_ = utils.Rc.Delete(key)
		}
	}

	return
}

func SaveEdbRefreshConfig(source, subSource int, classifyId, terminalCode, frequency, keyword, status, sysUserIdStr string, isSelectAll bool, list []request.RefreshConfigReq, edbSelectIdList []int, sysUserId int, sysUserRealName string) (err error, errMsg string, isSendEmail bool) {
	isSendEmail = true
	errMsg = `保存失败`

	if source <= 0 {
		errMsg = "来源不能为空"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	lenConf := len(list)
	if lenConf == 0 {
		errMsg = "至少需要一个刷新配置"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}
	if lenConf > 5 {
		errMsg = "刷新时间设置最多不超过5个"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	tmpArr := []string{"每周", "每旬", "每月", "每季", "每半年", "每年"}

	configMap := make(map[string]string)
	for _, v := range list {
		if utils.InArrayByStr(tmpArr, v.RefreshFrequency) && v.RefreshFrequencyDay < 0 {
			errMsg = "请选择具体日期"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}

		if v.RefreshTime == "" {
			errMsg = "请选择具体时间"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}

		key := fmt.Sprint(v.RefreshFrequency, "_", v.RefreshFrequencyDay, "_", v.RefreshTime)
		if _, ok := configMap[key]; ok {
			errMsg = "刷新频率和日期不能重复"
			err = errors.New(errMsg)
			isSendEmail = false
			return
		}
		configMap[key] = key
	}

	edbIdList := make([]int, 0)
	if isSelectAll {
		_, edbList, tmpErr := GetList(source, subSource, classifyId, terminalCode, sysUserIdStr, frequency, keyword, status, 0, 100000, "", "")
		if tmpErr != nil && !utils.IsErrNoRow(tmpErr) {
			err = tmpErr
			return
		}

		notIdMap := make(map[int]int, 0)
		for _, v := range edbSelectIdList {
			notIdMap[v] = v
		}

		for _, v := range edbList {
			_, ok := notIdMap[v.EdbInfoId]
			if ok {
				continue
			}

			edbIdList = append(edbIdList, v.EdbInfoId)
		}
	} else {
		edbIdList = edbSelectIdList
	}

	if len(edbIdList) <= 0 {
		errMsg = "指标不能为空"
		err = errors.New(errMsg)
		isSendEmail = false
		return
	}

	addMappingList := make([]*edb_refresh.EdbRefreshMapping, 0)

	addConfigList := make([]*edb_refresh.EdbRefreshConfig, 0)
	for _, v := range list {
		item, tmpErr := edb_refresh.GetEdbRefreshConfigListByCondition(v.RefreshFrequency, v.RefreshTime, v.RefreshFrequencyDay, v.RefreshAllData, v.RefreshDataNum)
		if tmpErr != nil {
			if !utils.IsErrNoRow(tmpErr) {
				err = tmpErr
				return
			}

			addConfigList = append(addConfigList, &edb_refresh.EdbRefreshConfig{
				RefreshFrequency:    v.RefreshFrequency,
				RefreshFrequencyDay: v.RefreshFrequencyDay,
				RefreshTime:         v.RefreshTime,
				RefreshAllData:      v.RefreshAllData,
				RefreshDataNum:      v.RefreshDataNum,
				ModifyTime:          time.Now(),
				CreateTime:          time.Now(),
			})
			continue
		}

		for _, edbId := range edbIdList {
			addMappingList = append(addMappingList, &edb_refresh.EdbRefreshMapping{
				EdbRefreshMappingId: 0,
				Source:              source,
				SubSource:           subSource,
				EdbInfoId:           edbId,
				EdbRefreshConfigId:  item.EdbRefreshConfigId,
				SysUserId:           sysUserId,
				SysUserRealName:     sysUserRealName,
				ModifyTime:          time.Now(),
				CreateTime:          time.Now(),
			})
		}
	}

	err = edb_refresh.SaveEdbRefreshConfig(source, subSource, sysUserId, sysUserRealName, addConfigList, addMappingList, edbIdList)

	return
}

func HandleRefreshTime(source, subSource int, list []*data_manage.BaseRefreshEdbInfo) (newList []*data_manage.BaseRefreshEdbInfo, err error, errMsg string, isSendEmail bool) {
	isSendEmail = true
	errMsg = "获取失败"
	newList = list

	defaultEdbRefreshConfigMap := make(map[string]string)
	{
		allDefaultEdbRefreshConfigList, tmpErr := GetAllDefaultEdbRefreshConfigListBySourceList(source, subSource)
		if tmpErr != nil {
			err = tmpErr
			return
		}

		defaultEdbRefreshConfigListMap := make(map[string][]string)
		for _, v := range allDefaultEdbRefreshConfigList {
			tmp, ok := defaultEdbRefreshConfigListMap[v.Frequency]
			if !ok {
				tmp = make([]string, 0)
			}
			defaultEdbRefreshConfigListMap[v.Frequency] = append(tmp, GetRefreshStr(v.RefreshFrequency, v.RefreshFrequencyDay, v.RefreshTime))
		}
		for k, v := range defaultEdbRefreshConfigListMap {
			defaultEdbRefreshConfigMap[k] = strings.Join(v, "，")
		}

	}

	edbInfoIdList := make([]int, 0)
	for _, v := range newList {
		edbInfoIdList = append(edbInfoIdList, v.EdbInfoId)
	}

	edbRefreshConfigMap := make(map[int]string)
	{
		configList, tmpErr := edb_refresh.GetEdbRefreshConfigAndEdbListBySourceAndeEdbInfoId(source, subSource, edbInfoIdList)
		if tmpErr != nil {
			err = tmpErr
			return
		}

		edbRefreshConfigListMap := make(map[int][]string)
		for _, v := range configList {
			tmp, ok := edbRefreshConfigListMap[v.EdbInfoId]
			if !ok {
				tmp = make([]string, 0)
			}
			edbRefreshConfigListMap[v.EdbInfoId] = append(tmp, GetRefreshStr(v.RefreshFrequency, v.RefreshFrequencyDay, v.RefreshTime))
		}
		for k, v := range edbRefreshConfigListMap {
			edbRefreshConfigMap[k] = strings.Join(v, "，")
		}
	}

	for _, v := range newList {
		refreshTime, ok := edbRefreshConfigMap[v.EdbInfoId]
		if ok {
			v.RefreshTime = refreshTime
			continue
		}
		if source == utils.DATA_SOURCE_YS {
			v.RefreshTime = defaultEdbRefreshConfigMap[``]
		} else {
			v.RefreshTime = defaultEdbRefreshConfigMap[v.Frequency]
		}
	}

	return
}

func GetList(source, subSource int, classifyId, terminalCode, sysUserId, frequency, keyword, status string, startSize, pageSize int, sortParam, sortType string) (total int, list []*data_manage.BaseRefreshEdbInfo, err error) {
	var pars []interface{}
	var condition string

	list = make([]*data_manage.BaseRefreshEdbInfo, 0)

	isStop := -1
	if status == `暂停` {
		isStop = 1
	} else if status == "启用" {
		isStop = 0
	}

	switch source {
	default:
		fmt.Println("line 530")
		condition += ` AND source = ? AND sub_source = ? `
		pars = append(pars, source, subSource)

		if isStop >= 0 {
			condition += " AND no_update = ? "
			pars = append(pars, isStop)
		}

		if classifyId != `` {
			classifyIdSlice := strings.Split(classifyId, ",")
			condition += ` AND classify_id IN (` + utils.GetOrmInReplace(len(classifyIdSlice)) + `)`
			pars = append(pars, classifyIdSlice)
		}
		if terminalCode != `` {
			condition += " AND terminal_code = ? "
			pars = append(pars, terminalCode)
		}
		if sysUserId != `` {
			sysUserIdSlice := strings.Split(sysUserId, ",")
			condition += ` AND sys_user_id IN (` + utils.GetOrmInReplace(len(sysUserIdSlice)) + `)`
			pars = append(pars, sysUserIdSlice)
		}
		if frequency != `` {
			frequencySlice := strings.Split(frequency, ",")
			condition += ` AND frequency IN (` + utils.GetOrmInReplace(len(frequencySlice)) + `)`
			pars = append(pars, frequencySlice)
		}
		if keyword != `` {
			keywordSlice := strings.Split(keyword, " ")
			if len(keywordSlice) > 0 {
				tmpConditionSlice := make([]string, 0)
				tmpConditionSlice = append(tmpConditionSlice, ` edb_name like ? or edb_code like ? `)
				pars = utils.GetLikeKeywordPars(pars, keyword, 2)

				for _, v := range keywordSlice {
					if v == ` ` || v == `` {
						continue
					}
					tmpConditionSlice = append(tmpConditionSlice, ` edb_name like ? or edb_code like ? `)
					pars = utils.GetLikeKeywordPars(pars, v, 2)
				}
				condition += ` AND (` + strings.Join(tmpConditionSlice, " or ") + `)`

			} else {
				condition += ` edb_name like ? or edb_code like ? `
				pars = utils.GetLikeKeywordPars(pars, keyword, 2)
			}
		}

		sortStr := ``
		if sortParam != `` {
			sortStr = fmt.Sprintf("%s %s,edb_info_id desc ", sortParam, sortType)
		}
		fmt.Println("line 582")
		fmt.Println(condition, pars, sortStr, startSize, pageSize)
		total, list, err = data_manage.GetEdbBaseInfoList(condition, pars, sortStr, startSize, pageSize)
	}

	return
}

func getAllDefaultEdbRefreshConfigKey(source, subSource int) string {
	return allDefaultEdbRefreshConfigKey + fmt.Sprintf("%d_%d", source, subSource)
}

func GetRefreshStr(refreshFrequency string, refreshFrequencyDay int, refreshTime string) string {
	refreshDayStr := ``
	switch refreshFrequency {
	case "每自然日", "每交易日":
	case "每周":
		switch refreshFrequencyDay {
		case 0:
			refreshDayStr = "日"
		case 1:
			refreshDayStr = "一"
		case 2:
			refreshDayStr = "二"
		case 3:
			refreshDayStr = "三"
		case 4:
			refreshDayStr = "四"
		case 5:
			refreshDayStr = "五"
		case 6:
			refreshDayStr = "六"
		case 7:
			refreshDayStr = "日"
		}
	default:
		if refreshFrequencyDay > 0 {
			refreshDayStr = fmt.Sprintf("第%d天", refreshFrequencyDay)
		} else {
			refreshDayStr = `最后一天`
		}
	}
	return refreshFrequency + refreshDayStr + " " + refreshTime
}