1
0
forked from lxh/go-wxhelper

Merge pull request '🆕 新增水群年度排行榜' (#6) from dev into main

Reviewed-on: lxh/go-wxhelper#6
This commit is contained in:
李寻欢 2024-01-01 00:59:54 +08:00
commit f5cc7953f4
5 changed files with 106 additions and 2 deletions

View File

@ -21,14 +21,15 @@ mysql:
task:
enable: false
syncFriends:
enable: true
enable: false
cron: '*/5 * * * *' # 五分钟一次
waterGroup:
enable: false
enable: true
cron:
yesterday: '30 9 * * *' # 每天9:30
week: '30 9 * * 1' # 每周一9:30
month: '30 9 1 * *' # 每月1号9:30
year: '55 0 1 1 *' # 每年1月1号9:30
# AI回复
ai:

View File

@ -28,4 +28,5 @@ type waterGroupCron struct {
Yesterday string `json:"yesterday" yaml:"yesterday"` // 昨日排行榜
Week string `json:"week" yaml:"week"` // 周排行榜
Month string `json:"month" yaml:"month"` // 月排行榜
Year string `json:"year" yaml:"year"` // 年排行榜
}

View File

@ -32,6 +32,9 @@ func InitTasks() {
if config.Conf.Task.WaterGroup.Cron.Month != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Month).Do(watergroup.Month)
}
if config.Conf.Task.WaterGroup.Cron.Year != "" {
_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Year).Do(watergroup.Year)
}
}
// 更新好友列表

View File

@ -33,6 +33,8 @@ func getRankData(groupId, date string) (rank []rankUser, err error) {
tx.Where("YEARWEEK(date_format(tm.create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1")
case "month":
tx.Where("PERIOD_DIFF(date_format(now(), '%Y%m'), date_format(create_at, '%Y%m')) = 1")
case "year":
tx.Where("YEAR(tm.create_at) = YEAR(NOW()) - 1")
}
// 查询指定时间段全部数据

97
tasks/watergroup/year.go Normal file
View File

@ -0,0 +1,97 @@
package watergroup
import (
"fmt"
"go-wechat/config"
"go-wechat/service"
"go-wechat/utils"
"log"
"strings"
"time"
)
// Year
// @description: 年排行榜
func Year() {
groups, err := service.GetAllEnableChatRank()
if err != nil {
log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
return
}
for _, group := range groups {
// 消息统计
dealYear(group.Wxid)
res, ok := config.Conf.Resource["wordcloud"]
if !ok {
continue
}
// 获取上周周数
year := time.Now().Local().AddDate(0, 0, -1).Year()
// 发送词云
fileName := fmt.Sprintf("%d_%s.png", year, group.Wxid)
utils.SendImage(group.Wxid, fmt.Sprintf(res.Path, fileName), 0)
}
}
// dealYear
// @description: 处理年度排行榜
// @param gid
func dealYear(gid string) {
notifyMsgs := []string{"#年度水群排行榜"}
// 获取上周消息总数
records, err := getRankData(gid, "year")
if err != nil {
log.Printf("获取去年消息排行失败, 错误信息: %v", err)
return
}
log.Printf("去年消息总数: %+v", records)
// 莫得消息,直接返回
if len(records) == 0 {
log.Printf("去年本群[%s]无对话记录", gid)
return
}
// 计算消息总数
var msgCount int64
for _, v := range records {
msgCount += v.Count
}
// 组装消息总数推送信息
notifyMsgs = append(notifyMsgs, " ")
notifyMsgs = append(notifyMsgs, "亲爱的群友们,新年已经悄悄来临,让我们一起迎接这充满希望和美好的时刻。在这个特殊的日子里,我要向你们致以最真挚的祝福。")
notifyMsgs = append(notifyMsgs, "首先,我想对去年在群中表现出色、积极参与的成员们表示衷心的祝贺和感谢!你们的活跃与奉献让群聊更加充满了生机和活力。你们的贡献不仅仅是为了自己,更是为了我们整个群体的进步与成长。")
notifyMsgs = append(notifyMsgs, "特此给去年年度活跃成员排行榜上的朋友们送上真诚的祝福。你们的热情、智慧和参与度,令我们很是钦佩。愿新的一年中,你们继续保持着你们的活力和激情,为群中带来更多的惊喜和启迪。")
notifyMsgs = append(notifyMsgs, "对于那些未上榜的朋友们,我要说,你们也是我们群聊中非常重要的一部分。你们或许没有在排行榜上留下痕迹,但你们的存在和参与同样不可或缺。你们为群聊注入了新的思维和观点,为我们提供了不同的视角和见解。")
notifyMsgs = append(notifyMsgs, "因此,我想特别鼓励未上榜的朋友们,继续发扬你们的热情和积极性。无论是在分享知识、讨论问题、还是互相支持鼓励,你们的贡献都是宝贵的。让我们共同创造一个更加活跃和有意义的群聊环境。")
notifyMsgs = append(notifyMsgs, "最后,让我们一起迈向新的一年,相信自己的潜力和可能性,用我们的友谊和互助支持彼此。愿新的一年给我们带来更多的快乐、成功和成长。")
notifyMsgs = append(notifyMsgs, "祝福你们新年快乐让我们一起迎接2022年的到来")
notifyMsgs = append(notifyMsgs, " -- OpnWechat")
notifyMsgs = append(notifyMsgs, " ")
notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ 去年本群 %d 位朋友共产生 %d 条发言", len(records), msgCount))
notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
notifyMsgs = append(notifyMsgs, " ")
for i, r := range records {
// 只取前十条
if i >= 10 {
break
}
log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
badge := "🏆"
switch i {
case 0:
badge = "🥇"
case 1:
badge = "🥈"
case 2:
badge = "🥉"
}
notifyMsgs = append(notifyMsgs, fmt.Sprintf("%s %s -> %d条", badge, r.Nickname, r.Count))
}
log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
}