eta_api/services/data_stat/edb_terminal.go

package data_stat

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

// SaveEdbTerminal 保存数据源终端信息
func SaveEdbTerminal(req *data_manage.AddEdbTerminalListReq) (errMsg string, err error) {
	nowTime := time.Now().Local()
	// 更新
	if req.TerminalId > 0 {
		item, e := data_manage.GetEdbTerminalById(req.TerminalId)
		if e != nil {
			if e.Error() == utils.ErrNoRow() {
				errMsg = "终端不存在"
				err = errors.New(errMsg)
				return
			}
			errMsg = "获取终端信息失败"
			err = errors.New("获取终端信息失败, Err:" + e.Error())
			return
		}
		updateCols := make([]string, 0)
		updateCols = append(updateCols, "Source", "Name", "ServerUrl", "DirPath", "Num", "Value", "ModifyTime")
		item.Source = req.Source
		item.Name = req.Name
		item.ServerUrl = req.ServerUrl
		item.DirPath = req.DirPath
		item.Num = req.Num
		item.Value = req.Value
		// todo 状态是否可修改
		//item.Status = 1
		item.ModifyTime = nowTime
		if e := item.Update(updateCols); e != nil {
			errMsg = "更新失败"
			err = errors.New("更新终端失败, Err:" + e.Error())
		}
		return
	}
	terminalCode := createTerminalCode(req.Source, req.Name)
	//item.TerminalCode = req.TerminalCode
	// 新增
	newTerminal := &data_manage.EdbTerminal{
		Source:       req.Source,
		Name:         req.Name,
		TerminalCode: terminalCode,
		ServerUrl:    req.ServerUrl,
		DirPath:      req.DirPath,
		Num:          req.Num,
		Status:       1,
		Value:        req.Value,
		CreateTime:   nowTime,
		ModifyTime:   nowTime,
	}
	if e := newTerminal.Add(); e != nil {
		errMsg = "新增失败"
		err = errors.New("新增终端失败, Err:" + e.Error())
	}
	return
}

// SetEdbTerminalStatus 启用/禁用 终端
func SetEdbTerminalStatus(id int, status int) (errMsg string, err error) {
	nowTime := time.Now().Local()
	// 更新
	item, e := data_manage.GetEdbTerminalById(id)
	if e != nil {
		if e.Error() == utils.ErrNoRow() {
			errMsg = "终端不存在"
			err = errors.New(errMsg)
			return
		}
		errMsg = "获取终端信息失败"
		err = errors.New("获取终端信息失败, Err:" + e.Error())
		return
	}
	updateCols := make([]string, 0)
	updateCols = append(updateCols, "Status", "ModifyTime")
	item.Status = status
	item.ModifyTime = nowTime
	if e := item.Update(updateCols); e != nil {
		errMsg = "更新失败"
		err = errors.New("更新终端失败, Err:" + e.Error())
	}
	return
}

func createTerminalCode(source int, sourceName string) (terminalCode string) {
	randInt := utils.GetRandNumber(1000)
	terminalCode = utils.MD5(fmt.Sprintf("%d_%s_%s_%d", source, sourceName, time.Now(), randInt))
	return
}

// GetEdbTerminalDirInfo 获取指标的终端信息
func GetEdbTerminalDirInfo(indexId, source int) (info *data_manage.EdbTerminalDirInfo, err error) {
	// 查询指标信息
	var terminalCode, filePath, name, dirPath string
	switch source {
	case utils.DATA_SOURCE_MYSTEEL_CHEMICAL:
		// 查询钢联指标信息
		indexInfo, e := data_manage.GetBaseFromMysteelChemicalIndexByIndexId(indexId)
		if e != nil {
			err = errors.New("获取指标信息失败, Err:" + e.Error())
			return
		}
		terminalCode = indexInfo.TerminalCode
		filePath = indexInfo.MergeFilePath
		if filePath == "" && indexInfo.FilePath != "" {
			filePath = indexInfo.FilePath
		}
		break
	case utils.DATA_SOURCE_YS:
		// 查询有色指标信息
		indexInfo, e := data_manage.GetBaseFromSmmIndexByIndexId(indexId)
		if e != nil {
			err = errors.New("获取指标信息失败, Err:" + e.Error())
			return
		}
		terminalCode = indexInfo.TerminalCode
		filePath = indexInfo.RenameFileName
		break
	case utils.DATA_SOURCE_SCI:
		// 查询SCI指标信息
		indexInfo, e := data_manage.GetBaseFromSciIndexByIndexId(indexId)
		if e != nil {
			err = errors.New("获取指标信息失败, Err:" + e.Error())
			return
		}
		terminalCode = indexInfo.TerminalCode
		filePath = indexInfo.FilePath
		break
	default:
		err = errors.New("不支持其他来源")
		return
	}
	// 查询终端信息
	if terminalCode == "" {
		err = errors.New("获取终端信息失败, Err:终端编码为空")
		return
	}
	terminalInfo, err := data_manage.GetEdbTerminalByCode(terminalCode)
	if err != nil {
		err = errors.New("获取终端信息失败, Err:" + err.Error())
		return
	}
	if terminalInfo.Source != source {
		err = errors.New("获取终端信息失败, Err:指标来源与终端来源不一致")
		return
	}
	name = terminalInfo.Name
	// 截取最后的文件名称
	if filePath != "" {
		// 使用 strings.LastIndex 和切片操作获取文件路径（不包括文件名）
		lastIndex := strings.LastIndex(filePath, `\`)
		if lastIndex != -1 {
			dirPath = filePath[:lastIndex]
			//fmt.Println("文件路径:", dirPath)
			fileName := filePath[lastIndex+1:]
			//fmt.Println("文件名:", fileName)
			filePath = fileName
		}
	}
	info = &data_manage.EdbTerminalDirInfo{
		TerminalCode: terminalCode,
		FilePath:     filePath,
		Name:         name,
		DirPath:      dirPath,
	}
	return
}