32
32
@ RunWith (SpringRunner .class )
33
33
public class CodeGenerator {
34
34
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" ;
35
38
private Logger log = LoggerFactory .getLogger (CodeGenerator .class );
36
39
37
40
@ Test
38
41
public void genCodeByCustomModelName () throws IOException , TemplateException {
39
- this .genCodeByCustomModelName ("demo_user" , "demoUser" , "DemoUser" );
42
+ this .genCodeByCustomModelName (TABLE_NAME , PACKAGE_NAME , ENTITIY_NAME );
40
43
}
41
44
42
45
/**
43
46
* 通过数据表名称,和自定义的 Model 名称生成代码
44
47
* 如输入表名称 "t_user_detail" 和自定义的 Model 名称 "User" 将生成 User、UserMapper、UserService ...
45
48
* @param tableName 数据表名称
46
- * @param modelName 自定义的 Model 名称
49
+ * @param entityName 自定义的 Entity 名称
47
50
*/
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 );
53
56
}
54
57
55
- public void genModelAndMapper (String tableName ,String packageName , String modelName ) {
58
+ public void genModelAndMapper (String tableName ,String packageName , String entityName ) {
56
59
CodeGenneratorService codeGenneratorService = SpringUtil .getBean (CodeGenneratorService .class );
57
60
GenTable tableInfo = codeGenneratorService .getTableInfo (tableName );
58
61
59
- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName ,modelName );
62
+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName ,entityName );
60
63
61
64
data .put ("table" ,tableInfo );
62
65
data .put ("mybatisIdParam" ," #{id}" );
@@ -82,9 +85,9 @@ public void genModelAndMapper(String tableName,String packageName, String model
82
85
}
83
86
}
84
87
85
- public void genService (String tableName ,String packageName , String modelName ) {
88
+ public void genService (String tableName ,String packageName , String entityName ) {
86
89
try {
87
- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , modelName );
90
+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , entityName );
88
91
String modelNameUpperCamel = (String )data .get ("modelNameUpperCamel" );
89
92
90
93
FreemarkerUtil freemarkerUtil = new FreemarkerUtil (data );
@@ -100,10 +103,10 @@ public void genService(String tableName,String packageName, String modelName) {
100
103
}
101
104
}
102
105
103
- public void genController (String tableName ,String packageName , String modelName ) {
106
+ public void genController (String tableName ,String packageName , String entityName ) {
104
107
try {
105
108
106
- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , modelName );
109
+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , entityName );
107
110
String modelNameUpperCamel = (String )data .get ("modelNameUpperCamel" );
108
111
data .put ("baseRequestMapping" , CodeNameConverUtil .modelNameConvertMappingPath (modelNameUpperCamel ));
109
112
@@ -117,10 +120,10 @@ public void genController(String tableName,String packageName, String modelName
117
120
118
121
}
119
122
120
- public void genControllerTest (String tableName ,String packageName , String modelName ) {
123
+ public void genControllerTest (String tableName ,String packageName , String entityName ) {
121
124
try {
122
125
123
- Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , modelName );
126
+ Map <String , Object > data = initBaseTemplateParamMap (tableName ,packageName , entityName );
124
127
String modelNameUpperCamel = (String )data .get ("modelNameUpperCamel" );
125
128
data .put ("baseRequestMapping" , CodeNameConverUtil .modelNameConvertMappingPath (modelNameUpperCamel ));
126
129
@@ -134,11 +137,11 @@ public void genControllerTest(String tableName,String packageName, String model
134
137
135
138
}
136
139
137
- private Map <String , Object > initBaseTemplateParamMap (String tableName ,String packageName ,String modelName ) {
140
+ private Map <String , Object > initBaseTemplateParamMap (String tableName ,String packageName ,String entityName ) {
138
141
Map <String , Object > data = new HashMap <>();
139
142
data .put ("date" , DATE );
140
143
data .put ("author" , AUTHOR );
141
- String modelNameUpperCamel = StringUtils .isEmpty (modelName ) ? CodeNameConverUtil .tableNameConvertUpperCamel (tableName ) : modelName ;
144
+ String modelNameUpperCamel = StringUtils .isEmpty (entityName ) ? CodeNameConverUtil .tableNameConvertUpperCamel (tableName ) : entityName ;
142
145
data .put ("modelNameUpperCamel" , modelNameUpperCamel );
143
146
data .put ("modelNameLowerCamel" , CodeNameConverUtil .tableNameConvertLowerCamel (tableName ));
144
147
data .put ("basePackage" , BASE_PACKAGE );
0 commit comments