opms_backend/opms_libary/plugin/wechat/base/base.go

package base

import (
	"dashoo.cn/opms_libary/plugin/wechat/context"
	"fmt"
	"github.com/gogf/gf/frame/g"
	"io/ioutil"
	"strings"
)

//WechatBase 微信公众平台,基本类
type WechatBase struct {
	*context.Context
}

//HTTPGetWithAccessToken 微信公众平台中，自动加上access_token变量的GET调用，
//如果失败，会清空AccessToken cache, 再试一次
func (c *WechatBase) HTTPGetWithAccessToken(url string) (resp []byte, err error) {
	retry := 1
Do:
	var accessToken string
	accessToken, err = c.GetAccessToken()
	if err != nil {
		return
	}

	var target = ""
	if strings.Contains(url, "?") {
		target = fmt.Sprintf("%s&access_token=%s", url, accessToken)
	} else {
		target = fmt.Sprintf("%s?access_token=%s", url, accessToken)
	}
	response, err := g.Client().Get(target)
	if err != nil {
		return
	}
	defer response.Body.Close()

	resp, err = ioutil.ReadAll(response.Body)
	err = context.CheckCommonError(resp)
	if err == context.ErrUnmarshall {
		return
	}
	if err != nil {
		if retry > 0 {
			retry--
			c.CleanAccessTokenCache()
			goto Do
		}
		return
	}
	return
}

//HTTPPostJSONWithAccessToken post json 自动加上access token, 并retry
func (c *WechatBase) HTTPPostJSONWithAccessToken(url string, obj interface{}) (resp []byte, err error) {
	retry := 1
Do:
	var accessToken string
	accessToken, err = c.GetAccessToken()
	if err != nil {
		return
	}

	var target = ""
	if strings.Contains(url, "?") {
		target = fmt.Sprintf("%s&access_token=%s", url, accessToken)
	} else {
		target = fmt.Sprintf("%s?access_token=%s", url, accessToken)
	}

	resp = g.Client().ContentJson().PostBytes(target, obj)

	err = context.CheckCommonError(resp)
	if err == context.ErrUnmarshall {
		return
	}
	if err != nil {
		if retry > 0 {
			retry--
			c.CleanAccessTokenCache()
			goto Do
		}
		return
	}
	return
}