mybatis plus代码生成工具的实现代码

mybatis plus代码生成工具的实现代码

前言：

原本想使用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

mysql-connector-java

8.0.22

单类代码实现，复制粘贴到编辑器里，主方法运行即可。

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 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;\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");

}

} catch (Exception e) {

e.printStackTrace();

}

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) {

StringBuilder sb = new StringBuilder();

try {

String beanName = formatBeanName(aliasName);

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" +

"}");

} catch (Exception e) {

e.printStackTrace();

}

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) {

StringBuilder sb = new StringBuilder();

try {

String beanName = formatBeanName(aliasName);

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");

sb.append("import "+packageName+".entity."+beanName+"Entity;\n");

sb.append( "/**\n" +

" * @author " + author + "\n" +

" * @date "+ format.format(new Date())+"\n" +

" */\n" +

"@Service\n" +

"public class "+beanName+"Service extends ServiceImpl<"+beanName+"Mapper, "+beanName+"Entity>{\n" +

"\n" +

"}");

} catch (Exception e) {

e.printStackTrace();

}

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

StringBuilder sb = new StringBuilder();

sb.append(str + "\n");

out.write(sb.toString().getBytes("utf-8"));//注意需要转换对应的字符集

out.close();

} catch (Exception e) {

e.printStackTrace();

}

}

//一次生产所有表

private static void tableNames() {

getConnection = getConnections();

try {

DatabaseMetaData dbmd = getConnection.getMetaDatahttp://();

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

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。