risingwave_expr_impl/scalar/
int256.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 risingwave_common::types::Int256;
16use risingwave_expr::ExprError::Parse;
17use risingwave_expr::{Result, function};
18use thiserror_ext::AsReport;
19const MAX_AVAILABLE_HEX_STR_LEN: usize = 66;
20
21/// Returns the integer value of the hexadecimal string.
22///
23/// # Example
24///
25/// ```slt
26/// query I
27/// select hex_to_int256('0xdeadbeef');
28/// ----
29/// 3735928559
30/// ```
31#[function("hex_to_int256(varchar) -> int256")]
32pub fn hex_to_int256(s: &str) -> Result<Int256> {
33    Int256::from_str_hex(s).map_err(|e| {
34        Parse(
35            if s.len() <= MAX_AVAILABLE_HEX_STR_LEN {
36                format!("failed to parse hex '{}', {}", s, e.as_report())
37            } else {
38                format!(
39                    "failed to parse hex '{}...'(truncated, total {} bytes), {}",
40                    &s[..MAX_AVAILABLE_HEX_STR_LEN],
41                    s.len(),
42                    e.as_report()
43                )
44            }
45            .into(),
46        )
47    })
48}
49
50#[cfg(test)]
51mod tests {
52    use risingwave_common::types::Int256;
53    use risingwave_expr::ExprError::Parse;
54
55    use crate::scalar::int256::hex_to_int256;
56
57    #[test]
58    fn test_hex_to_int256() {
59        assert_eq!(hex_to_int256("0x0").unwrap(), Int256::from(0));
60        assert_eq!(hex_to_int256("0x0000").unwrap(), Int256::from(0));
61        assert_eq!(hex_to_int256("0x1").unwrap(), Int256::from(1));
62
63        assert_eq!(hex_to_int256("0xf").unwrap(), Int256::from(15));
64        assert_eq!(hex_to_int256("0xff").unwrap(), Int256::from(255));
65
66        assert_eq!(
67            hex_to_int256("0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff")
68                .unwrap(),
69            Int256::from(-1)
70        );
71        assert_eq!(
72            hex_to_int256("0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff01")
73                .unwrap(),
74            Int256::from(-255)
75        );
76
77        // int256 max
78        assert_eq!(
79            hex_to_int256("0x7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff")
80                .unwrap(),
81            Int256::max_value(),
82        );
83
84        // int256 min
85        assert_eq!(
86            hex_to_int256("0x8000000000000000000000000000000000000000000000000000000000000000")
87                .unwrap(),
88            Int256::min_value(),
89        );
90    }
91
92    #[test]
93    fn test_failed() {
94        let failed_result = hex_to_int256("0xggggggg");
95        assert!(failed_result.is_err());
96        assert!(matches!(failed_result.as_ref().err(), Some(Parse(_))));
97    }
98}
risingwave_expr_impl/scalar/int256.rs

risingwave_expr_impl/scalar/
int256.rs
