rdbi-codegen 0.3.7

Generate Rust structs and rdbi DAO functions from MySQL schema DDL
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

//! Main code generator orchestrator

use crate::error::Result;

use crate::config::CodegenConfig;
use crate::parser::TableMetadata;

pub use super::dao_generator::generate_daos;
pub use super::struct_generator::generate_structs;

/// Main code generator that orchestrates struct and DAO generation
pub struct CodeGenerator<'a> {
    config: &'a CodegenConfig,
}

impl<'a> CodeGenerator<'a> {
    /// Create a new code generator with the given configuration
    pub fn new(config: &'a CodegenConfig) -> Self {
        Self { config }
    }

    /// Generate all code (structs and DAOs)
    pub fn generate(&self, tables: &[TableMetadata]) -> Result<()> {
        if self.config.generate_structs {
            self.generate_structs(tables)?;
        }
        if self.config.generate_dao {
            self.generate_daos(tables)?;
        }
        Ok(())
    }

    /// Generate struct files
    pub fn generate_structs(&self, tables: &[TableMetadata]) -> Result<()> {
        generate_structs(tables, self.config)
    }

    /// Generate DAO files
    pub fn generate_daos(&self, tables: &[TableMetadata]) -> Result<()> {
        generate_daos(tables, self.config)
    }
}