1115PM

controllers/component.go

@@ -46,7 +46,7 @@ func (c *WechatController) ComponentPush() {
 	r := c.Ctx.Request
 	defer r.Body.Close()
 	con, _ := ioutil.ReadAll(r.Body)
-	beego.Error(string(con))
+	utils.LogError("推送消息：", string(con))
 	// 解析xml
 	xp := &core.XMLParse{}
 	resMsg, err := xp.Parse(string(con))
@@ -54,15 +54,12 @@ func (c *WechatController) ComponentPush() {
 		utils.LogError("xml解析失败：", err)
 		c.ResponseRaw([]byte("success"))
 	}
-	utils.LogError("解析xml成功:", resMsg)
 	encrypt := resMsg["Encrypt"]
-	beego.Error(encrypt)
 	conf, err := c.wechatServ.GetComponentInfo()
 	if err != nil || conf == nil || conf.Appid == "" {
 		utils.LogError("读取微信配置文件失败")
 		c.ResponseRaw([]byte("success"))
 	}
-	beego.Error(conf.Aeskey)
 	EncodingAESKey := conf.Aeskey
 
 	AESKey, err := base64.StdEncoding.DecodeString(EncodingAESKey + "=")
@@ -70,10 +67,9 @@ func (c *WechatController) ComponentPush() {
 		utils.LogError("DecodeString失败：", err)
 		c.ResponseRaw([]byte("success"))
 	}
-	beego.Error(AESKey)
 	EncryptVal, err := base64.StdEncoding.DecodeString(encrypt)
 	if err != nil {
-		utils.LogError("密文解析失败：", err)
+		utils.LogError("密文base64解析", err)
 		c.ResponseRaw([]byte("success"))
 	}
 	msgbyte, err := utils.AesDecrypt(EncryptVal, AESKey)
@@ -81,15 +77,14 @@ func (c *WechatController) ComponentPush() {
 		utils.LogError("解密失败：", err)
 		c.ResponseRaw([]byte("success"))
 	}
-	beego.Error("解密数据：", string(msgbyte))
-	beego.Error("解密成功")
+	utils.LogError("解密成功")
 	// 解析xml
 	msg, err := xp.Parse(string(msgbyte))
 	if err != nil {
 		utils.LogError("xml解析失败：", err)
 		c.ResponseRaw([]byte("success"))
 	}
-	beego.Error("解析xml成功:", msg)
+	utils.LogError("xml解析成功：", msg)
 	switch msg["InfoType"] {
 	case INFOTYPE_TICKET:
 		// 更新ticket
@@ -106,7 +101,13 @@ func (c *WechatController) ComponentPush() {
 			"appid":              msg["AuthorizerAppid"],
 			"authorization_code": msg["AuthorizationCode"],
 		}
+		beego.Error("授权成功参数：", cert)
 		wxclient := utils.WechatInit(cert, c.wechatServ.UpdateToken)
+		if wxclient == nil {
+			utils.LogError("获取wxclient失败")
+			c.ResponseRaw([]byte("success"))
+		}
+		beego.Error(wxclient)
 		utils.AppendWxClient(wxclient)
 		// 插入数据库
 		mjson, _ := json.Marshal(msg)
@@ -115,6 +116,7 @@ func (c *WechatController) ComponentPush() {
 			AuthorizationCode: msg["AuthorizationCode"],
 			AuthorizationInfo: string(mjson),
 		}
+
 		err := c.wechatServ.SaveWechatConf(conf)
 		if err != nil {
 			utils.LogError("保存微信授权信息失败：", err)

models/wechat.go

@@ -95,15 +95,28 @@ func InitWechat() {
 }
 
 // UpdateToken 更新微信token
-func UpdateToken(token map[string]string) {
+func UpdateToken(token map[string]interface{}) {
 	var conf = model.SysWechatConf{
-		Token:        token["token"],
-		RefreshToken: token["refreshToken"],
-		Appid:        token["appid"],
+		Token:        token["token"].(string),
+		RefreshToken: token["refreshToken"].(string),
+		Appid:        token["appid"].(string),
 	}
-	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)
+	expire := token["expire_in"].(int64)
+	if expire > 0 {
+		beego.Error(time.Now())
+		expriesTime := time.Now().Local().Add(time.Duration(expire) * time.Second)
+		beego.Error(expriesTime)
+		conf.ExpiresDate = expriesTime
+		sql := `update sys_wechat_conf set token=?,refresh_token=?,expires_date=? where appid=?`
+		_, err := DBEngine.Exec(sql, conf.Token, conf.RefreshToken, expriesTime, conf.Appid)
+		if err != nil {
+			utils.LogError("更新微信token失败：", err)
+		}
+	} else {
+		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

@@ -5,6 +5,8 @@ import (
 	"wechat-conf/models/model"
 	"wechat-conf/utils"
 
+	"github.com/astaxie/beego"
+
 	"github.com/go-xorm/xorm"
 )
 
@@ -30,7 +32,7 @@ func (m *WechatDAO) GetWechatConfByAppID(appid string) (*model.SysWechatConf, er
 }
 
 // AddWechatConf 保存微信配置信息
-func (m *WechatDAO) AddWechatConf(conf *model.SysWechatConf) error {
+func (m *WechatDAO) AddWechatConf(conf model.SysWechatConf) error {
 	conf.ConfId = utils.GetGUID()
 	conf.Status = models.STATUS_READY
 	_, err := m.db.Insert(conf)
@@ -67,7 +69,8 @@ func (m *WechatDAO) SaveWechatConf(conf model.SysWechatConf) error {
 		}
 	} else {
 		// 新增
-		err := m.AddWechatConf(&conf)
+		beego.Error("开始新增：", conf)
+		err := m.AddWechatConf(conf)
 		if err != nil {
 			utils.LogError("新增微信配置失败：", err)
 			return err

service/wechat/wechat.go

@@ -28,7 +28,7 @@ func (s *WechatServ) SaveWechatConf(conf model.SysWechatConf) error {
 }
 
 // UpdateToken 更新微信token
-func (s *WechatServ) UpdateToken(token map[string]string) {
+func (s *WechatServ) UpdateToken(token map[string]interface{}) {
 	models.UpdateToken(token)
 }
 

utils/wechat.go

@@ -19,10 +19,8 @@ func RefreshComponentTicket(ticket string) {
 }
 
 // WechatInit 微信初始化
-func WechatInit(cert map[string]string, reflashToken func(map[string]string)) *component.WxClient {
-	var wechatClient *component.WxClient
-
-	wechatClient = component.NewWxClient(
+func WechatInit(cert map[string]string, reflashToken func(map[string]interface{})) *component.WxClient {
+	wechatClient := component.NewWxClient(
 		cert,
 		Component.GetToken,
 		Component.GetCertificate,