64 lines
2.1 KiB
Rust
64 lines
2.1 KiB
Rust
|
use log::{error, warn};
|
||
|
|
use lyphedb::{KVList, KeyDirectory, LDBNatsMessage, LypheDBCommand, PropagationStrategy};
|
||
|
|
use crate::ldb::{construct_path, hash, DBConn};
|
||
|
|
|
||
|
|
pub const WORDSTORE: &str = "wordstore";
|
||
|
|
|
||
|
|
pub async fn add_url_to_keywords(db: &DBConn, keywords: &[&str], url: &str) -> Result<(), ()> {
|
||
|
|
let mut key_sets = Vec::new();
|
||
|
|
for keyword in keywords {
|
||
|
|
let path = construct_path(&[WORDSTORE, keyword, &hash(url)]).as_bytes().to_vec();
|
||
|
|
let data = url.as_bytes().to_vec();
|
||
|
|
key_sets.push((path, data));
|
||
|
|
}
|
||
|
|
let cmd = LDBNatsMessage::Command(LypheDBCommand::SetKeys(KVList {
|
||
|
|
kvs: key_sets,
|
||
|
|
}, PropagationStrategy::OnRead));
|
||
|
|
|
||
|
|
match db.query(cmd).await {
|
||
|
|
LDBNatsMessage::Success => {
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
LDBNatsMessage::BadRequest => {
|
||
|
|
error!("bad request for add_url_to_keywords");
|
||
|
|
Err(())
|
||
|
|
}
|
||
|
|
LDBNatsMessage::NotFound => {
|
||
|
|
error!("not found for add_url_to_keywords");
|
||
|
|
Err(())
|
||
|
|
}
|
||
|
|
|
||
|
|
_ => {
|
||
|
|
warn!("lyphedb sent weird message as response, treating as error");
|
||
|
|
Err(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn get_urls_from_keyword(db: &DBConn, keyword: &str) -> Result<Vec<String>, ()> {
|
||
|
|
let path = construct_path(&[WORDSTORE, keyword]).as_bytes().to_vec();
|
||
|
|
let cmd = LDBNatsMessage::Command(LypheDBCommand::GetKeyDirectory(KeyDirectory {
|
||
|
|
key: path,
|
||
|
|
}));
|
||
|
|
|
||
|
|
match db.query(cmd).await {
|
||
|
|
LDBNatsMessage::Entries(kvs) => {
|
||
|
|
Ok(kvs.kvs.into_iter().map(|v| String::from_utf8_lossy(&v.1).to_string()).collect())
|
||
|
|
}
|
||
|
|
LDBNatsMessage::Success => {
|
||
|
|
warn!("lyphedb responded with \"success\" to get_urls_from_keywords, treating as error");
|
||
|
|
Err(())
|
||
|
|
}
|
||
|
|
LDBNatsMessage::BadRequest => {
|
||
|
|
warn!("bad request for get_urls_from_keywords");
|
||
|
|
Err(())
|
||
|
|
}
|
||
|
|
LDBNatsMessage::NotFound => {
|
||
|
|
Ok(vec![])
|
||
|
|
}
|
||
|
|
_ => {
|
||
|
|
warn!("lyphedb sent weird message as response, treating as error");
|
||
|
|
Err(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|