公众号菜单：上传微信公众号素材永久图片，图库

controllers/mpwechat/material_controller.go

@@ -3,13 +3,16 @@ package mpwechat
 import (
 	base_ctl "SCRM/controllers"
 	"SCRM/enums"
+	"SCRM/models"
 	"SCRM/service/wechat_service"
-	"SCRM/utils"
 	"bytes"
+	"encoding/json"
 	"mime/multipart"
 	"sync"
+	"time"
 
 	"io"
+	"io/ioutil"
 
 	"github.com/astaxie/beego"
 	"github.com/astaxie/beego/httplib"
@@ -17,6 +20,7 @@ import (
 
 func MpMaterialCtlRegistRouters() {
 	beego.Router("/api/mpwechat/media/add_material", &MpMaterialAPIController{}, "Post:AddMaterial")
+	beego.Router("/api/mpwechat/media/material", &MpMaterialAPIController{}, "Get:GetMaterials")
 }
 
 type MpMaterialAPIController struct {
@@ -46,13 +50,13 @@ func (c MpMaterialAPIController) AddMaterial() {
 		return
 	}
 	if authorization == nil || authorization.AuthorizerStatus != 1 {
-		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（公众号未授权）")
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（公众号未授权）")
 		return
 	}
 
 	file, head, err := c.GetFile("file")
 	if err != nil {
-		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（"+err.Error()+"）")
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err.Error()+"）")
 		return
 	}
 	defer file.Close()
@@ -75,28 +79,28 @@ func (c MpMaterialAPIController) AddMaterial() {
 		if field := &fields[i]; field.IsFile {
 			partWriter, err3 := multipartWriter.CreateFormFile(field.Name, field.FileName)
 			if err3 != nil {
-				c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（"+err3.Error()+"）")
+				c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err3.Error()+"）")
 				return
 			}
 			if _, err3 = io.Copy(partWriter, field.Value); err3 != nil {
-				c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（"+err3.Error()+"）")
+				c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err3.Error()+"）")
 				return
 			}
 		} else {
 			partWriter, err3 := multipartWriter.CreateFormField(field.Name)
 			if err3 != nil {
-				c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（"+err3.Error()+"）")
+				c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err3.Error()+"）")
 				return
 			}
 			if _, err3 = io.Copy(partWriter, field.Value); err3 != nil {
-				c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（"+err3.Error()+"）")
+				c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err3.Error()+"）")
 				return
 			}
 		}
 	}
 
 	if err = multipartWriter.Close(); err != nil {
-		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "上传失败:（"+err.Error()+"）")
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err.Error()+"）")
 		return
 	}
 	requestBodyBytes := buffer.Bytes()
@@ -104,13 +108,72 @@ func (c MpMaterialAPIController) AddMaterial() {
 	url := "https://api.weixin.qq.com/cgi-bin/material/add_material?type=image&access_token=" + authorization.AuthorizerAccessToken
 	res := httplib.Post(url)
 	res.Header("Content-Type", requestBodyType)
-	res.Body(requestBodyBytes)
-	response := make(map[string]interface{}, 0)
-	res.ToJSON(&response)
-	utils.InfoLog("%v", response)
+	response, err := res.Body(requestBodyBytes).Response()
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err.Error()+"）")
+		return
+	}
+	respBody, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err.Error()+"）")
+		return
+	}
+
+	var resMaterial wechat_service.ResAddMaterial
+	err = json.Unmarshal(respBody, &resMaterial)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err.Error()+"）")
+		return
+	}
+	if resMaterial.ErrCode != 0 {
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+resMaterial.ErrMsg+"）")
+		return
+	}
+	mediaID := resMaterial.MediaID
+	mediaUrl := resMaterial.URL
+	var media models.WechatMedias
+	media.UserOrgId = adminUserInfo.CurrentOrgId
+	media.MediaId = mediaID
+	media.MediaUrl = mediaUrl
+	media.Msgtype = "image"
+	media.MediaTitle = head.Filename
+	media.CreatedTime = time.Now().Unix()
+	media.UpdatedTime = time.Now().Unix()
+	media.MediaStatus = 1
+
+	err = wechat_service.SaveMaterial(&media)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBCreate, "上传失败:（"+err.Error()+"）")
+		return
+	}
+
+	returnData := make(map[string]interface{}, 0)
+	returnData["media"] = media
+	c.ServeSuccessJSON(returnData)
+	return
+}
+
+func (c *MpMaterialAPIController) GetMaterials() {
+	page, _ := c.GetInt64("page", 1)
+	limit, _ := c.GetInt64("limit", 10)
+	mediaType := c.GetString("media_type")
+	if page <= 0 {
+		page = 1
+	}
+	if limit <= 0 {
+		limit = 10
+	}
+
+	adminUserInfo := c.GetAdminUserInfo()
+	medias, total, err := wechat_service.GetPageMaterials(adminUserInfo.CurrentOrgId, page, limit, mediaType)
+	if err != nil {
+		c.ServeFailJsonSend(enums.ErrorCodeDBUpdate, "拉取失败:（"+err.Error()+"）")
+		return
+	}
 
 	returnData := make(map[string]interface{}, 0)
-	returnData["ok"] = "msg"
+	returnData["medias"] = medias
+	returnData["total"] = total
 	c.ServeSuccessJSON(returnData)
 	return
 }

controllers/mpwechat/menu_controller.go

@@ -35,6 +35,7 @@ func (c *MpMenusAPIController) GetMenus() {
 		buttons = nil
 	}
 
+	mediaIDs := make([]string, 0)
 	var menus wechat_service.SelfMenuInfo
 	if buttons != nil {
 		err = json.Unmarshal([]byte(buttons.ButtonFileds), &menus)
@@ -49,15 +50,50 @@ func (c *MpMenusAPIController) GetMenus() {
 		}
 
 		for index, menu := range menus.Button {
+			if menu.Type == "media_id" && len(menu.MediaId) > 0 {
+				mediaIDs = append(mediaIDs, menu.MediaId)
+			}
 			if _, exist := msgMap[menu.Key]; exist {
 				menus.Button[index].Message = msgMap[menu.Key]
 			}
 			for sindex, sub := range menus.Button[index].SubButton {
+				if sub.Type == "media_id" && len(sub.MediaId) > 0 {
+					mediaIDs = append(mediaIDs, sub.MediaId)
+				}
+
 				if _, exist := msgMap[sub.Key]; exist {
 					menus.Button[index].SubButton[sindex].Message = msgMap[sub.Key]
 				}
 			}
 		}
+		if len(mediaIDs) > 0 {
+			medias, err := wechat_service.FindMeterialsByMediaIDs(adminUserInfo.CurrentOrgId, "image", mediaIDs)
+			if err != nil {
+				c.ServeFailJsonSend(enums.ErrorCodeDataException, "获取菜单信息失败:（"+err.Error()+"）")
+				return
+			}
+			if len(medias) > 0 {
+				mediaMap := make(map[string]string, 0)
+				for _, media := range medias {
+					mediaMap[media.MediaId] = media.MediaUrl
+				}
+				for index, menu := range menus.Button {
+					if menu.Type == "media_id" && len(menu.MediaId) > 0 {
+						if _, exist := mediaMap[menu.MediaId]; exist {
+							menus.Button[index].MediaUrl = mediaMap[menu.MediaId]
+						}
+					}
+					for sindex, sub := range menus.Button[index].SubButton {
+						if sub.Type == "media_id" && len(sub.MediaId) > 0 {
+							if _, exist := mediaMap[sub.MediaId]; exist {
+								menus.Button[index].SubButton[sindex].MediaUrl = mediaMap[sub.MediaId]
+							}
+						}
+					}
+				}
+			}
+
+		}
 
 	}
 

jobcron/open_wechat_job.go

@@ -155,26 +155,26 @@ func RequestMpWechatAccessToken() {
 		var responseBytes []byte
 		responseBytes, err = util.PostJSON(uri, ReqRefreshToken)
 		if err != nil {
-			utils.ErrorLog(" error: %s", err)
+			utils.ErrorLog("error: %s", err)
 			continue
 		}
 
 		var res wechat_service.MPResult
 		err = json.Unmarshal(responseBytes, &res)
 		if err != nil {
-			utils.ErrorLog(" error: %s", err)
+			utils.ErrorLog("%s, error: %s", item.AuthorizerAppid, err)
 			continue
 		}
 
 		if res.ErrCode > 0 {
-			utils.ErrorLog(" error: %s", res.ErrMsg)
+			utils.ErrorLog("%s, error: %s", item.AuthorizerAppid, res.ErrMsg)
 			continue
 		}
 
 		var accessToken wechat_service.RefreshToken
 		err = json.Unmarshal(responseBytes, &accessToken)
 		if err != nil {
-			utils.ErrorLog("Unmarshal error: %s", err)
+			utils.ErrorLog("%s, Unmarshal error: %s", item.AuthorizerAppid, err)
 			continue
 		}
 
@@ -184,14 +184,14 @@ func RequestMpWechatAccessToken() {
 		info.UpdatedTime = time.Now().Unix()
 		err = wechat_service.SaveAuthorizationInfo(&info)
 		if err != nil {
-			utils.ErrorLog("Save error: %s", err)
+			utils.ErrorLog("%s,Save error: %s", item.AuthorizerAppid, err)
 			continue
 		}
 
 		atKey := fmt.Sprintf("Patient:AccessToken[%s]", item.AuthorizerAppid)
 		_, err = redisClient.Set(atKey, accessToken.AuthorizerAccessToken, 0).Result()
 		if err != nil {
-			utils.ErrorLog("Save redis error: %s", err)
+			utils.ErrorLog("%s,Save redis error: %s", item.AuthorizerAppid, err)
 			continue
 		}
 	}

main.go

@@ -18,7 +18,7 @@ func main() {
 
 	//微信开放平台
 	go func() {
-		jobcron.RequestComponentAccessToken()
+		// jobcron.RequestComponentAccessToken()
 		jobcron.RequestMpWechatAccessToken()
 		// jobcron.ReqJsapiTicket()
 	}()

models/wechat_models.go

@@ -72,3 +72,20 @@ type AuthorizationButtons struct {
 func (AuthorizationButtons) TableName() string {
 	return "sgj_patient_authorization_buttons"
 }
+
+type WechatMedias struct {
+	ID               int64  `gorm:"column:id" json:"id" form:"id"`
+	UserOrgId        int64  `gorm:"column:user_org_id" json:"user_org_id" form:"user_org_id"`
+	Msgtype          string `gorm:"column:msgtype" json:"msgtype" form:"msgtype"`
+	MediaId          string `gorm:"column:media_id" json:"media_id" form:"media_id"`
+	MediaTitle       string `gorm:"column:media_title" json:"media_title" form:"media_title"`
+	MediaDescription string `gorm:"column:media_description" json:"media_description" form:"media_description"`
+	MediaUrl         string `gorm:"column:media_url" json:"media_url" form:"media_url"`
+	UpdatedTime      int64  `gorm:"column:updated_time" json:"updated_time" form:"updated_time"`
+	MediaStatus      int64  `gorm:"column:media_status" json:"media_status" form:"media_status"`
+	CreatedTime      int64  `gorm:"column:created_time" json:"created_time" form:"created_time"`
+}
+
+func (WechatMedias) TableName() string {
+	return "sgj_patient_wechat_medias"
+}

service/wechat_service/material_service.go

@@ -0,0 +1,26 @@
+package wechat_service
+
+import (
+	"SCRM/models"
+	"SCRM/service"
+)
+
+func SaveMaterial(m *models.WechatMedias) (err error) {
+	err = service.PatientWriteDB().Save(m).Error
+	return err
+}
+
+func FindMeterialsByMediaIDs(orgID int64, mediaType string, mediaIDs []string) (medias []models.WechatMedias, err error) {
+	err = service.PatientReadDB().Model(&models.WechatMedias{}).Where("media_id IN (?) and user_org_id=? AND msgtype=? and media_status=1", mediaIDs, orgID, mediaType).Limit(len(mediaIDs)).Find(&medias).Error
+	return
+}
+
+func GetPageMaterials(orgID, page, limit int64, mediaType string) (medias []*models.WechatMedias, total int64, err error) {
+	db := service.PatientReadDB().Model(&models.WechatMedias{}).Where("user_org_id=? and msgtype=? and media_status=1", orgID, mediaType)
+
+	offset := (page - 1) * limit
+
+	err = db.Count(&total).Order("id desc").Offset(offset).Limit(limit).Find(&medias).Error
+
+	return
+}

service/wechat_service/open_wechat_model.go

@@ -82,6 +82,14 @@ type MPResult struct {
 	ErrMsg  string `json:"errmsg"`
 }
 
+//ResAddMaterial 永久性素材上传返回的结果
+type ResAddMaterial struct {
+	MPResult
+
+	MediaID string `json:"media_id"`
+	URL     string `json:"url"`
+}
+
 //ComponentAccessToken 第三方平台主动向微信服务器请求第三方平台自己的component_access_token返回的数据结构
 type ComponentAccessToken struct {
 	ComponentAccessToken string `json:"component_access_token"`
@@ -159,6 +167,7 @@ type SelfMenuButton struct {
 	Message   string           `json:"message,omitempty"`
 	URL       string           `json:"url,omitempty"`
 	MediaId   string           `json:"media_id,omitempty"`
+	MediaUrl  string           `json:"media_url,omitempty"`
 	SubButton []SelfMenuButton `json:"sub_button,omitempty"`
 }