reset PredictDb::value_trie_ after Build

rime · Nov 1, 2023 · 1d4b828 · 1d4b828
1 parent 0b512a5
commit 1d4b828
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/predict_db.cc b/src/predict_db.cc
@@ -51,8 +51,8 @@ bool PredictDb::Load() {
   }
   DLOG(INFO) << "found string table of size " << metadata_->value_trie.get()
              << ".";
-  value_trie_ = std::make_unique<StringTable>(metadata_->value_trie.get(),
-                                              metadata_->value_trie_size);
+  value_trie_ = make_unique<StringTable>(metadata_->value_trie.get(),
+                                         metadata_->value_trie_size);
 
   return true;
 }
@@ -166,6 +166,8 @@ bool PredictDb::Build(const predict::RawData& data) {
   string_table.Dump(value_trie_image, value_trie_image_size);
   metadata_->value_trie = value_trie_image;
   metadata_->value_trie_size = value_trie_image_size;
+  value_trie_ = make_unique<StringTable>(value_trie_image,
+                                         value_trie_image_size);
   // at last, complete the metadata
   std::strncpy(metadata_->format, kPredictFormat.c_str(),
                kPredictFormat.length());