## 代碼生成框架 演示代碼生成框架，以及生成代碼和數據庫文檔 ```java /** * 演示beetlsql 代碼生成框架 * * @author xiandafu */ public class S7CodeGen { SQLManager sqlManager; public S7CodeGen(SQLManager sqlManager) { this.sqlManager = sqlManager; initGroupTemplate(); } protected void initGroupTemplate(){ //指定模板文件路徑，正常情況下，不需要要指定，默認在classpath:templates，但idea的環境讀取不到 GroupTemplate groupTemplate = BaseTemplateSourceBuilder.getGroupTemplate(); String root = System.getProperty("user.dir"); //代碼模板在sql-gen，你可以指定自己的模板路徑 String templatePath = root+"/sql-gen/src/main/resources/templates/"; FileResourceLoader resourceLoader = new FileResourceLoader(templatePath); groupTemplate.setResourceLoader(resourceLoader); } public static void main(String[] args) throws Exception { //為了簡單起見，倆個sqlManager都來自同一個數據源，實際是不同數據庫，甚至是NOSQL SQLManager sqlManager = SampleHelper.init(); S7CodeGen gen = new S7CodeGen(sqlManager); gen.genCode(); gen.genDoc(); gen.genAllDoc(); } /** * 代碼生成，生成實體，mapper代碼 */ public void genCode(){ List<SourceBuilder> sourceBuilder = new ArrayList<>(); SourceBuilder entityBuilder = new EntitySourceBuilder(); SourceBuilder mapperBuilder = new MapperSourceBuilder(); SourceBuilder mdBuilder = new MDSourceBuilder(); sourceBuilder.add(entityBuilder); sourceBuilder.add(mapperBuilder); sourceBuilder.add(mdBuilder); SourceConfig config = new SourceConfig(sqlManager,sourceBuilder); //如果有錯誤，拋出異常而不是繼續運行1 EntitySourceBuilder.getGroupTemplate().setErrorHandler(new ReThrowConsoleErrorHandler() ); ConsoleOnlyProject project = new ConsoleOnlyProject(); String tableName = "USER"; config.gen(tableName,project); } /** * 生成數據庫文檔 */ public void genDoc(){ List<SourceBuilder> sourceBuilder = new ArrayList<>(); SourceBuilder docBuilder = new MDDocBuilder(); sourceBuilder.add(docBuilder); SourceConfig config = new SourceConfig(sqlManager,sourceBuilder); //如果有錯誤，拋出異常而不是繼續運行1 EntitySourceBuilder.getGroupTemplate().setErrorHandler(new ReThrowConsoleErrorHandler() ); ConsoleOnlyProject project = new ConsoleOnlyProject(); String tableName = "USER"; config.gen(tableName,project); } /** * 生成數據庫文檔 */ public void genAllDoc(){ List<SourceBuilder> sourceBuilder = new ArrayList<>(); SourceBuilder docBuilder = new MDDocBuilder(); sourceBuilder.add(docBuilder); SourceConfig config = new SourceConfig(sqlManager,sourceBuilder); //如果有錯誤，拋出異常而不是繼續運行1 EntitySourceBuilder.getGroupTemplate().setErrorHandler(new ReThrowConsoleErrorHandler() ); StringOnlyProject project = new StringOnlyProject(); config.genAll(project); String output = project.getContent(); System.out.println(output); } } ```