1
0
forked from lxh/go-wxhelper
go-wxhelper/tasks/hottop/hottop.go

89 lines
2.0 KiB
Go
Raw Normal View History

2024-07-15 14:14:24 +08:00
package hottop
import (
"fmt"
"go-wechat/client"
"go-wechat/common/types"
"go-wechat/model/entity"
"go-wechat/service"
"go-wechat/utils"
"log"
"slices"
"strings"
"time"
)
// HotTop
// @description: 热搜排行榜
func HotTop() {
// 发送到开启了的群
groups, err := service.GetAllEnableHotTop()
if err != nil {
log.Printf("获取启用了热榜的群组失败, 错误信息: %v", err)
return
}
// 获取热榜数据
news := getTopData()
if len(news) == 0 {
return
}
// 组装消息
msg := fmt.Sprintf("#热搜排行榜\n \n嘿朋友有新的新闻了喔快来康康吧\n \n%s", strings.Join(news, "\n-------\n"))
for _, group := range groups {
utils.SendMessage(group.Wxid, "", msg, 0)
// 休眠一秒,防止频繁发送
time.Sleep(time.Second)
}
}
// getTopData
// @description: 获取热榜数据
// @return data
func getTopData() (data []string) {
// 获取热榜数据
records := utils.NewsUtil().GetHotTop()
if len(records) == 0 {
log.Println("获取热榜数据失败")
return
}
var datas = make([]entity.HotTop, 0)
for _, item := range records {
var d = entity.HotTop{
CreatedAt: types.DateTime(time.Now().Local()),
Title: item.Title,
Hot: item.Hot,
Url: item.Url,
MobileUrl: item.MobilUrl,
}
datas = append(datas, d)
}
// 获取缓存数据
var oldTitles []string
err := client.MySQL.Model(&entity.HotTop{}).Order("created_at DESC").Limit(len(datas)).Pluck("title", &oldTitles).Error
if err != nil {
log.Println("获取历史热榜数据失败", err)
return
}
// 筛选出新数据
var newDatas []entity.HotTop
for _, d := range datas {
if slices.Contains(oldTitles, d.Title) {
continue
}
d.Channel = "百度"
newDatas = append(newDatas, d)
data = append(data, fmt.Sprintf("标题: %s\n热度: %s\n详情: %s", d.Title, d.Hot, d.Url))
}
// 保存新数据到数据库
if len(newDatas) > 0 {
err = client.MySQL.Create(&newDatas).Error
if err != nil {
log.Println("保存新热榜数据失败", err)
}
}
return
}