This commit is contained in:
散仙
2024-11-29 06:19:02 +08:00
parent 9b976ad402
commit f4a546c486

View File

@@ -8,9 +8,9 @@ export async function update(field, list) {
console.log('创建临时表') console.log('创建临时表')
await conn.query(`CREATE TEMPORARY TABLE temp_updates (id INT PRIMARY KEY, ${mysql.escapeId(field)} INT UNSIGNED)`) 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) { 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) const values = list.slice(i, i + 10000)
await conn.query(`INSERT INTO temp_updates (id, ${field}) VALUES ?`, [values]) await conn.query(`INSERT INTO temp_updates (id, ${field}) VALUES ?`, [values])
} }
@@ -20,3 +20,53 @@ export async function update(field, list) {
await conn.end() await conn.end()
console.log('更新完成') 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)]))
}