Compare commits

..

15 Commits

Author SHA1 Message Date
李寻欢
b4470b0888 水群排行榜新增周榜和月榜 2023-11-19 11:36:15 +08:00
李寻欢
500e241f8d 🐛 修复人设没传进去的BUG 2023-11-17 10:05:18 +08:00
李寻欢
6c2bf3fc9c AI新增模型和人设配置 2023-11-17 09:57:35 +08:00
李寻欢
91d2fc50e2 🎨 优化传递给AI的消息,去掉艾特机器人那一段字符串 2023-11-17 09:44:14 +08:00
李寻欢
119c2a5359 🎨 优化AI返回内容格式,加个换行 2023-11-13 16:58:52 +08:00
李寻欢
06a64c5e5a 🐛 Fix a bug. 2023-11-13 13:43:59 +08:00
李寻欢
60bfa0e8a0 新增一个简易的AI机器人 2023-11-13 13:32:42 +08:00
李寻欢
d08937563a 新增迎新操作 2023-11-03 11:59:40 +08:00
李寻欢
697f5560a4 🎨 语法优化 2023-11-03 11:48:51 +08:00
李寻欢
86daff5763 🐛 修复新成员无法入库的BUG 2023-11-03 11:48:34 +08:00
李寻欢
a6e935e233 📝 文档完善 2023-10-31 15:14:58 +08:00
李寻欢
ead2e06c4c 📝 文档完善 2023-10-31 15:12:15 +08:00
李寻欢
51078eba12 📝 文档完善 2023-10-31 14:51:34 +08:00
李寻欢
1611019673 📝 文档完善 2023-10-27 14:59:43 +08:00
李寻欢
11d79e4540 🐳 新增一个docker-compose文件 2023-10-26 10:10:48 +08:00
17 changed files with 561 additions and 38 deletions

View File

@@ -3,7 +3,7 @@ wechat:
# 微信HOOK接口地址
host: 10.0.0.73:19088
# 是否在启动的时候自动设置hook服务的回调
autoSetCallback: true
autoSetCallback: false
# 回调IP如果是Docker运行本参数必填如果Docker修改了映射格式为 ip:port
callback: 10.0.0.51
@@ -16,17 +16,33 @@ mysql:
db: wechat
task:
enable: false
enable: true
syncFriends:
enable: true
cron: '0 * * * *'
cron: '*/5 * * * *' # 五分钟一次
waterGroup:
enable: true
cron: '30 9 * * *'
enable: false
cron:
yesterday: '30 9 * * *' # 每天9:30
week: '30 9 * * 1' # 每周一9:30
month: '30 9 1 * *' # 每月1号9:30
# 需要发送水群排行榜的群Id
groups:
- '18958257758@chatroom'
- '49448748645@chatroom'
# 不计入统计范围的用户Id
blacklist:
- 'wxid_7788687886912'
- 'wxid_7788687886912'
# AI回复
ai:
# 是否启用
enable: false
# 模型不填默认gpt-3.5-turbo-0613
model: gpt-3.5-turbo-0613
# OpenAI Api key
apiKey: sk-xxxx
# 接口代理域名不填默认ChatGPT官方地址
baseUrl: https://sxxx
# 人设
personality: 你的名字叫张三,你是一个百科机器人,你的爱好是看电影,你的性格是开朗的,你的专长是讲故事,你的梦想是当一名童话故事作家。你对政治没有一点儿兴趣,也不会讨论任何与政治相关的话题,你甚至可以拒绝回答这一类话题。

11
config/ai.go Normal file
View File

@@ -0,0 +1,11 @@
package config
// ai
// @description: AI配置
type ai struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用AI
Model string `json:"model" yaml:"model"` // 模型
ApiKey string `json:"apiKey" yaml:"apiKey"` // API Key
BaseUrl string `json:"baseUrl" yaml:"baseUrl"` // API地址
Personality string `json:"personality" yaml:"personality"` // 人设
}

View File

@@ -8,20 +8,5 @@ type Config struct {
Task task `json:"task" yaml:"task"` // 定时任务配置
MySQL mysql `json:"mysql" yaml:"mysql"` // MySQL 配置
Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手
}
// task
// @description: 定时任务
type task struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
SyncFriends struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
Cron string `json:"cron" yaml:"cron"` // 定时任务表达式
} `json:"syncFriends" yaml:"syncFriends"` // 同步好友
WaterGroup struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
Cron string `json:"cron" yaml:"cron"` // 定时任务表达式
Groups []string `json:"groups" yaml:"groups"` // 启用的群Id
Blacklist []string `json:"blacklist" yaml:"blacklist"` // 黑名单
} `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜
Ai ai `json:"ai" yaml:"ai"` // AI配置
}

33
config/task.go Normal file
View File

@@ -0,0 +1,33 @@
package config
// task
// @description: 定时任务
type task struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友
WaterGroup waterGroup `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜
}
// syncFriends
// @description: 同步好友
type syncFriends struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
Cron string `json:"cron" yaml:"cron"` // 定时任务表达式
}
// waterGroup
// @description: 水群排行榜
type waterGroup struct {
Enable bool `json:"enable" yaml:"enable"` // 是否启用
Cron waterGroupCron `json:"cron" yaml:"cron"` // 定时任务表达式
Groups []string `json:"groups" yaml:"groups"` // 启用的群Id
Blacklist []string `json:"blacklist" yaml:"blacklist"` // 黑名单
}
// waterGroupCron
// @description: 水群排行榜定时任务
type waterGroupCron struct {
Yesterday string `json:"yesterday" yaml:"yesterday"` // 昨日排行榜
Week string `json:"week" yaml:"week"` // 周排行榜
Month string `json:"month" yaml:"month"` // 月排行榜
}

48
docker-compose.yaml Normal file
View File

@@ -0,0 +1,48 @@
version: '3.9'
services:
wechat:
image: lxh01/wxhelper-docker:3.9.5.81
container_name: gw-wechat
restart: unless-stopped
environment:
- WINEDEBUG=fixme-all
volumes:
- ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
ports:
- "8080:8080"
- "19088:19088"
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
interval: 60s
timeout: 10s
retries: 5
mysql:
image: mysql:8
container_name: gw-db
restart: unless-stopped
depends_on:
wechat:
condition: service_healthy
environment:
- MYSQL_ROOT_PASSWORD=wechat
- MYSQL_USER=wechat
- MYSQL_PASSWORD=wechat
- MYSQL_DATABASE=wechat
volumes:
- ./data/db:/var/lib/mysql
wxhelper:
image: gitee.ltd/lxh/go-wxhelper:latest
container_name: gw-service
restart: unless-stopped
depends_on:
- mysql
volumes:
# 配置文件请参阅项目根目录的config.yaml文件
- ./config/config.yaml:/app/config.yaml
ports:
- "19099:19099"

View File

@@ -19,13 +19,13 @@ func (Friend) TableName() string {
// 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)"` // 是否群成员
LeaveTime time.Time `json:"leaveTime"` // 离开时间
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)"` // 是否群成员
LeaveTime *time.Time `json:"leaveTime"` // 离开时间
}
func (GroupUser) TableName() string {

1
go.mod
View File

@@ -7,6 +7,7 @@ require (
github.com/fsnotify/fsnotify v1.6.0
github.com/go-co-op/gocron v1.34.1
github.com/go-resty/resty/v2 v2.8.0
github.com/sashabaranov/go-openai v1.17.5
github.com/spf13/viper v1.17.0
gorm.io/driver/mysql v1.5.1
gorm.io/gorm v1.25.4

2
go.sum
View File

@@ -177,6 +177,8 @@ github.com/sagikazarmark/locafero v0.3.0 h1:zT7VEGWC2DTflmccN/5T1etyKvxSxpHsjb9c
github.com/sagikazarmark/locafero v0.3.0/go.mod h1:w+v7UsPNFwzF1cHuOajOOzoq4U7v/ig1mpRjqV+Bu1U=
github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
github.com/sashabaranov/go-openai v1.17.5 h1:ItBzlrrfTtkFWOFlgfOhk3y/xRBC4PJol4gdbiK7hgg=
github.com/sashabaranov/go-openai v1.17.5/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
github.com/spf13/afero v1.10.0 h1:EaGW2JJh15aKOejeuJ+wpFSHnbd7GE6Wvp3TsNhb6LY=

74
handler/at_message.go Normal file
View File

@@ -0,0 +1,74 @@
package handler
import (
"context"
"fmt"
"github.com/sashabaranov/go-openai"
"go-wechat/config"
"go-wechat/entity"
"go-wechat/utils"
"log"
"regexp"
"strings"
)
// handleAtMessage
// @description: 处理At机器人的消息
// @param m
func handleAtMessage(m entity.Message) {
if !config.Conf.Ai.Enable {
return
}
// 预处理一下发送的消息,用正则去掉@机器人的内容
re := regexp.MustCompile(`@([^]+)`)
matches := re.FindStringSubmatch(m.Content)
if len(matches) > 0 {
// 过滤掉第一个匹配到的
m.Content = strings.Replace(m.Content, matches[0], "", 1)
}
// 组装消息体
messages := make([]openai.ChatCompletionMessage, 0)
if config.Conf.Ai.Personality != "" {
// 填充人设
messages = append(messages, openai.ChatCompletionMessage{
Role: openai.ChatMessageRoleSystem,
Content: config.Conf.Ai.Personality,
})
}
// 填充用户消息
messages = append(messages, openai.ChatCompletionMessage{
Role: openai.ChatMessageRoleUser,
Content: m.Content,
})
// 配置模型
model := openai.GPT3Dot5Turbo0613
if config.Conf.Ai.Model != "" {
model = config.Conf.Ai.Model
}
// 默认使用AI回复
conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
if config.Conf.Ai.BaseUrl != "" {
conf.BaseURL = fmt.Sprintf("%s/v1", config.Conf.Ai.BaseUrl)
}
client := openai.NewClientWithConfig(conf)
resp, err := client.CreateChatCompletion(
context.Background(),
openai.ChatCompletionRequest{
Model: model,
Messages: messages,
},
)
if err != nil {
log.Printf("OpenAI聊天发起失败: %v", err.Error())
utils.SendMessage(m.FromUser, m.GroupUser, "AI炸啦~", 0)
return
}
// 发送消息
utils.SendMessage(m.FromUser, m.GroupUser, "\n"+resp.Choices[0].Message.Content, 0)
}

View File

@@ -26,8 +26,14 @@ func Parse(remoteAddr net.Addr, msg []byte) {
groupUser := ""
msgStr := m.Content
if strings.Contains(m.FromUser, "@") {
// 系统消息不单独处理
if m.Type != types.MsgTypeRecalled && m.Type != types.MsgTypeSys {
switch m.Type {
case types.MsgTypeRecalled:
// 消息撤回
case types.MsgTypeSys:
// 系统消息
go handleSysMessage(m)
default:
// 默认消息处理
groupUser = strings.Split(m.Content, "\n")[0]
groupUser = strings.ReplaceAll(groupUser, ":", "")
@@ -50,5 +56,10 @@ func Parse(remoteAddr net.Addr, msg []byte) {
ent.DisplayFullContent = m.DisplayFullContent
ent.Raw = string(msg)
// 处理At机器人的消息
if strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你") {
go handleAtMessage(ent)
}
go service.SaveMessage(ent)
}

19
handler/sys_message.go Normal file
View File

@@ -0,0 +1,19 @@
package handler
import (
"go-wechat/model"
"go-wechat/utils"
"strings"
)
// handleSysMessage
// @description: 系统消息处理
// @param m
func handleSysMessage(m model.Message) {
// 有人进群
if strings.Contains(m.Content, "\"邀请\"") && strings.Contains(m.Content, "\"加入了群聊") {
// 发一张图乐呵乐呵
// 自己欢迎自己图片地址 D:\Share\emoticon\welcome-yourself.gif
utils.SendImage(m.FromUser, "D:\\Share\\emoticon\\welcome-yourself.gif", 0)
}
}

106
readme.md
View File

@@ -1,6 +1,104 @@
### 手动打包镜像
## 食用方式
0. 新建一个文件夹
```shell
docker build --push -t repo.lxh.io/lxh/go-wxhelper:latest .
mkdir wechat-hook # 名字随便写
cd wechat-hook
```
1. 创建配置文件`config.yaml`
```shell
mkdir config # 先创建一个文件夹保存配置文件,文件名不要变
vim config.yaml # 编辑配置文件,内容如下
```
```yaml
# 微信HOOK配置
wechat:
# 微信HOOK接口地址
host: wechat:19088
# 是否在启动的时候自动设置hook服务的回调
autoSetCallback: true
# 回调IP如果是Docker运行本参数必填如果Docker修改了映射格式为 ip:port如果使用项目提供的docker-compsoe.yaml文件启动可以不写
callback:
# 数据库
mysql:
host: mysql
port: 3306
user: wechat
password: wechat
db: wechat
task:
enable: false
syncFriends:
enable: true
cron: '0 * * * *'
waterGroup:
enable: true
cron: '30 9 * * *'
# 需要发送水群排行榜的群Id
groups:
- '11111@chatroom' # 自行配置
# 不计入统计范围的用户Id
blacklist:
- 'wxid_xxxx' # 自行配置
```
2. 创建`docker-compose.yaml`文件
```yaml
version: '3.9'
services:
wechat:
image: lxh01/wxhelper-docker:3.9.5.81
container_name: gw-wechat
restart: unless-stopped
environment:
- WINEDEBUG=fixme-all
volumes:
- ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
ports:
- "8080:8080"
- "19088:19088"
healthcheck:
test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]
interval: 60s
timeout: 10s
retries: 5
mysql:
image: mysql:8
container_name: gw-db
restart: unless-stopped
depends_on:
wechat:
condition: service_healthy
environment:
- MYSQL_ROOT_PASSWORD=wechat
- MYSQL_USER=wechat
- MYSQL_PASSWORD=wechat
- MYSQL_DATABASE=wechat
volumes:
- ./data/db:/var/lib/mysql
wxhelper:
image: gitee.ltd/lxh/go-wxhelper:latest
container_name: gw-service
restart: unless-stopped
depends_on:
- mysql
volumes:
# 配置文件请参阅项目根目录的config.yaml文件
- ./config/config.yaml:/app/config.yaml
ports:
- "19099:19099"
```
3. 启动
```shell
# 以下命令选个能用的就行
docker-compose up -d # 老版本
docker compose up -d # 新版本
```

86
tasks/month.go Normal file
View File

@@ -0,0 +1,86 @@
package tasks
import (
"fmt"
"go-wechat/client"
"go-wechat/config"
"go-wechat/entity"
"go-wechat/utils"
"log"
"strings"
)
// month
// @description: 月排行榜
func month() {
for _, id := range config.Conf.Task.WaterGroup.Groups {
dealMonth(id)
}
}
// dealMonth
// @description: 处理请求
// @param gid
func dealMonth(gid string) {
notifyMsgs := []string{"#上月水群排行榜"}
// 获取上月消息总数
var yesterdayMsgCount int64
err := client.MySQL.Model(&entity.Message{}).
Where("from_user = ?", gid).
Where("`type` < 10000").
Where("YEARWEEK(date_format(create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1").
Count(&yesterdayMsgCount).Error
if err != nil {
log.Printf("获取上月消息总数失败, 错误信息: %v", err)
return
}
log.Printf("上月消息总数: %d", yesterdayMsgCount)
if yesterdayMsgCount == 0 {
return
}
notifyMsgs = append(notifyMsgs, " ")
notifyMsgs = append(notifyMsgs, fmt.Sprintf("上月消息总数: %d", yesterdayMsgCount))
// 返回数据
type record struct {
GroupUser string
Nickname string
Count int64
}
var records []record
tx := client.MySQL.Table("t_message AS tm").
Joins("LEFT JOIN t_group_user AS tgu ON tgu.wxid = tm.group_user AND tm.from_user = tgu.group_id").
Select("tm.group_user", "tgu.nickname", "count( 1 ) AS `count`").
Where("tm.from_user = ?", gid).
Where("tm.type < 10000").
Where("YEARWEEK(date_format(tm.create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1").
Group("tm.group_user, tgu.nickname").Order("`count` DESC").
Limit(10)
// 黑名单
blacklist := config.Conf.Task.WaterGroup.Blacklist
// 如果有黑名单,过滤掉
if len(blacklist) > 0 {
tx.Where("tm.group_user NOT IN (?)", blacklist)
}
err = tx.Find(&records).Error
if err != nil {
log.Printf("获取上月消息失败, 错误信息: %v", err)
return
}
notifyMsgs = append(notifyMsgs, " ")
for i, r := range records {
log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
notifyMsgs = append(notifyMsgs, fmt.Sprintf("#%d: %s -> %d条", i+1, r.Nickname, r.Count))
}
notifyMsgs = append(notifyMsgs, " \n请未上榜的群友多多反思。")
log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
}

View File

@@ -20,8 +20,16 @@ func InitTasks() {
// 水群排行
if config.Conf.Task.WaterGroup.Enable {
log.Printf("水群排行任务已启用,执行表达式: %s", config.Conf.Task.WaterGroup.Cron)
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron).Do(yesterday)
log.Printf("水群排行任务已启用,执行表达式: %+v", config.Conf.Task.WaterGroup.Cron)
if config.Conf.Task.WaterGroup.Cron.Yesterday != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Yesterday).Do(yesterday)
}
if config.Conf.Task.WaterGroup.Cron.Week != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Week).Do(week)
}
if config.Conf.Task.WaterGroup.Cron.Month != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Month).Do(month)
}
}
// 更新好友列表

View File

@@ -30,6 +30,7 @@ func dealYesterday(gid string) {
var yesterdayMsgCount int64
err := client.MySQL.Model(&entity.Message{}).
Where("from_user = ?", gid).
Where("`type` < 10000").
Where("DATEDIFF(create_at,NOW()) = -1").
Count(&yesterdayMsgCount).Error
if err != nil {

86
tasks/week.go Normal file
View File

@@ -0,0 +1,86 @@
package tasks
import (
"fmt"
"go-wechat/client"
"go-wechat/config"
"go-wechat/entity"
"go-wechat/utils"
"log"
"strings"
)
// week
// @description: 周排行榜
func week() {
for _, id := range config.Conf.Task.WaterGroup.Groups {
dealWeek(id)
}
}
// dealWeek
// @description: 处理请求
// @param gid
func dealWeek(gid string) {
notifyMsgs := []string{"#上周水群排行榜"}
// 获取上周消息总数
var yesterdayMsgCount int64
err := client.MySQL.Model(&entity.Message{}).
Where("from_user = ?", gid).
Where("type < 10000").
Where("YEARWEEK(date_format(create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1").
Count(&yesterdayMsgCount).Error
if err != nil {
log.Printf("获取上周消息总数失败, 错误信息: %v", err)
return
}
log.Printf("上周消息总数: %d", yesterdayMsgCount)
if yesterdayMsgCount == 0 {
return
}
notifyMsgs = append(notifyMsgs, " ")
notifyMsgs = append(notifyMsgs, fmt.Sprintf("上周消息总数: %d", yesterdayMsgCount))
// 返回数据
type record struct {
GroupUser string
Nickname string
Count int64
}
var records []record
tx := client.MySQL.Table("t_message AS tm").
Joins("LEFT JOIN t_group_user AS tgu ON tgu.wxid = tm.group_user AND tm.from_user = tgu.group_id").
Select("tm.group_user", "tgu.nickname", "count( 1 ) AS `count`").
Where("tm.from_user = ?", gid).
Where("tm.type < 10000").
Where("YEARWEEK(date_format(tm.create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1").
Group("tm.group_user, tgu.nickname").Order("`count` DESC").
Limit(10)
// 黑名单
blacklist := config.Conf.Task.WaterGroup.Blacklist
// 如果有黑名单,过滤掉
if len(blacklist) > 0 {
tx.Where("tm.group_user NOT IN (?)", blacklist)
}
err = tx.Find(&records).Error
if err != nil {
log.Printf("获取上周消息失败, 错误信息: %v", err)
return
}
notifyMsgs = append(notifyMsgs, " ")
for i, r := range records {
log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
notifyMsgs = append(notifyMsgs, fmt.Sprintf("#%d: %s -> %d条", i+1, r.Nickname, r.Count))
}
notifyMsgs = append(notifyMsgs, " \n请未上榜的群友多多反思。")
log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
}

View File

@@ -18,18 +18,30 @@ func SendMessage(toId, atId, msg string, retryCount int) {
log.Printf("重试五次失败,停止发送")
return
}
// 组装参数
param := map[string]any{
"wxid": toId, // 群或好友Id
"msg": msg, // 消息
}
// 接口地址
apiUrl := config.Conf.Wechat.GetURL("/api/sendTextMsg")
if atId != "" {
apiUrl = config.Conf.Wechat.GetURL("/api/sendAtText")
param = map[string]any{
"chatRoomId": toId,
"wxids": atId,
"msg": msg, // 消息
}
}
pbs, _ := json.Marshal(param)
res := resty.New()
resp, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
Post(config.Conf.Wechat.GetURL("/api/sendTextMsg"))
Post(apiUrl)
if err != nil {
log.Printf("发送文本消息失败: %s", err.Error())
// 休眠五秒后重新发送
@@ -38,3 +50,35 @@ func SendMessage(toId, atId, msg string, retryCount int) {
}
log.Printf("发送文本消息结果: %s", resp.String())
}
// SendImage
// @description: 发送图片
// @param toId string 群或者好友Id
// @param imgPath string 图片路径
// @param retryCount int 重试次数
func SendImage(toId, imgPath string, retryCount int) {
if retryCount > 5 {
log.Printf("重试五次失败,停止发送")
return
}
// 组装参数
param := map[string]any{
"wxid": toId, // 群或好友Id
"imagePath": imgPath, // 图片地址
}
pbs, _ := json.Marshal(param)
res := resty.New()
resp, err := res.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
Post(config.Conf.Wechat.GetURL("/api/sendImagesMsg"))
if err != nil {
log.Printf("发送图片消息失败: %s", err.Error())
// 休眠五秒后重新发送
time.Sleep(5 * time.Second)
SendImage(toId, imgPath, retryCount+1)
}
log.Printf("发送图片消息结果: %s", resp.String())
}