根据MYSQL数据库生成注解类mybatisSQL语句(仅包含主键)

最新推荐文章于 2021-11-30 11:06:30 发布
最新推荐文章于 2021-11-30 11:06:30 发布
阅读量342 收藏 1
点赞数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_29395899/article/details/84236724
版权 
package com;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class GenBaseSQLMysql {

	private String packageOutPathEntity = "com.user.entity";// 指定实体生成所在包的路径
	private String packageOutPath = "com.user.sql";// 指定实体生成所在包的路径
	private String authorName = "linshiq";// 作者名字
	private String tablename = "";// 表名
	private boolean f_util = false; // 是否需要导入包java.util.*
	private boolean f_sql = false; // 是否需要导入包java.sql.*
	private boolean f_decimal = false; // 是否需要导入包java.sql.*

	// 数据库连接
	private static final String URL = "jdbc:mysql://localhost:3306/llll";
	private static final String NAME = "root";
	private static final String PASS = "root";
	private static final String DRIVER = "com.mysql.jdbc.Driver";

	/*
	 * 构造函数
	 */
	public GenBaseSQLMysql() {
		// 创建连接
		Connection con;

		PreparedStatement pStemt = null;
		try {
			try {
				Class.forName(DRIVER);
			} catch (ClassNotFoundException e1) {
				// TODO Auto-generated catch block
				e1.printStackTrace();
			}
			con = DriverManager.getConnection(URL, NAME, PASS);

			// 获取所有的表名
			DatabaseMetaData meta = con.getMetaData();
			ResultSet rs = meta.getTables(null, null, null, new String[] { "TABLE" });
			while (rs.next()) {
				System.out.println("------------------------------");
				System.out.println("表名：" + rs.getString(3));
				System.out.println("表所属用户名：" + rs.getString(2));

				String tablename = rs.getString(3);
				this.tablename = tablename;
				// 获取所有的主键
				// oracle 语句 select
				// table_name,dbms_metadata.get_ddl('TABLE','TABLE_NAME')from
				// dual,user_tables where table_name='TABLE_NAME';
				// TABLE_NAME为具体的表名,要求大写
				String sql_table = "SHOW CREATE TABLE " + tablename;
				PreparedStatement pre = con.prepareStatement(sql_table);
				ResultSet rs1 = pre.executeQuery();
				if (rs1.next()) {

					// 正则匹配数据
					Pattern pattern = Pattern.compile("PRIMARY KEY \\(\\`(.*)\\`\\)");
					Matcher matcher = pattern.matcher(rs1.getString(2));
					matcher.find();
					String data = "";
					try {
						data = matcher.group();
					} catch (IllegalStateException e) {
						System.out.println("没主键");
						continue;
					}

					// 过滤对于字符
					data = data.replaceAll("\\`|PRIMARY KEY \\(|\\)", "");
					// 拆分字符
					String[] stringArr = data.split(",");
					System.out.println("主键为:" + Arrays.toString(stringArr));

					System.out.println("------------------------------");

					// 查要生成实体类的表
					String sql = "select * from " + tablename;

					pStemt = con.prepareStatement(sql);
					ResultSetMetaData rsmd = pStemt.getMetaData();
					int size = rsmd.getColumnCount(); // 统计列
					String[] colnames = new String[size];
					String[] colTypes = new String[size];
					int[] colSizes = new int[size];
					for (int i = 0; i < size; i++) {
						colnames[i] = rsmd.getColumnName(i + 1).toLowerCase();
						colTypes[i] = rsmd.getColumnTypeName(i + 1);

						colSizes[i] = rsmd.getColumnDisplaySize(i + 1);
					}

					String content = parse(colnames, colTypes, colSizes, stringArr);

					try {
						File directory = new File("");
						// System.out.println("绝对路径："+directory.getAbsolutePath());
						// System.out.println("相对路径："+directory.getCanonicalPath());
						String path = this.getClass().getResource("").getPath();

						System.out.println(path);
						System.out.println("src/?/" + path.substring(path.lastIndexOf("/com/", path.length())));
						// String outputPath = directory.getAbsolutePath()+
						// "/src/"+path.substring(path.lastIndexOf("/com/",
						// path.length()), path.length()) + initcap(tablename) +
						// ".java";
						String outputPath = directory.getAbsolutePath() + "/src/main/java/"
								+ this.packageOutPath.replace(".", "/") + "/" + initcap(tablename) + "Dao.java";
						FileWriter fw = new FileWriter(outputPath);
						System.out.println(outputPath);
						PrintWriter pw = new PrintWriter(fw);
						pw.println(content);
						pw.flush();
						pw.close();
					} catch (IOException e) {
						e.printStackTrace();
					}
				}
			}

		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			// try {
			// con.close();
			// } catch (SQLException e) {
			// // TODO Auto-generated catch block
			// e.printStackTrace();
			// }
		}
	}

	/**
	 * 功能：生成实体类主体代码
	 * 
	 * @param colnames
	 * @param colTypes
	 * @param colSizes
	 * @param stringArr
	 * @return
	 */
	private String parse(String[] colnames, String[] colTypes, int[] colSizes, String[] stringArr) {

		// 确定导入包
		for (int i = 0; i < stringArr.length; i++) {

			for (int j = 0; j < colnames.length; j++) {

				if (colnames[j].equalsIgnoreCase(stringArr[i])) {

					if (colTypes[i].equalsIgnoreCase("datetime")) {
						f_util = true;
					}
					if (colTypes[i].equalsIgnoreCase("image") || colTypes[i].equalsIgnoreCase("text")) {
						f_sql = true;
					}

					String clone = colTypes[i].toLowerCase();

					if (clone.contains("decimal")) {
						f_decimal = true;
					}
				}

			}
		}

		StringBuffer sb = new StringBuffer();

		sb.append("package " + this.packageOutPath + ";\r\n");
		sb.append("\r\n");
		// 判断是否导入工具包
		if (f_util) {
			sb.append("import java.util.Date;\r\n");
		}
		if (f_sql) {
			sb.append("import java.sql.*;\r\n");
		}
		if (f_decimal) {
			sb.append("import java.math.BigDecimal;\r\n");
		}
		sb.append("import org.apache.ibatis.annotations.Delete;\r\n");
		sb.append("import org.apache.ibatis.annotations.Insert;\r\n");
		sb.append("import org.apache.ibatis.annotations.Select;\r\n");
		sb.append("import org.apache.ibatis.annotations.Update;\r\n");
		sb.append("import org.apache.ibatis.annotations.Mapper;\r\n");
		sb.append("import " + packageOutPathEntity + "." + initcap(tablename) + ";\r\n");
		sb.append("\r\n");

		// 注释部分
		sb.append("   /**\r\n");
		sb.append("    * " + tablename + " 基础SQL\r\n");
		sb.append("    * " + new Date() + " " + this.authorName + "\r\n");
		sb.append("    */ \r\n");
		// 实体部分
		sb.append("\r\n@Mapper");
		sb.append("\r\npublic interface " + initcap(tablename) + "Dao{\r\n");
		processDelete(sb, colnames, colTypes, colSizes, stringArr);
		sb.append("\r\n");
		processUpdate(sb, colnames, colTypes, colSizes, stringArr);
		sb.append("\r\n");
		processSelect(sb, colnames, colTypes, colSizes, stringArr);
		sb.append("\r\n");
		processInsert(sb, colnames, colTypes, colSizes, stringArr);

		sb.append("}\r\n");

		// System.out.println(sb.toString());
		return sb.toString();
	}

	/**
	 * @Author linshiq
	 *         <p>
	 *         <li>2018年11月15日-下午4:33:28</li>
	 *         <li>功能说明：补充基础Insert</li>
	 *         </p>
	 * @param sb
	 * @param colnames
	 * @param colTypes
	 * @param colSizes
	 * @param stringArr
	 */
	private void processInsert(StringBuffer sb, String[] colnames, String[] colTypes, int[] colSizes,
			String[] stringArr) {

		StringBuffer insertInfoSb = new StringBuffer();

		for (int i = 0; i < colnames.length; i++) {// 补充insert into部分
			insertInfoSb.append(colnames[i].toLowerCase()).append(",");
		}
		insertInfoSb.delete(insertInfoSb.length() - 1, insertInfoSb.length());

		StringBuffer valuesSb = new StringBuffer(); // 补充values部分
		for (int j = 0; j < colnames.length; j++) {

			valuesSb.append("#{" + colnames[j] + "}").append(",");

		}

		valuesSb.delete(valuesSb.length() - 1, valuesSb.length());

		// 开始组装
		StringBuffer overParm = new StringBuffer();
		overParm.append("INSERT INTO " + tablename + " ( " + insertInfoSb.toString() + " ) ");
		overParm.append(" VALUES (" + valuesSb.toString() + ")");

		sb.append("\t" + "@Insert(\"" + overParm.toString() + "\")" + "\r\n");// 补充方法注解

		sb.append("\tint insertByKey(" + initcap(tablename) + " " + tablename + ");\r\n");
		//System.out.println(parm);

	}

	/**
	 * @Author linshiq
	 *         <p>
	 *         <li>2018年11月15日-下午4:33:26</li>
	 *         <li>功能说明：补充基础Select</li>
	 *         </p>
	 * @param sb
	 * @param colnames
	 * @param colTypes
	 * @param colSizes
	 * @param stringArr
	 */
	private void processSelect(StringBuffer sb, String[] colnames, String[] colTypes, int[] colSizes,
			String[] stringArr) {

		StringBuffer resultSb = new StringBuffer();

		for (int i = 0; i < colnames.length; i++) {
			resultSb.append(colnames[i].toLowerCase()).append(",");
		}
		resultSb.delete(resultSb.length() - 1, resultSb.length());

		StringBuffer inParm = new StringBuffer();
		StringBuffer overParm = new StringBuffer();
		overParm.append("SELECT " + resultSb.toString() + " FROM " + tablename + " WHERE ");
		for (int i = 0; i < stringArr.length; i++) {

			for (int j = 0; j < colnames.length; j++) {

				if (colnames[j].equalsIgnoreCase(stringArr[i])) {
					inParm.append(sqlType2JavaType(colTypes[i]) + " " + stringArr[i] + ",");
					overParm.append(stringArr[i] + " = " + "#{" + stringArr[i] + "}").append(" and ");
				}

			}

		}
		overParm.delete(overParm.length() - 5, overParm.length());
		sb.append("\t" + "@Select(\"" + overParm.toString() + "\")" + "\r\n");
		inParm.delete(inParm.length() - 1, inParm.length());
		sb.append("\t" + initcap(tablename) + " selectByKey(" + inParm.toString() + ");\r\n");
		//System.out.println(parm);

	}

	/**
	 * @Author linshiq
	 *         <p>
	 *         <li>2018年11月15日-下午4:33:24</li>
	 *         <li>功能说明：补充基础Update</li>
	 *         </p>
	 * @param sb
	 * @param colnames
	 * @param colTypes
	 * @param colSizes
	 * @param stringArr
	 */
	private void processUpdate(StringBuffer sb, String[] colnames, String[] colTypes, int[] colSizes,
			String[] stringArr) {

		// 组装set部分
		StringBuffer setSb = new StringBuffer();
		
		List<String> mainKey = Arrays.asList(stringArr);
		
		for (int j = 0; j < colnames.length; j++) {
			
			if (mainKey.contains(colnames[j].toUpperCase()) || mainKey.contains(colnames[j].toLowerCase())) {
				continue;
			}
			
			setSb.append(colnames[j] + " = " + "#{" + colnames[j] + "}").append(",");

		}
		
		// 如果小于等于0说明都是主键
		if(setSb.length() <= 0){
			
			return;
		}
		
		setSb.delete(setSb.length() - 1, setSb.length());

		// 组装where
		StringBuffer whereSb = new StringBuffer();

		for (int i = 0; i < stringArr.length; i++) {

			whereSb.append(stringArr[i] + " = " + "#{" + stringArr[i].toLowerCase() + "}").append(" and ");

		}

		whereSb.delete(whereSb.length() - 5, whereSb.length());

		StringBuffer overParm = new StringBuffer();
		overParm.append("UPDATE " + tablename + " SET " + setSb.toString() + " WHERE "+whereSb.toString());

		sb.append("\t" + "@Update(\"" + overParm.toString() + "\")" + "\r\n");

		sb.append("\tint updateByKey(" + initcap(tablename) + " " + tablename + ");\r\n");

	}

	/**
	 * @Author linshiq
	 *         <p>
	 *         <li>2018年11月15日-下午4:13:47</li>
	 *         <li>功能说明：补充基础删除方法</li>
	 *         </p>
	 * @param sb
	 */
	private void processDelete(StringBuffer sb, String[] colnames, String[] colTypes, int[] colSizes,
			String[] stringArr) {

		StringBuffer inParm = new StringBuffer();// 组装入参
		
		StringBuffer deleteSql = new StringBuffer();
		deleteSql.append("DELETE FROM " + tablename + " WHERE ");
		for (int i = 0; i < stringArr.length; i++) {

			for (int j = 0; j < colnames.length; j++) {

				if (colnames[j].equalsIgnoreCase(stringArr[i])) {
					inParm.append(sqlType2JavaType(colTypes[i]) + " " + stringArr[i] + ",");
					deleteSql.append(stringArr[i] + " = " + "#{" + stringArr[i] + "}").append(" and "); // 组装主键限制
				}

			}

		}
	
		deleteSql.delete(deleteSql.length() - 5, deleteSql.length());
		
		sb.append("\t" + "@Delete(\"" + deleteSql.toString() + "\")" + "\r\n");
		inParm.delete(inParm.length() - 1, inParm.length());
		sb.append("\t" + "int deleteByKey(" + inParm.toString() + ");\r\n");
		//System.out.println(parm);

	}

	/**
	 * 功能：将输入字符串的首字母改成大写
	 * 
	 * @param str
	 * @return
	 */
	private String initcap(String str) {

		char[] ch = str.toCharArray();
		if (ch[0] >= 'a' && ch[0] <= 'z') {
			ch[0] = (char) (ch[0] - 32);
		}

		// for (int i = 1; i < ch.length; i++) {
		//
		// if (ch[i] >= 'a' && ch[i] <= 'z') {
		// ch[i] = (char) (ch[i] + 32);
		// }
		// }

		return new String(ch);
	}

	/**
	 * 功能：获得列的数据类型
	 * 
	 * @param sqlType
	 * @return
	 */
	private String sqlType2JavaType(String sqlType) {
		// System.out.println(sqlType);
		if (sqlType.equalsIgnoreCase("bit")) {
			return "boolean";
		} else if (sqlType.equalsIgnoreCase("tinyint")) {
			return "byte";
		} else if (sqlType.equalsIgnoreCase("smallint")) {
			return "short";
		} else if (sqlType.equalsIgnoreCase("int")) {
			return "int";
		} else if (sqlType.equalsIgnoreCase("bigint")) {
			return "long";
		} else if (sqlType.equalsIgnoreCase("float")) {
			return "float";
		} else if (sqlType.equalsIgnoreCase("numeric") || sqlType.equalsIgnoreCase("real")) {
			return "double";
		} else if (sqlType.equalsIgnoreCase("decimal") || sqlType.equalsIgnoreCase("money")
				|| sqlType.equalsIgnoreCase("smallmoney")) {
			return "BigDecimal";
		} else if (sqlType.equalsIgnoreCase("varchar") || sqlType.equalsIgnoreCase("char")
				|| sqlType.equalsIgnoreCase("nvarchar") || sqlType.equalsIgnoreCase("nchar")
				|| sqlType.equalsIgnoreCase("text")) {
			return "String";
		} else if (sqlType.equalsIgnoreCase("datetime") || sqlType.equalsIgnoreCase("timestamp")) {
			return "Date";
		} else if (sqlType.equalsIgnoreCase("image")) {
			return "Blod";
		} else {

			String clone = sqlType.toLowerCase();

			if (clone.contains("decimal")) {
				return "BigDecimal";
			}

			return null;
		}

	}

	/**
	 * 出口 TODO
	 * 
	 * @param args
	 */
	public static void main(String[] args) {

		new GenBaseSQLMysql();

	}

}

 

默默敲代码的码农
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
MyBatis注解开发---实现增删查改和动态SQL
qq_53317005的博客
04-17 2143
MyBatis注解开发---实现增删查改和动态SQL注解
MySQL表自动生成Java实体类
11-23
实体类通常包含对应于数据库表字段的属性，并且具有getter和setter方法，以及可能的构造函数和注解，如`@Entity`、`@Table`、`@Id`等，以定义表名和主键。 接下来，我们讨论如何自动生成这些实体类。这里提到的工具...
mybatis入门实例(mysql数据库，带建表语句)
01-31
最简单的mybatis实例，jar包齐全，执行数据库查询，带建表语句，只需在可用的mysql数据库中建表，即可运行。
配置表名称，自动生成mybatis中sql语句
qq_34004088的博客
09-10 506
/** * 配置表名称，自动生成mybatis中sql语句 */ import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import java.sql.Connection; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.s
mysql生成uui mybatis_mybatis----基础
weixin_32789583的博客
02-08 199
基础知识安装org.mybatismybatisx.x.x对原生态jdbc程序中问题总结环境java环境：jdkjdbc程序使用jdbc查询mysql数据库中用户表的记录问题总结数据库连接、使用时创建、不使用时就立即释放，对数据库进行频繁连接开启和关闭，造成数据库资源浪费，影响数据库性能(使用连接池管理数据库来连接进行优化)将sql语句硬编码到Java代码中，如果sql语句修改，需要重新编译代码，...
mysql中怎样自动生成代码_如何让Mybatis自动生成代码
weixin_35316354的博客
02-02 371
在使用 mybatis 过程中， 当手写 JavaBean 和 XML 写的越来越多的时候， 就越来越容易出错。这种重复性的工作， 我们当然不希望做那么多。还好， mybatis 为我们提供了强大的代码生成：MybatisGenerator。通过简单的配置，我们就可以生成各种类型的实体类，Mapper接口，MapperXML文件， Example对象等。通过这些生成的文件，我们就可以方便的进行单表...
mybatis sql自动生成
alvin_010的博客
11-11 2422
本文提供了一种自动生成sql语句的方法。
spring-boot-mybatis-interceptor:mybatis自定义主键插件,通过该插件生成分布式唯一雪花ID作为数据库表的主键
05-14
Mybatis自定义插件生成...通过拦截器拦截Mybatis的insert语句,通过自定义注解获取到主键，并为该主键赋值雪花ID,插入数据库中。 2、技术架构 项目总体技术选型 SpringBoot2.1.7 + Mybatis + Maven3.5.4 + Mysql + lom
Java自动生成数据库表实体和mybatis接口以及xml文件
12-05
在Java开发中，尤其在基于Oracle或MySQL数据库的项目中，使用MyBatis作为持久层框架时，手动编写数据库表的实体类、MyBatis接口和对应的XML文件是一项耗时且容易出错的工作。为了解决这个问题，开发者通常会利用代码...
SpringBoot使用mybatis批量新增500万数据到mysql数据库Demo
最新发布
07-02
7. **代码结构**：项目的`pom.xml`文件用于管理依赖，`src`目录包含源代码，`.idea`和`.mvn`目录则是IntelliJ IDEA和Maven的配置文件，`target`目录则是编译后生成的类文件和最终的可执行jar包。 综上所述，这个...
MySQL中REPLACE INTO和INSERT INTO的区别分析
09-11
REPLACE的运行与INSERT很相似。只有一点例外，假如表中的一个旧记录与一个用于PRIMARY KEY或一个UNIQUE索引的新记录具有相同的值，则在新记录被插入之前，旧记录被删除。
mybatis+mysql 通过sql脚本生成mapper的部分内容
weixin_30341745的博客
10-09 391
SQL mysql SELECT concat('<if test="', COLUMN_NAME, ' != null"> ', COLUMN_NAME, ',</if>') FROM information_schema.COLUMNS WHERE TABLE_NAME = 'view_opportunity' AND TABLE_SCHEMA = ...
MyBatis 3 自动生成 主键 针对不同的数据库（oracle/sqlserver/mysql）
weixin_33919941的博客
04-02 123
　　MyBatis自动生成的主键很多数据库支持自动生成主键的数据类型。不过这通常（并不总是）是个私有的特性。SQL Map 通过&lt;insert&gt;的子元素&lt;selectKey&gt;来支持自动生成的键值。它同时支持预生成（如Oracle）和后生成两种类型（如 MS-SQL Server MySQL）。 一、Oracle设置 &lt;insert id="saveUserIn...
简单介绍利用反射和注解从而生成SQL语句
weixin_34128839的博客
05-13 340
很多框架都是利用反射和注解从而简化配置。那么它是一个怎么的原理呢？这里我就以创建表为例，介绍一下如何利用反射和注解从而生成SQL语句。 1.先自定义两个注解XgcField和XgcTable XgcTable package com.xgc.test.annotation; import java.lang.annotation.ElementType; import java.lang.ann...
mysql生成mybatis配置文件xml
孤独灬行者的博客
09-26 387
1、主要语句 判断sql 语句： select CONCAT('<if test="', -- 驼峰 CONCAT( SUBSTRING_INDEX(LOWER(column_name),'_',1), substr(UPPER(column_name), length(SUBSTRING_INDEX(column_name,'_',1)) + 2,1), substr(LOWER(c...
MySQL之MyBatis初体验
tysearch的博客
11-30 1168
mysql之MyBatis初体验一、MyBatis概述1.1 直接操作jdbc的弊端1.2 MyBatis框架如何解决JDBC弊端1.3 MyBatis框架简介二、MyBatis入门2.1 入门Demo2.2 使用MyBatis完成传统表单CRUD2.3 抽取工具类2.4 动态代理方式实现CURD三、使用MyBatis完成复杂功能3.1 条件查询-包装类3.2 条件查询-arg或者param3.3 条件查询-使用@Param 一、MyBatis概述 1.1 直接操作jdbc的弊端 频繁创建和销毁数据库连接
达梦数据库用户和表空间常用SQL语句
lilv1989的博客
04-06 2427
达梦数据库用户和表空间常用SQL语句 我使用的环境是DM7，达梦数据库默认情况下可以使用SYSDBA用户使用默认表空间进行数据库操作。但是对于业务开发来说，使用普通用户和表空间进行使用保证数据的区分。 1、表空间 达梦数据库创建表空间的语法 CREATE TABLESPACE <表空间名> <数据文件子句>[<数据页缓冲池子句>][<存储加密子句>] <数据文件子句> ::= DATAFILE <文件说明项>{,&...
mybatis中使用replace into和 insert INTO … ON DUPLICATE KEY UPDATE批量操作
热门推荐
OK
11-17 4万+
一、replace into REPLACE INTO table_name (product_id,departs_date,price_value) VALUES <foreach collection="list" index="index" item="item" separator=","> (#{item.productId},#{item.depar
Spring Boot入门：注解式Mybatis集成教程与实战
文章可能提供了包含这些注解的实际代码片段，例如`@Entity`注解用于标注User类，`@Mapper`注解用于标注Mapper接口，以及如何利用`@Insert`、`@Select`等注解来声明SQL语句。 总结来说，这篇教程适合初学者快速上手...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值