diff --git a/main.js b/main.js index be6f0ac..d34f858 100644 --- a/main.js +++ b/main.js @@ -8,14 +8,30 @@ const 作品 = new Map() const 游戏 = new Map() const 截图 = new Map() const 格式 = 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+)/ -var 日期 = 格式.format(new Date()).replace(/\//g, '-') -var 存档 = new Date() +var 当前日期 = 格式.format(new Date()).replace(/\//g, '-') +var 正在存档 = false -if (fs.existsSync(`./data/${日期}.json`)) { - console.log(日期, '启动时读入数据..') - const file = fs.readFileSync(`./data/${日期}.json`, 'utf8') + +const 存档 = async (message = '存档中..') => { + if (正在存档) return + 正在存档 = true + console.log(message) + await fs.promises.writeFile(`./data/${当前日期}.json`, JSON.stringify({ + 作品: Object.fromEntries(作品), + 游戏: Object.fromEntries(游戏), + 截图: Object.fromEntries(截图), + }, null, 2)) + 正在存档 = false +} + + +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)) @@ -25,45 +41,39 @@ if (fs.existsSync(`./data/${日期}.json`)) { } -console.log(日期, '开始收集日志..') +console.log(当前日期, '开始收集日志..') +setInterval(() => 存档('10 分钟存档一次..'), 600000) - -tail.on('line', (line) => { +tail.on('line', async (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 (当前日期 !== 日志日期) { + await 存档('跨日期存档..') + 作品.clear() + 游戏.clear() + 截图.clear() + 当前日期 = 日志日期 } 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) ?? [] + const [uri, id, user_id] = item.request.uri.match(articleRegex) ?? [] 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) + if (item.request.headers.Referer[0].includes('/inspirationInfo/')) { + 游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1) + return + } + if (item.request.headers.Referer[0].includes('/game/')) { + 游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1) + return + } + console.log('未处理 Referer:', item.request.headers.Referer) } return } @@ -71,8 +81,7 @@ tail.on('line', (line) => { if (item.request.uri.includes('page=')) { return } - const reg = /\/api\/images\?similar=(\d+)/ - const [uri, id] = item.request.uri.match(reg) ?? [] + const [uri, id] = item.request.uri.match(imagesRegex) ?? [] if (uri && id) { 截图.set(id, 截图.has(id) ? 截图.get(id) + 1 : 1) } @@ -80,26 +89,11 @@ tail.on('line', (line) => { } }) -tail.on('end', () => { - console.log('没有更多内容,结束读取') - const sorted = new Map([...游戏.entries()].sort((a, b) => b[1] - a[1])) - const top10 = Array.from(sorted).slice(0, 10) - console.log(top10) -}) - -tail.on('error', (error) => { - console.error('错误:', error) -}) +tail.on('end', () => console.log('没有更多内容,结束读取')) +tail.on('error', (error) => console.error('错误:', error)) process.on('SIGINT', async () => { tail.unwatch() - - console.log('退出前储存数据..') - await fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({ - 作品: Object.fromEntries(作品), - 游戏: Object.fromEntries(游戏), - 截图: Object.fromEntries(截图), - }, null, 2)) - + await 存档('退出前储存数据..') process.exit() })