1113PM

models/wechat.go

@@ -20,7 +20,21 @@ func InitWechat() {
 			"authorization_code":       conf.AuthorizationCode,
 			"appid":                    conf.Appid,
 		}
-		client := utils.WechatInit(cert)
+		client := utils.WechatInit(cert, UpdateToken)
 		utils.AppendWxClient(client)
 	}
 }
+
+// UpdateToken 更新微信token
+func UpdateToken(token map[string]string) {
+	var conf = model.SysWechatConf{
+		Token:        token["token"],
+		RefreshToken: token["refreshToken"],
+		Appid:        token["appid"],
+	}
+	sql := `update sys_wechat_conf set token=?,refresh_token=? where appid=?`
+	_, err := DBEngine.Exec(sql, conf.Token, conf.RefreshToken, conf.Appid)
+	if err != nil {
+		utils.LogError("更新微信token失败：", err)
+	}
+}

models/wechat/wechat.go

@@ -29,8 +29,8 @@ func (m *WechatDAO) GetWechatConfByAppID(appid string) (*model.SysWechatConf, er
 	return &conf, err
 }
 
-// SaveWechatConf 保存微信配置信息
-func (m *WechatDAO) SaveWechatConf(conf *model.SysWechatConf) error {
+// AddWechatConf 保存微信配置信息
+func (m *WechatDAO) AddWechatConf(conf *model.SysWechatConf) error {
 	conf.ConfId = utils.GetGUID()
 	conf.Status = models.STATUS_READY
 	_, err := m.db.Insert(conf)
@@ -42,3 +42,36 @@ func (m *WechatDAO) UpdateWechatConf(conf model.SysWechatConf, cols []string) er
 	_, err := m.db.Cols(cols...).Where("conf_id = ?", conf.ConfId).Update(conf)
 	return err
 }
+
+// SaveWechatConf 保存微信配置
+func (m *WechatDAO) SaveWechatConf(conf model.SysWechatConf) error {
+	wechatConf, err := m.GetWechatConfByAppID(conf.Appid)
+	if err != nil {
+		utils.LogError("根据appid获取微信配置失败：", err)
+		return err
+	}
+	if wechatConf != nil && wechatConf.ConfId != "" {
+		// 修改
+		conf.ConfId = wechatConf.ConfId
+		var cols = []string{
+			"refresh_token",
+			"token",
+		}
+		if conf.AuthorizationInfo != "" {
+			cols = append(cols, "authorization_info")
+		}
+		err := m.UpdateWechatConf(conf, cols)
+		if err != nil {
+			utils.LogError("修改微信配置失败：", err)
+			return err
+		}
+	} else {
+		// 新增
+		err := m.AddWechatConf(&conf)
+		if err != nil {
+			utils.LogError("新增微信配置失败：", err)
+			return err
+		}
+	}
+	return nil
+}

service/wechat/wechat.go

@@ -1,6 +1,7 @@
 package wechat
 
 import (
+	"wechat-conf/models"
 	"wechat-conf/models/model"
 	"wechat-conf/models/wechat"
 	"wechat-conf/utils"
@@ -22,46 +23,11 @@ func NewWechatServ(ctx *utils.Context) *WechatServ {
 
 // SaveWechatConf 保存微信配置
 func (s *WechatServ) SaveWechatConf(conf model.SysWechatConf) error {
-	wechatConf, err := s.dao.GetWechatConfByAppID(conf.Appid)
-	if err != nil {
-		utils.LogError("根据appid获取微信配置失败：", err)
-		return err
-	}
-	if wechatConf != nil && wechatConf.ConfId != "" {
-		// 修改
-		conf.ConfId = wechatConf.ConfId
-		var cols = []string{
-			"refresh_token",
-			"token",
-		}
-		if conf.AuthorizationInfo != "" {
-			cols = append(cols, "authorization_info")
-		}
-		err := s.dao.UpdateWechatConf(conf, cols)
-		if err != nil {
-			utils.LogError("修改微信配置失败：", err)
-			return err
-		}
-	} else {
-		// 新增
-		err := s.dao.SaveWechatConf(&conf)
-		if err != nil {
-			utils.LogError("新增微信配置失败：", err)
-			return err
-		}
-	}
-	return nil
+	err := s.dao.SaveWechatConf(conf)
+	return err
 }
 
 // UpdateToken 更新微信token
 func (s *WechatServ) UpdateToken(token map[string]string) {
-	var conf = model.SysWechatConf{
-		Token:        token["token"],
-		RefreshToken: token["refreshToken"],
-		Appid:        token["appid"],
-	}
-	err := s.SaveWechatConf(conf)
-	if err != nil {
-		utils.LogError("更新微信token失败：", err)
-	}
+	models.UpdateToken(token)
 }