mysteel_watch/watch/edb_lib.go

package watch

import (
	"encoding/json"
	"errors"
	"eta/mysteel_watch/global"
	"eta/mysteel_watch/models/index"
	"eta/mysteel_watch/utils"
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)

type BaseResponse struct {
	Ret         int
	Msg         string
	ErrMsg      string
	ErrCode     string
	Data        interface{}
	Success     bool `description:"true 执行成功，false 执行失败"`
	IsSendEmail bool `json:"-" description:"true 发送邮件，false 不发送邮件"`
	IsAddLog    bool `json:"-" description:"true 新增操作日志，false 不新增操作日志" `
}

type IndexesResponse struct {
	Ret         int
	Msg         string
	ErrMsg      string
	ErrCode     string
	Data        []index.BaseFromMysteelChemicalIndex
	Success     bool `description:"true 执行成功，false 执行失败"`
	IsSendEmail bool `json:"-" description:"true 发送邮件，false 不发送邮件"`
	IsAddLog    bool `json:"-" description:"true 新增操作日志，false 不新增操作日志" `
}

type IndexResponse struct {
	Ret         int
	Msg         string
	ErrMsg      string
	ErrCode     string
	Data        index.BaseFromMysteelChemicalIndex
	Success     bool `description:"true 执行成功，false 执行失败"`
	IsSendEmail bool `json:"-" description:"true 发送邮件，false 不发送邮件"`
	IsAddLog    bool `json:"-" description:"true 新增操作日志，false 不新增操作日志" `
}

// postRefreshEdbData 刷新指标数据
//func postRefreshEdbData(param map[string]interface{}) (resp *BaseResponse, err error) {
//	urlStr := "mysteel_chemical/refresh"
//
//	if global.CONFIG.Serve.EdbLibUrl == `` {
//		err = errors.New("刷新未配置")
//		return
//	}
//	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
//	postData, err := json.Marshal(param)
//	if err != nil {
//		return
//	}
//	result, err := HttpPost(postUrl, string(postData), "application/json")
//	if err != nil {
//		return
//	}
//	if result != nil {
//		global.LOG.Info(" Refresh Result: " + string(result))
//		err = json.Unmarshal(result, &resp)
//		if err != nil {
//			return
//		}
//		return resp, nil
//	}
//	return nil, err
//}

// postHandleMysteelIndex 处理指标和指标数据
func postHandleMysteelIndex(req *HandleMysteelIndexReq) (resp *BaseResponse, err error) {
	urlStr := "/mysteel_chemical/handle/mysteel/index"

	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
	postData, err := json.Marshal(req)
	if err != nil {
		return
	}
	//fmt.Println("postData:" + string(postData))
	global.LOG.Debug(" postData: " + string(postData))
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}
	if result != nil {
		global.LOG.Debug(" postHandleMysteelIndex Result: " + string(result))
		err = json.Unmarshal(result, &resp)
		if err != nil {
			return
		}
		return resp, nil
	}
	return nil, err
}

// GetMaxFileIndexReq 获取最大文件数请求
type GetMaxFileIndexReq struct {
	Frequency    string
	EndDate      string
	TerminalCode string `description:"终端编码"`
}

// GetNoMergeIndexByFrequencyCount 获取未合并的指标总数量
func GetNoMergeIndexByFrequencyCount(req GetMaxFileIndexReq) (total float64, err error) {
	urlStr := "/mysteel_chemical/getNoMergeIndexByFrequencyCount"

	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
	postData, err := json.Marshal(req)
	if err != nil {
		return
	}
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp BaseResponse
	global.LOG.Debug(" GetNoMergeIndexByFrequencyCount Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	t, ok := resp.Data.(float64)
	if ok {
		total = t
	}
	return
}

// GetMaxFileIndex 获取最大的文件编号下标
func GetMaxFileIndex(req GetMaxFileIndexReq) (baseFromMysteelChemicalIndex index.BaseFromMysteelChemicalIndex, err error) {
	urlStr := "/mysteel_chemical/getMaxFileIndex"

	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
	postData, err := json.Marshal(req)
	if err != nil {
		return
	}
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp IndexResponse
	global.LOG.Debug(" Refresh Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	baseFromMysteelChemicalIndex = resp.Data
	return
}

// GetNoMergeIndexByFrequencyPageReq 获取最大文件数请求
type GetNoMergeIndexByFrequencyPageReq struct {
	Frequency string
	Limit     int
}

type GetNoMergeIndexByFrequencyPageReqV2 struct {
	Frequency    string
	EndDate      string
	StartSize    int
	PageSize     int
	TerminalCode string `description:"终端编码"`
}

// GetNoMergeIndexByFrequencyPage 当下需要合并到excel的指标
func GetNoMergeIndexByFrequencyPage(frequency string, limit int) (item []index.BaseFromMysteelChemicalIndex, err error) {
	urlStr := "/mysteel_chemical/getNoMergeIndexByFrequencyPage"

	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
	postData, err := json.Marshal(GetNoMergeIndexByFrequencyPageReq{
		Frequency: frequency,
		Limit:     limit,
	})
	if err != nil {
		return
	}
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetNoMergeIndexByFrequencyPage Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	item = resp.Data
	return
}

// GetIndexCountByMergeFilePathReq 获取该文件下标已经存在多少的指标请求体
type GetIndexCountByMergeFilePathReq struct {
	MergeFilePath string
}

// GetIndexCountByMergeFilePath 获取该文件下标已经存在多少的指标
func GetIndexCountByMergeFilePath(mergeFilePath string) (total float64, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	if mergeFilePath == "" {
		err = fmt.Errorf("文件路径为空")
		return
	}

	baseUrl := "/mysteel_chemical/getIndexCountByMergeFilePath"
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)
	postData, err := json.Marshal(GetIndexCountByMergeFilePathReq{
		MergeFilePath: mergeFilePath,
	})
	result, err := HttpPost(url, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp BaseResponse
	global.LOG.Debug(" GetIndexCountByMergeFilePath Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}
	t, ok := resp.Data.(float64)
	if ok {
		total = t
	}
	return
}

// GetIndexCreate 获取两分钟前新增的指标
func GetIndexCreate() (items []index.BaseFromMysteelChemicalIndex, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}

	baseUrl := "/mysteel_chemical/getIndexCreate?TerminalCode=" + global.CONFIG.Serve.TerminalCode
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)
	result, err := HttpPost(url, "", "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetIndexCreate Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	items = resp.Data
	return
}

// GetIndexRefreshAllByMergeFile 获取全部刷新所需文件
func GetIndexRefreshAllByMergeFile() (items []index.BaseFromMysteelChemicalIndex, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}

	baseUrl := "/mysteel_chemical/query/refresh"
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)

	result, err := HttpPost(url, "", "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetIndexRefreshAllByMergeFile Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	items = resp.Data
	return
}

// GetIndexRefreshMethanolByMergeFile 获取需要刷新的甲醇文件
func GetIndexRefreshMethanolByMergeFile() (items []index.BaseFromMysteelChemicalIndex, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}

	baseUrl := "/mysteel_chemical/getIndexRefreshMethanolByMergeFile"
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)
	result, err := HttpPost(url, "", "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetIndexRefreshMethanolByMergeFile Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	items = resp.Data
	return
}

// GetIndexRefreshMethanolByTimely 获取需要及时刷新的文件
func GetIndexRefreshMethanolByTimely() (items []index.BaseFromMysteelChemicalIndex, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}

	baseUrl := "/mysteel_chemical/getIndexRefreshMethanolByTimely"
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)
	result, err := HttpPost(url, "", "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetIndexRefreshMethanolByTimely Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	items = resp.Data
	return
}

type GetIndexByFrequencyReq struct {
	Frequency string
}

// GetIndexByFrequency 根据频率获取指标
func GetIndexByFrequency(frequency string) (items []index.BaseFromMysteelChemicalIndex, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	if frequency == "" {
		err = fmt.Errorf("频率为空")
		return
	}

	baseUrl := "/mysteel_chemical/getIndexByFrequency"
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)
	postData, err := json.Marshal(GetIndexByFrequencyReq{
		Frequency: frequency,
	})
	result, err := HttpPost(url, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetIndexByFrequency Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	items = resp.Data
	return
}

type GetIndexRefreshWeekReq struct {
	NowWeekZn string
	StartTime string
	EndTime   string
}

// GetIndexRefreshWeek 根据周获取指标
func GetIndexRefreshWeek(nowWeekZn, startTime, endTime string) (items []index.BaseFromMysteelChemicalIndex, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	if nowWeekZn == "" || startTime == "" || endTime == "" {
		err = fmt.Errorf("参数有误, nowWeekZn: %s, startTime: %s, endTime: %s", nowWeekZn, startTime, endTime)
		return
	}

	baseUrl := "/mysteel_chemical/getIndexRefreshWeek"
	url := fmt.Sprint(global.CONFIG.Serve.EdbLibUrl, baseUrl)
	postData, err := json.Marshal(GetIndexRefreshWeekReq{
		NowWeekZn: nowWeekZn,
		StartTime: startTime,
		EndTime:   endTime,
	})
	result, err := HttpPost(url, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp IndexesResponse
	global.LOG.Debug(" GetIndexRefreshWeek Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	items = resp.Data
	return
}

// UpdateIndexReq 更新指标
type UpdateIndexReq struct {
	Item       *index.BaseFromMysteelChemicalIndex
	UpdateCols []string
}

// MultiUpdateIndex 批量更新指标
func MultiUpdateIndex(req []UpdateIndexReq) (err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}
	if len(req) == 0 {
		err = fmt.Errorf("无指标需要更新")
		return
	}

	urlStr := "/mysteel_chemical/multiUpdateIndex"
	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
	postData, err := json.Marshal(req)
	if err != nil {
		return
	}
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp BaseResponse
	global.LOG.Debug(" MultiUpdateIndex Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}
	if resp.Ret != 200 {
		err = fmt.Errorf(resp.ErrMsg)
	}
	return
}

type RefreshCheckByDayReq struct {
	Source         int    `description:"来源id"`
	LatestDate     string `description:"数据最新日期"`
	FrequencyBatch string `description:"更新频度"`
}

type RefreshCheckByDayData struct {
	UpdateNum   int `description:"已更新指标数"`
	UnUpdateNum int `description:"未更新指标数"`
}

type RefreshCheckByDayResponse struct {
	Ret     int
	Msg     string
	ErrMsg  string
	ErrCode string
	Data    RefreshCheckByDayData
	Success bool `description:"true 执行成功，false 执行失败"`
}

// RefreshCheckByDay  每日更新检查
func RefreshCheckByDay(req RefreshCheckByDayReq) (data RefreshCheckByDayData, err error) {
	if global.CONFIG.Serve.EdbLibUrl == `` {
		err = errors.New("刷新未配置")
		return
	}

	urlStr := "/edb_info/refresh_check"
	postUrl := global.CONFIG.Serve.EdbLibUrl + urlStr
	postData, err := json.Marshal(req)
	if err != nil {
		return
	}
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}

	var resp RefreshCheckByDayResponse
	global.LOG.Debug(" RefreshCheckByDay Result: " + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}
	if resp.Ret != 200 {
		err = fmt.Errorf(resp.ErrMsg)
	}
	data = resp.Data
	return
}

func HttpPost(url, postData string, params ...string) ([]byte, error) {
	body := ioutil.NopCloser(strings.NewReader(postData))
	client := &http.Client{}
	req, err := http.NewRequest("POST", url, body)
	if err != nil {
		return nil, err
	}
	contentType := "application/x-www-form-urlencoded;charset=utf-8"
	if len(params) > 0 && params[0] != "" {
		contentType = params[0]
	}
	req.Header.Set("Content-Type", contentType)

	req.Header.Set("authorization", utils.MD5(global.CONFIG.Serve.AppEdbLibNameEn+global.CONFIG.Serve.EdbLibMd5Key))
	resp, err := client.Do(req)
	if resp != nil {
		defer resp.Body.Close()
		b, err := ioutil.ReadAll(resp.Body)
		//fmt.Println("HttpPost:" + string(b))
		return b, err
	}
	return nil, err
}