微信小程序 接口请求工具

Author: DeyiXu

net/http/http.go

@@ -0,0 +1,50 @@
+package http
+
+import (
+	"io/ioutil"
+	"net/http"
+	nurl "net/url"
+	"strings"
+)
+
+func Post(url, contentType string, args map[string]string) (result string, err error) {
+	if contentType == "" {
+		contentType = "application/x-www-form-urlencoded"
+	}
+	resp, err := http.Post(url, contentType, strings.NewReader(argsEncode(args)))
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return
+	}
+	result = string(body)
+	return
+}
+
+func Get(url string, args map[string]string) (result string, err error) {
+	if args != nil {
+		url += "?" + argsEncode(args)
+	}
+	resp, err := http.Get(url)
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return
+	}
+	result = string(body)
+	return
+}
+
+func argsEncode(params map[string]string) string {
+	args := nurl.Values{}
+	for k, v := range params {
+		args.Set(k, v)
+	}
+	return args.Encode()
+}

wechat/applet/applet.go

@@ -0,0 +1,32 @@
+package applet
+
+import (
+	"errors"
+
+	"github.com/Undesoft/pkg/net/http"
+)
+
+var (
+	AppId     = ""
+	AppSecret = ""
+)
+
+func Execute(url string, args map[string]string) (result string, err error) {
+	err = checkConfig()
+	if err != nil {
+		return
+	}
+	args["appid"] = AppId
+	args["secret"] = AppSecret
+	return http.Get(url, args)
+}
+
+func checkConfig() error {
+	if AppId == "" {
+		return errors.New("AppKey 不能为空")
+	}
+	if AppSecret == "" {
+		return errors.New("AppSecret 不能为空")
+	}
+	return nil
+}

wechat/applet/applet_test.go

@@ -0,0 +1,25 @@
+package applet
+
+import (
+	"fmt"
+	"testing"
+)
+
+func init() {
+	AppId = "123"
+	AppSecret = "abc"
+}
+
+func TestExecute(t *testing.T) {
+	// 小程序登录
+	url := "https://api.weixin.qq.com/sns/jscode2session"
+	args := map[string]string{
+		"js_code":    "登录时获取的 code",
+		"grant_type": "authorization_code",
+	}
+	result, err := Execute(url, args)
+	if err != nil {
+		t.Fatal(err)
+	}
+	fmt.Println(result)
+}