前言：
原本想使用AutoGenerator 是 MyBatis-Plus 的官方代码生成器 ，尝试了一下，竟然报错，原因可能是MyBatis-Plus和mybatis-plus-generator 的版本不一致，因为我用的MyBatis-Plus 的版本是3.42 ，但是mybatis-plus-generator的3.4.2不知道怎么了，下载不下来，只能下载3.4.1，发现运行起来老是报错，还有一堆配置说明要看，于是自己手写生成代码的工具类，觉得更简单些。分享给大家，请多多指教。
pom文件引入java-mysql 驱动依赖
mysql-connector-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 {
return “Long”;
}
}
/**
* 驼峰转换
*/
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);
} else {
// 用下划线将原始字符串分割
String[] columns = column.split(“_”);
for (String columnSplit : columns) {
// 跳过原始字符串中开头、结尾的下换线或双重下划线
if (columnSplit.isEmpty()) {
continue;
}
// 处理真正的驼峰片段
if (result.length() == 0) {
// 第一个驼峰片段，全部字母都小写
result.append(columnSplit.toLowerCase());
} else {
// 其他的驼峰片段，首字母大写
result.append(columnSplit.substring(0, 1).toUpperCase()).append(columnSplit.substring(1).toLowerCase());
}
}
return result.toString();
}
}
/**
* 实体名称转换
*/
private static String formatBeanName(String column) {
StringBuilder result = new StringBuilder();
// 快速检查
if (column == null || column.isEmpty()) {
// 没必要转换
return “”;
} else if (!column.contains(“_”)) {
// 不含下划线，仅将首字母大写
return column.substring(0, 1).toUpperCase() + column.substring(1);
} else {
// 用下划线将原始字符串分割
String[] columns = column.split(“_”);
for (String columnSplit : columns) {
// 跳过原始字符串中开头、结尾的下换线或双重下划线
if (columnSplit.isEmpty()) {
continue;
}
// 处理真正的驼峰片段
result.append(columnSplit.substring(0, 1).toUpperCase()).append(columnSplit.substring(1).toLowerCase());
}
return result.toString();
}
}
/**
* 实体类字段
*/
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; ”);
sb.append(“import com.baomidou.mybatisplus.annotation.TableName; ”);
sb.append(“import lombok.Data; ”);
int length=sb.length();
boolean dateFlag=false;
sb.append( ” /** ” +
” * @author ” + author + “ ” +
” * @date “+ format.format(new Date())+” ” +
” */ ” +
“@Data ” +
“@TableName(\””+tableName+”\”) ” +
“public class “+beanName+”Entity { ”);
while (rs.next()) {
if(formatType(rs.getString(“TYPE_NAME”)).equals(“Date”)){
dateFlag=true;
}
sb.append(“ //”).append(rs.getString(“REMARKS”)).append(defaultValue(rs.getString(“COLUMN_DEF”))).append(“ ”);
sb.append(“ private “).append(formatType(rs.getString(“TYPE_NAME”))).append(” “).append(columnToProperty(rs.getString(“COLUMN_NAME”))).append(“; ”);
}
sb.append(“} “);
if(dateFlag){
sb.insert(length, “import java.util.Date; ”);
}
} catch (Exception e) {
e.printStackTrace();
}
write(sb.toString(),”Entity.java”,”entity”);
System.err.println(“ 类型：JAVA数据层实体类（bean.java）” + “ 状态：成功” + “ 时间：” + format.format(new Date()) + “ ”);
}
/**
* 生成DAO层接口
*/
private static void getMapper(String tableName,String aliasName) {
StringBuilder sb = new StringBuilder();
try {
String beanName = formatBeanName(aliasName);
sb.append(“package “+packageName+”.mapper; ”);
sb.append(“import com.baomidou.mybatisplus.core.mapper.BaseMapper; ”);
sb.append(“import “+packageName+”.entity.”+beanName+”Entity; ”);
sb.append( “/** ” +
” * @author ” + author + “ ” +
” * @date “+ format.format(new Date())+” ” +
” */ ” +
“public interface “+beanName+”Mapper extends BaseMapper<“+beanName+”Entity>{ ” +
” ” +
“}”);
} catch (Exception e) {
e.printStackTrace();
}
write(sb.toString(),”Mapper.java”,”mapper”);
System.err.println(“ 类型：JAVA数据持久层接口（dao.java）” + “ 状态：成功” + “ 时间：” + format.format(new Date()) + “ ”);
}
/**
* 生成SERVICE层接口
*/
private static void getService(String tableName,String aliasName) {
StringBuilder sb = new StringBuilder();
try {
String beanName = formatBeanName(aliasName);
sb.append(“package “+packageName+”.service; ”);
sb.append(“import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; ”);
sb.append(“import org.springframework.stereotype.Service; ”);
sb.append(“import “+packageName+”.mapper.”+beanName+”Mapper; ”);
sb.append(“import “+packageName+”.entity.”+beanName+”Entity; ”);
sb.append( “/** ” +
” * @author ” + author + “ ” +
” * @date “+ format.format(new Date())+” ” +
” */ ” +
“@Service ” +
“public class “+beanName+”Service extends ServiceImpl<“+beanName+”Mapper, “+beanName+”Entity>{ ” +
“ ” +
“}”);
} catch (Exception e) {
e.printStackTrace();
}
write(sb.toString(),”Service.java”,”service”);
System.err.println(“ 类型：JAVA业务层接口（service.java）” + “ 状态：成功” + “ 时间：” + format.format(new Date()) + “ ”);
}
/**
* 写文件，支持中文字符，在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
StringBuilder sb = new StringBuilder();
sb.append(str + “ ”);
out.write(sb.toString().getBytes(“utf-8”));//注意需要转换对应的字符集
out.close();
} catch (Exception e) {
e.printStackTrace();
}
}
//一次生产所有表
private static void tableNames() {
getConnection = getConnections();
try {
DatabaseMetaData dbmd = getConnection.getMetaData();
ResultSet rs = dbmd.getTables(getConnection.getCatalog(), null, null, new String[] { “TABLE” });
while (rs.next()) {
tableName=rs.getString(“TABLE_NAME”);
aliasName=getAliasName(tableName,”of_”);
//实体
getBean(tableName,aliasName);
//dao层接口
getMapper(tableName,aliasName);
//业务类接口
getService(tableName,aliasName);
}
} catch (Exception e) {
e.printStackTrace();
}
}
public static void main(String[] args) {
// tableNames();
//实体
getBean(tableName,aliasName);
//mapper接口
getMapper(tableName,aliasName);
//业务类接口
getService(tableName,aliasName);
}
}
生成文件截图
实体
mapper
service
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，请发送邮件至 ZLME@ZLME.COM 举报，一经查实，立刻删除。