hongze_cygx/services/chart.go

package services

import (
	"context"
	"encoding/json"
	"fmt"
	"hongze/hongze_cygx/models"
	"hongze/hongze_cygx/utils"
	"io/ioutil"
	nhttp "net/http"
	"strconv"
	"strings"
	"time"
)

//get公共请求方法
func PublicGetDate(url, authorization string) (body []byte, err error) {
	defer func() {
		if err != nil {
			go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", url+"Get ErrMsg:"+err.Error(), utils.EmailSendToUsers)
		}
	}()
	method := "GET"
	client := &nhttp.Client{}
	req, err := nhttp.NewRequest(method, url, nil)
	if err != nil {
		return
	}
	req.Header.Add("Authorization", authorization)
	res, err := client.Do(req)
	if err != nil {
		return
	}
	defer res.Body.Close()
	body, err = ioutil.ReadAll(res.Body)
	if err != nil {
		return
	}
	return
}

//获取列表
func GetChartListByApi(cont context.Context) (err error) {
	defer func() {
		if err != nil {
			go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "GetArticleListByApi ErrMsg:"+err.Error(), utils.EmailSendToUsers)
		}
	}()
	url := utils.ApiUrl + "charts/mp?take=100&skip=0"
	authorization := utils.ApiAuthorization
	body, err := PublicGetDate(url, authorization)
	if err != nil {
		return
	}
	var chartResult models.ChartResultApi
	err = json.Unmarshal(body, &chartResult)
	if err != nil {
		return err
	}
	for _, v := range chartResult.Data {
		item := new(models.CygxChart)
		item.ChartId = v.ChartId
		item.PtagId = v.PtagId
		item.CtagId = v.CtagId
		item.Title = v.Title
		item.TitleEn = v.TitleEn
		item.CreateDateApi = time.Now()
		item.CreateDate = v.CreateDate
		item.PublishStatus = v.PublishStatus
		item.PtagName = v.Ptag.Name
		item.CtagName = v.Ctag.Name
		item.Cover = v.Cover
		item.Iframe = v.Iframe
		count, err := models.GetChartCountById(v.ChartId)
		if err != nil && err.Error() != utils.ErrNoRow() {
			return err
		}
		if count == 0 {
			_, err := models.AddCygxChart(item)
			if err != nil {
				return err
			}
		} else {
			updateParams := make(map[string]interface{})
			updateParams["Title"] = v.Title
			updateParams["PtagId"] = v.PtagId
			updateParams["CtagId"] = v.CtagId
			updateParams["TitleEn"] = v.TitleEn
			updateParams["CreateDate"] = v.CreateDate
			updateParams["PublishStatus"] = v.PublishStatus
			updateParams["PtagName"] = v.Ptag.Name
			updateParams["CtagName"] = v.Ctag.Name
			updateParams["Cover"] = v.Cover
			updateParams["Iframe"] = v.Iframe
			whereParam := map[string]interface{}{"chart_id": v.ChartId}
			err = models.UpdateByExpr(models.CygxChart{}, whereParam, updateParams)
		}
	}
	return

}

//获取图表分类
func GetChartPtagByApi() (items []*models.ChartPtagResp, err error) {
	defer func() {
		if err != nil {
			go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "GetChartPtagByApi ErrMsg:"+err.Error(), utils.EmailSendToUsers)
		}
	}()
	url := utils.ApiUrl + "charts/ptag"
	authorization := utils.ApiAuthorization
	body, err := PublicGetDate(url, authorization)
	if err != nil {
		return
	}
	var chartResult models.ChartPtagResultApi
	err = json.Unmarshal(body, &chartResult)
	if err != nil {
		return
	}
	for _, v := range chartResult.Data {
		item := new(models.ChartPtagResp)
		item.PermissionName = v.ChartPermissionName
		item.ChartPermissionId = v.ChartPermissionId
		if len(v.Ctag) > 0 {
			for _, v2 := range v.Ctag {
				itemCtag := new(models.CtagResp)
				itemCtag.CtagId = v2.Id
				itemCtag.Name = v2.Name
				item.List = append(item.List, itemCtag)
			}
		}
		items = append(items, item)
	}
	return
}

//获取用户的Token
func GetUserTokenByMobile(mobile string) (token string, err error) {
	//缓存校验
	cacheKey := fmt.Sprint("xygx_chart:chart_token:add:", "Mobile", mobile)
	ttlTime := utils.Rc.GetRedisTTL(cacheKey)
	if ttlTime > 0 {
		token, _ = utils.Rc.RedisString(cacheKey)
	}
	if token == "" {
		url := utils.ApiUrl + "auth/login"
		method := "POST"
		payload := strings.NewReader(`{
    "phone_number":"` + mobile + `",
    "password":"hz123456"}`)
		client := &nhttp.Client{}
		req, errReq := nhttp.NewRequest(method, url, payload)
		if errReq != nil {
			err = errReq
			return
		}
		req.Header.Add("Content-Type", "application/json")
		req.Header.Add("Cookie", "sessionid=naj5j5kl1jjynh7og1rsaxkl1vrsl829")
		res, errReq := client.Do(req)
		if errReq != nil {
			err = errReq
			return
		}
		defer res.Body.Close()
		body, errReq := ioutil.ReadAll(res.Body)
		if errReq != nil {
			err = errReq
			return
		}
		var chartResult models.ChartUserTokenResultApi
		errReq = json.Unmarshal(body, &chartResult)
		if errReq != nil {
			err = errReq
			return
		}
		token = chartResult.Data.AccessToken
		utils.Rc.Put(cacheKey, token, time.Hour*24)
	}
	return
}

//获取图表收藏
func GetChartCollectionByApi(mobile string, take, skip int) (items []*models.HomeChartListResp, err error, total int) {
	defer func() {
		if err != nil {
			go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "GetChartPtagByApi ErrMsg:"+err.Error(), utils.EmailSendToUsers)
		}
	}()
	url := utils.ApiUrl + "charts/favorites?take=" + strconv.Itoa(take) + "&skip=" + strconv.Itoa(skip)
	authorization, err := GetUserTokenByMobile(mobile)
	if err != nil {
		return
	}
	authorization = "bearer " + authorization
	body, err := PublicGetDate(url, authorization)
	if err != nil {
		return
	}
	var chartResult models.ChartFavoritesResultApi
	err = json.Unmarshal(body, &chartResult)
	if err != nil {
		return
	}
	for _, v := range chartResult.Data {
		item := new(models.HomeChartListResp)
		item.ChartId = v.ChartId
		item.Title = v.ChartInfo.Title
		item.TitleEn = v.ChartInfo.TitleEn
		item.CreateDate = v.ChartInfo.CreateDate
		item.PtagName = v.ChartInfo.Ptag.Name
		item.CtagName = v.ChartInfo.Ctag.Name
		item.BodyHtml = v.ChartInfo.Cover
		item.HttpUrl = "https://vmp.hzinsights.com/v2/charts/" + strconv.Itoa(v.ChartId)
		item.IsNeedJump = true
		items = append(items, item)
	}
	total = chartResult.Pagination.Total
	return
}

//判断策略平台是否已经添加过收藏
func GetIsCollectionChart(mobile string, chartId int) (isCollection bool, err error) {
	//获取所有的收藏列表，进行比对看看是否收藏，调用三方接口详情没有是否收藏的字段
	list, err, _ := GetChartCollectionByApi(mobile, 9999, 0)
	if err != nil {
		return
	}
	for _, v := range list {
		if v.ChartId == chartId {
			isCollection = true
		}
	}
	return
}

//添加收藏
func AddCollectionChart(mobile string, chartId int) (err error) {
	authorization, err := GetUserTokenByMobile(mobile)
	if err != nil {
		return
	}
	authorization = "bearer " + authorization
	url := utils.ApiUrl + "charts/favorites"
	method := "POST"
	payload := strings.NewReader(`{
    "chart_id":` + strconv.Itoa(chartId) + `
}`)
	client := &nhttp.Client{}
	req, err := nhttp.NewRequest(method, url, payload)
	if err != nil {
		return
	}
	req.Header.Add("Authorization", authorization)
	req.Header.Add("Content-Type", "application/json")
	req.Header.Add("Cookie", "sessionid=naj5j5kl1jjynh7og1rsaxkl1vrsl829")
	res, err := client.Do(req)
	if err != nil {
		return
	}
	defer res.Body.Close()
	_, err = ioutil.ReadAll(res.Body)
	if err != nil {
		return
	}
	return
}

//移除收藏
func DeleteCollectionChart(mobile string, chartId int) (err error) {
	authorization, err := GetUserTokenByMobile(mobile)
	if err != nil {
		return
	}
	authorization = "bearer " + authorization
	url := utils.ApiUrl + "charts/favorites/" + strconv.Itoa(chartId)
	method := "DELETE"
	client := &nhttp.Client{}
	req, err := nhttp.NewRequest(method, url, nil)
	if err != nil {
		return
	}
	req.Header.Add("Authorization", authorization)
	req.Header.Add("Cookie", "sessionid=naj5j5kl1jjynh7og1rsaxkl1vrsl829")
	res, err := client.Do(req)
	if err != nil {
		return
	}
	defer res.Body.Close()
	_, err = ioutil.ReadAll(res.Body)
	if err != nil {
		return
	}
	return
}