通过POJO生成MySQL的DDL语句

news2024/11/15 2:18:06

背景

有时候下载的源码没有数据库的DDL语句，需要手动去创建，这就很麻烦了，这时需要一个利器通过POJO对象生成DDL语句，一键生成，直接执行即可。

工程结构示例

pom.xml文件

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>org.example</groupId>
    <artifactId>pojoToMysql</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>11</maven.compiler.source>
        <maven.compiler.target>11</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.reflections</groupId>
            <artifactId>reflections</artifactId>
            <version>0.9.11</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.10</version>
        </dependency>
        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>4.2.1</version>
        </dependency>
    </dependencies>

</project>

相关注解

@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface FieldPlus {

    boolean isExist() default true;
    // 注释
    String comment() default "";
    // not null 标识
    boolean isNotNull() default true;

    String defaultValue() default "";

    int length() default 0;
}
@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Indexes {

    /**
     * 命名规则 1-索引类型 2-字段名称组合(以|分隔) 3-索引名称 以:分隔
     * 示例 index:anchor_id|merchant_id:index_ac_mc
     *     unique:anchor_id|merchant_id:uq_ac_mc
     * @return 索引数组
     */
    String[] index() default "";
    
}
@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface PrimaryKey {

    String value() default "id";
    
}

相关的实体类

@Data
public class Entity {

    private String tableName;

    private List<Property> properties;

    /**
     * 索引
     * primaryKey 主键
     * uniqueIndex 唯一索引
     * index 普通索引
     */
    private List<Tuple> indexes;
}
@Data
public class Property {

    /**
     * 字段名字
     */
    private String fieldName;

    /**
     * 字段定义
     */
    private String fieldDefinition;

    /**
     * 注释
     */
    private String comment;

    /**
     * 非空标识
     */
    private boolean notNullSign;

}

扫描包的类

public class PackageScanner {

    public List<Class<?>> scanPackage(String packageName) throws IOException, ClassNotFoundException {
        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
        String path = packageName.replace('.', '/');
        Enumeration<URL> resources = classLoader.getResources(path);
        List<Class<?>> classes = new ArrayList<>();

        while (resources.hasMoreElements()) {
            URL resource = resources.nextElement();
            File file = new File(resource.getFile());
            if (file.isDirectory()) {
                for (File classFile : Objects.requireNonNull(file.listFiles())) {
                    if (classFile.getName().endsWith(".class")) {
                        String className = packageName + "." + classFile.getName().replace(".class", "");
                        Class<?> clazz = Class.forName(className);
                        classes.add(clazz);
                    }
                }
            }
        }

        return classes;
    }
}

两个需要生成DDL的示例类

@PrimaryKey()
@Indexes(index = {"index:merchant_id|anchor_id:index_mc_ac","unique:id|anchor_id|merchant_id:uq_id_ac_mc"})
public class LiveInfo {

    @FieldPlus(length = 20)
    private Long id;
    @FieldPlus(comment = "商户ID")
    private Integer merchantId;
    @FieldPlus(comment = "主播ID")
    private Long anchorId;
    @FieldPlus(comment = "房间地址", length = 256)
    private String pullAddress;
    private Integer liveType;

    @FieldPlus(comment = "用户名称")
    private String username;

    @FieldPlus(comment = "开始时间")
    private LocalDateTime createTime;
}
@PrimaryKey(value = "id")
public class UserInfo {


    private Long id;

    @FieldPlus(comment = "用户名称")
    private String username;

    private Date updateTime;

}

驼峰转下横杠工具类

public class StringUtils {

    public static String toUnderScoreCase(String camelCaseStr) {
        StringBuilder builder = new StringBuilder();
        for (int i = 0; i < camelCaseStr.length(); i++) {
            char c = camelCaseStr.charAt(i);
            if (Character.isUpperCase(c)) {
                if (i > 0) {
                    builder.append('_');
                }
                builder.append(Character.toLowerCase(c));
            } else {
                builder.append(c);
            }
        }
        return builder.toString();
    }

}

核心方法

public class Main {
    public static void main(String[] args) throws IOException, ClassNotFoundException {
        PackageScanner packageScanner = new PackageScanner();
        // 获取pojo clazz文件
        List<Class<?>> classes = packageScanner.scanPackage("org.example.pojo");

        List<Entity> entityList = new ArrayList<>();
        // 遍历完封装成对象
        for (Class<?> clazz : classes) {
            Entity entity = new Entity();
            String substring = clazz.getName().substring(clazz.getName().lastIndexOf(".") + 1);
            String tableName = StringUtils.toUnderScoreCase(substring);
            entity.setTableName(tableName);
            // 获取类上的索引
            List<Tuple> indexList = new ArrayList<>();
            PrimaryKey primaryKey = clazz.getAnnotation(PrimaryKey.class);
            if (Objects.nonNull(primaryKey)){
                //indexHashMap.put(primaryKey.value(), "primaryKey");
                indexList.add(new Tuple("primaryKey", primaryKey.value()));
            }
            Indexes indexes = clazz.getAnnotation(Indexes.class);
            if (Objects.nonNull(indexes)){
                for (String index : indexes.index()) {
                    String[] split = index.split(":");
                    indexList.add(new Tuple(split[0], split[1], split[2]));
                }
            }

            entity.setIndexes(indexList);

            // 遍历属性
            Field[] fields = clazz.getDeclaredFields();
            List<Property> propertyList = new ArrayList<>();
            for (Field field : fields) {
                Property property = new Property();
                String fieldName = StringUtils.toUnderScoreCase(field.getName());
                property.setFieldName(fieldName);
                String typeName = field.getType().getName();
                //System.out.println("属性名" + name);
                // 注解处理
                FieldPlus fieldPlus = field.getAnnotation(FieldPlus.class);
                if (Objects.equals("java.lang.String",typeName)){
                    if (Objects.nonNull(fieldPlus) && fieldPlus.length() > 0){
                        property.setFieldDefinition("varchar(" + fieldPlus.length() + ")");
                    } else {
                        property.setFieldDefinition("varchar(128)");
                    }
                } else if (Objects.equals("java.lang.Long",typeName) || Objects.equals("java.lang.Integer",typeName)){
                    if (Objects.nonNull(fieldPlus) && fieldPlus.length() > 0 && fieldPlus.length() <= 3){
                        property.setFieldDefinition("tinyint(" + fieldPlus.length() + ")");
                    } else if (Objects.nonNull(fieldPlus) && fieldPlus.length() > 3 && fieldPlus.length() <= 10) {
                        property.setFieldDefinition("int(" + fieldPlus.length() + ")");
                    } else if (Objects.nonNull(fieldPlus) && fieldPlus.length() > 10) {
                        property.setFieldDefinition("bigint(" + fieldPlus.length() + ")");
                    } else {
                        property.setFieldDefinition("int(10)");
                    }
                } else if (Objects.equals("java.time.LocalDateTime", typeName) || Objects.equals("java.util.Date", typeName)) {
                    property.setFieldDefinition("timestamp");
                }
                // 注解处理
                if (Objects.nonNull(fieldPlus)){
                    property.setComment(Objects.nonNull(fieldPlus.comment()) && !("").equals(fieldPlus.comment())
                            ? fieldPlus.comment() : null );
                    property.setNotNullSign(fieldPlus.isNotNull());
                }

                propertyList.add(property);
            }
            entity.setProperties(propertyList);
            entityList.add(entity);

        }

        // 遍历对象生成DDL语句输出txt文档
        FileOutputStream fos = new FileOutputStream("init.txt");

        for (Entity entity : entityList) {
            String beforeSql = "CREATE TABLE `{}` (";
            String tableName = StrUtil.format(beforeSql, entity.getTableName());
            fos.write(tableName.getBytes());
            // 写入换行符（根据操作系统的不同，换行符可能会有所差异，Windows为"\r\n"，Unix/Linux为"\n"）
            fos.write(System.getProperty("line.separator").getBytes());

            // 填充字段
            for (Property property : entity.getProperties()) {
                String field = "  " + property.getFieldName() + " " + property.getFieldDefinition();
                if (property.isNotNullSign()){
                    field = field + " NOT NULL";
                }
                if (Objects.nonNull(property.getComment())){
                    field = field + " COMMENT '" + property.getComment() +"'";
                }
                field = field + ",";
                fos.write(field.getBytes());
                fos.write(System.getProperty("line.separator").getBytes());
            }

            if (entity.getIndexes() != null && entity.getIndexes().size() > 0){
                for (int i = 0; i < entity.getIndexes().size(); i++) {
                    Tuple tuple = entity.getIndexes().get(i);
                    if ("primaryKey".equals(tuple.get(0))){
                        String primaryKeyFormat = "  PRIMARY KEY (`{}`) USING BTREE,";
                        primaryKeyFormat = processEndComma(entity, i, primaryKeyFormat);
                        String primaryKey = StrUtil.format(primaryKeyFormat, (String)tuple.get(1));
                        fos.write(primaryKey.getBytes());
                        fos.write(System.getProperty("line.separator").getBytes());
                    } else if ("index".equals(tuple.get(0))){
                        String indexFormat = "  KEY `{}` ({}) USING BTREE,";
                        indexFormat = processEndComma(entity, i, indexFormat);
                        String[] split = ((String) tuple.get(1)).split("\\|");
                        StringBuilder fieldArray = new StringBuilder();
                        for (String s : split) {
                            fieldArray.append("`").append(s).append("`").append(",");
                        }
                        String index = StrUtil.format(indexFormat, tuple.get(2), fieldArray.substring(0, fieldArray.length()-1));
                        fos.write(index.getBytes());
                        fos.write(System.getProperty("line.separator").getBytes());
                    } else if ("unique".equals(tuple.get(0))){
                        String uniqueIndexFormat = "  UNIQUE KEY `{}` ({}) USING BTREE,";
                        uniqueIndexFormat = processEndComma(entity, i, uniqueIndexFormat);
                        String[] split = ((String) tuple.get(1)).split("\\|");
                        StringBuilder fieldArray = new StringBuilder();
                        for (String s : split) {
                            fieldArray.append("`").append(s).append("`").append(",");
                        }
                        String index = StrUtil.format(uniqueIndexFormat, tuple.get(2), fieldArray.substring(0, fieldArray.length()-1));
                        fos.write(index.getBytes());
                        fos.write(System.getProperty("line.separator").getBytes());
                    }
                }
            }


            String endSql = ") ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;";
            fos.write(endSql.getBytes());
            fos.write(System.getProperty("line.separator").getBytes());
            fos.write(System.getProperty("line.separator").getBytes());
        }

        fos.close();
    }

    private static String processEndComma(Entity entity, int i, String indexFormat) {
        if (i == entity.getIndexes().size()-1){
            indexFormat = indexFormat.replace(",","");
        }
        return indexFormat;
    }

}

最后的成果

CREATE TABLE `live_info` (
  id bigint(20) NOT NULL,
  merchant_id int(10) NOT NULL COMMENT '商户ID',
  anchor_id int(10) NOT NULL COMMENT '主播ID',
  pull_address varchar(256) NOT NULL COMMENT '房间地址',
  live_type int(10),
  username varchar(128) NOT NULL COMMENT '用户名称',
  create_time timestamp NOT NULL COMMENT '开始时间',
  PRIMARY KEY (`id`) USING BTREE,
  KEY `index_mc_ac` (`merchant_id`,`anchor_id`) USING BTREE,
  UNIQUE KEY `uq_id_ac_mc` (`id`,`anchor_id`,`merchant_id`) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;

CREATE TABLE `user_info` (
  id int(10),
  username varchar(128) NOT NULL COMMENT '用户名称',
  update_time timestamp,
  PRIMARY KEY (`id`) USING BTREE
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/2057788.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！

相关文章

如何在 Git 中安全撤销提交与更改

如何在 Git 中安全撤销提交与更改

文章目录 前言一、Git Reset1. --soft&#xff1a;保留变更在暂存区2. --mixed&#xff08;默认选项&#xff09;&#xff1a;保留变更在工作区3. --hard&#xff1a;彻底丢弃所有变更 二、Git Revert1. 撤销单个提交2. 撤销多个提交3. 撤销合并提交 三、实际例子总结 前言 在…
阅读更多...
你知道手机零部件尺寸检测的重要性吗？

你知道手机零部件尺寸检测的重要性吗？

手机零部件作为手机制造行业的基础&#xff0c;其品质的优劣直接关系到行业的发展&#xff0c;所以加强手机精密零部件尺寸检测非常重要。如今&#xff0c;手机零部件变得更加精细&#xff0c;对质量的要求也在不断提高&#xff0c;随着生产规模逐渐扩大&#xff0c;传统的检测…
阅读更多...
java ssl使用自定义证书

java ssl使用自定义证书

1.证书错误 Caused by: sun.security.provider.certpath.SunCertPathBuilderException: unable to find valid certification path to requested target 2.生成客户端证书 openssl x509 -in <(openssl s_client -connect 192.168.11.19:8101 -prexit 2>/dev/null) -ou…
阅读更多...
C语言 | Leetcode C语言题解之第355题设计推特

C语言 | Leetcode C语言题解之第355题设计推特

题目&#xff1a; 题解&#xff1a; typedef struct {int tweetId;int userId; } Tweet;typedef struct {int* dict[501];Tweet* tweetList;int tweetListLen; } Twitter;Twitter* twitterCreate() {Twitter* obj malloc(sizeof(Twitter));for (int i 0; i < 501; i) {ob…
阅读更多...
【Linux】 gdb-调试器初入门（简单版使用）

【Linux】 gdb-调试器初入门（简单版使用）

&#x1f525;系列文章&#xff1a;《Linux入门》 目录 一、背景 二、什么是GDB &#x1f337;定义 &#x1f337;GDB调试工具---提供的帮助 三、GDB的安装教程-Ubuntu &#x1f337;gdb的安装 四、哪类程序可被调试 &#x1f337;程序的发布方式 &#x1f337;Debug版…
阅读更多...
力扣 | 背包dp | 279. 完全平方数、518. 零钱兑换 II、474. 一和零、377. 组合总和 Ⅳ

力扣 | 背包dp | 279. 完全平方数、518. 零钱兑换 II、474. 一和零、377. 组合总和 Ⅳ

文章目录 一、279. 完全平方数二、518. 零钱兑换 II三、474. 一和零四、377. 组合总和 Ⅳ 一、279. 完全平方数 LeetCode&#xff1a;279. 完全平方数 朴素想法&#xff1a; 这个题目最简单的想法是&#xff0c;可以用 O ( n n ) O(n\sqrt{}n) O(n ​n)的动态规划解决&#x…
阅读更多...
OpenCV几何图像变换（1）映射转换函数convertMaps()的使用

OpenCV几何图像变换（1）映射转换函数convertMaps()的使用

操作系统&#xff1a;ubuntu22.04 OpenCV版本&#xff1a;OpenCV4.9 IDE:Visual Studio Code 编程语言&#xff1a;C11 算法描述 将图像变换映射从一种表示形式转换为另一种表示形式。 该函数将用于 remap 的映射对从一种表示形式转换为另一种表示形式。以下选项 ((map1.type…
阅读更多...
车辆类型检测算法、车辆类型源码及其样本与模型解析

车辆类型检测算法、车辆类型源码及其样本与模型解析

车辆类型检测算法是利用计算机视觉和深度学习技术&#xff0c;对车辆图像进行自动分析和识别&#xff0c;以判断车辆的类型&#xff08;如轿车、SUV、货车等&#xff09;的一种算法。以下是对车辆类型检测算法的详细解析&#xff1a; 一、算法基础 车辆类型检测算法的基础是图…
阅读更多...
区间预测|基于长短期记忆网络LSTM分位数单变量时间序列区间预测Matlab程序QRLSTM

区间预测|基于长短期记忆网络LSTM分位数单变量时间序列区间预测Matlab程序QRLSTM

区间预测|基于长短期记忆网络LSTM分位数单变量时间序列区间预测Matlab程序QRLSTM 文章目录 前言区间预测|基于长短期记忆网络LSTM分位数单变量时间序列区间预测Matlab程序QRLSTM 一、QRLSTM模型1. 基本原理1.1 LSTM (Long Short-Term Memory)1.2 量化回归&#xff08;Quantile …
阅读更多...
移动端GenAI应用的崛起：从市场规模到成功案例分析

移动端GenAI应用的崛起：从市场规模到成功案例分析

随着生成式人工智能&#xff08;GenAI&#xff09;技术的飞速发展&#xff0c;移动应用市场正经历一场前所未有的变革。从图像编辑到聊天机器人&#xff0c;这些基于AI的应用不仅满足了用户日益增长的需求&#xff0c;也为企业带来了巨大的商业机遇。本文将探讨这一领域的最新趋…
阅读更多...
网站建设中：高效利用Robots.txt文件的策略与实践

网站建设中：高效利用Robots.txt文件的策略与实践

原文&#xff1a;网站建设中&#xff1a;高效利用Robots.txt文件的策略与实践 - 孔乙己大叔 (rebootvip.com) 在网站中使用robots.txt文件是一种控制搜索引擎爬虫访问网站内容的方法。以下是关于如何在网站中使用robots.txt的详细步骤和注意事项&#xff1a; 一、创建robots.t…
阅读更多...
集团数字化转型方案（四）

集团数字化转型方案（四）

集团数字化转型方案通过全面部署人工智能&#xff08;AI&#xff09;、大数据分析、云计算和物联网&#xff08;IoT&#xff09;技术&#xff0c;创建了一个智能化的企业运营平台&#xff0c;涵盖从业务流程自动化、实时数据监控、精准决策支持&#xff0c;到个性化客户服务和高…
阅读更多...
PV、UV、IP：网站流量分析的关键指标

PV、UV、IP：网站流量分析的关键指标

原文&#xff1a;PV、UV、IP&#xff1a;网站流量分析的关键指标 - 孔乙己大叔 (rebootvip.com) 摘要&#xff1a; 在浩瀚的互联网海洋中&#xff0c;PV&#xff08;Page View&#xff0c;页面浏览量&#xff09;、UV&#xff08;Unique Visitor&#xff0c;独立访客数…
阅读更多...
Eclipse SVN 插件在线下载地址

Eclipse SVN 插件在线下载地址

Eclipse SVN 插件 Subversive 在线安装 1、选择help下的install new software 2、点击 add 3、Name随便写&#xff0c;Location输入&#xff1a; https://download.eclipse.org/technology/subversive/4.8/release/latest/ 点击Add 4、然后一直下一步&#xff0c;Finish&am…
阅读更多...
【QT】——1_QT学习笔记

【QT】——1_QT学习笔记

一、QT是什么&#xff1f; QT 是一个功能强大、应用广泛的跨平台 C 应用程序开发框架&#xff0c;它不仅提供了丰富多样、美观实用的图形界面组件&#xff0c;还具备高效灵活的信号与槽通信机制&#xff0c;能够帮助开发者轻松构建出复杂且性能优越的应用程序&#xff0c;广泛…
阅读更多...
VS Code中基于MSTest编写和运行测试

VS Code中基于MSTest编写和运行测试

MS Test&#xff08;Microsoft Test Framework&#xff09;是微软提供的一个用于.NET应用程序的单元测试框架。以下是一个使用MS Test进行单元测试的示例&#xff0c;该示例将涵盖测试的基本步骤和概念。 项目搭建 在VS Code中开发C#时&#xff0c;创建solution&#xff08;解…
阅读更多...
AI绘画Stable Diffusion画全身图总是人脸扭曲？ADetailer插件实现一键解决！商业级AI人物生成教程

AI绘画Stable Diffusion画全身图总是人脸扭曲？ADetailer插件实现一键解决！商业级AI人物生成教程

大家好&#xff0c;我是灵魂画师向阳 你是否遇到过SD生成的人物脸部扭曲、甚至令人恶心的情况&#xff1f;也曾感到束手无策&#xff1f;别担心&#xff0c;这份教程专为你而来。 在使用SD生成人物全身照时&#xff0c;你可能经常发现人物的脸部会出现扭曲问题。这是因为人物…
阅读更多...
整体思想以及取模

整体思想以及取模

前言&#xff1a;一开始由于失误&#xff0c;误以为分数相加取模不能&#xff0c;但是其实是可以取模的 这个题目如果按照一般方法&#xff0c;到达每个节点再进行概率统计&#xff0c;但是不知道为什么只过了百分之十五的测试集 题目地址 附上没过关的代码 #include<bits…
阅读更多...
如何在IIS中为typecho博客启用HTTPS访问

如何在IIS中为typecho博客启用HTTPS访问

在上篇文章中&#xff0c;介绍了如何安装typecho博客系统&#xff0c;默认是没有启用https访问的&#xff0c;这篇文章介绍如何 在IIS中开启 https访问。 开启https访问需要两个步骤&#xff1a; 1、申请 一个ssl证书&#xff0c;我这里以阿里云上面的申请流程为例。其它云服务…
阅读更多...
[Linux网络】基本网络命令socket编写TCP应用层实现简易计算器

[Linux网络】基本网络命令socket编写TCP应用层实现简易计算器

W...Y的主页 &#x1f60a; 代码仓库分享&#x1f495; 前言&#xff1a;我们在上篇博客中学习了使用socket套接字完成了UDP的网络编程&#xff0c;今天我们继续使用套接字完成TCP的学习。 首先我们先来了解一些网络指令&#xff0c;让大家可以在实现网络编程后查看一些与网…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023