update wechatminiprogram

service/wechatminiprogram/save_img.go

@@ -0,0 +1,31 @@
+package wechatminiprogram
+
+import (
+	"gopkg.in/dtapps/go-library.v3/utils/gohttp"
+	"os"
+)
+
+func (app *App) SaveImg(resp gohttp.Response, dir, saveName string) string {
+	// 返回是二进制图片，或者json错误
+	if resp.Header.Get("Content-Type") == "image/jpeg" || resp.Header.Get("Content-Type") == "image/png" {
+		// 保存在output目录
+		outputFileName := saveName
+
+		if resp.Header.Get("Content-Type") == "image/jpeg" {
+			outputFileName = outputFileName + ".jpg"
+		} else {
+			outputFileName = outputFileName + ".png"
+		}
+	here:
+		f, err := os.OpenFile(dir+outputFileName, os.O_CREATE|os.O_RDWR, 0666)
+		if err != nil {
+			os.Mkdir(dir, 0666)
+			goto here
+		}
+		f.Write(resp.Body)
+		f.Close()
+		return dir + outputFileName
+	} else {
+		return ""
+	}
+}

utils/gohttp/gohttp.go

@@ -16,7 +16,7 @@ import (
 type Response struct {
 	Status        string
 	StatusCode    int
-	Header        map[string][]string
+	Header        http.Header
 	Body          []byte
 	ContentLength int64
 }