wechat-api/internal/logic/WorkPhone/wx_refresh_logic.go

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 WxRefreshLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewWxRefreshLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WxRefreshLogic {
	return &WxRefreshLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx}
}

type GetWeChatsResp struct {
	Code int64 `json:"code"`
	Data []struct {
		Accountid   string      `json:"accountid"`
		Avatar      string      `json:"avatar"`
		Brand       string      `json:"brand"`
		Cid         int64       `json:"cid"`
		City        string      `json:"city"`
		Cname       interface{} `json:"cname"`
		Country     string      `json:"country"`
		Deviceid    string      `json:"deviceid"`
		Devnickname string      `json:"devnickname"`
		Gender      int64       `json:"gender"`
		Groupid     interface{} `json:"groupid"`
		ID          int64       `json:"id"`
		Isonline    int64       `json:"isonline"`
		LoginTime   int64       `json:"login_time"`
		ModifyTime  int64       `json:"modify_time"`
		Module      string      `json:"module"`
		Province    string      `json:"province"`
		Snumber     int64       `json:"snumber"`
		Wechatid    string      `json:"wechatid"`
		Wechatnick  string      `json:"wechatnick"`
		Wechatno    string      `json:"wechatno"`
	} `json:"data"`
	Msg string `json:"msg"`
}

func (l *WxRefreshLogic) WxRefresh() (resp *types.BaseMsgResp, err error) {
	var result GetWeChatsResp
	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://chat.gkscrm.com:13086/pc/GetWeChatsReq?id=13")

	if err != nil {
		return
	}
	if !res.IsSuccessState() {
		err = fmt.Errorf("GetWeChats failed with status code %d", res.StatusCode)
		return
	}

	for _, account := range result.Data {
		wxinfo, err := l.svcCtx.DB.Wx.Query().
			Where(
				wx.WxidEQ(account.Wechatid),
			).
			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(account.Wechatno).
				SetNickname(account.Wechatnick).
				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.Wechatid).
				SetAccount(account.Wechatno).
				SetHeadBig(account.Avatar).
				SetNickname(account.Wechatnick).
				SetStatus(status).
				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)
			}
		}
	}

	return &types.BaseMsgResp{Msg: errormsg.Success}, nil

	return
}