fix: 报告详情公众号分享

controllers/wechat.go

@@ -52,21 +52,48 @@ func (this *WechatCommonController) GetWxSign() {
 		WxAppId:     appId,
 		WxAppSecret: appSecret,
 	}
-	accessToken, err, errMsg := wechat.GetAccessToken(tReq)
-	if err != nil {
-		br.Msg = "获取微信信息失败"
-		br.ErrMsg = "获取access_token失败，err:" + errMsg
-		return
+
+	var accessToken string
+	if utils.BusinessCode == utils.BusinessCodeRelease {
+		// 弘则ETA从eta_pub统一获取
+		token, e, msg := wechat.GetAccessToken(tReq)
+		if e != nil {
+			br.Msg = "获取微信配置失败"
+			br.ErrMsg = "获取access_token失败，err:" + msg
+			return
+		}
+		accessToken = token
+	} else {
+		// 从eta_report获取
+		token, e, msg := wechat.GetWxAccessToken(tReq.WxAppId, tReq.WxAppSecret)
+		if e != nil {
+			br.Msg = "获取微信信息失败"
+			br.ErrMsg = "获取access_token失败，err:" + msg
+			return
+		}
+		accessToken = token
 	}
 
 	errCode, ticket, err := services.GetWxTicket(accessToken)
 	if err != nil {
 		if errCode == 40001 {
-			accessToken, err, errMsg = wechat.GetAccessToken(tReq)
-			if err != nil {
-				br.Msg = "获取微信配置失败"
-				br.ErrMsg = "获取access_token失败，err:" + errMsg
-				return
+			// 重新获取
+			if utils.BusinessCode == utils.BusinessCodeRelease {
+				token, e, msg := wechat.GetAccessToken(tReq)
+				if e != nil {
+					br.Msg = "获取微信配置失败"
+					br.ErrMsg = "获取access_token失败，err:" + msg
+					return
+				}
+				accessToken = token
+			} else {
+				token, e, msg := wechat.GetWxAccessToken(tReq.WxAppId, tReq.WxAppSecret)
+				if e != nil {
+					br.Msg = "获取微信信息失败"
+					br.ErrMsg = "获取access_token失败，err:" + msg
+					return
+				}
+				accessToken = token
 			}
 
 			errCode, ticket, err = services.GetWxTicket(accessToken)

models/wechat.go

@@ -1,5 +1,12 @@
 package models
 
+import (
+	"encoding/json"
+	"eta/eta_report/utils"
+	"fmt"
+	"github.com/rdlucklib/rdluck_tools/http"
+)
+
 type WechatSign struct {
 	AppId     string
 	NonceStr  string
@@ -14,3 +21,29 @@ type WxTicket struct {
 	Errmsg  string `json:"errmsg"`
 	Ticket  string `json:"ticket"`
 }
+
+type WxAccessToken struct {
+	AccessToken string `json:"access_token"`
+	ExpiresIn   int    `json:"expires_in"`
+	Errcode     int    `json:"errcode"`
+	Errmsg      string `json:"errmsg"`
+}
+
+func GetWxToken(wxAppId, wxAppSecret string) (item *WxAccessToken, err error) {
+	getUrl := "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=" + wxAppId + "&secret=" + wxAppSecret
+	result, err := http.Get(getUrl)
+	if err != nil {
+		utils.FileLog.Info("GetWxToken Err:", err.Error())
+		return
+	}
+	utils.FileLog.Info(fmt.Sprintf("GetWxAccessToken :%s", string(result)))
+	err = json.Unmarshal(result, &item)
+	if err != nil {
+		fmt.Println(fmt.Sprintf("GetWxToken Unmarshal Err:%s", err.Error()))
+		return
+	}
+	if item.Errmsg != "" {
+		utils.FileLog.Info(fmt.Sprintf("GetWxToken fail result:%s", string(result)))
+	}
+	return
+}

services/wechat/token.go

@@ -3,12 +3,14 @@ package wechat
 import (
 	"encoding/json"
 	"errors"
+	"eta/eta_report/models"
 	"eta/eta_report/services/alarm_msg"
 	"eta/eta_report/utils"
 	"fmt"
 	"io/ioutil"
 	"net/http"
 	"strings"
+	"time"
 )
 
 type WxTokenResp struct {
@@ -87,3 +89,112 @@ func GetAccessToken(tokenReq WxTokenReq) (accessToken string, err error, errMsg
 	}
 	return
 }
+
+// GetWxAccessToken 获取微信token
+func GetWxAccessToken(wxAppId, wxAppSecret string) (accessToken string, err error, errMsg string) {
+	accessToken, err, errMsg = getWxAccessToken(wxAppId, wxAppSecret)
+	if err != nil {
+		utils.FileLog.Info(fmt.Sprintf("获取Token失败,err：%s,errMsg：%s", err.Error(), errMsg))
+		return
+	}
+	return
+}
+
+// getWxAccessToken 获取微信token
+func getWxAccessToken(wxAppId, wxAppSecret string) (accessToken string, err error, errMsg string) {
+	redisKey := getRedisKeyByAppid(wxAppId)
+	if redisKey == `` {
+		errMsg = "未配置缓存key"
+		err = errors.New(errMsg)
+		return
+	}
+
+	accessToken, err = utils.Rc.RedisString(redisKey)
+	//fmt.Println(err)
+	//fmt.Println(accessToken)
+	//if err != nil {
+	//	errMsg = "GetWxAccessToken Err:" + err.Error()
+	//	utils.FileLog.Info("获取Token失败,msg:" + errMsg)
+	//	return
+	//}
+	//取到数据后就直接返回了，没有后续了
+	if accessToken != "" {
+		return
+	}
+
+	//缓存中没有取到数据，那么需要去强制刷新新的accessToken
+	return refreshWxAccessToken(wxAppId, wxAppSecret)
+}
+
+// refreshWxAccessToken 强制刷新微信token
+func refreshWxAccessToken(wxAppId, wxAppSecret string) (accessToken string, err error, errMsg string) {
+	fmt.Println("强制刷新" + wxAppId + "微信token")
+	defer func() {
+		if errMsg != `` {
+			utils.FileLog.Info(fmt.Sprintf("强制刷新%s微信token异常：%s", wxAppId, errMsg))
+		}
+	}()
+
+	if wxAppId == "" || wxAppSecret == "" {
+		errMsg = fmt.Sprintf("必要信息为空, wxAppId: %s, wxAppSecret: %s", wxAppId, wxAppSecret)
+		err = errors.New(errMsg)
+		return
+	}
+
+	redisKey := getRedisKeyByAppid(wxAppId)
+	if redisKey == `` {
+		errMsg = "未配置缓存key"
+		err = errors.New(errMsg)
+		return
+	}
+
+	//调用微信官方接口获取新的accessToken
+	wxAccessToken, tmpErr := models.GetWxToken(wxAppId, wxAppSecret)
+	if tmpErr != nil {
+		err = tmpErr
+		errMsg = "通过微信接口获取accessToken失败 Err:" + err.Error()
+		return
+	}
+
+	//如果没有token数据
+	if wxAccessToken.AccessToken == "" {
+		errMsg = "微信返回的accessToken异常： Err:" + wxAccessToken.Errmsg
+		err = errors.New(errMsg)
+		return
+	}
+
+	accessToken = wxAccessToken.AccessToken
+	////如果是弘则研究的appid，那么需要更新mysql的accessToken
+	//if wxAppId == utils.WxAppId {
+	//	expiresIn := time.Now().Add(time.Duration(wxAccessToken.ExpiresIn) * time.Second).Unix()
+	//	err = models.ModifyAccessToken(wxAccessToken.AccessToken, expiresIn)
+	//	if err != nil {
+	//		errMsg = "更新mysql中的accessToken失败 Err:" + err.Error()
+	//		return
+	//	}
+	//}
+
+	//更新redis的accessToken（过期时间提前十分钟）
+	redisTimeExpire := time.Duration(wxAccessToken.ExpiresIn-600) * time.Second
+	err = utils.Rc.Put(redisKey, accessToken, redisTimeExpire)
+	if err != nil {
+		errMsg = "更新redis中的accessToken失败 Err:" + err.Error()
+		return
+	}
+
+	return
+}
+
+// 根据微信appid获取对应的缓存key
+func getRedisKeyByAppid(wxAppId string) (redisKey string) {
+	redisKey = utils.ETA_WX_ACCESS_TOEKN + wxAppId
+	//switch wxAppId {
+	//case utils.WxAppId:
+	//	redisKey = utils.CACHE_WX_ACCESS_TOKEN_HZ
+	//case utils.AdminWxAppId:
+	//	redisKey = utils.HZ_ADMIN_WX_ACCESS_TOEKN + wxAppId
+	//default:
+	//	redisKey = utils.HZ_ADMIN_WX_ACCESS_TOEKN + wxAppId
+	//}
+	return redisKey
+}

utils/constants.go

@@ -79,3 +79,17 @@ const (
 const (
 	CACHE_CREATE_REPORT_IMGPDF_QUEUE = "eta_report:report_img_pdf_queue" // 生成报告长图PDF队列
 )
+
+// 缓存key
+const (
+	CACHE_WX_ACCESS_TOKEN_HZ   = "wx:accesstoken:hzyj" //弘则研究公众号 微信accessToken
+	HZ_ADMIN_WX_ACCESS_TOEKN   = "hz_admin:wx:access_token:"
+	CACHE_WX_ACCESS_TOKEN_CYGX = "xygxxzs_wxtoken" //查研观向小助手公众号 微信accessToken
+	ETA_WX_ACCESS_TOEKN        = "eta:wx:access_token:"
+)
+
+// 自用商户号
+const (
+	BusinessCodeSandbox = "E2023080700" // 试用平台
+	BusinessCodeRelease = "E2023080900" // 生产环境
+)