forked from lxh/go-wxhelper
✨ 新增同步好友列表任务
This commit is contained in:
parent
40e4dc48c6
commit
3e8051906f
@ -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"
|
||||||
|
}
|
||||||
|
101
tasks/friends.go
101
tasks/friends.go
@ -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
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -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)
|
||||||
|
|
||||||
// 开启定时任务
|
// 开启定时任务
|
||||||
|
Loading…
Reference in New Issue
Block a user