From d7aee50cc5e50d08aa5f48b268bda0686a760bd2 Mon Sep 17 00:00:00 2001 From: BTMuli Date: Thu, 5 Mar 2026 00:32:47 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20=E7=B2=BE=E7=AE=80=E4=BB=A3?= =?UTF-8?q?=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/Sqlite/modules/userGacha.ts | 239 ++++++++++-------------- 1 file changed, 97 insertions(+), 142 deletions(-) diff --git a/src/plugins/Sqlite/modules/userGacha.ts b/src/plugins/Sqlite/modules/userGacha.ts index 8dcc09e3..d0e5993f 100644 --- a/src/plugins/Sqlite/modules/userGacha.ts +++ b/src/plugins/Sqlite/modules/userGacha.ts @@ -6,10 +6,95 @@ import showDialog from "@comp/func/dialog.js"; import showLoading from "@comp/func/loading.js"; import showSnackbar from "@comp/func/snackbar.js"; +import Database from "@tauri-apps/plugin-sql"; import { getUtc8Time, getWikiBrief, timestampToDate } from "@utils/toolFunc.js"; +import { ref, Ref } from "vue"; import TGSqlite from "../index.js"; +/** + * 批量插入祈愿数据 + * @since Beta v0.9.8 + * @param db - 数据库 + * @param uid - 用户UID + * @param data - 祈愿数据 + * @param size - batchSize + * @param cnt - cntRef + */ +async function insertGachaList( + db: Database, + uid: string, + data: Array, + size: number, + cnt: Ref, +): Promise { + await db.execute("PRAGMA busy_timeout = 5000;"); + for (let i = 0; i < data.length; i += size) { + await db.execute("BEGIN IMMEDIATE;"); + try { + const batch = data.slice(i, i + size); + let batchSql = ""; + const batchParams = []; + for (const item of batch) { + const updateTime = timestampToDate(Date.now()); + batchSql += ` + INSERT INTO GachaRecords(uid, gachaType, itemId, count, time, + name, type, rank, id, uigfType, updated) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) + ON CONFLICT(id) + DO UPDATE + SET uid=?, + gachaType=?, + itemId=?, + count=?, + time=?, + name=?, + type=?, + rank=?, + uigfType=?, + updated=?; + `; + batchParams.push( + uid, + item.gacha_type, + item.item_id ?? null, + item.count ?? 1, + item.time, + item.name, + item.item_type ?? null, + item.rank_type ?? null, + item.id, + item.uigf_gacha_type, + updateTime, + // update 部分 + uid, + item.gacha_type, + item.item_id ?? null, + item.count ?? 1, + item.time, + item.name, + item.item_type ?? null, + item.rank_type ?? null, + item.uigf_gacha_type, + updateTime, + ); + cnt.value++; + } + await db.execute(batchSql, batchParams); + await db.execute("COMMIT;"); + } catch (e) { + const msg = String(e); + if (/BUSY|LOCKED|SQLITE_BUSY|SQLITE_LOCKED/i.test(msg)) { + await showDialog.check(`数据库锁定`, `请刷新页面(F5)后重试操作`); + return; + } + // 其他错误直接抛出 + await db.execute("ROLLBACK;"); + throw e; + } + } +} + /** * 转换祈愿数据，防止多语言 * @since Beta v0.9.5 @@ -213,86 +298,21 @@ async function mergeUIGF( ): Promise { const db = await TGSqlite.getDB(); const len = data.length; - let cnt = 0; + const cnt = ref(0); let timer: NodeJS.Timeout | null = null; if (showProgress) { timer = setInterval(async () => { - const progress = Math.round((cnt / len) * 100 * 100) / 100; - const current = data[cnt]?.time ?? ""; - const name = data[cnt]?.name ?? ""; - const rank = data[cnt]?.rank_type ?? "0"; + const progress = Math.round((cnt.value / len) * 100 * 100) / 100; + const current = data[cnt.value]?.time ?? ""; + const name = data[cnt.value]?.name ?? ""; + const rank = data[cnt.value]?.rank_type ?? "0"; await showLoading.update(`[${progress}%][${current}] ${"⭐".repeat(Number(rank))}-${name}`, { timeout: 0, }); }, 1000); } const transformed = data.map((g) => transGacha(g)); - const BATCH_SIZE = 100; - await db.execute("PRAGMA busy_timeout = 5000;"); - for (let i = 0; i < transformed.length; i += BATCH_SIZE) { - await db.execute("BEGIN IMMEDIATE;"); - try { - const batch = transformed.slice(i, i + BATCH_SIZE); - let batchSql = ""; - const batchParams = []; - for (const item of batch) { - const updateTime = timestampToDate(Date.now()); - batchSql += ` - INSERT INTO GachaRecords(uid, gachaType, itemId, count, time, - name, type, rank, id, uigfType, updated) - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) - ON CONFLICT(id) - DO UPDATE - SET uid=?, - gachaType=?, - itemId=?, - count=?, - time=?, - name=?, - type=?, - rank=?, - uigfType=?, - updated=?; - `; - batchParams.push( - uid, - item.gacha_type, - item.item_id ?? null, - item.count ?? 1, - item.time, - item.name, - item.item_type ?? null, - item.rank_type ?? null, - item.id, - item.uigf_gacha_type, - updateTime, - // update 部分 - uid, - item.gacha_type, - item.item_id ?? null, - item.count ?? 1, - item.time, - item.name, - item.item_type ?? null, - item.rank_type ?? null, - item.uigf_gacha_type, - updateTime, - ); - cnt++; - } - await db.execute(batchSql, batchParams); - await db.execute("COMMIT;"); - } catch (e) { - const msg = String(e); - if (/BUSY|LOCKED|SQLITE_BUSY|SQLITE_LOCKED/i.test(msg)) { - await showDialog.check(`数据库锁定`, `请刷新页面(F5)后重试操作`); - return; - } - // 其他错误直接抛出 - await db.execute("ROLLBACK;"); - throw e; - } - } + await insertGachaList(db, uid, transformed, 100, cnt); if (timer) { clearInterval(timer); await showLoading.update(`[100%] 完成`, { timeout: 0 }); @@ -312,14 +332,14 @@ async function mergeUIGF4( ): Promise { const db = await TGSqlite.getDB(); const len = data.list.length; - let cnt: number = 0; + const cnt = ref(0); let timer: NodeJS.Timeout | null = null; if (showProgress) { timer = setInterval(async () => { - const progress = Math.round((cnt / len) * 100 * 100) / 100; - const current = data.list[cnt]?.time ?? ""; - const name = data.list[cnt]?.name ?? data.list[cnt]?.item_id; - const rank = data.list[cnt]?.rank_type ?? "0"; + const progress = Math.round((cnt.value / len) * 100 * 100) / 100; + const current = data.list[cnt.value]?.time ?? ""; + const name = data.list[cnt.value]?.name ?? data.list[cnt.value]?.item_id; + const rank = data.list[cnt.value]?.rank_type ?? "0"; await showLoading.update(`[${progress}%][${current}] ${"⭐".repeat(Number(rank))}-${name}`, { timeout: 0, }); @@ -327,72 +347,7 @@ async function mergeUIGF4( } const uid = data.uid.toString(); const transformed = data.list.map((g) => transGacha(g, data.timezone)); - const BATCH_SIZE = 100; - await db.execute("PRAGMA busy_timeout = 5000;"); - for (let i = 0; i < transformed.length; i += BATCH_SIZE) { - await db.execute("BEGIN IMMEDIATE;"); - try { - const batch = transformed.slice(i, i + BATCH_SIZE); - let batchSql = ""; - const batchParams = []; - for (const item of batch) { - const updateTime = timestampToDate(Date.now()); - batchSql += ` - INSERT INTO GachaRecords(uid, gachaType, itemId, count, time, - name, type, rank, id, uigfType, updated) - VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) - ON CONFLICT(id) - DO UPDATE - SET uid=?, - gachaType=?, - itemId=?, - count=?, - time=?, - name=?, - type=?, - rank=?, - uigfType=?, - updated=?; - `; - batchParams.push( - uid, - item.gacha_type, - item.item_id ?? null, - item.count ?? 1, - item.time, - item.name, - item.item_type ?? null, - item.rank_type ?? null, - item.id, - item.uigf_gacha_type, - updateTime, - // update 部分 - uid, - item.gacha_type, - item.item_id ?? null, - item.count ?? 1, - item.time, - item.name, - item.item_type ?? null, - item.rank_type ?? null, - item.uigf_gacha_type, - updateTime, - ); - cnt++; - } - await db.execute(batchSql, batchParams); - await db.execute("COMMIT;"); - } catch (e) { - const msg = String(e); - if (/BUSY|LOCKED|SQLITE_BUSY|SQLITE_LOCKED/i.test(msg)) { - await showDialog.check(`数据库锁定`, `请刷新页面(F5)后重试操作`); - return; - } - // 其他错误直接抛出 - await db.execute("ROLLBACK;"); - throw e; - } - } + await insertGachaList(db, uid, transformed, 100, cnt); if (timer) { clearInterval(timer); await showLoading.update(`[100%] 完成`, { timeout: 0 });