reload after db upgrade logic

This commit is contained in:
polv committed 2026-03-07 09:03:11 +07:00
1 parent afeb43fc2c
commit a38a28a836
3 files changed
+33 -19

No files matched your search

+22 -6
View File
@@ -176,7 +176,7 @@ impl YomitanWriter {
self.create_materialized_views()?; self.create_materialized_views()?;
self.attach_glossary()?; self.attach_glossary()?;
let schema_version = "2"; let schema_version = "2026-03-07";
let current_schema_version = { let current_schema_version = {
let mut stmt = self let mut stmt = self
@@ -193,7 +193,7 @@ impl YomitanWriter {
.query_row("SELECT MAX(id) FROM glossaries", [], |r| r.get::<_, i64>(0))? .query_row("SELECT MAX(id) FROM glossaries", [], |r| r.get::<_, i64>(0))?
as usize; as usize;
let message = &format!("migrate to {}", YOMITAN_GLOSSARY_DBFILE); let message = &format!("Migrate to {}", YOMITAN_GLOSSARY_DBFILE);
progress_callback(YomitanProgress { progress_callback(YomitanProgress {
message: message.to_string(), message: message.to_string(),
current: 0, current: 0,
@@ -232,15 +232,29 @@ impl YomitanWriter {
message: message.to_string(), message: message.to_string(),
current: i, current: i,
total, total,
steps: 0, steps: i,
}); });
} }
} }
} }
tx.commit()?; tx.commit()?;
progress_callback(YomitanProgress {
message: message.to_string(),
current: total,
total,
steps: total,
});
self.hash_glossary()?; self.hash_glossary()?;
progress_callback(YomitanProgress {
message: "Cleaning up old glossary".to_string(),
current: 0,
total: 0,
steps: total,
});
let _timer = Timer::new(format!("cleanup for {}", YOMITAN_GLOSSARY_DBFILE)); let _timer = Timer::new(format!("cleanup for {}", YOMITAN_GLOSSARY_DBFILE));
self.conn.execute_batch( self.conn.execute_batch(
" "
@@ -250,9 +264,11 @@ impl YomitanWriter {
)?; )?;
} }
self.conn if v < schema_version.to_string() {
.prepare("UPDATE schema_meta SET value = ?2 WHERE key = ?1")? self.conn
.execute(["schema_version", schema_version])?; .prepare("UPDATE schema_meta SET value = ?2 WHERE key = ?1")?
.execute(["schema_version", schema_version])?;
}
} else { } else {
self.conn self.conn
.prepare("INSERT INTO schema_meta (key, value) VALUES (?1, ?2)")? .prepare("INSERT INTO schema_meta (key, value) VALUES (?1, ?2)")?
+4 -2
View File
@@ -34,6 +34,7 @@ interface Progress {
message: string; message: string;
current?: number; current?: number;
total?: number; total?: number;
unit?: string;
} }
function LoadingDialog() { function LoadingDialog() {
@@ -139,9 +140,10 @@ function LoadingDialog() {
listen<DownloadProgress>("download-url-progress", ({ payload }) => { listen<DownloadProgress>("download-url-progress", ({ payload }) => {
updateProgress({ updateProgress({
message: `Downloading ${payload.filepath} (MiB)`, message: `Downloading ${payload.filepath}`,
current: payload.downloaded >> 20, current: payload.downloaded >> 20,
total: payload.content_length >> 20, total: payload.content_length >> 20,
unit: "MiB",
}); });
}); });
}, []); }, []);
@@ -175,7 +177,7 @@ function LoadingDialog() {
? messages.map((m, i) => ( ? messages.map((m, i) => (
<p key={i}> <p key={i}>
{m.total {m.total
? `${m.message} (${m.current}/${m.total})` ? `${m.message} (${(m.current || 0).toLocaleString()}/${m.total.toLocaleString()}${m.unit ? ` ${m.unit}` : ""})`
: m.message} : m.message}
</p> </p>
)) ))
+7 -11
View File
@@ -1,4 +1,5 @@
use std::{ use std::{
cell::Cell,
fs::{File, remove_file}, fs::{File, remove_file},
path::PathBuf, path::PathBuf,
}; };
@@ -25,9 +26,6 @@ pub async fn init_yomitan(
lang: String, lang: String,
state: tauri::State<'_, AppService>, state: tauri::State<'_, AppService>,
) -> Result<(), CJDicError> { ) -> Result<(), CJDicError> {
let mut new_dicts: Vec<String> = vec![];
let mut to_be_removed_dicts: Vec<String> = vec![];
let app_dir = app let app_dir = app
.path() .path()
.app_data_dir() .app_data_dir()
@@ -56,25 +54,23 @@ pub async fn init_yomitan(
.map(|d| app_dir.join(&d.filepath)) .map(|d| app_dir.join(&d.filepath))
.collect(); .collect();
let mut r = state.load_yomitan_zip_dir( let update_count = Cell::new(0usize);
state.load_yomitan_zip_dir(
zip_files, zip_files,
lang.as_str(), lang.as_str(),
|r| { |r| {
update_count.set(update_count.get() + r.new_dicts.len() + r.to_be_removed_dicts.len());
app.emit("load-yomitan-dir", r).unwrap(); app.emit("load-yomitan-dir", r).unwrap();
}, },
|r| { |r| {
update_count.set(update_count.get() + 1);
app.emit("yomitan-import-progress", r).unwrap(); app.emit("yomitan-import-progress", r).unwrap();
}, },
)?; )?;
new_dicts.append(&mut r.new_dicts); if update_count.get() > 0 {
to_be_removed_dicts.append(&mut r.to_be_removed_dicts);
if new_dicts.len() + to_be_removed_dicts.len() > 0 {
println!("Imported {:?}", new_dicts);
println!("Removed {:?}", to_be_removed_dicts);
println!("Request app restart?",); println!("Request app restart?",);
// Request app restart anyway // Request app restart anyway
app.request_restart(); app.request_restart();
} }