jdbc工具类-生成domain实体

最新推荐文章于 2024-02-02 20:44:30 发布

都市桃源

最新推荐文章于 2024-02-02 20:44:30 发布

阅读量1.6k

点赞数

分类专栏： java 文章标签： jdbc freemarker 代码生成器

本文链接：https://blog.csdn.net/do_bset_yourself/article/details/71036225

版权

java 专栏收录该内容

17 篇文章 1 订阅

订阅专栏

最近两天项目里遇到一个需求，要求将数据库里某些表的某些字段，加密，但是还要保证能及时还原回去.

需求，很简单，但是写的时候，也遇到了一些问题，主要是oracle数据库使用数据库查询数据库时，resultset返回的结果集，我们的数据库最大1329条，当然这个参数可以修改，但是，我们遇到问题的时候不能去修改配置，治标不治本,所以,还是分页查询吧,最终问题解决了.

其中比较坑的就是,注意使用jdbc的批量操作,每个一定量的数据批量提交一次,我们的数据较小,所以我每隔一百条记录提交一次就好,测试1300记录,导出,加密,还原不到一分钟,(还是字段较多的哦),总体性能不错.

当时,写这个工具的时候,考虑到后期数据的变动,采用了反射,将指定的sql语句查询出来的resultset映射成javabean对象,更新时,将javabean转换为list参数列表,然后生成更新语句,总的来说这设计,帮我减少了很多工作.
但是整个写代码的过程中,很麻烦的就是,属性顺序要和sql查询顺序一致.

当然,javabean还是提前手写好了.
这一步,要是表多了,自己手写很累,所以考虑一下,正好也有时间了,就稍微学习了下,利用freeemark生成javabean,当初本来想看看mybatis的逆向工程的源码的,但是我的maven不知道咋回事,暂时不能用了,所以暂时也没心情整他,所以就自己写一个吧.
使用的技术:freemark模板引擎,jdbc
好了,不废话了,我们来一步步实现吧.

需求:根据数据库生成javabean

用到的jar包,freemarker 2.3和mysqljdbc-connector

1.第一步jdbc查询出所有的表,和表的列信息和列类型信息

具体实现如下:

package com.taoyuan.jdbc;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 * @author 都市桃源 读取数据库信息 2017年4月30日
 */
public class ConvertUtil {
    public static final Map<String,String> columnToJavaType=new HashMap<String,String>();
    static{//mysql类型到java类型的转换
        columnToJavaType.put("VARCHAR", "String");
        columnToJavaType.put("INT", "Integer");
        columnToJavaType.put("BIGINT", "Long");
        columnToJavaType.put("TIME", "Date");
        columnToJavaType.put("DATETIME", "Date");
        columnToJavaType.put("TEXT", "String");
    }
    public static List<Table> getTables(Connection con) throws SQLException {
        DatabaseMetaData metaData = con.getMetaData();// 获取数据库信息
        //获取数据库信息
        ResultSet rs = metaData.getTables(null, null, null, new String[] { "TABLE", "VIEW" });
        List<Table> tables=new LinkedList<Table>();
        Table tab=null;
        List<Attribute> columnList=null;
        while (rs.next()) {//遍历表和列
            if (rs.getString(4) != null
                    && (rs.getString(4).equalsIgnoreCase("TABLE") || rs.getString(4).equalsIgnoreCase("VIEW"))) {
                String tableName = rs.getString(3).toLowerCase();
                tab=new Table();
                tab.setTableName(toClassCase(tableName));
                columnList=new ArrayList<Attribute>();
                ResultSet colRet = metaData.getColumns(null, "%", tableName, "%");//获取列
                Attribute col=null;
                while (colRet.next()) {//遍历列
                    col=new Attribute();
                    String columnName = colRet.getString("COLUMN_NAME");
                    String columnType = colRet.getString("TYPE_NAME");
                    col.setName(toCamelCase(columnName));
                    col.setType(columnToJavaType.get(columnType));
                    columnList.add(col);
                }
                tab.setColumns(columnList);
            }
            tables.add(tab);

        }
        return tables;
    }

    public static void main(String[] args) throws Exception {
    /*  String driver = "com.mysql.jdbc.Driver";
        String url = "jdbc:mysql://localhost:3306/note?useUnicode=true&amp;characterEncoding=UTF-8";
        String user = "root";
        String password = "";
        Class.forName(driver);
        Connection conn = DriverManager.getConnection(url, user, password);*/

    }
    /**将字符串转换为驼峰命名
     * @param str
     * @return
     */
    public static String toCamelCase(String str){
        if(str==null||!str.contains("_"))return str;
        StringBuffer buf=new StringBuffer();
        char[] charArray = str.toCharArray();
        for(int i=0,len=charArray.length;i<len-1;i++){
            if(charArray[i]=='_'&&(charArray[i+1]>='a'&&charArray[i+1]<'z')){
                charArray[i+1]=(char) (charArray[i+1]-32);
            }
            if(charArray[i]!='_')
            buf.append(charArray[i]);
        }
        buf.append(charArray[charArray.length-1]);
        return buf.toString();
    }
    /**转换类名
     * @param str
     * @return
     */
    public static String toClassCase(String str){
        String camelCase = toCamelCase(str);
        String substring = camelCase.substring(0,1);
        return camelCase.replaceFirst(substring, substring.toUpperCase());
    }

}
封装下对象table和属性:

package com.taoyuan.jdbc;

public class Attribute {
private String type;
private String name;
public Attribute() {
super();
}
public Attribute(String name,String type) {
super();
this.type = type;
this.name = name;
}
public String getType() {
return type;
}

public void setType(String type) {
    this.type = type;
}

public String getName() {
    return name;
}

public void setName(String name) {
    this.name = name;
}

}

package com.taoyuan.jdbc;

import java.util.List;

/**
 * @author 都市桃源
 *2017年4月30日
 *表实体
 */
public class Table {
    private String tableName;
    private List<Attribute> columns;
    public String getTableName() {
        return tableName;
    }
    public void setTableName(String tableName) {
        this.tableName = tableName;
    }
    public List<Attribute> getColumns() {
        return columns;
    }
    public void setColumns(List<Attribute> columns) {
        this.columns = columns;
    }



}

调用方法ConvertUtil.getTbales,就可以获取所有表和视图的信息,jdbc的类型对应,各位可以自己添加,我测试的时候就用到这个几个

2,第二步,代码生成

首先准备模板,一个正常的freemarker模板

package ${packageName};


/**
 *  @author ${author}
 */
public class ${className} {
    <#list attrs as attr> 
    private ${attr.type} ${attr.name};
    </#list>

    <#list attrs as attr>
    public void set${attr.name?cap_first}(${attr.type} ${attr.name}){
        this.${attr.name} = ${attr.name};
    }
    public ${attr.type} get${attr.name?cap_first}(){
        return this.${attr.name};
    }

    </#list>

}

编写模板生成代码,官网上有例子http://freemarker.org/
参考博客:
http://blog.csdn.net/whithorse/article/details/19206087

我的需求很简单,生成dao的我就不写了,而且就算生成也很鸡肋.

基础配置

#jdbc配置
url=jdbc:mysql://localhost:3306/note?useUnicode=true&amp;characterEncoding=UTF-8
driver=com.mysql.jdbc.Driver
user=root
password=
#生成类的包名
packageName=com.toyuanx.pojo
#作者
author=taoyuan
#采用模板名称
templatePath=pojo.ftl
#模板所在目录
templateDir=D:/freemarker

代码生成工具类

package com.taoyuan.test;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.sql.Connection;
import java.sql.DriverManager;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;

import com.taoyuan.jdbc.Attribute;
import com.taoyuan.jdbc.ConvertUtil;
import com.taoyuan.jdbc.Table;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import freemarker.template.TemplateExceptionHandler;

public class DomainGenerator {
    static Properties pro=new Properties();
    static{
        try {
            pro.load(DomainGenerator.class.getClassLoader().getResourceAsStream("jdbc.properties"));
        } catch (IOException e) {
        System.out.println("加载配置文件失败");
        }
    }
    public static void main(String[] args) throws Exception {
        String driver = pro.getProperty("driver");
        String url = pro.getProperty("url");
        String user = pro.getProperty("user");
        String password = pro.getProperty("password");
        String packageName = pro.getProperty("packageName");
        String author = pro.getProperty("author");
        Class.forName(driver);
        Connection conn = DriverManager.getConnection(url, user, password);
        List<Table> tables = ConvertUtil.getTables(conn);
        gen(tables,packageName,author);
    }

    public static  void gen(List<Table> tables,String packageName,String author ) throws IOException, TemplateException{
        Configuration cfg = new Configuration(Configuration.VERSION_2_3_22);
        cfg.setDirectoryForTemplateLoading(new File(pro.getProperty("templateDir")));   
        cfg.setDefaultEncoding("UTF-8");
        cfg.setTemplateExceptionHandler(TemplateExceptionHandler.RETHROW_HANDLER);
        Template temp = cfg.getTemplate(pro.getProperty("templatePath")); 
        for(int i=0,length=tables.size();i<length;i++){
            Table table = tables.get(i);
        Map<String, Object> root = new HashMap<String, Object>();
        root.put("packageName", packageName);
        root.put("author", "taoyuan");
        root.put("className", table.getTableName());
        root.put("attrs", table.getColumns());

        File dir = new File("D:/freemarker");
        if(!dir.exists()){
            dir.mkdirs();
        }
        OutputStream fos = new  FileOutputStream( new File(dir, table.getTableName()+".java")); //java文件的生成目录   
        Writer out = new OutputStreamWriter(fos);
        temp.process(root, out);

        fos.flush();  
        fos.close();
        }
        System.out.println("代码生成成功");
    }


}

最终效果:
这里写图片描述

这里写图片描述

都市桃源

关注

0
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
jdbc工具类-生成domain实体

最近两天项目里遇到一个需求，要求将数据库里某些表的某些字段，加密，但是还要保证能及时还原回去.需求，很简单，但是写的时候，也遇到了一些问题，主要是oracle数据库使用数据库查询数据库时，resultset返回的结果集，我们的数据库最大1329条，当然这个参数可以修改，但是，我们遇到问题的时候不能去修改配置，治标不治本,所以,还是分页查询吧,最终问题解决了.其中比较坑的就是,注意使用jdbc的批量操
复制链接

扫一扫