qdrant_client/builders/
max_optimization_threads_builder.rs

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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
use crate::qdrant::*;

/// Max number of threads (jobs) for running optimizations per shard.
/// Each optimization job will also use `max_indexing_threads` threads by itself for index building.
///
/// - If `auto` - have no limit and choose dynamically to saturate CPU.
/// - If `disabled` or `0` - no optimization threads, optimizations will be disabled.
pub struct MaxOptimizationThreadsBuilder {
    pub(crate) inner: MaxOptimizationThreads,
}

impl MaxOptimizationThreadsBuilder {
    /// Use specific number of optimization threads.
    ///
    /// - If `0` - no optimization threads, optimizations will be disabled.
    #[allow(unused_mut)]
    #[inline]
    pub fn threads(threads: u64) -> Self {
        Self {
            inner: MaxOptimizationThreads::from(threads),
        }
    }

    /// No optimization threads, optimizations will be disabled.
    #[allow(unused_mut)]
    #[inline]
    pub fn disabled() -> Self {
        Self::threads(0)
    }

    /// Have no limit and choose dynamically to saturate CPU.
    #[allow(unused_mut)]
    #[inline]
    pub fn auto() -> Self {
        Self {
            inner: MaxOptimizationThreads::from(max_optimization_threads::Variant::Setting(
                max_optimization_threads::Setting::Auto as i32,
            )),
        }
    }
}

impl From<MaxOptimizationThreadsBuilder> for MaxOptimizationThreads {
    fn from(value: MaxOptimizationThreadsBuilder) -> Self {
        value.build()
    }
}

impl MaxOptimizationThreadsBuilder {
    pub fn build(self) -> MaxOptimizationThreads {
        self.inner
    }
}

impl Default for MaxOptimizationThreadsBuilder {
    fn default() -> Self {
        Self::auto()
    }
}

impl From<u64> for MaxOptimizationThreads {
    fn from(threads: u64) -> Self {
        MaxOptimizationThreads {
            variant: Some(max_optimization_threads::Variant::from(threads)),
        }
    }
}

impl From<max_optimization_threads::Variant> for MaxOptimizationThreads {
    fn from(setting: max_optimization_threads::Variant) -> Self {
        MaxOptimizationThreads {
            variant: Some(setting),
        }
    }
}

impl From<u64> for max_optimization_threads::Variant {
    fn from(threads: u64) -> Self {
        Self::Value(threads)
    }
}

impl From<max_optimization_threads::Setting> for max_optimization_threads::Variant {
    fn from(setting: max_optimization_threads::Setting) -> Self {
        Self::Setting(setting as i32)
    }
}