mz_expr/scalar/func/impls/
pg_legacy_char.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
// Copyright Materialize, Inc. and contributors. All rights reserved.
//
// Use of this software is governed by the Business Source License
// included in the LICENSE file.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0.

use std::str;

use mz_ore::fmt::FormatBuffer;
use mz_repr::adt::char::Char;
use mz_repr::adt::system::PgLegacyChar;
use mz_repr::adt::varchar::{VarChar, VarCharMaxLength};

use crate::EvalError;

pub fn format_pg_legacy_char<B>(buf: &mut B, c: u8) -> Result<(), EvalError>
where
    B: FormatBuffer,
{
    // PostgreSQL is willing to hold invalid UTF-8 in a `Datum::String`, but
    // we are not.
    match str::from_utf8(&[c]) {
        Ok(s) => {
            buf.write_str(s);
            Ok(())
        }
        Err(_) => Err(EvalError::InvalidByteSequence {
            byte_sequence: format!("{:#02x}", c).into(),
            encoding_name: "UTF8".into(),
        }),
    }
}

sqlfunc!(
    #[sqlname = "\"char\"_to_text"]
    #[preserves_uniqueness = true]
    #[inverse = to_unary!(super::CastStringToPgLegacyChar)]
    fn cast_pg_legacy_char_to_string(a: PgLegacyChar) -> Result<String, EvalError> {
        let mut buf = String::new();
        format_pg_legacy_char(&mut buf, a.0)?;
        Ok(buf)
    }
);

sqlfunc!(
    #[sqlname = "\"char\"_to_char"]
    #[preserves_uniqueness = true]
    #[inverse = to_unary!(super::CastStringToPgLegacyChar)]
    fn cast_pg_legacy_char_to_char(a: PgLegacyChar) -> Result<Char<String>, EvalError> {
        let mut buf = String::new();
        format_pg_legacy_char(&mut buf, a.0)?;
        Ok(Char(buf))
    }
);

sqlfunc!(
    #[sqlname = "\"char\"_to_varchar"]
    #[preserves_uniqueness = true]
    #[inverse = to_unary!(super::CastStringToVarChar{fail_on_len: false, length: Some(VarCharMaxLength::try_from(1).unwrap())})]
    fn cast_pg_legacy_char_to_var_char(a: PgLegacyChar) -> Result<VarChar<String>, EvalError> {
        let mut buf = String::new();
        format_pg_legacy_char(&mut buf, a.0)?;
        Ok(VarChar(buf))
    }
);

sqlfunc!(
    #[sqlname = "\"char\"_to_integer"]
    #[preserves_uniqueness = true]
    #[inverse = to_unary!(super::CastInt32ToPgLegacyChar)]
    fn cast_pg_legacy_char_to_int32(a: PgLegacyChar) -> i32 {
        // Per PostgreSQL, casts to `i32` are performed as if `PgLegacyChar` is
        // signed.
        // See: https://github.com/postgres/postgres/blob/791b1b71da35d9d4264f72a87e4078b85a2fcfb4/src/backend/utils/adt/char.c#L91-L96
        i32::from(i8::from_ne_bytes([a.0]))
    }
);