qw_refresh_logic.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
  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 QwRefreshLogic struct {
  17. logx.Logger
  18. ctx context.Context
  19. svcCtx *svc.ServiceContext
  20. }
  21. func NewQwRefreshLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QwRefreshLogic {
  22. return &QwRefreshLogic{
  23. Logger: logx.WithContext(ctx),
  24. ctx: ctx,
  25. svcCtx: svcCtx}
  26. }
  27. func (l *QwRefreshLogic) QwRefresh() (resp *types.BaseMsgResp, err error) {
  28. var result types.WeworkGetResp
  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://wecom.gkscrm.com:15086/pc/GetWeChatsReq?id=19")
  34. if err != nil {
  35. return nil, err
  36. }
  37. if !res.IsSuccessState() {
  38. err = fmt.Errorf("GetWeChats failed with status code %d", res.StatusCode)
  39. return nil, err
  40. }
  41. for _, account := range result.Data {
  42. if account.WxId == "" {
  43. continue
  44. }
  45. wxinfo, err := l.svcCtx.DB.Wx.Query().
  46. Where(
  47. wx.WxidEQ(account.WxId),
  48. wx.Ctype(3),
  49. ).
  50. Only(l.ctx)
  51. if err != nil && !ent.IsNotFound(err) {
  52. return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
  53. }
  54. var status uint8
  55. if account.Isonline == 0 {
  56. status = 1
  57. } else {
  58. status = 0
  59. }
  60. if wxinfo != nil {
  61. err = l.svcCtx.DB.Wx.UpdateOneID(wxinfo.ID).
  62. SetServerID(0).
  63. SetPort(account.Deviceid).
  64. SetProcessID(strconv.FormatInt(account.Cid, 10)).
  65. SetAccount("").
  66. SetNickname(account.Name).
  67. SetHeadBig(account.Avatar).
  68. SetStatus(status).
  69. Exec(l.ctx)
  70. if err != nil {
  71. return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
  72. }
  73. } else {
  74. l.Debug("wxinfo is nil")
  75. _, err := l.svcCtx.DB.Wx.Create().
  76. SetServerID(0).
  77. SetPort(account.Deviceid).
  78. SetProcessID(strconv.FormatInt(account.Cid, 10)).
  79. SetWxid(account.WxId).
  80. SetAccount("").
  81. SetHeadBig(account.Avatar).
  82. SetNickname(account.Name).
  83. SetStatus(status).
  84. SetCtype(3).
  85. SetAllowList([]string{}).SetBlockList([]string{}).SetGroupAllowList([]string{}).SetGroupBlockList([]string{}).
  86. Save(l.ctx)
  87. l.Error("wxinfo save", err)
  88. if err != nil {
  89. return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
  90. }
  91. }
  92. }
  93. l.svcCtx.Rds.Del(l.ctx, "crontask_wx_server_info")
  94. return &types.BaseMsgResp{Msg: errormsg.Success}, nil
  95. }