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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
| <dependency>
<groupId>MySQL</groupId>
<artifactId>MySQL-connector-java</artifactId>
<version>8.0.22</version>
</dependency>
`单类代码实现，复制粘贴到编辑器里，主方法运行即可。```java
import org.apache.commons.lang3.StringUtils;
import java.io.File;
import java.io.FileOutputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.text.SimpleDateFormat;
import java.util.Date;
/**
* mybatis plus通用生成工具
* 可生成业务接口、mapper接口、实体类
*
* @author tarzan Liu
* @date 2021/4/10 19:44
*/
public class MyBatisPlusTools {
private static final String driver = "com.MySQL.cj.jdbc.Driver";//驱动
private static final String user = "root"; //数据库账号
private static final String pwd = "123456"; //数据库密码
private static final String url = "jdbc:MySQL://127.0.0.1:3306/ofcms" + "?user=" + user + "&password=" + pwd+"&useUnicode=true&characterEncoding=UTF-8&zeroDateTimeBehavior=convertToNull"; //链接参数
private static String tableName = "of_cms_ad"; // 数据库表名
private static String aliasName = "cms_ad"; // 数据库别名,可以与数据库表名相同
private static final String packagePath = "com/tarzan/cms"; //mapper.xml命名空间路径
private static final String packageName = "com.tarzan.cms"; //mapper.xml命名空间路径
private static final String author = "tarzan"; // 作者
private static final String rootPathName = "src/main/java/"; // 默认生成主文件夹路径
private static Connection getConnection = null;
static SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
/**
* 链接数据库
*/
private static Connection getConnections() {
try {
Class.forName(driver);
getConnection = DriverManager.getConnection(url);
} catch (Exception e) {
e.printStackTrace();
}
return getConnection;
}
/**
* 格式化默认值
*/
private static String defaultValue(String value) {
if (StringUtils.isNotBlank(value)) {
return "；默认值：" + value;
}
{
return "";
}
private static String getAliasName(String tableName,String prefix){
return tableName.substring(prefix.length());
}
/**
* 格式化数据类型
* 返回的是基本类型的包装类
* 如果使用基本数据类型long
*/
private static String formatType(String typeValue) {
if ("bit".equalsIgnoreCase(typeValue)) {
return "Boolean";
}
else if (
typeValue.equalsIgnoreCase("int") || typeValue.equalsIgnoreCase("int unsigned")
|| typeValue.equalsIgnoreCase("tinyint") || typeValue.equalsIgnoreCase("tinyint unsigned")
|| typeValue.equalsIgnoreCase("smallint") || typeValue.equalsIgnoreCase("smallint unsigned")
|| typeValue.equalsIgnoreCase("mediumint") || typeValue.equalsIgnoreCase("mediumint unsigned")
) {
return "Integer";
} else if (typeValue.equalsIgnoreCase("bigint") || typeValue.equalsIgnoreCase("bigint unsigned")) {
return "Long";
} else if (typeValue.equalsIgnoreCase("float") || typeValue.equalsIgnoreCase("float unsigned")) {
return "Float";
} else if (typeValue.equalsIgnoreCase("decimal") || typeValue.equalsIgnoreCase("decimal unsigned") || typeValue.equalsIgnoreCase("numeric") || typeValue.equalsIgnoreCase("numeric unsigned")
|| typeValue.equalsIgnoreCase("real") || typeValue.equalsIgnoreCase("real unsigned") || typeValue.equalsIgnoreCase("money") || typeValue.equalsIgnoreCase("money unsigned")
|| typeValue.equalsIgnoreCase("smallmoney") || typeValue.equalsIgnoreCase("smallmoney unsigned")) {
return "Double";
} else if (typeValue.equalsIgnoreCase("varchar") || typeValue.equalsIgnoreCase("char")
|| typeValue.equalsIgnoreCase("nvarchar") || typeValue.equalsIgnoreCase("nchar")
|| typeValue.equalsIgnoreCase("text")) {
return "String";
} else if (typeValue.equalsIgnoreCase("datetime")) {
return "Date";
} else if (typeValue.equalsIgnoreCase("image")) {
return "Blod";
} else {
}
}
/**
* 驼峰转换
*/
private static String columnToProperty(String column) {
StringBuilder result = new StringBuilder();
// 快速检查
if (column == null || column.isEmpty()) {
// 没必要转换
return "";}
else column =column.toLowerCase();
if (!column.contains("_")) {
// 不含下划线，仅将首字母小写
return column.substring(0, 1).toLowerCase() + column.substring(1);
// 用下划线将原始字符串分割
String[] columns = column.split("_");
for (String columnSplit : columns) {
// 跳过原始字符串中开头、结尾的下换线或双重下划线
if (columnSplit.isEmpty()) {
continue;
}
// 处理真正的驼峰片段
if (result.length() == 0) {
// 第一个驼峰片段，全部字母都小写
result.append(columnSplit.toLowerCase());
// 其他的驼峰片段，首字母大写
result.append(columnSplit.substring(0, 1).toUpperCase()).append(columnSplit.substring(1).toLowerCase());
}
return result.toString();
}
/**
* 实体名称转换
*/
private static String formatBeanName(String column) {
} else if (!column.contains("_")) {
// 不含下划线，仅将首字母大写
return column.substring(0, 1).toUpperCase() + column.substring(1);
}
}
}
/**
* 实体类字段
*/
private static void getBean(String tableName,String aliasName) {
getConnection = getConnections();
StringBuilder sb = new StringBuilder();
try {
DatabaseMetaData dbmd = getConnection.getMetaData();
ResultSet rs = dbmd.getColumns(null, "%", tableName, "%");
String beanName = formatBeanName(aliasName);
sb.append("package "+packageName+".entity;\n\n");
sb.append("import com.baomidou.mybatisplus.annotation.TableName;\n");
sb.append("import lombok.Data;\n");
int length=sb.length();
boolean dateFlag=false;
sb.append( " /**\n" +
" * @author " + author + "\n" +
" * @date "+ format.format(new Date())+"\n" +
" */\n" +
"@Data\n" +
"@TableName(\""+tableName+"\")\n" +
"public class "+beanName+"Entity {\n");
while (rs.next()) {
if(formatType(rs.getString("TYPE_NAME")).equals("Date")){
dateFlag=true;
}
sb.append("\t//").append(rs.getString("REMARKS")).append(defaultValue(rs.getString("COLUMN_DEF"))).append("\n");
sb.append("\tprivate ").append(formatType(rs.getString("TYPE_NAME"))).append(" ").append(columnToProperty(rs.getString("COLUMN_NAME"))).append(";\n");
}
sb.append("} ");
if(dateFlag){
sb.insert(length, "import java.util.Date;\n");
}
}
write(sb.toString(),"Entity.java","entity");
System.err.println("\n类型：JAVA数据层实体类（bean.java）" + "\n状态：成功" + "\n时间：" + format.format(new Date()) + "\n");
}
/**
* 生成DAO层接口
*/
private static void getMapper(String tableName,String aliasName) {
try {
sb.append("package "+packageName+".mapper;\n\n");
sb.append("import com.baomidou.mybatisplus.core.mapper.BaseMapper;\n");
sb.append("import "+packageName+".entity."+beanName+"Entity;\n");
sb.append( "/**\n" +
" * @author " + author + "\n" +
" * @date "+ format.format(new Date())+"\n" +
" */\n" +
"public interface "+beanName+"Mapper extends BaseMapper<"+beanName+"Entity>{\n" +
" \n" +
"}");
}
write(sb.toString(),"Mapper.java","mapper");
System.err.println("\n类型：JAVA数据持久层接口（dao.java）" + "\n状态：成功" + "\n时间：" + format.format(new Date()) + "\n");
}
/**
* 生成SERVICE层接口
*/
private static void getService(String tableName,String aliasName) {
try {
sb.append("package "+packageName+".service;\n\n");
sb.append("import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;\n");
sb.append("import org.springframework.stereotype.Service;\n");
sb.append("import "+packageName+".mapper."+beanName+"Mapper;\n");
"@Service\n" +
"public class "+beanName+"Service extends ServiceImpl<"+beanName+"Mapper, "+beanName+"Entity>{\n" +
"\n" +
"}");
}
write(sb.toString(),"Service.java","service");
System.err.println("\n类型：JAVA业务层接口（service.java）" + "\n状态：成功" + "\n时间：" + format.format(new Date()) + "\n");
}
/**
* 写文件，支持中文字符，在linux redhad下测试过
* @param str 文本内容
* @param name 文本名称
* */
private static void write(String str, String name,String type) {
try {
File dir = new File(rootPathName +packagePath+ "/" + type);
dir.mkdirs();
String path = dir.getPath() + "/" + formatBeanName(aliasName)+name;
File file = new File(path);
if (!file.exists())
file.createNewFile();
FileOutputStream out = new FileOutputStream(file, false); //如果追加方式用true
sb.append(str + "\n");
out.write(sb.toString().getBytes("utf-8"));//注意需要转换对应的字符集
out.close();
}
//一次生产所有表
private static void tableNames() {
try {
ResultSet rs = dbmd.getTables(getConnection.getCatalog(), null, null, new String[] { "TABLE" });
tableName=rs.getString("TABLE_NAME");
aliasName=getAliasName(tableName,"of_");
//实体
getBean(tableName,aliasName);
//dao层接口
getMapper(tableName,aliasName);
//业务类接口
getService(tableName,aliasName);
}
}
public static void main(String[] args) {
// tableNames();
//实体
//mapper接口
}
}
|
