123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107 |
- package WorkPhone
- import (
- "context"
- "fmt"
- "github.com/imroc/req/v3"
- "github.com/suyuan32/simple-admin-common/msg/errormsg"
- "strconv"
- "time"
- "wechat-api/ent"
- "wechat-api/ent/wx"
- "wechat-api/internal/utils/dberrorhandler"
- "wechat-api/internal/svc"
- "wechat-api/internal/types"
- "github.com/zeromicro/go-zero/core/logx"
- )
- type QwRefreshLogic struct {
- logx.Logger
- ctx context.Context
- svcCtx *svc.ServiceContext
- }
- func NewQwRefreshLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QwRefreshLogic {
- return &QwRefreshLogic{
- Logger: logx.WithContext(ctx),
- ctx: ctx,
- svcCtx: svcCtx}
- }
- func (l *QwRefreshLogic) QwRefresh() (resp *types.BaseMsgResp, err error) {
- var result types.WeworkGetResp
- client := req.C().DevMode()
- client.SetCommonRetryCount(2).
- SetCommonRetryBackoffInterval(1*time.Second, 5*time.Second).
- SetCommonRetryFixedInterval(2 * time.Second).SetTimeout(30 * time.Second)
- res, err := client.R().SetSuccessResult(&result).Post("http://wecom.gkscrm.com:15086/pc/GetWeChatsReq?id=19")
- if err != nil {
- return nil, err
- }
- if !res.IsSuccessState() {
- err = fmt.Errorf("GetWeChats failed with status code %d", res.StatusCode)
- return nil, err
- }
- for _, account := range result.Data {
- if account.WxId == "" {
- continue
- }
- wxinfo, err := l.svcCtx.DB.Wx.Query().
- Where(
- wx.WxidEQ(account.WxId),
- wx.Ctype(3),
- ).
- Only(l.ctx)
- if err != nil && !ent.IsNotFound(err) {
- return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
- }
- var status uint8
- if account.Isonline == 0 {
- status = 1
- } else {
- status = 0
- }
- if wxinfo != nil {
- err = l.svcCtx.DB.Wx.UpdateOneID(wxinfo.ID).
- SetServerID(0).
- SetPort(account.Deviceid).
- SetProcessID(strconv.FormatInt(account.Cid, 10)).
- SetAccount("").
- SetNickname(account.Name).
- SetHeadBig(account.Avatar).
- SetStatus(status).
- Exec(l.ctx)
- if err != nil {
- return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
- }
- } else {
- l.Debug("wxinfo is nil")
- _, err := l.svcCtx.DB.Wx.Create().
- SetServerID(0).
- SetPort(account.Deviceid).
- SetProcessID(strconv.FormatInt(account.Cid, 10)).
- SetWxid(account.WxId).
- SetAccount("").
- SetHeadBig(account.Avatar).
- SetNickname(account.Name).
- SetStatus(status).
- SetCtype(3).
- SetAllowList([]string{}).SetBlockList([]string{}).SetGroupAllowList([]string{}).SetGroupBlockList([]string{}).
- Save(l.ctx)
- l.Error("wxinfo save", err)
- if err != nil {
- return nil, dberrorhandler.DefaultEntError(l.Logger, err, account)
- }
- }
- }
- l.svcCtx.Rds.Del(l.ctx, "crontask_wx_server_info")
- return &types.BaseMsgResp{Msg: errormsg.Success}, nil
- }
|