当前位置:首页> 正文

mybatisplus代码生成器使用(java代码生成工具)

mybatisplus代码生成器使用(java代码生成工具)

  前言:

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

  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: + “?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());

  }

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

  “@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”);

  }

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

  “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”);

  }

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

  “@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”);

  }

  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.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

展开全文阅读

相关内容