39 lines
967 B
Rust
39 lines
967 B
Rust
use crate::{
|
|
error::{Error, Result},
|
|
utils::Peer,
|
|
};
|
|
|
|
mod valkey;
|
|
|
|
pub use valkey::RegisterRequest;
|
|
|
|
pub enum Storage {
|
|
Valkey(valkey::ValkeyStorage),
|
|
}
|
|
|
|
pub trait StorageImpl {
|
|
async fn register_device(&self, request: &RegisterRequest) -> Result<()>;
|
|
async fn get_peers(&self) -> Result<Vec<Peer>>;
|
|
}
|
|
|
|
impl StorageImpl for Storage {
|
|
async fn register_device(&self, request: &RegisterRequest) -> Result<()> {
|
|
match self {
|
|
Self::Valkey(storage) => storage.register_device(request).await,
|
|
}
|
|
}
|
|
|
|
async fn get_peers(&self) -> Result<Vec<Peer>> {
|
|
match self {
|
|
Self::Valkey(storage) => storage.get_peers().await,
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn get_storage_from_env() -> Result<Storage> {
|
|
Ok(Storage::Valkey(valkey::ValkeyStorage {
|
|
valkey_client: redis::Client::open("redis://127.0.0.1:6379/")
|
|
.map_err(|e| Error::valkey_connect(e, "127.0.0.1:6379/".to_string()))?,
|
|
}))
|
|
}
|