120 lines
4.7 KiB
JavaScript
120 lines
4.7 KiB
JavaScript
import fs from 'fs'
|
||
import { Tail } from 'tail'
|
||
import { update } from './update.js'
|
||
|
||
// 用于存储文章、游戏、截图相关的统计数据
|
||
const articles = new Map() // 文章统计
|
||
const games = new Map() // 游戏统计
|
||
const screenshots = new Map() // 截图统计
|
||
|
||
// 日期格式化工具
|
||
const dateFormat = new Intl.DateTimeFormat('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })
|
||
|
||
// 匹配文章和截图的正则表达式
|
||
const articleRegex = /\/web\/v1\/article\/get\?id=(\d+)&userId=(\d+)/
|
||
const imagesRegex = /\/api\/images\?similar=(\d+)/
|
||
|
||
let currentDate = dateFormat.format(new Date()).replace(/\//g, '-') // 当前日期
|
||
let isArchiving = false // 是否正在存档标记
|
||
|
||
// 存档函数：将数据存入文件
|
||
const archive = async (message = '正在存档...') => {
|
||
if (isArchiving) return // 防止重复存档
|
||
isArchiving = true
|
||
console.log(message)
|
||
await fs.promises.writeFile(`./data/${currentDate}.json`, JSON.stringify({
|
||
articles: Object.fromEntries(articles), // 将 Map 转换为普通对象
|
||
games: Object.fromEntries(games),
|
||
screenshots: Object.fromEntries(screenshots),
|
||
}, null, 2))
|
||
isArchiving = false
|
||
}
|
||
|
||
// 主函数
|
||
export default function () {
|
||
// 启动时加载当天的数据文件
|
||
if (fs.existsSync(`./data/${currentDate}.json`)) {
|
||
console.log(currentDate, '启动时加载数据...')
|
||
const file = fs.readFileSync(`./data/${currentDate}.json`, 'utf8')
|
||
if (file) {
|
||
const data = JSON.parse(file)
|
||
// 将文件中的数据加载到对应的 Map 中
|
||
Object.entries(data.articles).forEach(([key, value]) => articles.set(key, value))
|
||
Object.entries(data.games).forEach(([key, value]) => games.set(key, value))
|
||
Object.entries(data.screenshots).forEach(([key, value]) => screenshots.set(key, value))
|
||
}
|
||
}
|
||
|
||
console.log(currentDate, '开始收集日志...')
|
||
|
||
// 定时存档（每10分钟）
|
||
setInterval(() => {
|
||
archive('每10分钟自动存档...')
|
||
update('day_rank', Object.entries(Object.fromEntries(screenshots))) // 更新数据库中 day_rank 字段
|
||
}, 600000)
|
||
|
||
// 实时读取日志文件
|
||
const tail = new Tail('/opt/log/caddy/access.log')
|
||
tail.on('line', async (line) => {
|
||
const item = JSON.parse(line) // 解析日志行
|
||
if (item.level !== 'debug') return // 仅处理 debug 级别的日志
|
||
if (item.msg !== 'upstream roundtrip') return // 仅处理指定类型的日志
|
||
|
||
// 检查日志的日期是否跨天
|
||
const logDate = dateFormat.format(new Date(item.ts * 1000)).replace(/\//g, '-')
|
||
if (currentDate !== logDate) {
|
||
await archive('跨日期存档...') // 保存当前数据
|
||
articles.clear() // 清空统计数据
|
||
games.clear()
|
||
screenshots.clear()
|
||
currentDate = logDate // 更新当前日期
|
||
}
|
||
|
||
// 处理文章相关日志
|
||
if (item.request.uri.startsWith('/web/v1/article/get')) {
|
||
const [uri, id, userId] = item.request.uri.match(articleRegex) ?? []
|
||
if (uri && id && userId && item.request.headers.Referer) {
|
||
const referer = item.request.headers.Referer[0]
|
||
// 根据 Referer 的不同路径，分类计数
|
||
if (referer.includes('/articleDetails/')) {
|
||
articles.set(id, articles.has(id) ? articles.get(id) + 1 : 1)
|
||
return
|
||
}
|
||
if (referer.includes('/inspirationInfo/')) {
|
||
games.set(id, games.has(id) ? games.get(id) + 1 : 1)
|
||
return
|
||
}
|
||
if (referer.includes('/game/')) {
|
||
games.set(id, games.has(id) ? games.get(id) + 1 : 1)
|
||
return
|
||
}
|
||
console.log('未处理的 Referer:', referer) // 打印未处理的 Referer
|
||
}
|
||
return
|
||
}
|
||
|
||
// 处理截图相关日志
|
||
if (item.request.uri.startsWith('/api/images')) {
|
||
if (item.request.uri.includes('page=')) {
|
||
return // 忽略分页请求
|
||
}
|
||
const [uri, id] = item.request.uri.match(imagesRegex) ?? []
|
||
if (uri && id) {
|
||
screenshots.set(id, screenshots.has(id) ? screenshots.get(id) + 1 : 1)
|
||
}
|
||
return
|
||
}
|
||
})
|
||
|
||
// 日志尾部事件监听
|
||
tail.on('end', () => console.log('没有更多内容，停止读取'))
|
||
tail.on('error', (error) => console.error('错误:', error))
|
||
|
||
// 捕获退出信号，确保数据存档
|
||
process.on('SIGINT', async () => {
|
||
tail.unwatch() // 停止监听日志
|
||
await archive('退出前存档数据...')
|
||
process.exit()
|
||
})
|
||
}
|