根据表名生成实体类

package test;

import java.sql.Connection;

import java.sql.DatabaseMetaData;

import java.sql.ResultSet;

import org.junit.Test;

import com.cmb.ccrm.base.BaseTest;

import com.google.common.base.CaseFormat;

public class test extends BaseTest {

 @Test  public void 生成实体类() throws Exception {

  String tableName = "S55_TEST";

  Connection conn = connection.getConnection();  

 DatabaseMetaData databaseMetaData = conn.getMetaData();  

 ResultSet tableRet = databaseMetaData.getTables(null, "%", tableName, new String[] { "TABLE" });

  while (tableRet.next()) {    

String columnName;    String columnType;    

ResultSet colRet = databaseMetaData.getColumns(null, "%", tableName, "%");    

while (colRet.next()) {     columnName = colRet.getString("COLUMN_NAME");     

columnType = colRet.getString("TYPE_NAME").toLowerCase();     

String remarks = colRet.getString("REMARKS");     

String columType = "";     

switch (columnType) {     

case "varchar":      

columType = "String";    

  break;    

 case "int":     

 columType = "int";      break;

    default:     

 columType = "String";   

  }

    System.out.println("/**");

    System.out.println("*" + remarks);

    System.out.println("*/");

    System.out.println("private " + columType + "\t"       + CaseFormat.UPPER_UNDERSCORE.to(CaseFormat.UPPER_CAMEL, columnName) + ";");    

 System.out.println();

   }

  }

 }

}

猜你喜欢

转载自www.cnblogs.com/wu1254/p/10718964.html