diff --git a/update.js b/update.js index ee177d9..863a64f 100644 --- a/update.js +++ b/update.js @@ -8,9 +8,9 @@ export async function update(field, list) { console.log('创建临时表') await conn.query(`CREATE TEMPORARY TABLE temp_updates (id INT PRIMARY KEY, ${mysql.escapeId(field)} INT UNSIGNED)`) - console.log(`数据总量 ${list.length} 条, 每次 10000 条写入 ${Math.ceil(list.length/10000)} 次`) + console.log(`数据总量 ${list.length} 条, 每次 10000 条写入 ${Math.ceil(list.length / 10000)} 次`) for (let i = 0; i < list.length; i += 10000) { - console.log(`第 ${i/10000+1} 次写入`) + console.log(`第 ${i / 10000 + 1} 次写入`) const values = list.slice(i, i + 10000) await conn.query(`INSERT INTO temp_updates (id, ${field}) VALUES ?`, [values]) } @@ -20,3 +20,53 @@ export async function update(field, list) { await conn.end() console.log('更新完成') } + +export async function likes(day = 1) { + const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8')) + const conn = await mysql.createConnection(config) + const [rows] = await conn.query(`SELECT * FROM web_praise WHERE create_time >= CURDATE() AND create_time < CURDATE() + INTERVAL ${day} DAY`) + await conn.end() + return rows +} + +export async function collects(day = 1) { + const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8')) + const conn = await mysql.createConnection(config) + const [rows] = await conn.query(`SELECT * FROM web_collect WHERE create_time >= CURDATE() AND create_time < CURDATE() + INTERVAL ${day} DAY`) + await conn.end() + return rows +} + +export async function views(day = 1) { + const startDate = new Date(Date.now() - day * 24 * 60 * 60 * 1000) + const currentDate = new Date() + + const files = fs.readdirSync('./data').filter(name => /\d{4}-\d{2}-\d{2}\.json$/.test(name)).filter(name => { + const date = new Date(name.match(/\d{4}-\d{2}-\d{2}/)[0]) + return date >= startDate && date <= currentDate + }) + + const result = { articles: new Map(), games: new Map(), screenshots: new Map() } + + files.forEach(name => { + const data = JSON.parse(fs.readFileSync(`./data/${name}`, 'utf8')) + for (const type of ['articles', 'games', 'screenshots']) { + Object.entries(data[type] || {}).forEach(([key, value]) => { + result[type].set(key, (result[type].get(key) || 0) + value) + }) + } + }) + + return Object.fromEntries( + Object.entries(result).map(([key, map]) => [ + key, + [...map.entries()].sort((a, b) => b[1] - a[1]).map(([k]) => k) + ]) + ) +} + +// 计算总权重 +export function totalWeight(data) { + const total = Object.values(data).reduce((acc, cur) => acc + cur, 0) + return Object.fromEntries(Object.entries(data).map(([key, value]) => [key, Math.round(value / total * 100)])) +}