package com.databi.utils;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.Serializable;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* 为了链接oracle数据库 生成表对应的javabean
* @author Administrator
*
*/
public class JavaBeanUtils implements Serializable{
private PreparedStatement ps = null;
private ResultSet rs = null;
private static Connection con = null;
private CallableStatement cst = null;
static class Ora{
static final String DRIVER_CLASS = "oracle.jdbc.driver.OracleDriver";
static final String DATABASE_URL = "jdbc:oracle:thin:@127.0.0.1:1521:orcl";
static final String DATABASE_USER = "databi";
static final String DATABASE_PASSWORD = "123456";
static final String DATABASE_TABLE = "bas_dept"; //需要生成的表名
}
static class MySql{
static final String DRIVER_CLASS = "com.mysql.jdbc.Driver";
static final String DATABASE_URL = "jdbc:mysql://localhost/plusoft_test?useUnicode=true&characterEncoding=GBK";
static final String DATABASE_USER = "root";
static final String DATABASE_PASSWORD = "1234";
}
public static Connection getOracleConnection() {
try {
Class.forName(Ora.DRIVER_CLASS);
con=DriverManager.getConnection(Ora.DATABASE_URL,Ora.DATABASE_USER,Ora.DATABASE_PASSWORD);
return con;
} catch (Exception ex) {
System.out.println(ex.getMessage());
}
return con;
}
public static Connection getMySqlConnection() {
try {
Class.forName(MySql.DRIVER_CLASS);
con=DriverManager.getConnection(MySql.DATABASE_URL,MySql.DATABASE_USER,MySql.DATABASE_PASSWORD);
return con;
} catch (Exception ex) {
ex.printStackTrace();
}
return con;
}
public static List<Map> getOracleTable(String Table) throws SQLException{
getOracleConnection();
List<Map> list = new ArrayList<Map>();
try {
DatabaseMetaData m_DBMetaData = con.getMetaData();
//getColumns(java.lang.String catalog, java.lang.String schema,java.lang.String table, java.lang.String col)
ResultSet colrs = m_DBMetaData.getColumns(null,Ora.DATABASE_USER.toUpperCase(), Table.toUpperCase(),"%");
while(colrs.next()) {
Map map = new HashMap();
String columnName = colrs.getString("COLUMN_NAME");
String columnType = colrs.getString("TYPE_NAME");
int datasize = colrs.getInt("COLUMN_SIZE");
int digits = colrs.getInt("DECIMAL_DIGITS");
int nullable = colrs.getInt("NULLABLE");
String remarks = colrs.getString("REMARKS");
//System.out.println(columnName+" "+columnType+" "+datasize+" "+digits+" "+ nullable);
map.put("columnName", columnName);
map.put("columnType", columnType);
map.put("datasize", datasize);
map.put("remarks", remarks);
list.add(map);
// System.out.println("TABLE_CAT" + "===" + colrs.getString("TABLE_CAT"));
// System.out.println("TABLE_SCHEM" + "===" + colrs.getString("TABLE_SCHEM"));
// System.out.println("TABLE_NAME" + "===" + colrs.getString("TABLE_NAME"));
// System.out.println("COLUMN_NAME" + "===" + colrs.getString("COLUMN_NAME"));
// System.out.println("DATA_TYPE" + "===" + colrs.getString("DATA_TYPE"));
// System.out.println("TYPE_NAME" + "===" + colrs.getString("TYPE_NAME"));
// System.out.println("COLUMN_SIZE" + "===" + colrs.getString("COLUMN_SIZE"));
// System.out.println("BUFFER_LENGTH" + "===" + colrs.getString("BUFFER_LENGTH"));
// System.out.println("DECIMAL_DIGITS" + "===" + colrs.getString("DECIMAL_DIGITS"));
// System.out.println("NUM_PREC_RADIX" + "===" + colrs.getString("NUM_PREC_RADIX"));
// System.out.println("NULLABLE" + "===" + colrs.getString("NULLABLE"));
// System.out.println("REMARKS" + "===" + colrs.getString("REMARKS"));
// System.out.println("COLUMN_DEF" + "===" + colrs.getString("COLUMN_DEF"));
// System.out.println("SQL_DATA_TYPE" + "===" + colrs.getString("SQL_DATA_TYPE"));
// System.out.println("SQL_DATETIME_SUB" + "===" + colrs.getString("SQL_DATETIME_SUB"));
// System.out.println("CHAR_OCTET_LENGTH" + "===" + colrs.getString("CHAR_OCTET_LENGTH"));
// System.out.println("ORDINAL_POSITION" + "===" + colrs.getString("ORDINAL_POSITION"));
// System.out.println("IS_NULLABLE" + "===" + colrs.getString("IS_NULLABLE"));
}
// while(colRet.next()){
// System.out.print("列名："+colRet.getString("COLUMN_NAME"));
// System.out.print(" 数据类型是："+colRet.getString("DATA_TYPE"));
// System.out.print(" 类型名称是："+colRet.getString("TYPE_NAME"));
// System.out.print(" 列大小是："+colRet.getString("COLUMN_SIZE"));
// System.out.println(" 注释是："+colRet.getString("REMARKS"));
// }
} catch (Exception e) {
e.printStackTrace();
}finally{
con.close();
}
return list;
}
/**
* 把Oracle字段类型 转化为 java类型
* @param sqlType 字段类型
* @param size 字段大小
* @param scale 默认=0
* @return
*/
public static String oracleSqlType2JavaType(String sqlType,int size,int scale){
if (sqlType.equals("integer")) {
return "Integer";
} else if (sqlType.equals("long")) {
return "Long";
} else if (sqlType.equals("float")
|| sqlType.equals("float precision")
|| sqlType.equals("double")
|| sqlType.equals("double precision")
) {
return "BigDecimal";
}else if (sqlType.equals("number")
||sqlType.equals("decimal")
|| sqlType.equals("numeric")
|| sqlType.equals("real")) {
return scale==0? (size<10? "Integer" : "Long") : "BigDecimal";
}else if (sqlType.equals("varchar")
|| sqlType.equals("varchar2")
|| sqlType.equals("char")
|| sqlType.equals("nvarchar")
|| sqlType.equals("nchar")) {
return "String";
} else if (sqlType.equals("datetime")
|| sqlType.equals("date")
|| sqlType.equals("timestamp")) {
return "Date";
}
return "String";
}
public static String getItems(List<Map> map,String tablename){
//记得转化成小写
StringBuffer sb = new StringBuffer();
sb.append("package com.databi.bean;");
sb.append("\r\n");
sb.append("import java.util.Date;\r\n");
sb.append("/** \r\n "
+" * \r\n"
+" * @author lsp \r\n"
+" *\r\n"
+ "*/\r\n" );
sb.append("\r\n");
sb.append("public class "+ getUpperOne(tablename.toLowerCase()) + " implements java.io.Serializable {\r\n");
//得到私有属性
for (Map map0 : map) {
String columnname = map0.get("columnName").toString();
String columntype = map0.get("columnType").toString();
String columnsize = map0.get("datasize").toString();
String remarks = map0.get("remarks")==null?"":map0.get("remarks").toString();
String javaType = oracleSqlType2JavaType(columntype.toLowerCase(),Integer.parseInt(columnsize),0);
String temp = "\tprivate "+javaType+" "+columnname.toLowerCase()+"; //"+remarks+"\r\n";
sb.append(temp);
}
//得到getter和setter
for (Map map0 : map) {
String columnname = map0.get("columnName").toString();
String columntype = map0.get("columnType").toString();
String columnsize = map0.get("datasize").toString();
String javaType = oracleSqlType2JavaType(columntype.toLowerCase(),Integer.parseInt(columnsize),0);
String temp = "\tpublic "+javaType+" "+"get"+getUpperOne(columnname.toLowerCase())+"(){\r\n";
String temp1 = "\t\treturn "+columnname.toLowerCase()+";\r\n";
String temp2 = "\t}\r\n";
sb.append(temp+temp1+temp2);
temp = "\tpublic void "+"set"+getUpperOne(columnname.toLowerCase())+"("+javaType+" "+columnname.toLowerCase()+"){\r\n";
temp1 = "\t\tthis."+columnname.toLowerCase()+" = "+columnname.toLowerCase()+";\r\n";
temp2 = "\t}\r\n";
sb.append(temp+temp1+temp2);
}
sb.append("}");
return sb.toString();
}
/**
* 把输入字符串的首字母改成大写
* @param str
* @return
*/
public static String getUpperOne(String str){
char[] ch = str.toCharArray();
if (ch[0] >= 'a' && ch[0] <= 'z') {
ch[0] = (char) (ch[0] - 32);
}
return new String(ch);
}
public static void main(String[] args) throws IOException {
//JavaBeanUtils.sysoutOracleTCloumns("pexam_items_title", "his_yhkf");
try {
String tables = "bi_bas_dept";
String[] arr = tables.split(",");
for (String string : arr) {
String name = getUpperOne(string.toLowerCase());
List<Map> map = getOracleTable(string);
String a = getItems(map,string);
File file = new File("D:\\bbb\\"+name+".java");
if (!file.exists()) {
file.createNewFile();
}
FileOutputStream fos = new FileOutputStream(file,true);//true表示在文件末尾追加
fos.write(a.getBytes());
fos.close();//流要及时关闭
}
System.out.println("生成java完成");
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
