iis服务器助手广告广告
返回顶部
首页 > 资讯 > 后端开发 > Python >如何让java只根据数据库表名自动生成实体类
  • 361
分享到

如何让java只根据数据库表名自动生成实体类

2024-04-02 19:04:59 361人浏览 八月长安

Python 官方文档:入门教程 => 点击学习

摘要

根据数据库表名生成实体类 公司用的jpa,没有用mybatis。所以也没有用mybatis自动生成。但有些数据库表字段太多,就想着一劳永逸了,连数据库注释都搞上去 第一种 这里使用的

根据数据库表名生成实体类

公司用的jpa,没有用mybatis。所以也没有用mybatis自动生成。但有些数据库表字段太多,就想着一劳永逸了,连数据库注释都搞上去

第一种

这里使用的是jdbcTemplate+Junit测试生成,方式可变。

SpringBoot版本是2.4.4,只需要加上@springBootTest就可以了。不用@RunWith

pom:


<dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>

代码


package com.shinedata.bims.WEB;
 
import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang3.StringUtils;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
 
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 

@SpringBootTest
public class TestController {
 
    @Autowired
    private JdbcTemplate jdbcTemplate;
 
    @Test
    public void database(){
        createEntity("1ceping", "t_evaluation_template_info_copy","com.shinedata.bims.entity",false);
    }
 
    
    public void createEntity(String dataBaseName,String tableName,String packageName,boolean isAddEntitySuffix){
        String className=tableName;
        if(tableName.substring(0,2).equals("t_")){
            StringBuilder stringBuilder = new StringBuilder(tableName);
            stringBuilder.replace(0, 2, "");
            String initialsUpperCase = stringBuilder.substring(0, 1).toUpperCase();
            className=initialsUpperCase+stringBuilder.substring(1);
        }
        className=removeUnderline(className)+(isAddEntitySuffix?"Entity":"");
 
        StringBuffer classBuffer=new StringBuffer();
        classBuffer.append("import java.util.Date;\r\n");
        classBuffer.append("import java.time.LocalDateTime;\r\n");
        classBuffer.append("import com.alibaba.fastJSON.jsONObject;\r\n");
        classBuffer.append("import java.lang.*;\r\n");
        classBuffer.append("import java.math.*;\r\n");
        classBuffer.append("import java.sql.*;\r\n");
        classBuffer.append("import lombok.Data;\r\n\r\n\r\n");
        classBuffer.append("@Data\r\n");
        classBuffer.append("public class " + className + " {\r\n\r\n");
        List<Map> filedMaps = getFiledMaps(dataBaseName, tableName);
        processAllAttrs(classBuffer,filedMaps);
        classBuffer.append("}\r\n");
        markerBean(className,classBuffer.toString(),packageName);
    }
 
    
    public void markerBean(String className, String content, String packageName) {
        String folder = System.getProperty("user.dir") + "/src/main/java/" + packageName.replace(".","/") + "/";
        File file = new File(folder);
        if (!file.exists()) {
            file.mkdirs();
        }
        String fileName = folder + className + ".java";
        try {
            File newjava = new File(fileName);
            FileWriter fw = new FileWriter(newjava);
            fw.write("package\t" + packageName + ";\r\n");
            fw.write(content);
            fw.flush();
            fw.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
 
    
    private void processAllAttrs(StringBuffer sb,List<Map> filedMaps) {
        for (int i = 0; i < filedMaps.size(); i++) {
            Map map = filedMaps.get(i);
            String fieldType = MapUtils.getString(map, "fieldType");
            String fieldName = MapUtils.getString(map, "fieldName");
            String fieldComment = MapUtils.getString(map, "fieldComment");
            if(StringUtils.isNotBlank(fieldComment)){
                sb.append("\t\r\n");
            }
            sb.append("\tprivate " + fieldType + " " + fieldName + ";\r\n\r\n");
        }
    }
 
    
    private List<Map> getFiledMaps(String dataBaseName,String tableName) {
        String sql="SELECT * FROM infORMation_schema.`COLUMNS` WHERE TABLE_SCHEMA ='"+dataBaseName+"' AND TABLE_NAME = '"+tableName+"'";
        List<Map> tableFieldList = jdbcTemplate.query(sql, new RowMapper<Map>() {
            @Override
            public Map<String,String> mapRow(ResultSet resultSet, int rowNum) throws SQLException {
                Map<String,String> fieldMap = new HashMap();
                String column_name = resultSet.getString("COLUMN_NAME");
                setFieldName(column_name.toLowerCase(),fieldMap);
                String data_type = resultSet.getString("DATA_TYPE");
                setFieldType(data_type.toUpperCase(),fieldMap);
                fieldMap.put("fieldComment",resultSet.getString("COLUMN_COMMENT"));
                return fieldMap;
            }
        });
        return tableFieldList;
    }
 
 
    private void setFieldName(String columnName, Map fieldMap) {
        fieldMap.put("fieldName",removeUnderline(columnName));
    }
 
    
    public String removeUnderline(String string){
        StringBuilder columnNameBuilder=new StringBuilder(string);
        if(!string.contains("_")){
            return string;
        }else {
            int i = columnNameBuilder.indexOf("_");
            columnNameBuilder.replace(i,i+1, "").replace(i,i+1,columnNameBuilder.substring(i,i+1).toUpperCase());
            return removeUnderline(columnNameBuilder.toString());
        }
    }
 
    private void setFieldType(String columnType,Map fieldMap){
        String fieldType="String";
        if(columnType.equals("INT")||columnType.equals("INTEGER")){
            fieldType="Integer";
        }else if(columnType.equals("BIGINT")){
            fieldType="Long";
        }else if(columnType.equals("DATETIME")){
            fieldType="Date";
        }else if(columnType.equals("TEXT")||columnType.equals("VARCHAR")||columnType.equals("TINYTEXT")||columnType.equals("LONGTEXT")){
            fieldType="String";
        }else if(columnType.equals("DOUBLE")){
            fieldType="Double";
        }else if(columnType.equals("BIT")){
            fieldType="Boolean";
        }else if(columnType.equals("FLOAT")){
            fieldType="Float";
        }else if(columnType.equals("DECIMAL")){
            fieldType="BigDecimal";
        }else if(columnType.equals("DATE")){
            fieldType="Date";
        }else if(columnType.equals("TIMESTAMP")){
            fieldType="LocalDateTime";
        }else if(columnType.equals("CHAR")){
            fieldType="Char";
        }else if(columnType.equals("JSON")){//Mysql5.7版本才开始有的
            fieldType="JSONObject";
        }
        fieldMap.put("fieldType",fieldType);
    }
 
}

生成的类

用的lombok,就懒得去搞getset了

第二种

搞的是直接main方法运行下就生成,大同小异。不用Junit,方便一些,就是个工具。和第一种结果一样的


package com.shinedata.bims.web;
 
import org.apache.commons.collections.MapUtils;
import org.apache.commons.lang3.StringUtils;
 
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.*;
import java.util.*;
 

public class TableToEntityUtils {
 
 
    static final String USER = "root";
    static final String PASS = "xxxxxxxxxxx";
//    static final String JDBC_DRIVER = "com.mysql.cj.jdbc.Driver";
    // MySQL 8.0 以下版本 - JDBC 驱动名及数据库 URL
    static final String JDBC_DRIVER = "com.mysql.jdbc.Driver";
    static final String DB_URL = "jdbc:mysql://xxxxxxxxxxxxxxxxxxxxxxx:3306/1ceping?useUnicode=true&characterEncoding=UTF8&zeroDateTimeBehavior=convertToNull&autoReconnect=true&allowMultiQueries=true&rewriteBatchedStatements=true";
 
    public static void main(String[] args) throws Exception {
        createEntity("1ceping", "t_evaluation_template_info_copy","com.shinedata.bims.entity",false);
    }
 
    
    public static void createEntity(String dataBaseName, String tableName, String packageName, boolean isAddEntitySuffix) throws Exception{
        String className=tableName;
        if(tableName.substring(0,2).equals("t_")){
            StringBuilder stringBuilder = new StringBuilder(tableName);
            stringBuilder.replace(0, 2, "");
            String initialsUpperCase = stringBuilder.substring(0, 1).toUpperCase();
            className=initialsUpperCase+stringBuilder.substring(1);
        }
        className=removeUnderline(className)+(isAddEntitySuffix?"Entity":"");
 
        StringBuffer classBuffer=new StringBuffer();
        classBuffer.append("import java.util.Date;\r\n");
        classBuffer.append("import java.time.LocalDateTime;\r\n");
        classBuffer.append("import com.alibaba.fastjson.JSONObject;\r\n");
        classBuffer.append("import java.lang.*;\r\n");
        classBuffer.append("import java.math.*;\r\n");
        classBuffer.append("import java.sql.*;\r\n");
        classBuffer.append("import lombok.Data;\r\n\r\n\r\n");
        classBuffer.append("@Data\r\n");
        classBuffer.append("public class " + className + " {\r\n\r\n");
        List<Map> filedMaps = getFiledMaps(dataBaseName, tableName);
        processAllAttrs(classBuffer,filedMaps);
        classBuffer.append("}\r\n");
        markerBean(className,classBuffer.toString(),packageName);
    }
 
    
    public static void markerBean(String className, String content, String packageName) throws Exception {
//      这里不使用System.getProperty("user.dir")了。user.dir是根据运行时环境来的
        File f2 = new File(TableToEntityUtils.class.getResource("/").getPath());
        String homePath=f2.getCanonicalPath().replace("\\target\\classes", "");
        String folder = homePath + "/src/main/java/" + packageName.replace(".","/") + "/";
        File file = new File(folder);
        if (!file.exists()) {
            file.mkdirs();
        }
        String fileName = folder + className + ".java";
        try {
            File newjava = new File(fileName);
            FileWriter fw = new FileWriter(newjava);
            fw.write("package\t" + packageName + ";\r\n");
            fw.write(content);
            fw.flush();
            fw.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
 
    
    public static String removeUnderline(String string){
        StringBuilder columnNameBuilder=new StringBuilder(string);
        if(!string.contains("_")){
            return string;
        }else {
            int i = columnNameBuilder.indexOf("_");
            columnNameBuilder.replace(i,i+1, "").replace(i,i+1,columnNameBuilder.substring(i,i+1).toUpperCase());
            return removeUnderline(columnNameBuilder.toString());
        }
    }
 
    
    private static List<Map> getFiledMaps(String dataBaseName, String tableName) {
        Connection conn = null;
        Statement stmt = null;
        List<Map> tableFieldList=new ArrayList<>();
        try{
            // 注册 JDBC 驱动
            Class.forName(JDBC_DRIVER);
            conn = DriverManager.getConnection(DB_URL,USER,PASS);
            // 执行查询
            stmt = conn.createStatement();
            String sql= "SELECT * FROM information_schema.`COLUMNS` WHERE TABLE_SCHEMA ='"+dataBaseName+"' AND TABLE_NAME= '"+tableName+"'";
            ResultSet rs = stmt.executeQuery(sql);
            while (rs.next()) {
                Map<String,String> fieldMap = new HashMap();
                String column_name = rs.getString("COLUMN_NAME");
                setFieldName(column_name.toLowerCase(),fieldMap);
                String data_type = rs.getString("DATA_TYPE");
                setFieldType(data_type.toUpperCase(),fieldMap);
                fieldMap.put("fieldComment",rs.getString("COLUMN_COMMENT"));
                tableFieldList.add(fieldMap);
            }
            rs.close();
            stmt.close();
            conn.close();
        }catch(SQLException se){
            // 处理 JDBC 错误
            se.printStackTrace();
        }catch(Exception e){
            // 处理 Class.forName 错误
            e.printStackTrace();
        }finally{
            // 关闭资源
            try{
                if(stmt!=null){
                    stmt.close();
                }
            }catch(SQLException se2){
            }// 什么都不做
            try{
                if(conn!=null) conn.close();
            }catch(SQLException se){
                se.printStackTrace();
            }
        }
        return tableFieldList;
    }
 
    
    private static void processAllAttrs(StringBuffer sb, List<Map> filedMaps) {
        for (int i = 0; i < filedMaps.size(); i++) {
            Map map = filedMaps.get(i);
            String fieldType = MapUtils.getString(map, "fieldType");
            String fieldName = MapUtils.getString(map, "fieldName");
            String fieldComment = MapUtils.getString(map, "fieldComment");
            if(StringUtils.isNotBlank(fieldComment)){
                sb.append("\t\r\n");
            }
            sb.append("\tprivate " + fieldType + " " + fieldName + ";\r\n\r\n");
        }
    }
 
    private static void setFieldName(String columnName, Map fieldMap) {
        fieldMap.put("fieldName",removeUnderline(columnName));
    }
 
    private static void setFieldType(String columnType, Map fieldMap){
        String fieldType="String";
        if(columnType.equals("INT")||columnType.equals("INTEGER")){
            fieldType="Integer";
        }else if(columnType.equals("BIGINT")){
            fieldType="Long";
        }else if(columnType.equals("DATETIME")){
            fieldType="Date";
        }else if(columnType.equals("TEXT")||columnType.equals("VARCHAR")||columnType.equals("TINYTEXT")||columnType.equals("LONGTEXT")){
            fieldType="String";
        }else if(columnType.equals("DOUBLE")){
            fieldType="Double";
        }else if(columnType.equals("BIT")){
            fieldType="Boolean";
        }else if(columnType.equals("FLOAT")){
            fieldType="Float";
        }else if(columnType.equals("DECIMAL")){
            fieldType="BigDecimal";
        }else if(columnType.equals("DATE")){
            fieldType="Date";
        }else if(columnType.equals("TIMESTAMP")){
            fieldType="LocalDateTime";
        }else if(columnType.equals("CHAR")){
            fieldType="Char";
        }else if(columnType.equals("JSON")){//mysql5.7版本才开始有的
            fieldType="JSONObject";
        }
        fieldMap.put("fieldType",fieldType);
    }
}

到此这篇关于如何让java只根据数据库表名自动生成实体类的文章就介绍到这了,更多相关根据数据库表名生成实体类内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网!

--结束END--

本文标题: 如何让java只根据数据库表名自动生成实体类

本文链接: https://www.lsjlt.com/news/127953.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
猜你喜欢
  • 如何让java只根据数据库表名自动生成实体类
    根据数据库表名生成实体类 公司用的jpa,没有用mybatis。所以也没有用mybatis自动生成。但有些数据库表字段太多,就想着一劳永逸了,连数据库注释都搞上去 第一种 这里使用的...
    99+
    2024-04-02
  • 怎么让java只根据数据库表名自动生成实体类
    这篇文章将为大家详细讲解有关怎么让java只根据数据库表名自动生成实体类,小编觉得挺实用的,因此分享给大家做个参考,希望大家阅读完这篇文章后可以有所收获。根据数据库表名生成实体类公司用的jpa,没有用mybatis。所以也没有用mybati...
    99+
    2023-06-15
  • IDEA中如何根据数据库自动生成实体类
    这篇文章主要介绍“IDEA中如何根据数据库自动生成实体类”的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇“IDEA中如何根据数据库自动生成实体类”文章能帮助大家解决问题。1. IDEA连接数据库点击右侧...
    99+
    2023-07-05
  • Springboot如何根据实体类生成数据库表
    目录Springboot 实体类生成数据库表第一步:添加springboot-data-jpa和数据库的依赖关系第二步:编写yml文件的配置第三步:实体类中使用的注解第四步:启动项目...
    99+
    2024-04-02
  • 如何使用Hibernate根据实体类自动生成表
    本篇内容介绍了“如何使用Hibernate根据实体类自动生成表”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!第一种方法这种方法需要配置 hi...
    99+
    2023-07-05
  • Mybatis如何自动生成数据库表的实体类
    第一步引入jar 第二步,配置文本文件 # 数据库驱动jar 路径 本地创库的包 drive.class.path=C:/Users/Administrator/.m2/re...
    99+
    2024-04-02
  • idea如何根据表生成实体类
    您可以按照以下步骤根据表生成实体类: 分析表结构:仔细研究表的字段、数据类型、约束等信息,以便准确地生成实体类。 创建实体类...
    99+
    2023-10-24
    idea
  • mybatis plus generator 根据数据库自动生成实体类的实现示例
    目录1、添加依赖2、编写代码生成器3、运行主程序,输入表名1、添加依赖 <dependency> <groupId>com.baomidou<...
    99+
    2024-04-02
  • 数据库怎么自动生成实体类
    有多种方法可以自动生成数据库实体类,以下是常见的几种方法:1. 使用ORM框架:许多ORM框架都有自动生成实体类的功能,例如Hibe...
    99+
    2023-08-15
    数据库
  • Mybatis如何通过数据库表自动生成实体类和xml映射文件
    本篇内容主要讲解“Mybatis如何通过数据库表自动生成实体类和xml映射文件”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“Mybatis如何通过数据库表自动生成实体类和xml映射文件”吧!环境...
    99+
    2023-06-20
  • Mybatis通过数据库表自动生成实体类和xml映射文件
    环境:maven+idea。 1. 需要的jar包 基本的spring和mybatis依赖包就不说了,在pom文件的build->plugins节点下需要添加(两个依赖包也可以...
    99+
    2024-04-02
  • Java通过数据库表生成实体类详细过程
    目录项目背景项目代码使用说明配置相关swagger操作目前的缺点项目背景 最近在做的项目,涉及到数据库的操作了,之前做的是直接调用接口,不用做存库操作。 因此要增加大量特殊格式的实体...
    99+
    2023-02-07
    Java数据库表生成实体类 Java生成实体类
  • SpringBoot+jpa配置如何根据实体类自动创建表
    目录jpa配置根据实体类自动创建表1.配置文件application.properties2.pom.xml引入包3.编写实体类4.运行项目5.针对项目启动以后数据库并未生成数据库表...
    99+
    2024-04-02
  • mybatis如何根据表逆向自动化生成代码实例
    前言  自动生成代码总结(1)——(mybatis自动生成实体类、mapper文件、mapper.xml文件) 若采用mybatis框架,数据库新建表,手动编写的话,需要编...
    99+
    2024-04-02
  • Gorm + Gen自动生成数据库结构体
    文章目录 安装创建数据表生成model运行效果 使用gorm操作数据库的时候,需要自己来创建用于映射数据表的结构体,比较繁琐。现在gorm官方提供了一个可以自动生成数据库结构体的工具:Gen。 这个工具不仅仅可以用来生成数据...
    99+
    2023-08-16
    mysql go
  • django如何根据现有数据库表生成model详解
    目录一,在setting.py中配置好连接数据库的参数二,打开cmd或pycharm的Terminal输入命令三,managed = False 删除或改为 managed = Tr...
    99+
    2024-04-02
  • java如何实现自动生成数据库设计文档
    目录前言实现步骤main方法类全部代码表qrtz_blob_triggers表qrtz_calendars表qrtz_cron_triggers前言 以前我们还需要手写数据库设计文档...
    99+
    2024-04-02
  • Java实现根据模板自动生成新的PPT
    目录项目需求模板文件如下实现过程1.引入第三方依赖2.编写业务代码3.根据模板生成新的PPT项目需求 最近项目中有一个需求就是让Java代码去代替人工操作,自动生成PPT,具体就是查...
    99+
    2024-04-02
  • IDEA如何自动生成JPA实体类
    小编给大家分享一下IDEA如何自动生成JPA实体类,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!1、连接数据库(mysql)1.1 打开 MySQL 1.2 配置...
    99+
    2023-06-20
  • IDEA自动把数据库表映射为实体类
    1 打开视图菜单,选择工具窗口,选择数据库 2 找到打开的窗口,选择,+,选择数据源,选择MySQL(这里以MySQL为例)   3 填写相应的信息,最后点击测试,测试通过后点击应用,确定  4 还是回到刚刚的窗口,点击你的数据库名,点...
    99+
    2023-09-15
    数据库 intellij-idea mysql
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作