sql-fun 0.1.0

SQL query/statement execution 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

use proc_macro2::TokenStream;
use quote::quote;
use syn::ItemFn;

use crate::{Connector, SqlFunError, TokioPostgresCodeGenerator, get_result_type_args};
use sql_fun_core::{SqlFunArgs, SqlFunMetadata};

/// generate sql_query_one
pub fn generate_sql_query_one(
    sql_str: String,
    item: TokenStream,
) -> Result<TokenStream, SqlFunError> {
    let args = SqlFunArgs::try_from_env()?;
    let metadata = SqlFunMetadata::load_from(args.metadata_file()?)?;
    let _connector = Connector::get_connector(&metadata)?;

    let mut code_generator_builder = TokioPostgresCodeGenerator::builder();
    let mut function_ast: ItemFn = syn::parse2(item)?;

    if function_ast.sig.asyncness.is_none() {
        return Err(SqlFunError::custom("function must be async"));
    }

    code_generator_builder.set_sql(&function_ast, &sql_str)?;

    let (row_type, error_type) = get_result_type_args(&function_ast)?;

    code_generator_builder.row_type(row_type);
    code_generator_builder.error_type(error_type);

    if function_ast.sig.asyncness.is_none() {
        return Err(SqlFunError::custom("function must be async"));
    }
    let generator = code_generator_builder.build()?;
    let sql_execution_code = generator.generate_query_one();

    function_ast
        .block
        .stmts
        .insert(0, syn::parse2(sql_execution_code)?);
    Ok(quote! { #function_ast })
}