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

View File

@@ -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)]))
}