修正数据写入

README.md

@@ -7,8 +7,17 @@
 导出接口
 
 ```bash
+# 安装依赖
 npm install --registry=https://registry.npmmirror.com
-
 node main.js
+
+# 连接数据库
+ssh -NCPf ai -L 3306:localhost:3306
 ```
 
+```sql
+-- 创建字段
+ALTER TABLE web_images ADD COLUMN day_rank INT UNSIGNED;
+ALTER TABLE web_images ADD COLUMN week_rank INT UNSIGNED;
+ALTER TABLE web_images ADD COLUMN month_rank INT UNSIGNED;
+```

main.js

@@ -50,7 +50,7 @@ export default function () {
     // 定时存档（每10分钟）
     setInterval(() => {
 	    archive('每10分钟自动存档...')
-	    update('day_rank', Object.fromEntries(screenshots)) // 更新数据库中 day_rank 字段
+	    update('day_rank', Object.entries(Object.fromEntries(screenshots))) // 更新数据库中 day_rank 字段
     }, 600000)
 
     // 实时读取日志文件

update.js

@@ -1,19 +1,22 @@
-import mysql from 'mysql2/promise'
 import fs from 'fs'
+import mysql from 'mysql2/promise'
 
-export async function update(list, field) {
+export async function update(field, list) {
   const { mysql: config } = JSON.parse(fs.readFileSync('config.json', 'utf8'))
   const conn = await mysql.createConnection(config)
 
-  await conn.query(`CREATE TEMPORARY TABLE temp_updates (id INT PRIMARY KEY, ${field} INT)`)
+  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)} 次`)
   for (let i = 0; i < list.length; i += 10000) {
-    const values = list.slice(i, i + 10000).map(({ id, ...rest }) => [id, rest[field]])
+    console.log(`第 ${i/10000+1} 次写入`)
+    const values = list.slice(i, i + 10000)
     await conn.query(`INSERT INTO temp_updates (id, ${field}) VALUES ?`, [values])
   }
 
+  console.log('更新数据从临时表转入生产库')
   await conn.query(`UPDATE web_images t JOIN temp_updates tu ON t.id = tu.id SET t.${field} = tu.${field}`)
-  console.log('更新完成')
   await conn.end()
+  console.log('更新完成')
 }
-