3232@ RunWith (SpringRunner .class )
3333public class CodeGenerator {
3434
35+ public static final String TABLE_NAME = "demo_user" ;
36+ public static final String PACKAGE_NAME = "demoUser" ;
37+ public static final String ENTITIY_NAME = "DemoUser" ;
3538 private Logger log = LoggerFactory .getLogger (CodeGenerator .class );
3639
3740 @ Test
3841 public void genCodeByCustomModelName () throws IOException , TemplateException {
39- this .genCodeByCustomModelName ("demo_user" , "demoUser" , "DemoUser" );
42+ this .genCodeByCustomModelName (TABLE_NAME , PACKAGE_NAME , ENTITIY_NAME );
4043 }
4144
4245 /**
4346 * 通过数据表名称,和自定义的 Model 名称生成代码
4447 * 如输入表名称 "t_user_detail" 和自定义的 Model 名称 "User" 将生成 User、UserMapper、UserService ...
4548 * @param tableName 数据表名称
46- * @param modelName 自定义的 Model 名称
49+ * @param entityName 自定义的 Entity 名称
4750 */
48- public void genCodeByCustomModelName (String tableName ,String packageName , String modelName ) throws IOException , TemplateException {
49- genModelAndMapper (tableName ,packageName , modelName );
50- genService (tableName ,packageName , modelName );
51- genController (tableName ,packageName , modelName );
52- genControllerTest (tableName ,packageName , modelName );
51+ public void genCodeByCustomModelName (String tableName ,String packageName , String entityName ) throws IOException , TemplateException {
52+ genModelAndMapper (tableName ,packageName , entityName );
53+ genService (tableName ,packageName , entityName );
54+ genController (tableName ,packageName , entityName );
55+ genControllerTest (tableName ,packageName , entityName );
5356 }
5457
55- public void genModelAndMapper (String tableName ,String packageName , String modelName ) {
58+ public void genModelAndMapper (String tableName ,String packageName , String entityName ) {
5659 CodeGenneratorService codeGenneratorService = SpringUtil .getBean (CodeGenneratorService .class );
5760 GenTable tableInfo = codeGenneratorService .getTableInfo (tableName );
5861
59- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName ,modelName );
62+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName ,entityName );
6063
6164 data .put ("table" ,tableInfo );
6265 data .put ("mybatisIdParam" ," #{id}" );
@@ -82,9 +85,9 @@ public void genModelAndMapper(String tableName,String packageName, String model
8285 }
8386 }
8487
85- public void genService (String tableName ,String packageName , String modelName ) {
88+ public void genService (String tableName ,String packageName , String entityName ) {
8689 try {
87- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , modelName );
90+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , entityName );
8891 String modelNameUpperCamel = (String )data .get ("modelNameUpperCamel" );
8992
9093 FreemarkerUtil freemarkerUtil = new FreemarkerUtil (data );
@@ -100,10 +103,10 @@ public void genService(String tableName,String packageName, String modelName) {
100103 }
101104 }
102105
103- public void genController (String tableName ,String packageName , String modelName ) {
106+ public void genController (String tableName ,String packageName , String entityName ) {
104107 try {
105108
106- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , modelName );
109+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , entityName );
107110 String modelNameUpperCamel = (String )data .get ("modelNameUpperCamel" );
108111 data .put ("baseRequestMapping" , CodeNameConverUtil .modelNameConvertMappingPath (modelNameUpperCamel ));
109112
@@ -117,10 +120,10 @@ public void genController(String tableName,String packageName, String modelName
117120
118121 }
119122
120- public void genControllerTest (String tableName ,String packageName , String modelName ) {
123+ public void genControllerTest (String tableName ,String packageName , String entityName ) {
121124 try {
122125
123- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , modelName );
126+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , entityName );
124127 String modelNameUpperCamel = (String )data .get ("modelNameUpperCamel" );
125128 data .put ("baseRequestMapping" , CodeNameConverUtil .modelNameConvertMappingPath (modelNameUpperCamel ));
126129
@@ -134,11 +137,11 @@ public void genControllerTest(String tableName,String packageName, String model
134137
135138 }
136139
137- private Map <String , Object > initBaseTemplateParamMap (String tableName ,String packageName ,String modelName ) {
140+ private Map <String , Object > initBaseTemplateParamMap (String tableName ,String packageName ,String entityName ) {
138141 Map <String , Object > data = new HashMap <>();
139142 data .put ("date" , DATE );
140143 data .put ("author" , AUTHOR );
141- String modelNameUpperCamel = StringUtils .isEmpty (modelName ) ? CodeNameConverUtil .tableNameConvertUpperCamel (tableName ) : modelName ;
144+ String modelNameUpperCamel = StringUtils .isEmpty (entityName ) ? CodeNameConverUtil .tableNameConvertUpperCamel (tableName ) : entityName ;
142145 data .put ("modelNameUpperCamel" , modelNameUpperCamel );
143146 data .put ("modelNameLowerCamel" , CodeNameConverUtil .tableNameConvertLowerCamel (tableName ));
144147 data .put ("basePackage" , BASE_PACKAGE );
0 commit comments