Skip to content

Commit 19db397

Browse files
authored
feat: add clear() method (#17)
1 parent 6a98939 commit 19db397

1 file changed

Lines changed: 12 additions & 3 deletions

File tree

src/lib.rs

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ where
139139
/// engine.insert_tokens("Arya Stark", &["Arya Stark", "a fictional
140140
/// character in American author George R. R", "portrayed by English actress."]);
141141
pub fn insert_tokens(&mut self, id: Id, tokens: &[&str]) {
142-
self.delete(&id);
142+
self.remove(&id);
143143

144144
let id_num = self.id_num_counter;
145145
self.ids_map.insert(id.clone(), id_num);
@@ -272,8 +272,8 @@ where
272272
result_ids
273273
}
274274

275-
/// Deletes entry by id.
276-
pub fn delete(&mut self, id: &Id) {
275+
/// Remove an entry by id.
276+
pub fn remove(&mut self, id: &Id) {
277277
if let Some(id_num) = self.ids_map.get(id) {
278278
for token in &self.forward_map[id_num] {
279279
self.reverse_map
@@ -287,6 +287,15 @@ where
287287
};
288288
}
289289

290+
/// Clear all entries.
291+
pub fn clear(&mut self) {
292+
self.forward_map.clear();
293+
self.reverse_map.clear();
294+
self.ids_map.clear();
295+
self.reverse_ids_map.clear();
296+
self.id_num_counter = 0;
297+
}
298+
290299
fn tokenize(&self, tokens: &[&str]) -> Vec<String> {
291300
let tokens: Vec<String> = tokens
292301
.iter()

0 commit comments

Comments
 (0)