wx_refresh_logic.go 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  1. package WorkPhone
  2. import (
  3. "context"
  4. "fmt"
  5. "github.com/imroc/req/v3"
  6. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  7. "strconv"
  8. "time"
  9. "wechat-api/ent"
  10. "wechat-api/ent/wx"
  11. "wechat-api/internal/utils/dberrorhandler"
  12. "wechat-api/internal/svc"
  13. "wechat-api/internal/types"
  14. "github.com/zeromicro/go-zero/core/logx"
  15. )
  16. type WxRefreshLogic struct {
  17. logx.Logger
  18. ctx context.Context
  19. svcCtx *svc.ServiceContext
  20. }
  21. func NewWxRefreshLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WxRefreshLogic {
  22. return &WxRefreshLogic{
  23. Logger: logx.WithContext(ctx),
  24. ctx: ctx,
  25. svcCtx: svcCtx}
  26. }
  27. func (l *WxRefreshLogic) WxRefresh() (resp *types.BaseMsgResp, err error) {
  28. var result types.WorkPhoneGetWeChatsResp
  29. client := req.C().DevMode()
  30. client.SetCommonRetryCount(2).
  31. SetCommonRetryBackoffInterval(1*time.Second, 5*time.Second).
  32. SetCommonRetryFixedInterval(2 * time.Second).SetTimeout(30 * time.Second)
  33. res, err := client.R().SetSuccessResult(&result).Post("http://chat.gkscrm.com:13086/pc/GetWeChatsReq?id=13")
  34. if err != nil {
  35. return
  36. }
  37. if !res.IsSuccessState() {
  38. err = fmt.Errorf("GetWeChats failed with status code %d", res.StatusCode)
  39. return
  40. }
  41. for _, account := range result.Data {
  42. if account.Wechatid == "" {
  43. continue
  44. }
  45. wxinfo, err := l.svcCtx.DB.Wx.Query().
  46. Where(
  47. wx.And(
  48. wx.Or(
  49. wx.WxidEQ(account.Wechatid),
  50. wx.PortEQ(account.Deviceid),
  51. ),
  52. wx.CtypeEQ(1),
  53. ),
  54. ).
  55. Only(l.ctx)
  56. if err != nil && !ent.IsNotFound(err) {
  57. return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
  58. }
  59. var status uint8
  60. if account.Isonline == 0 {
  61. status = 1
  62. } else {
  63. status = 0
  64. }
  65. if wxinfo != nil {
  66. err = l.svcCtx.DB.Wx.UpdateOneID(wxinfo.ID).
  67. SetServerID(0).
  68. SetPort(account.Deviceid).
  69. SetProcessID(strconv.FormatInt(account.Cid, 10)).
  70. SetAccount(account.Wechatno).
  71. SetNickname(account.Wechatnick).
  72. SetHeadBig(account.Avatar).
  73. SetStatus(status).
  74. Exec(l.ctx)
  75. if err != nil {
  76. return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
  77. }
  78. } else {
  79. l.Debug("wxinfo is nil")
  80. _, err := l.svcCtx.DB.Wx.Create().
  81. SetServerID(0).
  82. SetPort(account.Deviceid).
  83. SetProcessID(strconv.FormatInt(account.Cid, 10)).
  84. SetWxid(account.Wechatid).
  85. SetAccount(account.Wechatno).
  86. SetHeadBig(account.Avatar).
  87. SetNickname(account.Wechatnick).
  88. SetStatus(status).
  89. SetAllowList([]string{}).SetBlockList([]string{}).SetGroupAllowList([]string{}).SetGroupBlockList([]string{}).
  90. Save(l.ctx)
  91. l.Error("wxinfo save", err)
  92. if err != nil {
  93. return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
  94. }
  95. }
  96. }
  97. l.svcCtx.Rds.Del(l.ctx, "crontask_wx_server_info")
  98. return &types.BaseMsgResp{Msg: errormsg.Success}, nil
  99. return
  100. }