flare_dht/metadata/
mod.rs

1#[cfg(all(feature = "raft", feature = "rkyv"))]
2pub mod raft;
3
4use crate::error::FlareError;
5use crate::proto::{
6    ClusterMetadata, CreateCollectionRequest, CreateCollectionResponse,
7    JoinRequest, JoinResponse,
8};
9use crate::shard::ShardMetadata;
10use crate::NodeId;
11use std::u32;
12
13#[async_trait::async_trait]
14pub trait MetadataManager: Send + Sync {
15    async fn initialize(&mut self) -> Result<(), FlareError>;
16    async fn get_shard_ids(
17        &self,
18        col_name: &str,
19    ) -> Option<Vec<ShardGroupState>>;
20    async fn get_shard_id(
21        &self,
22        col_name: &str,
23        key: &[u8],
24    ) -> Option<ShardGroupState>;
25    async fn leave(&self);
26    async fn other_leave(&self, node_id: NodeId) -> Result<(), FlareError>;
27    async fn other_join(
28        &self,
29        join_request: JoinRequest,
30    ) -> Result<JoinResponse, FlareError>;
31    async fn get_metadata(&self) -> Result<ClusterMetadata, FlareError>;
32
33    async fn local_shards(&self) -> Vec<ShardMetadata>;
34    async fn create_collection(
35        &self,
36        mut request: CreateCollectionRequest,
37    ) -> Result<CreateCollectionResponse, FlareError>;
38
39    fn create_watch(&self) -> tokio::sync::watch::Receiver<u64>;
40}
41
42#[cfg_attr(
43    feature = "rkyv",
44    derive(rkyv::Archive, rkyv::Serialize, rkyv::Deserialize),
45    rkyv(compare(PartialEq), derive(Debug))
46)]
47#[derive(serde::Serialize, serde::Deserialize, Debug, Default, Clone)]
48pub struct CollectionMetadataState {
49    pub name: String,
50    pub shards: Vec<ShardGroupState>,
51    pub seed: u32,
52    pub replication: u8,
53}
54
55#[cfg_attr(
56    feature = "rkyv",
57    derive(rkyv::Archive, rkyv::Serialize, rkyv::Deserialize),
58    rkyv(compare(PartialEq), derive(Debug))
59)]
60#[derive(serde::Serialize, serde::Deserialize, Debug, Default, Clone)]
61pub struct ShardGroupState {
62    pub shard_ids: Vec<u64>,
63}
flare_dht/metadata/mod.rs

flare_dht/metadata/
mod.rs
