63 lines
1.9 KiB
JavaScript
63 lines
1.9 KiB
JavaScript
import fs from 'fs'
|
|
import mysql from 'mysql2/promise'
|
|
|
|
// 更新指标到数据库
|
|
export async function update(table, rank, views) {
|
|
if (!['day_rank', 'week_rank', 'month_rank', 'year_rank', 'aeon_rank'].includes(rank)) {
|
|
throw new Error('字段类型错误')
|
|
}
|
|
|
|
const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8'))
|
|
const conn = await mysql.createConnection(config)
|
|
|
|
console.log('创建临时表')
|
|
await conn.query(`CREATE TEMPORARY TABLE temp_updates (id INT PRIMARY KEY, views_count INT UNSIGNED)`)
|
|
|
|
console.log(`数据总量 ${views.length} 条, 每次 10000 条写入 ${Math.ceil(views.length / 10000)} 次`)
|
|
for (let i = 0; i < views.length; i += 10000) {
|
|
console.log(`第 ${i / 10000 + 1} 次写入`)
|
|
const values = views.slice(i, i + 10000)
|
|
await conn.query(`INSERT INTO temp_updates (id, views_count) VALUES ?`, [values])
|
|
}
|
|
|
|
|
|
|
|
for (let name of ['comment', 'praise', 'collect']) {
|
|
console.log(`统计当天时间段内 ${name} 数到临时表`)
|
|
await conn.query(`
|
|
UPDATE temp_updates t
|
|
JOIN (
|
|
SELECT
|
|
${mame}_id AS article_id,
|
|
COUNT(*) AS ${name}_count
|
|
FROM
|
|
web_${name}
|
|
WHERE
|
|
DATE(create_time) = CURDATE()
|
|
GROUP BY
|
|
${field}_id
|
|
) stats
|
|
ON t.id = stats.article_id
|
|
SET t.${name}_count = stats.${name}_count;
|
|
`)
|
|
}
|
|
|
|
console.log('统计当天时间段内排行榜到临时表')
|
|
await conn.query(`
|
|
UPDATE temp_updates
|
|
SET day_rank =
|
|
collect_count * 35 +
|
|
praise_count * 20 +
|
|
comment_count * 20 +
|
|
views_count * 25;
|
|
`)
|
|
|
|
console.log('更新前清空当日记录')
|
|
await conn.query(`UPDATE ${table} SET ${rank} = 0`)
|
|
|
|
console.log('更新数据从临时表转入生产库')
|
|
await conn.query(`UPDATE ${table} t JOIN temp_updates tu ON t.id = tu.id SET t.${rank} = tu.${rank}`)
|
|
await conn.end()
|
|
console.log('更新完成')
|
|
}
|