🎉 first commit
This commit is contained in:
李寻欢
111
tasks/friends.go
Normal file
111
tasks/friends.go
Normal file
@@ -0,0 +1,111 @@
|
||||
package tasks
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"github.com/go-resty/resty/v2"
|
||||
"go-wechat/constant"
|
||||
"go-wechat/entity"
|
||||
"go-wechat/model"
|
||||
"log"
|
||||
"slices"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// 同步群成员
|
||||
|
||||
// syncFriends
|
||||
// @description: 同步好友列表
|
||||
func syncFriends() {
|
||||
var base model.Response[[]entity.Friend]
|
||||
|
||||
client := resty.New()
|
||||
resp, err := client.R().
|
||||
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
||||
SetResult(&base).
|
||||
Post("http://10.0.0.73:19088/api/getContactList")
|
||||
if err != nil {
|
||||
log.Printf("获取好友列表失败: %s", err.Error())
|
||||
return
|
||||
}
|
||||
log.Printf("获取好友列表结果: %s", resp.String())
|
||||
for _, friend := range base.Data {
|
||||
if strings.Contains(friend.Wxid, "gh_") || strings.Contains(friend.Wxid, "@openim") {
|
||||
continue
|
||||
}
|
||||
// 特殊Id跳过
|
||||
if slices.Contains(constant.SpecialId, friend.Wxid) {
|
||||
continue
|
||||
}
|
||||
log.Printf("昵称: %s -> 类型: %d -> 微信号: %s -> 微信原始Id: %s", friend.Nickname, friend.Type, friend.CustomAccount, friend.Wxid)
|
||||
|
||||
// 群成员，同步一下成员信息
|
||||
if strings.Contains(friend.Wxid, "@chatroom") {
|
||||
syncGroupUsers(friend.Wxid)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
// syncGroupUsers
|
||||
// @description: 同步群成员
|
||||
// @param gid
|
||||
func syncGroupUsers(gid string) {
|
||||
var baseResp model.Response[model.GroupUser]
|
||||
|
||||
// 组装参数
|
||||
param := map[string]any{
|
||||
"chatRoomId": gid, // 群Id
|
||||
}
|
||||
pbs, _ := json.Marshal(param)
|
||||
|
||||
client := resty.New()
|
||||
_, err := client.R().
|
||||
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
||||
SetBody(string(pbs)).
|
||||
SetResult(&baseResp).
|
||||
Post("http://10.0.0.73:19088/api/getMemberFromChatRoom")
|
||||
if err != nil {
|
||||
log.Printf("获取群成员信息失败: %s", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// 昵称Id
|
||||
wxIds := strings.Split(baseResp.Data.Members, "^G")
|
||||
|
||||
log.Printf(" 群成员数: %d", len(wxIds))
|
||||
for _, wxid := range wxIds {
|
||||
// 获取成员信息
|
||||
cp, _ := getContactProfile(wxid)
|
||||
if cp.Wxid != "" {
|
||||
log.Printf(" 微信Id: %s -> 昵称: %s -> 微信号: %s", wxid, cp.Nickname, cp.Account)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// getContactProfile
|
||||
// @description: 获取成员详情
|
||||
// @param wxid
|
||||
// @return ent
|
||||
// @return err
|
||||
func getContactProfile(wxid string) (ent model.ContactProfile, err error) {
|
||||
var baseResp model.Response[model.ContactProfile]
|
||||
|
||||
// 组装参数
|
||||
param := map[string]any{
|
||||
"wxid": wxid, // 群Id
|
||||
}
|
||||
pbs, _ := json.Marshal(param)
|
||||
|
||||
client := resty.New()
|
||||
_, err = client.R().
|
||||
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
||||
SetBody(string(pbs)).
|
||||
SetResult(&baseResp).
|
||||
Post("http://10.0.0.73:19088/api/getContactProfile")
|
||||
if err != nil {
|
||||
log.Printf("获取成员详情失败: %s", err.Error())
|
||||
return
|
||||
}
|
||||
ent = baseResp.Data
|
||||
return
|
||||
}
|
||||
24
tasks/tasks.go
Normal file
24
tasks/tasks.go
Normal file
@@ -0,0 +1,24 @@
|
||||
package tasks
|
||||
|
||||
import (
|
||||
"github.com/go-co-op/gocron"
|
||||
"log"
|
||||
"time"
|
||||
)
|
||||
|
||||
// InitTasks
|
||||
// @description: 初始化定时任务
|
||||
func InitTasks() {
|
||||
// 定时任务发送消息
|
||||
s := gocron.NewScheduler(time.Local)
|
||||
|
||||
// 每天早上九点半发送前一天的水群排行
|
||||
_, _ = s.Every(1).Day().At("09:30").Do(yesterday)
|
||||
|
||||
// 每小时更新一次好友列表
|
||||
_, _ = s.Every(1).Minute().Do(syncFriends)
|
||||
|
||||
// 开启定时任务
|
||||
s.StartAsync()
|
||||
log.Println("定时任务初始化成功")
|
||||
}
|
||||
46
tasks/water_group.go
Normal file
46
tasks/water_group.go
Normal file
@@ -0,0 +1,46 @@
|
||||
package tasks
|
||||
|
||||
import (
|
||||
"go-wechat/client"
|
||||
"go-wechat/entity"
|
||||
"log"
|
||||
)
|
||||
|
||||
// 水群排行榜
|
||||
|
||||
// yesterday
|
||||
// @description: 昨日排行榜
|
||||
func yesterday() {
|
||||
// 获取昨日消息总数
|
||||
var yesterdayMsgCount int64
|
||||
err := client.MySQL.Model(&entity.Message{}).
|
||||
Where("from_user = ?", "18958257758@chatroom").
|
||||
Where("DATEDIFF(create_at,NOW()) = -1").
|
||||
Count(&yesterdayMsgCount).Error
|
||||
if err != nil {
|
||||
log.Printf("获取昨日消息总数失败, 错误信息: %v", err)
|
||||
return
|
||||
}
|
||||
log.Printf("昨日消息总数: %d", yesterdayMsgCount)
|
||||
|
||||
// 返回数据
|
||||
type record struct {
|
||||
GroupUser string
|
||||
Count int64
|
||||
}
|
||||
|
||||
var records []record
|
||||
err = client.MySQL.Model(&entity.Message{}).
|
||||
Select("group_user", "count( 1 ) AS `count`").
|
||||
Where("from_user = ?", "18958257758@chatroom").
|
||||
Where("DATEDIFF(create_at,NOW()) = -1").
|
||||
Group("group_user").Order("`count` DESC").
|
||||
Limit(5).Find(&records).Error
|
||||
if err != nil {
|
||||
log.Printf("获取昨日消息失败, 错误信息: %v", err)
|
||||
return
|
||||
}
|
||||
for _, r := range records {
|
||||
log.Printf("账号: %s -> %d", r.GroupUser, r.Count)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user