risingwave_common/session_config/
non_zero64.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::num::{NonZeroU64, ParseIntError};
16use std::str::FromStr;
17
18/// When set this config as `0`, the value is `None`, otherwise the value is
19/// `Some(val)`
20#[derive(Copy, Default, Debug, Clone, PartialEq, Eq)]
21pub struct ConfigNonZeroU64(pub Option<NonZeroU64>);
22
23impl FromStr for ConfigNonZeroU64 {
24    type Err = ParseIntError;
25
26    fn from_str(s: &str) -> Result<Self, Self::Err> {
27        let parsed = s.parse::<u64>()?;
28        if parsed == 0 {
29            Ok(Self(None))
30        } else {
31            Ok(Self(NonZeroU64::new(parsed)))
32        }
33    }
34}
35
36impl std::fmt::Display for ConfigNonZeroU64 {
37    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
38        if let ConfigNonZeroU64(Some(inner)) = self {
39            write!(f, "{}", inner)
40        } else {
41            write!(f, "0")
42        }
43    }
44}
45
46impl ConfigNonZeroU64 {
47    pub fn map<U, F>(self, f: F) -> Option<U>
48    where
49        F: FnOnce(NonZeroU64) -> U,
50    {
51        self.0.map(f)
52    }
53}
risingwave_common/session_config/non_zero64.rs

risingwave_common/session_config/
non_zero64.rs
