1919import java .util .Map ;
2020
2121/**
22- * sso server (for web)
23- *
24- * @author xuxueli 2017-08-01 21:39:47
22+ * spring boot code generator
23+ * @author zhengk/moshow
2524 */
2625@ Controller
2726public class IndexController {
@@ -37,7 +36,11 @@ public String index() {
3736
3837 @ RequestMapping ("/genCode" )
3938 @ ResponseBody
40- public ReturnT <Map <String , String >> codeGenerate (String tableSql ) {
39+ public ReturnT <Map <String , String >> codeGenerate (String tableSql ,String authorName ,String packageName ) {
40+
41+ if (StringUtils .isBlank (authorName )) authorName ="大狼狗" ;
42+
43+ if (StringUtils .isBlank (packageName )) packageName ="com.softdev.system" ;
4144
4245 try {
4346
@@ -51,23 +54,26 @@ public ReturnT<Map<String, String>> codeGenerate(String tableSql) {
5154 // code genarete
5255 Map <String , Object > params = new HashMap <String , Object >();
5356 params .put ("classInfo" , classInfo );
57+ params .put ("authorName" , authorName );
58+ params .put ("packageName" , packageName );
5459
5560 // result
5661 Map <String , String > result = new HashMap <String , String >();
5762
58- result .put ("controller_code" , freemarkerTool .processString ("xxl-code-generator/controller.ftl" , params ));
59- result .put ("service_code" , freemarkerTool .processString ("xxl-code-generator/service.ftl" , params ));
60- result .put ("service_impl_code" , freemarkerTool .processString ("xxl-code-generator/service_impl.ftl" , params ));
61-
62- result .put ("dao_code" , freemarkerTool .processString ("xxl-code-generator/dao.ftl" , params ));
63- result .put ("mybatis_code" , freemarkerTool .processString ("xxl-code-generator/mybatis.ftl" , params ));
64- result .put ("model_code" , freemarkerTool .processString ("xxl-code-generator/model.ftl" , params ));
63+ result .put ("controller" , freemarkerTool .processString ("xxl-code-generator/controller.ftl" , params ));
64+ result .put ("service" , freemarkerTool .processString ("xxl-code-generator/service.ftl" , params ));
65+ result .put ("service_impl" , freemarkerTool .processString ("xxl-code-generator/service_impl.ftl" , params ));
66+ result .put ("dao" , freemarkerTool .processString ("xxl-code-generator/dao.ftl" , params ));
67+ result .put ("mybatis" , freemarkerTool .processString ("xxl-code-generator/mybatis.ftl" , params ));
68+ result .put ("model" , freemarkerTool .processString ("xxl-code-generator/model.ftl" , params ));
6569
66- result .put ("entity_code" , freemarkerTool .processString ("xxl-code-generator/entity.ftl" , params ));
67- result .put ("swaggerui_code" , freemarkerTool .processString ("xxl-code-generator/swaggerui.ftl" , params ));
68- result .put ("repository_code" , freemarkerTool .processString ("xxl-code-generator/repository.ftl" , params ));
70+ result .put ("entity" , freemarkerTool .processString ("xxl-code-generator/entity.ftl" , params ));
71+ result .put ("swaggerui" , freemarkerTool .processString ("xxl-code-generator/swagger-ui.ftl" , params ));
72+ result .put ("repository" , freemarkerTool .processString ("xxl-code-generator/repository.ftl" , params ));
73+ result .put ("jpacontroller" , freemarkerTool .processString ("xxl-code-generator/jpacontroller.ftl" , params ));
6974
70- result .put ("jpacontroller_code" , freemarkerTool .processString ("xxl-code-generator/jpacontroller.ftl" , params ));
75+ result .put ("jtdao" , freemarkerTool .processString ("xxl-code-generator/jtdao.ftl" , params ));
76+ result .put ("jtdaoimpl" , freemarkerTool .processString ("xxl-code-generator/jtdaoimpl.ftl" , params ));
7177
7278 // 计算,生成代码行数
7379 int lineNum = 0 ;
@@ -81,7 +87,7 @@ public ReturnT<Map<String, String>> codeGenerate(String tableSql) {
8187 return new ReturnT <Map <String , String >>(result );
8288 } catch (IOException | TemplateException e ) {
8389 logger .error (e .getMessage (), e );
84- return new ReturnT <Map <String , String >>(ReturnT .FAIL_CODE , "表结构解析失败" );
90+ return new ReturnT <Map <String , String >>(ReturnT .FAIL_CODE , "表结构解析失败" + e . getMessage () );
8591 }
8692
8793 }
0 commit comments