risingwave_expr_impl/scalar/
sha.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_expr::function;
16use sha1::{Digest, Sha1};
17use sha2::{Sha224, Sha256, Sha384, Sha512};
18
19#[function("sha1(bytea) -> bytea")]
20pub fn sha1(data: &[u8]) -> impl AsRef<[u8]> {
21    Sha1::digest(data)
22}
23
24#[function("sha224(bytea) -> bytea")]
25pub fn sha224(data: &[u8]) -> impl AsRef<[u8]> {
26    Sha224::digest(data)
27}
28
29#[function("sha256(bytea) -> bytea")]
30pub fn sha256(data: &[u8]) -> impl AsRef<[u8]> {
31    Sha256::digest(data)
32}
33
34#[function("sha384(bytea) -> bytea")]
35pub fn sha384(data: &[u8]) -> impl AsRef<[u8]> {
36    Sha384::digest(data)
37}
38
39#[function("sha512(bytea) -> bytea")]
40pub fn sha512(data: &[u8]) -> impl AsRef<[u8]> {
41    Sha512::digest(data)
42}
43
44#[cfg(test)]
45mod tests {
46    use super::{sha1, sha224, sha256, sha384, sha512};
47    #[test]
48    fn test_sha1() {
49        let cases = [(
50            r#"hello world"#.as_bytes(),
51            b"\x2a\xae\x6c\x35\xc9\x4f\xcf\xb4\x15\xdb\xe9\x5f\x40\x8b\x9c\xe9\x1e\xe8\x46\xed",
52        )];
53
54        for (ori, encoded) in cases {
55            let t = sha1(ori);
56            assert_eq!(t.as_ref(), encoded);
57        }
58    }
59
60    #[test]
61    fn test_sha224() {
62        let cases = [
63            (r#"hello world"#.as_bytes(), b"\x2f\x05\x47\x7f\xc2\x4b\xb4\xfa\xef\xd8\x65\x17\x15\x6d\xaf\xde\xce\xc4\x5b\x8a\xd3\xcf\x25\x22\xa5\x63\x58\x2b"),
64        ];
65
66        for (ori, encoded) in cases {
67            let t = sha224(ori);
68            assert_eq!(t.as_ref(), encoded);
69        }
70    }
71
72    #[test]
73    fn test_sha256() {
74        let cases = [
75            (r#"hello world"#.as_bytes(), b"\xb9\x4d\x27\xb9\x93\x4d\x3e\x08\xa5\x2e\x52\xd7\xda\x7d\xab\xfa\xc4\x84\xef\xe3\x7a\x53\x80\xee\x90\x88\xf7\xac\xe2\xef\xcd\xe9"),
76        ];
77
78        for (ori, encoded) in cases {
79            let t = sha256(ori);
80            assert_eq!(t.as_ref(), encoded);
81        }
82    }
83
84    #[test]
85    fn test_sha384() {
86        let cases = [
87            (r#"hello world"#.as_bytes(), b"\xfd\xbd\x8e\x75\xa6\x7f\x29\xf7\x01\xa4\xe0\x40\x38\x5e\x2e\x23\x98\x63\x03\xea\x10\x23\x92\x11\xaf\x90\x7f\xcb\xb8\x35\x78\xb3\xe4\x17\xcb\x71\xce\x64\x6e\xfd\x08\x19\xdd\x8c\x08\x8d\xe1\xbd"),
88        ];
89
90        for (ori, encoded) in cases {
91            let t = sha384(ori);
92            assert_eq!(t.as_ref(), encoded);
93        }
94    }
95
96    #[test]
97    fn test_sha512() {
98        let cases = [
99            (r#"hello world"#.as_bytes(), b"\x30\x9e\xcc\x48\x9c\x12\xd6\xeb\x4c\xc4\x0f\x50\xc9\x02\xf2\xb4\xd0\xed\x77\xee\x51\x1a\x7c\x7a\x9b\xcd\x3c\xa8\x6d\x4c\xd8\x6f\x98\x9d\xd3\x5b\xc5\xff\x49\x96\x70\xda\x34\x25\x5b\x45\xb0\xcf\xd8\x30\xe8\x1f\x60\x5d\xcf\x7d\xc5\x54\x2e\x93\xae\x9c\xd7\x6f"),
100        ];
101
102        for (ori, encoded) in cases {
103            let t = sha512(ori);
104            assert_eq!(t.as_ref(), encoded);
105        }
106    }
107}
risingwave_expr_impl/scalar/sha.rs

risingwave_expr_impl/scalar/
sha.rs
