基本实现

main.js

@@ -1,22 +1,69 @@
+import fs from 'fs'
 import { Tail } from 'tail'
 
-const tail = new Tail('./access.log')
+const tail = new Tail('/opt/log/caddy/access.log')
 
 // 指标(日榜, 周榜, 月榜, 总榜)
 const 作品 = new Map()
 const 游戏 = new Map()
 const 截图 = new Map()
+const 格式 = new Intl.DateTimeFormat('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })
+
+
+var 日期 = 格式.format(new Date()).replace(/\//g, '-')
+var 存档 = new Date()
+
+if (fs.existsSync(`./data/${日期}.json`)) {
+    console.log(日期, '启动时读入数据..')
+    const file = fs.readFileSync(`./data/${日期}.json`, 'utf8')
+    if (file) {
+        const data = JSON.parse(file)
+        Object.entries(data.作品).forEach(([key, value]) => 作品.set(key, value))
+        Object.entries(data.游戏).forEach(([key, value]) => 游戏.set(key, value))
+        Object.entries(data.截图).forEach(([key, value]) => 截图.set(key, value))
+    }
+}
+
+
+console.log(日期, '开始收集日志..')
+
 
 tail.on('line', (line) => {
     const item = JSON.parse(line)
     if (item.level !== 'debug') return
     if (item.msg !== 'upstream roundtrip') return
+
+    if ((new Date()).getTime() - 存档.getTime() > 600000) {
+        console.log('10 分钟存档一次..')
+        fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({
+            作品: Object.fromEntries(作品),
+            游戏: Object.fromEntries(游戏),
+            截图: Object.fromEntries(截图),
+        }, null, 2))
+        存档 = new Date()
+    }
+
+    const 日志日期 = 格式.format(new Date(item.ts * 1000)).replace(/\//g, '-')
+    if (日期 !== 日志日期) {
+        fs.promises.writeFile(`./data/${日期} 作品.json`, JSON.stringify(Object.fromEntries(作品), null, 2))
+        fs.promises.writeFile(`./data/${日期} 游戏.json`, JSON.stringify(Object.fromEntries(游戏), null, 2))
+        fs.promises.writeFile(`./data/${日期} 截图.json`, JSON.stringify(Object.fromEntries(截图), null, 2))
+        日期 = 日志日期
+    }
+
     if (item.request.uri.startsWith('/web/v1/article/get')) {
         const reg = /\/web\/v1\/article\/get\?id=(\d+)&userId=(\d+)/
         const [uri, id, user_id] = item.request.uri.match(reg) ?? []
-        if (uri && id && user_id) {
-            // 判断类型
-            游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1)
+        if (uri && id && user_id && item.request.headers.Referer) {
+            if (item.request.headers.Referer[0].includes('/inspirationInfo/')) {
+                游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1)
+                return
+            }
+            if (item.request.headers.Referer[0].includes('/articleDetails/')) {
+                作品.set(id, 作品.has(id) ? 作品.get(id) + 1 : 1)
+                return
+            }
+            console.log(item.request.headers.Referer)
         }
         return
     }
@@ -44,7 +91,15 @@ tail.on('error', (error) => {
     console.error('错误:', error)
 })
 
-process.on('SIGINT', () => {
+process.on('SIGINT', async () => {
     tail.unwatch()
+
+    console.log('退出前储存数据..')
+    await fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({
+        作品: Object.fromEntries(作品),
+        游戏: Object.fromEntries(游戏),
+        截图: Object.fromEntries(截图),
+    }, null, 2))
+
     process.exit()
 })