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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
use std::fmt::{self, Display, Formatter};
use super::cleanup_sst::{Runner as CleanupSSTRunner, Task as CleanupSSTTask};
use super::compact::{Runner as CompactRunner, Task as CompactTask};
use crate::store::StoreRouter;
use engine_traits::KvEngine;
use pd_client::PdClient;
use tikv_util::worker::Runnable;
pub enum Task {
Compact(CompactTask),
CleanupSST(CleanupSSTTask),
}
impl Display for Task {
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
match self {
Task::Compact(ref t) => t.fmt(f),
Task::CleanupSST(ref t) => t.fmt(f),
}
}
}
pub struct Runner<E, C, S>
where
E: KvEngine,
S: StoreRouter<E>,
{
compact: CompactRunner<E>,
cleanup_sst: CleanupSSTRunner<E, C, S>,
}
impl<E, C, S> Runner<E, C, S>
where
E: KvEngine,
C: PdClient,
S: StoreRouter<E>,
{
pub fn new(
compact: CompactRunner<E>,
cleanup_sst: CleanupSSTRunner<E, C, S>,
) -> Runner<E, C, S> {
Runner {
compact,
cleanup_sst,
}
}
}
impl<E, C, S> Runnable for Runner<E, C, S>
where
E: KvEngine,
C: PdClient,
S: StoreRouter<E>,
{
type Task = Task;
fn run(&mut self, task: Task) {
match task {
Task::Compact(t) => self.compact.run(t),
Task::CleanupSST(t) => self.cleanup_sst.run(t),
}
}
}