eta_api/services/data/supply_analysis/base_supply_analysis_lib.go

package supply_analysis

import (
	"encoding/json"
	"eta/eta_api/models"
	"eta/eta_api/utils"
	"io/ioutil"
	"net/http"
	"strings"
)

type CalculateEdbDataResponse 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 不新增操作日志" `
}

// Calculate 分析品种
func Calculate(paramStr string) (resp *CalculateEdbDataResponse, err error) {
	_, resultByte, err := postAddEdbData(paramStr, "stock_plant/calculate")
	err = json.Unmarshal(resultByte, &resp)
	if err != nil {
		return
	}
	return
}

// postAddEdbData 新增指标数据
func postAddEdbData(paramStr string, urlStr string) (resp *models.BaseResponse, result []byte, err error) {
	postUrl := utils.EDB_LIB_URL + urlStr
	result, err = HttpPost(postUrl, paramStr, "application/json")
	if err != nil {
		return
	}
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}

	return
}

// RefreshEdbData 刷新指标数据
func RefreshEdbData(futureGoodEdbInfoId int, futureGoodEdbCode, startDate string) (resp *models.BaseResponse, err error) {
	param := make(map[string]interface{})
	param["FutureGoodEdbCode"] = futureGoodEdbCode
	param["FutureGoodEdbInfoId"] = futureGoodEdbInfoId
	param["StartDate"] = startDate
	urlStr := `future_good/refresh`
	resp, err = postRefreshEdbData(param, urlStr)
	return
}

// postRefreshEdbData 刷新指标数据
func postRefreshEdbData(param map[string]interface{}, urlStr string) (resp *models.BaseResponse, err error) {
	postUrl := utils.EDB_LIB_URL + urlStr
	postData, err := json.Marshal(param)
	if err != nil {
		return
	}
	result, err := HttpPost(postUrl, string(postData), "application/json")
	if err != nil {
		return
	}
	utils.FileLog.Info("postRefreshEdbData:" + postUrl + ";" + string(postData) + ";result:" + string(result))
	err = json.Unmarshal(result, &resp)
	if err != nil {
		return
	}
	return resp, nil
}

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(utils.APP_EDB_LIB_NAME_EN+utils.EDB_LIB_Md5_KEY))
	resp, err := client.Do(req)
	defer resp.Body.Close()
	b, err := ioutil.ReadAll(resp.Body)
	utils.FileLog.Debug("HttpPost:" + string(b))
	return b, err
}