opms_backend/opms_libary/plugin/dingtalk/base/base.go

package base

import (
	"bytes"
	"dashoo.cn/opms_libary/plugin/dingtalk/context"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)

const BaseApiUrl = "https://api.dingtalk.com"

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

//HTTPGetWithAccessToken 自动加上access_token变量的GET调用，
//如果失败，会清空AccessToken cache, 再试一次
func (c *Base) HTTPGetWithAccessToken(url string, param map[string]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, "http") {
		target = fmt.Sprintf("%s?access_token=%s", url, accessToken)
	} else {
		if strings.Contains(url, "?") {
			target = fmt.Sprintf("%s%s&x-acs-dingtalk-access-token=%s", BaseApiUrl, url, accessToken)
		} else {
			target = fmt.Sprintf("%s%s?x-acs-dingtalk-access-token=%s", BaseApiUrl, url, accessToken)
		}
	}
	//双重for-range
	for k, v := range param {
		target = fmt.Sprintf("%s&%s=%v", target, k, v)
	}

	var response *http.Response
	response, err = http.Get(target)
	if err != nil {
		return
	}
	defer response.Body.Close()

	resp, err = ioutil.ReadAll(response.Body)

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

//HTTPPostJSONWithAccessToken post json 自动加上access token, 并retry
func (c *Base) 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, "http") {
		target = fmt.Sprintf("%s?access_token=%s", url, accessToken)
	} else {
		if strings.Contains(url, "?") {
			target = fmt.Sprintf("%s%s", BaseApiUrl, url)
		} else {
			target = fmt.Sprintf("%s%s", BaseApiUrl, url)
		}
	}
	resp, err = PostJSON(target, obj, accessToken)

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

//PostJSON post json 数据请求
func PostJSON(url string, obj interface{}, token string) ([]byte, error) {
	jsonData, err := json.Marshal(obj)
	if err != nil {
		return nil, err
	}

	jsonData = bytes.Replace(jsonData, []byte("\\u003c"), []byte("<"), -1)
	jsonData = bytes.Replace(jsonData, []byte("\\u003e"), []byte(">"), -1)
	jsonData = bytes.Replace(jsonData, []byte("\\u0026"), []byte("&"), -1)

	body := bytes.NewBuffer(jsonData)

	req, _ := http.NewRequest("POST", url, body)
	// 比如说设置个token
	req.Header.Set("x-acs-dingtalk-access-token", token)
	// 再设置个json
	req.Header.Set("Content-Type", "application/json")

	response, err := (&http.Client{}).Do(req)
	if err != nil {
		return nil, err
	}
	defer response.Body.Close()

	if response.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", url, response.StatusCode)
	}
	return ioutil.ReadAll(response.Body)
}