123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. package models
  2. import (
  3. "time"
  4. "wechat-conf/models/model"
  5. "wechat-conf/utils"
  6. "github.com/astaxie/beego"
  7. )
  8. // ComponentInit 初始化第三方
  9. func ComponentInit() {
  10. var conf = model.SysComponentConf{}
  11. _, err := DBEngine.Get(&conf)
  12. if err != nil {
  13. utils.LogError("初始化第三方失败:", err)
  14. return
  15. }
  16. utils.LogError("获取第三方信息:", conf)
  17. if conf.Appid == "" {
  18. utils.LogError("初始化第三方数据失败,数据为空")
  19. return
  20. }
  21. d := "0"
  22. if !conf.ExpiresDate.IsZero() {
  23. now := time.Now()
  24. subd := conf.ExpiresDate.Sub(now)
  25. beego.Error(d)
  26. d = string(subd)
  27. }
  28. var cert = map[string]string{
  29. "appid": conf.Appid,
  30. "aeskey": conf.Aeskey,
  31. "secret": conf.Secret,
  32. "componentVerifyTicket": conf.Ticket,
  33. "token": conf.Token,
  34. "expires_in": d,
  35. }
  36. utils.ComponentInit(cert, UpdateComponentToken)
  37. }
  38. // UpdateComponentToken 更新第三方token
  39. func UpdateComponentToken(token map[string]interface{}) {
  40. beego.Error("更新第三方token:", token)
  41. if token["accessToken"].(string) == "" {
  42. return
  43. }
  44. var conf = model.SysComponentConf{
  45. Token: token["accessToken"].(string),
  46. }
  47. expire := token["expire_in"].(int64)
  48. if expire > 0 {
  49. expriesTime := time.Now().Local().Add(time.Duration(expire) * time.Second)
  50. conf.ExpiresDate = expriesTime
  51. sql := `update sys_component_conf set token=?,expires_date=?`
  52. _, err := DBEngine.Exec(sql, conf.Token, conf.ExpiresDate)
  53. if err != nil {
  54. utils.LogError("更新第三方token失败:", err)
  55. }
  56. } else {
  57. sql := `update sys_component_conf set token=?`
  58. _, err := DBEngine.Exec(sql, conf.Token, conf.ExpiresDate)
  59. if err != nil {
  60. utils.LogError("更新第三方token失败:", err)
  61. }
  62. }
  63. }
  64. // InitWechat 初始化微信
  65. func InitWechat() {
  66. var wxconfs []model.SysWechatConf
  67. err := DBEngine.Where("status>?", STATUS_DEL).Find(&wxconfs)
  68. if err != nil {
  69. utils.LogError("初始化微信失败:", err)
  70. return
  71. }
  72. for _, conf := range wxconfs {
  73. var cert = map[string]string{
  74. "authorizer_access_token": conf.Token,
  75. "authorizer_refresh_token": conf.RefreshToken,
  76. "authorization_code": conf.AuthorizationCode,
  77. "appid": conf.Appid,
  78. }
  79. client := utils.WechatInit(cert, UpdateToken)
  80. beego.Error(client)
  81. utils.AppendWxClient(client)
  82. }
  83. }
  84. // UpdateToken 更新微信token
  85. func UpdateToken(token map[string]interface{}) {
  86. var conf = model.SysWechatConf{
  87. Token: token["token"].(string),
  88. RefreshToken: token["refreshToken"].(string),
  89. Appid: token["appid"].(string),
  90. }
  91. expire := token["expire_in"].(int64)
  92. if expire > 0 {
  93. beego.Error(time.Now())
  94. expriesTime := time.Now().Local().Add(time.Duration(expire) * time.Second)
  95. beego.Error(expriesTime)
  96. conf.ExpiresDate = expriesTime
  97. sql := `update sys_wechat_conf set token=?,refresh_token=?,expires_date=? where appid=?`
  98. _, err := DBEngine.Exec(sql, conf.Token, conf.RefreshToken, expriesTime, conf.Appid)
  99. if err != nil {
  100. utils.LogError("更新微信token失败:", err)
  101. }
  102. } else {
  103. sql := `update sys_wechat_conf set token=?,refresh_token=? where appid=?`
  104. _, err := DBEngine.Exec(sql, conf.Token, conf.RefreshToken, conf.Appid)
  105. if err != nil {
  106. utils.LogError("更新微信token失败:", err)
  107. }
  108. }
  109. }