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

//LICENSE Portions Copyright 2019-2021 ZomboDB, LLC.
//LICENSE
//LICENSE Portions Copyright 2021-2023 Technology Concepts & Design, Inc.
//LICENSE
//LICENSE Portions Copyright 2023-2023 PgCentral Foundation, Inc. <contact@pgcentral.org>
//LICENSE
//LICENSE All rights reserved.
//LICENSE
//LICENSE Use of this source code is governed by the MIT license that can be found in the LICENSE file.
use pgrx_sql_entity_graph::metadata::{
    ArgumentError, Returns, ReturnsError, SqlMapping, SqlTranslatable,
};

use crate::{AnyNumeric, Numeric};

unsafe impl<const P: u32, const S: u32> SqlTranslatable for Numeric<P, S> {
    fn argument_sql() -> Result<SqlMapping, ArgumentError> {
        match (P, S) {
            (0, 0) => Ok(SqlMapping::literal("NUMERIC")),
            (p, 0) => Ok(SqlMapping::As(format!("NUMERIC({p})"))),
            (p, s) => Ok(SqlMapping::As(format!("NUMERIC({p}, {s})"))),
        }
    }

    fn return_sql() -> Result<Returns, ReturnsError> {
        match (P, S) {
            (0, 0) => Ok(Returns::One(SqlMapping::literal("NUMERIC"))),
            (p, 0) => Ok(Returns::One(SqlMapping::As(format!("NUMERIC({p})")))),
            (p, s) => Ok(Returns::One(SqlMapping::As(format!("NUMERIC({p}, {s})")))),
        }
    }
}

unsafe impl SqlTranslatable for AnyNumeric {
    fn argument_sql() -> Result<SqlMapping, ArgumentError> {
        Ok(SqlMapping::literal("NUMERIC"))
    }

    fn return_sql() -> Result<Returns, ReturnsError> {
        Ok(Returns::One(SqlMapping::literal("NUMERIC")))
    }
}