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

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

公司用的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;

/**
 * @ClassName TestController
 * @Author yupanpan
 * @Date 2021/4/12 14:24
 */
@SpringBootTest
public class TestController {

    @Autowired
    private JdbcTemplate jdbcTemplate;

    @Test
    public void database(){
        createEntity("1ceping", "t_evaluation_template_info_copy","com.shinedata.bims.entity",false);
    }

    /**
     *
     * @param dataBaseName 数据库名
     * @param tableName 表名
     * @param packageName 包名
     * @param isAddEntitySuffix 类名是否添加Entity后缀 true-添加 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);
    }

    /**
     * 创建实体类文件
     * @param className 类名(不包含.java文件名后缀) 根据表名首字母大写并去掉开头t_和所有下划线-驼峰命名
     * @param content 添加的内容(字段注释等)
     * @param packageName 包名(com.xxx.xxx.xxx)
     */
    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();
        }
    }

    /**
     * 解析输出属性
     *
     * @return
     */
    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").append("\t* ").append(fieldComment).append("\n").append("\t*/\r\n");
            }
            sb.append("\tprivate " + fieldType + " " + fieldName + ";\r\n\r\n");
        }
    }

    /**
     * 获取表字段信息(列名、类型、注释等)
     * @param dataBaseName
     * @param tableName
     * @return
     */
    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));
    }

    /**
     * 去下划线
     * @param string
     * @return
     */
    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.*;

/**
 * @ClassName TableToEntityUtils
 * @Author yupanpan
 * @Date 2021/6/8 17:25
 */
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);
    }

    /**
     *
     * @param dataBaseName 数据库名
     * @param tableName 表名
     * @param packageName 包名
     * @param isAddEntitySuffix 类名是否添加Entity后缀 true-添加 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);
    }

    /**
     * 创建实体类文件
     * @param className 类名(不包含.java文件名后缀) 根据表名首字母大写并去掉开头t_和所有下划线-驼峰命名
     * @param content 添加的内容(字段注释等)
     * @param packageName 包名(com.xxx.xxx.xxx)
     */
    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();
        }
    }

    /**
     * 去下划线
     * @param string
     * @return
     */
    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());
        }
    }

    /**
     * 获取表字段信息(列名、类型、注释等)
     * @param dataBaseName
     * @param tableName
     * @return
     */
    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;
    }

    /**
     * 解析输出属性
     *
     * @return
     */
    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").append("\t* ").append(fieldComment).append("\n").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只根据数据库表名自动生成实体类的文章就介绍到这了,更多相关根据数据库表名生成实体类内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们!

(0)

相关推荐

  • java反射机制给实体类相同字段自动赋值实例

    一.封装一个工具类 1.简易版 package net.aexit.construct.acceptance.websky.utils; import java.lang.reflect.Field; import java.lang.reflect.Method; import java.util.HashMap; import java.util.Iterator; import java.util.Map; public class ClassReflection { /** * @par

  • java 实现反射 json动态转实体类--fastjson

    我就废话不多说了,大家还是直接看代码吧~ package test.refect; public class Student { // 姓名 private String name; // 年龄 private String age; // 住址 private String address; public String getName() { return name; } public void setName(String name) { this.name = name; } public

  • idea中Java实体类怎样生成序列化的版本号的方法

    例如: 单击File->单击Settings, 在对话框左侧目录中找到,Editor->Inspections,并单击选中: 在右边的输入框里输入serializable 找到 Serializable class without 'serialVersionUID并在后面打上勾: 于是乎在实体类的类名上面,alt+enter 点击生成序列化版本号 到此这篇关于idea中Java实体类怎样生成序列化的版本号的方法的文章就介绍到这了,更多相关idea实体类序列化内容请搜索我们以前的文章或继续浏览

  • Java中实体类为什么要实现Serializable序列化的作用

    客户端访问了某个能开启会话功能的资源, web服务器就会创建一个与该客户端对应的HttpSession对象,每个HttpSession对象都要站用一定的内存空间.如果在某一时间段内访问站点的用户很多,web服务器内存中就会积累大量的HttpSession对象,消耗大量的服务器内存,即使用户已经离开或者关闭了浏览器,web服务器仍要保留与之对应的HttpSession对象,在他们超时之前,一直占用web服务器内存资源. web服务器通常将那些暂时不活动但未超时的HttpSession对象转移到文件

  • JavaBean实体类处理外键过程解析

    数据库中的表在映射为JavaBean的实体类时,实体类的字段名应与数据库表中的字段名一致.那么外键要怎么处理? 1. JavaBean中实体类中不应该出现外键字段,弊端如下 : 如果出现外键对应的字段,那么将会出现一个对象与另一个对象的属性相对应,这将不是面向对象中的对象与对象对应. 单个外键只能存储一个字符串,用处不大 2.外键处理方式 :数据库表中的外键字段应当转换成外键对应的表的对象,也就是JavaBean的实体类不声明外键的字段,而是声明外键对应数据表类的类型.举个例子,如下 : 数据库

  • Java实体类不要使用基本类型的知识点总结

    今天来记录一下,在项目中因为基本类型,所产生的bug 包装类:8种基本类型的包装类 应用场景:数据库建立实体映射多用包装类 这两句话是重点:就是建立实体类禁止使用基本数据量类型!!!而用对应的包装类, 为什么呢,看以下场景. JAVA代码 <font style="color:rgb(77, 77, 77)"><font face="&quot"><font style="font-size:16px">

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

    根据数据库表名生成实体类 公司用的jpa,没有用mybatis.所以也没有用mybatis自动生成.但有些数据库表字段太多,就想着一劳永逸了,连数据库注释都搞上去 第一种 这里使用的是jdbcTemplate+Junit测试生成,方式可变. SpringBoot版本是2.4.4,只需要加上@SpringBootTest就可以了.不用@RunWith pom: <dependency> <groupId>org.springframework.boot</groupId>

  • 利用JAVA反射,读取数据库表名,自动生成对应实体类的操作

    本代码是利用java反射,读取数据库表自动根据表名生成实体类,数据库采用老牌SQLSERVER 2000,驱动为JTDS,其他数据库可根据情况自定修改. 代码中包含了大部分数据库类型与JAVA类型的转换,少数未包含进去的会在生成代码时打印出来,方面后期查找修改. import java.io.File; import java.io.FileWriter; import java.io.IOException; import java.io.PrintWriter; import java.sq

  • Python使用win32com模块实现数据库表结构自动生成word表格的方法

    本文实例讲述了Python使用win32com模块实现数据库表结构自动生成word表格的方法.分享给大家供大家参考,具体如下: 下载win32模块 下载链接:https://sourceforge.net/projects/pywin32/files/pywin32/ 连接mysql import MySQLdb db_host = "" db_port = 3306 db_name = "" db_user = "" db_pwd = &quo

  • mybatis plus generator 根据数据库自动生成实体类的实现示例

    目录 1.添加依赖 2.编写代码生成器 3.运行主程序,输入表名 1.添加依赖 <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>3.3.2</version> </dependency> <dependency> <groupId&g

  • vbs adox提取数据库表名和列名的类

    <!--#include virtual="/include/conn.inc"--> <% '********************************* '* classname: ADOXDBOPR '* 制作人: 刘晓逸 '* 作用: 用adox进行数据库的操作 '* 版本: 1.0 '*www.jb51.net 我们'********************************** Class ADOXDBOPR dim objADOX Priva

  • Mybatis通过数据库表自动生成实体类和xml映射文件

    环境:maven+idea. 1. 需要的jar包 基本的spring和mybatis依赖包就不说了,在pom文件的build->plugins节点下需要添加(两个依赖包也可以直接添加到pom的依赖里面去,这里是为了直接通过maven的插件来生成.如果不是使用maven,自行百度下): <plugin> <groupId>org.mybatis.generator</groupId> <artifactId>mybatis-generator-mave

  • SQL提取数据库表名及字段名等信息代码示例

    本文向大家介绍了使用SQL语句提取数据库所有表的表名.字段名的实例代码,在SQLserver 中进行了测试,具体内容如下: --查询所有用户表所有字段的特征 SELECT D.Name as TableName, A.colorder AS ColOrder, A.name AS Name, COLUMNPROPERTY(A.ID,A.Name, 'IsIdentity') AS IsIdentity, CASE WHEN EXISTS (SELECT 1 FROM dbo.sysobjects

  • ASP获取数据库表名、库名、字段名的方法

    ASP获取数据库表名,字段名 以SQLServer为例: 复制代码 代码如下: < %   SET Conn=Server.CreateObject("ADODB.Connection")   Conn.Open "Server=IP地址;Provider=sqloledb;Database=库名称;UID=用户名;PWD=密码;"   %> 读SqlServer库中的表名: 复制代码 代码如下: < %   Set rs=Conn.OpenSch

  • ASP获取ACCESS数据库表名及结构的代码

    <html> <head> <title>获取ACCESS数据库表名_www.jb51.net</title> </head> <body style="text-align:left;margin-left:50px;font-family:'arial';font-size:12px"> <form style="padding:5px;margin:5px;margin-left:0px&qu

  • SQL Server数字开头的数据库表名的解决方法

    今天遇到了个郁闷的问题,关于数据库表名的问题. SQL Server的数据库的表名可以用数字开头,但是在查询的时候,不能直接写表名,如下: select *         form 1_tablename 在执行过程中,报错.但是给表名加上中括号,就可以了,如下: select *         from [1_tablename] 用VS中的DataSet开发时,会自动生成SQL语句,但是生成的SQL语句中,有的加上方括号,但是有的没有加, 这在程序执行过程中,就不能确定语句的正确性,而且

随机推荐