crabka-broker 0.3.6

Single-node Apache Kafka-compatible broker (MVP)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

//! Server-side assignors (KIP-848). Each implementation maps a set of
//! members + subscriptions + topic metadata to per-member partition
//! assignments.

pub mod range;
pub mod uniform;

pub use range::RangeAssignor;
pub use uniform::UniformAssignor;

use std::collections::HashMap;

use crabka_protocol::primitives::uuid::Uuid;

#[derive(Debug, Clone)]
pub struct MemberSubscription {
    pub member_id: String,
    pub rack_id: Option<String>,
    pub subscribed_topic_ids: Vec<Uuid>,
}

#[derive(Debug, Clone, Default)]
pub struct TopicMetadata {
    pub partitions_per_topic: HashMap<Uuid, i32>,
    /// Per-`(topic_id, partition_index)` set of racks on which the
    /// partition has at least one replica. Empty (or the key missing
    /// entirely) for partitions whose replicas have no rack info — the
    /// assignor then falls back to its non-rack-aware behavior.
    /// Populated by the coordinator's metadata snapshot.
    pub partition_racks: HashMap<(Uuid, i32), Vec<String>>,
}

pub type Assignment = HashMap<String, HashMap<Uuid, Vec<i32>>>;

pub trait Assignor: Send + Sync + std::fmt::Debug {
    fn name(&self) -> &'static str;
    fn assign(&self, members: &[MemberSubscription], topics: &TopicMetadata) -> Assignment;
}