go-wxhelper/tasks/water_group.go

104 lines
2.6 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package tasks
import (
"fmt"
"go-wechat/client"
"go-wechat/entity"
"go-wechat/utils"
"io"
"log"
"os"
"strings"
)
// 水群排行榜
// yesterday
// @description: 昨日排行榜
func yesterday() {
// 从环境变量读取需要处理的群Id
gid := strings.Split(os.Getenv("GROUP_ID"), ",")
for _, id := range gid {
dealYesterday(id)
}
}
// dealYesterday
// @description: 处理请求
// @param gid
func dealYesterday(gid string) {
notifyMsgs := []string{"#昨日水群排行榜"}
// 读取黑名单文件名单内的Id不上榜
var blacklist []string
file, err := os.Open("blacklist.txt")
if err != nil {
log.Printf("读取黑名单失败: %v", err)
} else {
defer file.Close()
var content []byte
if content, err = io.ReadAll(file); err != nil {
log.Printf("读取黑名单失败: %v", err)
} else {
blacklist = strings.Split(string(content), "\n")
}
}
// 获取昨日消息总数
var yesterdayMsgCount int64
err = client.MySQL.Model(&entity.Message{}).
Where("from_user = ?", gid).
Where("DATEDIFF(create_at,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("DATEDIFF(tm.create_at,NOW()) = -1").
Group("tm.group_user, tgu.nickname").Order("`count` DESC").
Limit(10)
// 如果有黑名单,过滤掉
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)
}