在工具中把sql转换为数据库类型

package com.czb.entity;

import com.czb.util.DBHelper;

import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

public class TableToPo {
//注意 链接数据库 表名
    private String tablename = "bank";//表名
    private String[] colnames;//列名数组
    private String[] colTypes;//列名类型数组
    private int[] colSizes;//列名大小数组
    private boolean f_util = false;// 是否需要导入包java.util.*
    private boolean f_sql = false;//是否需要导入包java.sql.*
    private boolean f_math = false;// 是否需要导入包java.math.*
    public TableToPo(){
        Connection conn = DBHelper.getConn();
        PreparedStatement ps = null;
        ResultSetMetaData rs = null;
        String sql = "select * from "+tablename;
        //ResultSetMetaData 类
        //可用于获取关于 ResultSet 对象中列的类型和属性信息的对象:
        //?getColumnName(int column):获取指定列的名称
        //?getColumnCount():返回当前 ResultSet 对象中的列数。
        //?getColumnTypeName(int column):检索指定列的数据库特定的类型名称。
        //?getColumnDisplaySize(int column):指示指定列的最大标准宽度,以字符为单位。
        //?isNullable(int column):指示指定列中的值是否可以为 null。
        //?isAutoIncrement(int column):指示是否自动为指定列进行编号,这样这些列仍然是只读的。
        try {
            //获取预编译对象
            ps = conn.prepareStatement(sql);
            //获取关于 ResultSet 对象中列的类型和属性信息的对象
            rs = ps.getMetaData();
            //返回当前 ResultSet 对象中的列数
            int size = rs.getColumnCount();
            //列名数组
            colnames = new String[size];
            //列名类型数组
            colTypes = new String[size];
            //列名大小数组
            colSizes = new int[size];
            //遍历ResultSet 列的名称和检索指定列的数据库特定的类型名称
            for (int i = 0; i < rs.getColumnCount(); i++) {
                //列的名称
                colnames[i] = rs.getColumnName(i+1);
                //检索指定列的数据库特定的类型名称
                colTypes[i] = rs.getColumnTypeName(i+1);
                //equalsIgnoreCase() 方法用于将字符串与指定的对象比较,不考虑大小写。
                //判断索指定列的数据库特定的类型名称是否有时间类,是否需要导入包java.sql.*
                if(colTypes[i].equalsIgnoreCase("datetime")
                        || colTypes[i].equalsIgnoreCase("date")
                        || colTypes[i].equalsIgnoreCase("smalldatetime")
                        || colTypes[i].equalsIgnoreCase("time")){
                    f_sql = true;
                }
                //判断索指定列的数据库特定的类型名称是否有数值类型类// 是否需要导入包java.math.*
                if (colTypes[i].equalsIgnoreCase("money")
                        || colTypes[i].equalsIgnoreCase("decimal")
                        || colTypes[i].equalsIgnoreCase("smallmoney")
                        || colTypes[i].equalsIgnoreCase("numeric")) {
                    f_math = true;
                }
                //getColumnDisplaySize()ResultSetMetaData接口的方法,获取索引为某某的列的显示大小-
                //列名大小数组
                colSizes[i] = rs.getColumnDisplaySize(i + 1);
            }
            //解析处理(生成实体类主体代码)
            String content = parse(colnames, colTypes, colSizes);
            //按字符向流中写入数据并写入对应的类中
            FileWriter fileWriter = new FileWriter(initcap(tablename) + ".java");
            // 使用指定文件创建不具有自动行刷新的新 PrintWriter。
            PrintWriter pw = new PrintWriter(fileWriter);
            //println在要输出的内容后自动加了回车换行 ,这个函数不能用输出格式。
            pw.println(content);
            //flush()方法是输出储存在内存中全部的内容(批量输出)。
            pw.flush();
            //关闭资源
            pw.close();
        } catch (SQLException | IOException e) {
            e.printStackTrace();
        }
    }
    /**
     * 解析处理(生成实体类主体代码)
     */
    private String parse(String[] colNames, String[] colTypes, int[] colSizes) {
        StringBuffer sb = new StringBuffer();
        //判断追加包类
        if (f_util) {
            sb.append("import java.util.Date;\r\n");
        }
        if (f_sql) {
            sb.append("import java.sql.*;\r\n");
        }
        if (f_math) {
            sb.append("import java.math.BigDecimal;\r\n\r\n\r\n");
        }
        //方法体
        sb.append("public class " + initcap(tablename) + " {\r\n");
        //属性
        processAllAttrs(sb);
        //方法
        processAllMethod(sb);
        sb.append("}\r\n");
        //输出判断
        System.out.println(sb.toString());
        return sb.toString();
    }
    /**
     * 生成所有的方法
     *
     * @param sb
     */
    private void processAllMethod(StringBuffer sb) {
        //遍历生成get,set方法
        //列名称数组遍历
        for (int i = 0; i < colnames.length; i++) {
            //\t 的意思是 横向跳到下一制表符位置 \r 的意思是 回车 \n 的意思是回车换行
            sb.append("\tpublic " + sqlType2JavaType(colTypes[i]) + " get"
                    + initcap(colnames[i]) + "(){\r\n");
            sb.append("\t\treturn " + colnames[i] + ";\r\n");
            sb.append("\t}\r\n");
            sb.append("\tpublic void set" + initcap(colnames[i]) + "("
                    + sqlType2JavaType(colTypes[i]) + " " + colnames[i]
                    + "){\r\n");
            sb.append("\t\tthis." + colnames[i] + "=" + colnames[i] + ";\r\n");
            sb.append("\t}\r\n");

        }
    }
    /**
     * 解析输出属性
     *
     * @return
     */
    private void processAllAttrs(StringBuffer sb) {
        //遍历列名数组 类型 参数;
        for (int i = 0; i < colnames.length; i++) {
            sb.append("\tprivate " + sqlType2JavaType(colTypes[i]) + " "
                    + colnames[i] + ";\r\n");
        }
    }
    /**
     * 把输入字符串的首字母改成大写
     *
     * @param str
     * @return
     */
    private String initcap(String str) {
        char[] ch = str.toCharArray();
        if (ch[0] >= 'a' && ch[0] <= 'z') {
            ch[0] = (char) (ch[0] - 32);
        }
        return new String(ch);
    }

    /**
     * 获取数组类型 和数组名
     * @param sqlType
     * @return
     */

    private String sqlType2JavaType(String sqlType) {
        if (sqlType.equalsIgnoreCase("bit")) {
            return "Boolean";
        } else if (sqlType.equalsIgnoreCase("timestamp")
                || sqlType.equalsIgnoreCase("binary")
                || sqlType.equalsIgnoreCase("image")
                || sqlType.equalsIgnoreCase("varbinary")) {
            return "byte[]";
        } else if (sqlType.equalsIgnoreCase("smallint")
                || sqlType.equalsIgnoreCase("tinyint")) {
            return "short";
        } else if (sqlType.equalsIgnoreCase("int")) {
            return "Integer";
        } else if (sqlType.equalsIgnoreCase("bigint")) {
            return "long";
        } else if (sqlType.equalsIgnoreCase("real")) {
            return "float";
        } else if (sqlType.equalsIgnoreCase("real")
                || sqlType.equalsIgnoreCase("float")) {
            return "double";
        } else if (sqlType.equalsIgnoreCase("money")
                || sqlType.equalsIgnoreCase("decimal")
                || sqlType.equalsIgnoreCase("smallmoney")
                || sqlType.equalsIgnoreCase("numeric")) {
            return "BigDecimal";
        } else if (sqlType.equalsIgnoreCase("varchar")
                || sqlType.equalsIgnoreCase("char")
                || sqlType.equalsIgnoreCase("nvarchar")
                || sqlType.equalsIgnoreCase("nchar")
                || sqlType.equalsIgnoreCase("text")
                || sqlType.equalsIgnoreCase("ntext")
                || sqlType.equalsIgnoreCase("uniqueidentifier")) {
            return "String";
        } else if (sqlType.equalsIgnoreCase("date")) {
            return "Date";//Date
        } else if (sqlType.equalsIgnoreCase("smalldatetime")
                || sqlType.equalsIgnoreCase("datetime")) {
            return "Timestamp";
        } else if (sqlType.equalsIgnoreCase("time")) {
            return "Time";
        }
        return null;
    }
    public static void main(String[] args) {
        new TableToPo();
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值