snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-23 10:37:41 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

squash-me: Add set/get/del_document_attribute to Index methods

Browse Source
This commit is contained in:
Clément Renault 2019-04-12 18:46:36 +02:00
parent 58c020a2e1
commit 7338e522bd
No known key found for this signature in database
GPG Key ID: 0151CDAB43460DAE
1 changed files with 45 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
meilidb-data/src/database.rs
View File

@ -1,12 +1,12 @@
use std::sync::Arc;
use std::path::Path;
use meilidb_core::Index as WordIndex;
use meilidb_core::{DocumentId, Index as WordIndex};
use meilidb_core::shared_data_cursor::{FromSharedDataCursor, SharedDataCursor};
use meilidb_core::write_to_bytes::WriteToBytes;
use sled::IVec;
use crate::Schema;
use crate::{Schema, SchemaAttr};
#[derive(Debug)]
pub enum Error {
@ -32,6 +32,17 @@ fn index_name(name: &str) -> Vec<u8> {
format!("index-{}", name).into_bytes()
}
fn document_key(id: DocumentId, attr: SchemaAttr) -> Vec<u8> {
let DocumentId(document_id) = id;
let SchemaAttr(schema_attr) = attr;
let mut bytes = Vec::new();
bytes.extend_from_slice(b"document-");
bytes.extend_from_slice(&document_id.to_be_bytes()[..]);
bytes.extend_from_slice(&schema_attr.to_be_bytes()[..]);
bytes
}
fn ivec_into_arc(ivec: IVec) -> Arc<[u8]> {
match ivec {
IVec::Inline(len, bytes) => Arc::from(&bytes[..len as usize]),
@ -123,4 +134,36 @@ impl Index {
pub fn word_index(&self) -> &WordIndex {
&self.word_index
}
pub fn set_document_attribute<V>(
&self,
id: DocumentId,
attr: SchemaAttr,
value: V,
) -> Result<Option<IVec>, Error>
where IVec: From<V>,
{
let key = document_key(id, attr);
Ok(self.inner.set(key, value)?)
}
pub fn get_document_attribute(
&self,
id: DocumentId,
attr: SchemaAttr
) -> Result<Option<IVec>, Error>
{
let key = document_key(id, attr);
Ok(self.inner.get(key)?)
}
pub fn del_document_attribute(
&self,
id: DocumentId,
attr: SchemaAttr
) -> Result<Option<IVec>, Error>
{
let key = document_key(id, attr);
Ok(self.inner.del(key)?)
}
}