Skip to content

Commit

Permalink
Fix: 空の辞書でクラッシュするバグを修正 (#733)
Browse files Browse the repository at this point in the history
* Fix: 空の辞書でクラッシュするバグを修正

* Fix: blocking APIの方も修正

* Fix: コミット忘れ

* Change: テキストを変更

* Add: コメントを追加

* Change: use_user_dictに統合

* Delete: is_emptyを削除

* Refactor: 共通化

* コメント変更、load_with_userdic、変数の定義を元commitに戻す

---------

Co-authored-by: Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>
  • Loading branch information
sevenc-nanashi and Hiroshiba authored Feb 6, 2024
1 parent b9c953c commit 8f4dcce
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions crates/voicevox_core/src/engine/open_jtalk.rs
Original file line number Diff line number Diff line change
Expand Up @@ -128,9 +128,12 @@ pub(crate) mod blocking {
}

impl Inner {
// FIXME: 中断可能にする
// TODO: 中断可能にする
pub(super) fn use_user_dict(&self, words: &str) -> crate::result::Result<()> {
let result = {
// 空の辞書を読み込もうとするとクラッシュするのでユーザー辞書なしでロード
if words.is_empty() {
self.load_with_userdic(None)
} else {
// ユーザー辞書用のcsvを作成
let mut temp_csv =
NamedTempFile::new().map_err(|e| ErrorRepr::UseUserDict(e.into()))?;
Expand Down Expand Up @@ -158,17 +161,17 @@ pub(crate) mod blocking {
"-q",
]);

let Resources { mecab, .. } = &mut *self.resources.lock().unwrap();
self.load_with_userdic(Some(temp_dict_path.as_ref()))
}
}
fn load_with_userdic(&self, dict_path: Option<&Path>) -> crate::result::Result<()> {
let Resources { mecab, .. } = &mut *self.resources.lock().unwrap();

mecab.load_with_userdic(self.dict_dir.as_ref(), Some(Path::new(&temp_dict_path)))
};
let result = mecab.load_with_userdic(self.dict_dir.as_ref(), dict_path);

if !result {
return Err(
ErrorRepr::UseUserDict(anyhow!("辞書のコンパイルに失敗しました")).into(),
);
return Err(ErrorRepr::UseUserDict(anyhow!("辞書を読み込めませんでした。")).into());
}

Ok(())
}
}
Expand Down

0 comments on commit 8f4dcce

Please sign in to comment.