arysn 0.5.3

ORM code generator
Documentation 
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

use crate::error::Result;
use crate::generator::Column;
use proc_macro2::{Ident, TokenStream};
use quote::{format_ident, quote};
use std::collections::HashMap;
use tokio_postgres::Client;

pub async fn definitions(
    columns: &Vec<Column>,
    client: &Client,
) -> Result<HashMap<String, TokenStream>> {
    let mut result = HashMap::new();
    for column in columns.iter() {
        if column.data_type != "USER-DEFINED" || column.udt_name == "geography" {
            continue;
        }
        let sql = r"
SELECT e.enumlabel
FROM pg_type t
   JOIN pg_enum e ON t.oid = e.enumtypid
   JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
WHERE t.typname = $1
ORDER BY e.enumsortorder
";
        let rows = client.query(sql, &[&column.udt_name]).await?;
        let enumlabels_pg: Vec<String> = rows.iter().map(|row| row.get(0)).collect();
        let enumlabels: Vec<Ident> = enumlabels_pg
            .iter()
            .map(|x| {
                let rust_name = crate::utils::title_case(x);
                format_ident!("{}", rust_name)
            })
            .collect();
        let enum_name = &column.rust_type;
        let enum_name_pg = &column.udt_name;
        result.insert(
            enum_name.to_string(),
            quote! {
                #[derive(Debug, Clone, Deserialize, Serialize, PartialEq)]
                #[cfg_attr(
                    target_arch = "x86_64",
                    derive(FromSql, ToSql),
                    postgres(name = #enum_name_pg)
                )]
                pub enum #enum_name {
                    #(#[cfg_attr(
                        target_arch = "x86_64",
                        postgres(name = #enumlabels_pg)
                    )]
                    #enumlabels,)*
                }
            },
        );
    }
    Ok(result)
}