前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Java代码实体类生成SQL语句(Java实体类转数据库)

Java代码实体类生成SQL语句(Java实体类转数据库)

作者头像
全栈程序员站长
发布2022-08-23 12:57:21
3.8K1
发布2022-08-23 12:57:21
举报
文章被收录于专栏:全栈程序员必看

大家好,又见面了,我是你们的朋友全栈君。

有的时候把数据删了,如果照着实体类重新创建数据库的话比较麻烦,可以使用这个工具,把代码复制到项目里面设置一下即可把Java代码中的实体类转换为SQL语句输出为一个文件,打开执行命令即可。

下载:https://download.csdn.net/download/weixin_44893902/20367422

在这里插入图片描述
在这里插入图片描述

代码如下:

代码语言:javascript
复制
package Main;

import javax.xml.bind.annotation.XmlElement;
import java.io.*;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;

public class GenerateSqlFromEntityUtil { 
   

	public static void main(String[] a) { 
   
		// 实体类的位置
		Class klass = cn.ac.azure.model.User.class;
		// 生成的sql语句的位置
		String outputPath = "D:/outSql/User.txt";
		generateTableSql(klass, outputPath, null);
		System.out.println("生成结束");
	}

	public static void writeFile(String content, String outputPath) { 
   
		File file = new File(outputPath);
		System.out.println("文件路径: " + file.getAbsolutePath());
		// 输出文件的路径
		if (!file.getParentFile().exists()) { 
   
			file.getParentFile().mkdirs();
		}
		FileOutputStream fos = null;
		OutputStreamWriter osw = null;
		BufferedWriter out = null;

		try { 
   
			// 如果文件存在,就删除
			if (file.exists()) { 
   
				file.delete();
			}
			file.createNewFile();
			fos = new FileOutputStream(file, true);
			osw = new OutputStreamWriter(fos);
			out = new BufferedWriter(osw);
			out.write(content);
			// 清空缓冲流,把缓冲流里的文本数据写入到目标文件里
			out.flush();
		} catch (FileNotFoundException e) { 
   
			e.printStackTrace();
		} catch (IOException e) { 
   
			e.printStackTrace();
		} finally { 
   
			try { 
   
				fos.close();
			} catch (IOException e) { 
   
				e.printStackTrace();
			}

			try { 
   
				osw.close();
			} catch (IOException e) { 
   
				e.printStackTrace();
			}

			try { 
   
				out.close();
			} catch (IOException e) { 
   
				e.printStackTrace();
			}
		}
	}

	public static void generateTableSql(Class obj, String outputPath, String tableName) { 
   
		// tableName 如果是 null,就用类名做表名
		if (tableName == null || tableName.equals("")) { 
   
			tableName = obj.getName();
			tableName = tableName.substring(tableName.lastIndexOf(".") + 1);
		}
		// 表名用大写字母
		tableName = tableName.toUpperCase();

		Field[] fields = obj.getDeclaredFields();
		Object param;
		String column;

		StringBuilder sb = new StringBuilder();

		sb.append("drop table if exists ").append(tableName).append(";\r\n");

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

		sb.append("create table ").append(tableName).append("(\r\n");

		System.out.println(tableName);

		boolean firstId = true;

		for (int i = 0; i < fields.length; i++) { 
   
			Field f = fields[i];

			column = f.getName();

			System.out.println(column + ", " + f.getType().getSimpleName());

			param = f.getType();
			sb.append(column); // 一般第一个是主键

			if (param instanceof Integer) { 
   
				sb.append(" INTEGER ");
			} else { 
   
				// 注意:根据需要,自行修改 varchar 的长度。这里设定为长度等于 50
				int length = 50;
				sb.append(" VARCHAR(" + length + ")");
			}

			if (firstId == true) { 
   
				sb.append(" PRIMARY KEY ");
				firstId = false;
			}

			// 获取字段中包含 fieldMeta 的注解

			// 获取属性的所有注释
			Annotation[] allAnnotations = f.getAnnotations();

			XmlElement xmlElement = null;
			Class annotationType = null;

			for (Annotation an : allAnnotations) { 
   
				sb.append(" COMMIT '");
				xmlElement = (XmlElement) an;
				annotationType = an.annotationType();
				param = ((XmlElement) an).name();
				System.out.println("属性 " + f.getName() + " ----- 的注释类型有: " + param);
				sb.append(param).append("'");
			}

			if (i != fields.length - 1) { 
    // 最后一个属性后面不加逗号
				sb.append(", ");
			}

			sb.append("\n");
		}

		String sql = sb.toString();

		sql = sb.substring(0, sql.length() - 1) + "\n) " + "ENGINE = INNODB DEFAULT CHARSET = utf8;";

		writeFile(sql, outputPath);
	}
}

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/137938.html原文链接:https://javaforall.cn

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022年5月6,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 代码如下:
相关产品与服务
数据库
云数据库为企业提供了完善的关系型数据库、非关系型数据库、分析型数据库和数据库生态工具。您可以通过产品选择和组合搭建,轻松实现高可靠、高可用性、高性能等数据库需求。云数据库服务也可大幅减少您的运维工作量,更专注于业务发展,让企业一站式享受数据上云及分布式架构的技术红利!
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档