Java 实体类 创建表 ---生成sql语句

package cn.com.onezerobit.oms.po;

import java.io.File;
import java.io.FileOutputStream;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;


/**
 * sql自动生成
 * 
 * @author
 */
public class SqlGenerator {

	public static void main(String[] args) {

		String className = "com.model.Studet";
		List<String> allClasses = getAllClasses(
				"E:\\ss");
		// 获取包下的所有类名称
		String prefix = "cn.xxx.";
		String sql = generateSql(className);
		System.out.println(sql);
//		for (String str : allClasses) {
//			className = prefix + str.substring(0, str.lastIndexOf("."));
//			String sql = generateSql(className);
//			System.out.println(sql.toString());
//		}

	}

	/**
	 * 根据实体类生成建表语句
	 * 
	 * @author
	 * @param className 全类名
	 * @param filePath  磁盘路径 如 : d:/workspace/
	 */
	public static String generateSql(String className) {
		try {
			Class<?> clz = Class.forName(className);
			className = clz.getSimpleName();
			Field[] fields = clz.getDeclaredFields();
			StringBuffer column = new StringBuffer();
//			String varchar = " varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci DEFAULT NULL,";
			String varchar = " int(4) NULL ,";
			
			for (Field f : fields) {
				
				column.append(" \n `" + camelToUnderline(f.getName()) + "`").append(varchar);
			}
			StringBuffer sql = new StringBuffer();
			String tableName=camelToUnderline("TUps"+className);
			sql.append("\n DROP TABLE IF EXISTS `" + tableName + "`; ")
					.append(" \n CREATE TABLE `" + tableName + "`  (")
					.append(" \n `id` int(11) NOT NULL AUTO_INCREMENT,").append(" \n " + column)
					.append(" \n PRIMARY KEY (`id`) USING BTREE,").append("\n INDEX `id`(`id`) USING BTREE")
					.append(" \n ) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8 COLLATE = utf8_general_ci;");
			return sql.toString();
		} catch (ClassNotFoundException e) {
			return null;
		}

	}

	public static final char UNDERLINE = '_';

	public static String camelToUnderline(String param) {
		if (param == null || "".equals(param.trim())) {
			return "";
		}
		int len = param.length();
		StringBuilder sb = new StringBuilder(len);
		for (int i = 0; i < len; i++) {
			char c = param.charAt(i);
			if (Character.isUpperCase(c)) {
				sb.append(UNDERLINE);
				sb.append(Character.toLowerCase(c));
			} else {
				sb.append(c);
			}
		}
		return sb.toString();
	}

	/**
	 * 获取包下的所有类名称,获取的结果类似于 XXX.java
	 * 
	 * @author
	 * @param packageName
	 * @return
	 */
	public static List<String> getAllClasses(String packageName) {
		List<String> classList = new ArrayList<String>();
		String className = "";
		File f = new File(packageName);
		if (f.exists() && f.isDirectory()) {
			File[] files = f.listFiles();
			for (File file : files) {
				className = file.getName();
				classList.add(className);
			}
			return classList;
		} else {
			return null;
		}
	}

	/**
	 * 将string 写入sql文件
	 * 
	 * @author
	 * @param str
	 * @param path
	 */
	public static void StringToSql(String str, String path) {
		byte[] sourceByte = str.getBytes();
		if (null != sourceByte) {
			try {
				File file = new File(path); // 文件路径(路径+文件名)
				if (!file.exists()) { // 文件不存在则创建文件,先创建目录
					File dir = new File(file.getParent());
					dir.mkdirs();
					file.createNewFile();
				}
				FileOutputStream outStream = new FileOutputStream(file); // 文件输出流用于将数据写入文件
				outStream.write(sourceByte);
				outStream.flush();
				outStream.close(); // 关闭文件输出流
				System.out.println("生成成功");
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}
}

支持驼峰    缺点 不能智能识别类型 

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
Java实体类生成SQL语句的方法有很多种,下面介绍两种比较常见的方法。 方法一:使用反射机制 使用反射机制可以获取Java实体类中的属性和对应的值,然后拼接成SQL语句。 示例代码: ``` public static String createInsertSql(Object obj, String tableName) throws IllegalAccessException { Field[] fields = obj.getClass().getDeclaredFields(); StringBuilder sb = new StringBuilder(); sb.append("insert into ").append(tableName).append("("); for (int i = 0; i < fields.length; i++) { if (i > 0) { sb.append(","); } String fieldName = fields[i].getName(); sb.append(fieldName); } sb.append(") values ("); for (int i = 0; i < fields.length; i++) { if (i > 0) { sb.append(","); } fields[i].setAccessible(true); Object value = fields[i].get(obj); if (value instanceof String) { sb.append("'").append(value).append("'"); } else { sb.append(value); } } sb.append(")"); return sb.toString(); } ``` 方法二:使用第三方库 可以使用第三方库,如MyBatis Generator,自动生成Java实体类对应的SQL语句。 示例代码: ``` <generatorConfiguration> <context id="Mysql" targetRuntime="MyBatis3"> <jdbcConnection driverClass="com.mysql.jdbc.Driver" connectionURL="jdbc:mysql://localhost:3306/test" userId="root" password="root"> </jdbcConnection> <javaModelGenerator targetPackage="com.example.model" targetProject="src/main/java"> <property name="enableSubPackages" value="true" /> <property name="trimStrings" value="true" /> </javaModelGenerator> <sqlMapGenerator targetPackage="com.example.mapper" targetProject="src/main/resources"> <property name="enableSubPackages" value="true" /> </sqlMapGenerator> <javaClientGenerator type="XMLMAPPER" targetPackage="com.example.mapper" targetProject="src/main/java"> <property name="enableSubPackages" value="true" /> </javaClientGenerator> <table tableName="user"></table> </context> </generatorConfiguration> ``` 以上是两种常见的Java实体类生成SQL语句的方法,具体使用哪种方法可以根据实际情况选择。
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值