menu img

controllers/wechatmenu/wechatmenu.go

@@ -7,6 +7,7 @@ import (
 	"wechat-conf/models/model"
 	"wechat-conf/service/keyvalue"
 	"wechat-conf/service/sysuser"
+	"wechat-conf/service/wechatimg"
 	"wechat-conf/service/wechatmenu"
 	"wechat-conf/utils"
 
@@ -18,6 +19,7 @@ type MenuController struct {
 	dao  *wechatmenu.WechatmenuServ
 	sdao *sysuser.SysuserServ
 	kdao *keyvalue.KeyvalueServ
+	idao *wechatimg.WechatImgServ
 	controllers.BaseController
 }
 
@@ -28,6 +30,7 @@ func (c *MenuController) Constructor() {
 	c.dao = wechatmenu.NewWechatmenuServ(c.Context)
 	c.sdao = sysuser.NewSysuserServ(c.Context)
 	c.kdao = keyvalue.NewKeyvalueServ(c.Context)
+	c.idao = wechatimg.NewWechatImgServ(c.Context)
 }
 
 type Wehchatmenu struct {
@@ -70,6 +73,7 @@ func (c *MenuController) GetMenuList() {
 	menu, err := client.GetMenu()
 	beego.Error(menu)
 	var keyList []*model.TaWechatKeyValue
+	var imgList []*model.TaWechatImg
 	if menu != nil {
 		menus := menu["menu"].(map[string]interface{})
 		wechatmenu := menus["button"].([]interface{})
@@ -83,7 +87,15 @@ func (c *MenuController) GetMenuList() {
 					}
 					keyList = append(keyList, key)
 				}
+				if m["type"].(string) == "media_id" {
+					img, err := c.idao.GetWechatImgById(m["media_id"].(string))
+					if err != nil {
+						c.ResponseError(err)
+					}
+					imgList = append(imgList, img)
+				}
 			}
+
 			if m["sub_button"] != nil {
 				subbuttons := m["sub_button"].([]interface{})
 				for j := 0; j < len(subbuttons); j++ {
@@ -95,6 +107,13 @@ func (c *MenuController) GetMenuList() {
 						}
 						keyList = append(keyList, key)
 					}
+					if subb["type"].(string) == "media_id" {
+						img, err := c.idao.GetWechatImgById(subb["media_id"].(string))
+						if err != nil {
+							c.ResponseError(err)
+						}
+						imgList = append(imgList, img)
+					}
 				}
 			}
 		}
@@ -111,6 +130,7 @@ func (c *MenuController) GetMenuList() {
 	c.ResponseJSON(map[string]interface{}{
 		"menu":    menu,
 		"keyList": keyList,
+		"imgList": imgList,
 	})
 }
 

service/wechatimg/wechatimg.go

@@ -95,3 +95,11 @@ func (s *WechatImgServ) GetWechatImgList(orgId string, page, pageSize int) (map[
 		"page":     page,
 	}, nil
 }
+func (s *WechatImgServ) GetWechatImgById(mediaId string) (*model.TaWechatImg, error) {
+	imgList, err := s.dao.GetWechatImgByMediaId(mediaId)
+	if err != nil {
+		utils.LogError("获取菜单图片失败: " + err.Error())
+		return nil, errors.New("获取菜单图片失败")
+	}
+	return imgList, nil
+}