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

use proc_macro2::TokenStream;
use quote::quote;
use sql_fun_core::{SqlFunArgs, SqlFunMetadata};
use syn::ItemFn;

use crate::{Connector, SqlFunError, TokioPostgresCodeGenerator};

/// generate for sql_execture
pub fn generate_sql_execute(
    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 generator = code_generator_builder.build()?;
    let sql_execution_code = generator.generate_execute();
    function_ast
        .block
        .stmts
        .insert(0, syn::parse2(sql_execution_code)?);
    Ok(quote! { #function_ast })
}