2019-10-18 13:05:28 +02:00
|
|
|
use super::BEU64;
|
|
|
|
use crate::DocumentId;
|
2019-10-03 16:13:09 +02:00
|
|
|
use std::sync::Arc;
|
2019-10-18 13:05:28 +02:00
|
|
|
use zlmdb::types::{ByteSlice, OwnedType};
|
2019-10-16 17:05:24 +02:00
|
|
|
use zlmdb::Result as ZResult;
|
2019-10-03 15:04:11 +02:00
|
|
|
|
|
|
|
#[derive(Copy, Clone)]
|
|
|
|
pub struct DocsWords {
|
2019-10-16 17:05:24 +02:00
|
|
|
pub(crate) docs_words: zlmdb::Database<OwnedType<BEU64>, ByteSlice>,
|
2019-10-03 15:04:11 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
impl DocsWords {
|
|
|
|
pub fn put_doc_words(
|
|
|
|
&self,
|
2019-10-16 17:05:24 +02:00
|
|
|
writer: &mut zlmdb::RwTxn,
|
2019-10-03 15:04:11 +02:00
|
|
|
document_id: DocumentId,
|
|
|
|
words: &fst::Set,
|
2019-10-18 13:05:28 +02:00
|
|
|
) -> ZResult<()> {
|
2019-10-16 17:05:24 +02:00
|
|
|
let document_id = BEU64::new(document_id.0);
|
2019-10-03 16:13:09 +02:00
|
|
|
let bytes = words.as_fst().as_bytes();
|
2019-10-16 17:05:24 +02:00
|
|
|
self.docs_words.put(writer, &document_id, bytes)
|
2019-10-03 15:04:11 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn del_doc_words(
|
|
|
|
&self,
|
2019-10-16 17:05:24 +02:00
|
|
|
writer: &mut zlmdb::RwTxn,
|
2019-10-03 15:04:11 +02:00
|
|
|
document_id: DocumentId,
|
2019-10-18 13:05:28 +02:00
|
|
|
) -> ZResult<bool> {
|
2019-10-16 17:05:24 +02:00
|
|
|
let document_id = BEU64::new(document_id.0);
|
|
|
|
self.docs_words.delete(writer, &document_id)
|
2019-10-03 15:04:11 +02:00
|
|
|
}
|
2019-10-03 16:13:09 +02:00
|
|
|
|
2019-10-16 17:05:24 +02:00
|
|
|
pub fn doc_words(
|
2019-10-03 16:13:09 +02:00
|
|
|
&self,
|
2019-10-16 17:05:24 +02:00
|
|
|
reader: &zlmdb::RoTxn,
|
2019-10-03 16:13:09 +02:00
|
|
|
document_id: DocumentId,
|
2019-10-18 13:05:28 +02:00
|
|
|
) -> ZResult<Option<fst::Set>> {
|
2019-10-16 17:05:24 +02:00
|
|
|
let document_id = BEU64::new(document_id.0);
|
|
|
|
match self.docs_words.get(reader, &document_id)? {
|
|
|
|
Some(bytes) => {
|
2019-10-03 16:13:09 +02:00
|
|
|
let len = bytes.len();
|
|
|
|
let bytes = Arc::from(bytes);
|
2019-10-16 17:05:24 +02:00
|
|
|
let fst = fst::raw::Fst::from_shared_bytes(bytes, 0, len).unwrap();
|
2019-10-03 16:13:09 +02:00
|
|
|
Ok(Some(fst::Set::from(fst)))
|
2019-10-18 13:05:28 +02:00
|
|
|
}
|
2019-10-03 16:13:09 +02:00
|
|
|
None => Ok(None),
|
|
|
|
}
|
|
|
|
}
|
2019-10-03 15:04:11 +02:00
|
|
|
}
|