Oracle 数据库脚本解析生成相应的bean

最近工作遇到一个问题

用power design 生成的oracle sql脚本中,有几张表字段特别多，其中有一张表字段高达一百多个,平时开发需求中,我们一般都创建一个与数据库字段相对应的java bean对象，对于字段特别多的表,如果手工去建这样一个对象的话,工作量很大，很枯燥，浪费时间.

为了解决一个问题,我写了一段代码去解析oracle sql生成java bean.代码如下:

 

 

 

 package util; import java.io.File; import java.io.IOException; import java.io.InputStream; import java.util.List; import java.util.regex.Pattern; import org.apache.commons.io.FileUtils; import org.apache.commons.io.IOUtils; import org.apache.commons.lang.StringUtils; import org.apache.commons.lang.SystemUtils; import org.apache.commons.lang.WordUtils; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; public class SqlScriptParser { private static Log log = LogFactory.getLog(SqlScriptParser.class); private static final String CREATE_TABLE_EXPRESSSION= ".*create//s+table//s+.*"; private static final String END_TABLE_EXPRESSSION= ".*//);//s*"; private static final String CLASS_NAME= "@class_name@"; private static final String FILED_LIST= "@filed_list@"; private static final String FIELD_TYPE = "@field_type@"; private static final String FIELD_NAME = "@file_name@"; private static final String FIELD_TEMPLATE = "private @field_type@ @file_name@;"; @SuppressWarnings("unchecked") public static void parse(String filename,String output){ File file = new File(filename); try { List<String> list = FileUtils.readLines(file, "utf-8"); boolean active = false; String template = getClassTpl(); StringBuilder filedbuilder = null; String tablename = null; for(String line : list){ if(line.contains("constraint")) continue; line = StringUtils.strip(line); String[] fieldProperty = line.split("//s+"); if(Pattern.matches(CREATE_TABLE_EXPRESSSION, line)){ active = true; filedbuilder = new StringBuilder(); tablename = line.replaceAll(".*create//s+table//s+", ""); tablename = StringUtils.strip(tablename); log.info(tablename); tablename = tablename.replace("T_LOT_", "");//去掉前缀 tablename = tablename.toLowerCase(); tablename = WordUtils.capitalize(tablename ,new char[]{'_'});//首字线大写 template = getClassTpl().replaceAll(CLASS_NAME, tablename); } else if(active && fieldProperty.length >= 2){ String fieldName= fieldProperty[0].toLowerCase(); String fieldType = fieldProperty[1].toLowerCase(); if(fieldType.matches(".*number//(//d+,//d+//).*")){ fieldType = "double"; }else if(fieldType.matches(".*number//(//d//).*")){ fieldType = "int"; }else if(fieldType.matches(".*number//(//d{2,}//).*")){ fieldType = "long"; }else if(fieldType.matches(".*varchar.*") || fieldType.matches(".*char.*")){ fieldType = "String"; }else if(fieldType.matches(".*date.*")){ fieldType = "Date"; } filedbuilder.append(FIELD_TEMPLATE.replaceAll(FIELD_TYPE, fieldType).replaceAll(FIELD_NAME, fieldName)); filedbuilder.append(SystemUtils.LINE_SEPARATOR); } else if(active && Pattern.matches(END_TABLE_EXPRESSSION, line)){ log.info(line); active = false; File outdir = new File(output); if(!outdir.exists()){ outdir.mkdirs(); } template = template.replaceAll(FILED_LIST, filedbuilder.toString()); File outfile = new File(output+File.separator+tablename+".java"); FileUtils.writeStringToFile(outfile, template,"utf-8"); } } } catch (IOException e) { log.error(e.getMessage()); } } public static void main(String[] args){ parse("d://report//report.sql","d://report//bean"); } private static String getClassTpl(){ InputStream in = SqlScriptParser.class.getClassLoader().getResourceAsStream("util/ClassTpl.xdt"); String template = null; try { template = IOUtils.toString(in); } catch (IOException e) { log.error(e.getMessage()); }finally{ IOUtils.closeQuietly(in); } return template; } }

解析过程中用到的模板文件ClassTpl.xdt如下

package com.joyveb.lottery.domain; import java.util.*; public class @class_name@ { @filed_list@ }

最后，我们把代码拷贝到eclipse下,生成getter,setter就可以了，当然代码很多地方还有待改进