新增同步好友列表任务

This commit is contained in:
李寻欢 2023-09-22 08:39:09 +08:00
parent 40e4dc48c6
commit 3e8051906f
3 changed files with 107 additions and 15 deletions

View File

@ -4,17 +4,27 @@ package entity
// @description: 好友列表 // @description: 好友列表
type Friend struct { type Friend struct {
CustomAccount string `json:"customAccount"` // 微信号 CustomAccount string `json:"customAccount"` // 微信号
EncryptName string `json:"encryptName"` // 不知道
Nickname string `json:"nickname"` // 昵称 Nickname string `json:"nickname"` // 昵称
Pinyin string `json:"pinyin"` // 昵称拼音大写首字母 Pinyin string `json:"pinyin"` // 昵称拼音大写首字母
PinyinAll string `json:"pinyinAll"` // 昵称全拼 PinyinAll string `json:"pinyinAll"` // 昵称全拼
Reserved1 int `json:"reserved1"` // 未知
Reserved2 int `json:"reserved2"` // 未知
Type int `json:"type"` // 类型
VerifyFlag int `json:"verifyFlag"` // 未知
Wxid string `json:"wxid"` // 微信原始Id Wxid string `json:"wxid"` // 微信原始Id
} }
func (Friend) TableName() string { func (Friend) TableName() string {
return "t_friend" return "t_friend"
} }
// GroupUser
// @description: 群成员
type GroupUser struct {
GroupId string `json:"groupId"` // 群Id
Account string `json:"account"` // 账号
HeadImage string `json:"headImage"` // 头像
Nickname string `json:"nickname"` // 昵称
Wxid string `json:"wxid"` // 微信Id
IsMember bool `json:"isMember" gorm:"type:tinyint(1)"` // 是否群成员
}
func (GroupUser) TableName() string {
return "t_group_user"
}

View File

@ -3,9 +3,11 @@ package tasks
import ( import (
"encoding/json" "encoding/json"
"github.com/go-resty/resty/v2" "github.com/go-resty/resty/v2"
"go-wechat/client"
"go-wechat/constant" "go-wechat/constant"
"go-wechat/entity" "go-wechat/entity"
"go-wechat/model" "go-wechat/model"
"gorm.io/gorm"
"log" "log"
"slices" "slices"
"strings" "strings"
@ -16,10 +18,10 @@ import (
// syncFriends // syncFriends
// @description: 同步好友列表 // @description: 同步好友列表
func syncFriends() { func syncFriends() {
var base model.Response[[]entity.Friend] var base model.Response[[]model.FriendItem]
client := resty.New() res := resty.New()
resp, err := client.R(). resp, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8"). SetHeader("Content-Type", "application/json;chartset=utf-8").
SetResult(&base). SetResult(&base).
Post("http://10.0.0.73:19088/api/getContactList") Post("http://10.0.0.73:19088/api/getContactList")
@ -28,6 +30,10 @@ func syncFriends() {
return return
} }
log.Printf("获取好友列表结果: %s", resp.String()) log.Printf("获取好友列表结果: %s", resp.String())
tx := client.MySQL.Begin()
defer tx.Commit()
for _, friend := range base.Data { for _, friend := range base.Data {
if strings.Contains(friend.Wxid, "gh_") || strings.Contains(friend.Wxid, "@openim") { if strings.Contains(friend.Wxid, "gh_") || strings.Contains(friend.Wxid, "@openim") {
continue continue
@ -38,18 +44,52 @@ func syncFriends() {
} }
log.Printf("昵称: %s -> 类型: %d -> 微信号: %s -> 微信原始Id: %s", friend.Nickname, friend.Type, friend.CustomAccount, friend.Wxid) log.Printf("昵称: %s -> 类型: %d -> 微信号: %s -> 微信原始Id: %s", friend.Nickname, friend.Type, friend.CustomAccount, friend.Wxid)
// 群成员,同步一下成员信息 // 判断是否存在,不存在的话就新增,存在就修改一下名字
if strings.Contains(friend.Wxid, "@chatroom") { var count int64
syncGroupUsers(friend.Wxid) err = tx.Model(&entity.Friend{}).Where("wxid = ?", friend.Wxid).Count(&count).Error
if err != nil {
continue
}
if count == 0 {
// 新增
err = tx.Create(&entity.Friend{
CustomAccount: friend.CustomAccount,
Nickname: friend.Nickname,
Pinyin: friend.Pinyin,
PinyinAll: friend.PinyinAll,
Wxid: friend.Wxid,
}).Error
if err != nil {
log.Printf("新增好友失败: %s", err.Error())
continue
}
} else {
pm := map[string]any{
"nickname": friend.Nickname,
"custom_account": friend.CustomAccount,
"pinyin": friend.Pinyin,
"pinyin_all": friend.PinyinAll,
}
err = tx.Model(&entity.Friend{}).Where("wxid = ?", friend.Wxid).Updates(pm).Error
if err != nil {
log.Printf("修改好友失败: %s", err.Error())
continue
}
} }
// 群成员,同步一下成员信息
if strings.Contains(friend.Wxid, "@chatroom") {
syncGroupUsers(tx, friend.Wxid)
} }
} }
log.Println("同步好友列表完成")
}
// syncGroupUsers // syncGroupUsers
// @description: 同步群成员 // @description: 同步群成员
// @param gid // @param gid
func syncGroupUsers(gid string) { func syncGroupUsers(tx *gorm.DB, gid string) {
var baseResp model.Response[model.GroupUser] var baseResp model.Response[model.GroupUser]
// 组装参数 // 组装参数
@ -58,8 +98,8 @@ func syncGroupUsers(gid string) {
} }
pbs, _ := json.Marshal(param) pbs, _ := json.Marshal(param)
client := resty.New() res := resty.New()
_, err := client.R(). _, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8"). SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)). SetBody(string(pbs)).
SetResult(&baseResp). SetResult(&baseResp).
@ -71,13 +111,54 @@ func syncGroupUsers(gid string) {
// 昵称Id // 昵称Id
wxIds := strings.Split(baseResp.Data.Members, "^G") wxIds := strings.Split(baseResp.Data.Members, "^G")
log.Printf(" 群成员数: %d", len(wxIds)) log.Printf(" 群成员数: %d", len(wxIds))
// 修改不在数组的群成员状态为不在
err = tx.Model(&entity.GroupUser{}).Where("wxid NOT IN (?)", wxIds).Update("is_member", false).Error
if err != nil {
log.Printf("修改群成员状态失败: %s", err.Error())
return
}
for _, wxid := range wxIds { for _, wxid := range wxIds {
// 获取成员信息 // 获取成员信息
cp, _ := getContactProfile(wxid) cp, _ := getContactProfile(wxid)
if cp.Wxid != "" { if cp.Wxid != "" {
log.Printf(" 微信Id: %s -> 昵称: %s -> 微信号: %s", wxid, cp.Nickname, cp.Account) log.Printf(" 微信Id: %s -> 昵称: %s -> 微信号: %s", wxid, cp.Nickname, cp.Account)
// 查询成员是否存在,不在就新增,否则修改
var count int64
err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("wxid = ?", wxid).Count(&count).Error
if err != nil {
log.Printf("查询群成员失败: %s", err.Error())
continue
}
if count == 0 {
// 新增
err = tx.Create(&entity.GroupUser{
GroupId: gid,
Account: cp.Account,
HeadImage: cp.HeadImage,
Nickname: cp.Nickname,
Wxid: cp.Wxid,
IsMember: true,
}).Error
if err != nil {
log.Printf("新增群成员失败: %s", err.Error())
continue
}
} else {
// 修改
pm := map[string]any{
"account": cp.Account,
"head_image": cp.HeadImage,
"nickname": cp.Nickname,
}
err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("wxid = ?", wxid).Updates(pm).Error
if err != nil {
log.Printf("修改群成员失败: %s", err.Error())
continue
}
}
} }
} }
} }

View File

@ -16,6 +16,7 @@ func InitTasks() {
_, _ = s.Every(1).Day().At("09:30").Do(yesterday) _, _ = s.Every(1).Day().At("09:30").Do(yesterday)
// 每小时更新一次好友列表 // 每小时更新一次好友列表
//_, _ = s.Every(5).Minute().Do(syncFriends)
_, _ = s.Every(1).Hour().Do(syncFriends) _, _ = s.Every(1).Hour().Do(syncFriends)
// 开启定时任务 // 开启定时任务