risingwave_sqlparser/parser_v2/
number.rs

1// Licensed under the Apache License, Version 2.0 (the "License");
2// you may not use this file except in compliance with the License.
3// You may obtain a copy of the License at
4//
5//     http://www.apache.org/licenses/LICENSE-2.0
6//
7// Unless required by applicable law or agreed to in writing, software
8// distributed under the License is distributed on an "AS IS" BASIS,
9// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10// See the License for the specific language governing permissions and
11// limitations under the License.
12
13use core::ops::RangeBounds;
14
15use winnow::combinator::{cut_err, delimited};
16use winnow::error::{ContextError, StrContext};
17use winnow::{ModalParser, ModalResult, Parser};
18
19use super::{TokenStream, token};
20use crate::tokenizer::Token;
21
22/// Consume a [number][Token::Number] from token.
23pub fn token_number<S>(input: &mut S) -> ModalResult<String>
24where
25    S: TokenStream,
26{
27    token
28        .verify_map(|t| {
29            if let Token::Number(number) = t.token {
30                Some(number)
31            } else {
32                None
33            }
34        })
35        .context(StrContext::Label("number"))
36        .parse_next(input)
37}
38
39/// Consume an unsigned literal integer/long
40pub fn literal_uint<S>(input: &mut S) -> ModalResult<u64>
41where
42    S: TokenStream,
43{
44    token_number
45        .try_map(|s| s.parse::<u64>())
46        .context(StrContext::Label("u64"))
47        .parse_next(input)
48}
49
50/// Consume an unsigned literal integer
51pub fn literal_u32<S>(input: &mut S) -> ModalResult<u32>
52where
53    S: TokenStream,
54{
55    token_number
56        .try_map(|s| s.parse::<u32>())
57        .context(StrContext::Label("u32"))
58        .parse_next(input)
59}
60
61/// Consume an literal integer
62pub fn literal_i64<S>(input: &mut S) -> ModalResult<i64>
63where
64    S: TokenStream,
65{
66    token_number
67        .try_map(|s| s.parse::<i64>())
68        .context(StrContext::Label("i64"))
69        .parse_next(input)
70}
71
72/// Consume a precision definition in some types, e.g. `FLOAT(32)`.
73///
74/// The precision must be in the given range.
75pub fn precision_in_range<S>(
76    range: impl RangeBounds<u64> + std::fmt::Debug,
77) -> impl ModalParser<S, u64, ContextError>
78where
79    S: TokenStream,
80{
81    #[derive(Debug, thiserror::Error)]
82    #[error("precision must be in range {0}")]
83    struct OutOfRange(String);
84
85    delimited(
86        Token::LParen,
87        cut_err(literal_uint.try_map(move |v| {
88            if range.contains(&v) {
89                Ok(v)
90            } else {
91                Err(OutOfRange(format!("{:?}", range)))
92            }
93        })),
94        cut_err(Token::RParen),
95    )
96}
risingwave_sqlparser/parser_v2/number.rs

risingwave_sqlparser/parser_v2/
number.rs
