`

freemaker自动生成源代码

javafreemaker 
阅读更多

自己写了一个自动生成源代码的工具，代码自动生成，以及一个完整项目的成功实践，还在继续完善当中，采用freemaker配置模板，可以根据自己的需要生成自己想要的代码

 

先来看看怎么使用,很简单

先配置config.xml 配置设置模板路径，以及项目生成路径，可以自己定义

 

Xml代码 复制代码 收藏代码
  1. <?xml version="1.0" encoding="UTF-8"?>  
  2. <xml-body>  
  3.  <code>  
  4.   <projectpath>D:/vobuilder/vobuilder</projectpath>  
  5.      
  6.   <model>com.hyj.model</model>  
  7.   <action>com.hyj.action</action>  
  8.   <dao>com.hyj.dao</dao>  
  9.   <daoimpl>com.hyj.dao.jdbc</daoimpl>  
  10.   <service>com.hyj.service</service>  
  11.   <serviceimpl>com.hyj.service.impl</serviceimpl>    
  12.   <jsp>/freemarker/jsp</jsp>  
  13.       
  14.  </code>  
  15. </xml-body>   
<?xml version="1.0" encoding="UTF-8"?>
<xml-body>
 <code>
  <projectpath>D:/vobuilder/vobuilder</projectpath>
  
  <model>com.hyj.model</model>
  <action>com.hyj.action</action>
  <dao>com.hyj.dao</dao>
  <daoimpl>com.hyj.dao.jdbc</daoimpl>
  <service>com.hyj.service</service>
  <serviceimpl>com.hyj.service.impl</serviceimpl> 
  <jsp>/freemarker/jsp</jsp>
   
 </code>
</xml-body>

 

 

 

Java代码 复制代码 收藏代码
  1. CodeGenerator cg = new CodeGenerator();   
  2. cg.generate("tableName");   
  3.   
  4. //这样就生成了所有项目的文件   
CodeGenerator cg = new CodeGenerator();
cg.generate("tableName");

//这样就生成了所有项目的文件

  

 

 1.生成代码的引擎

 

Java代码 复制代码 收藏代码
  1. package com.hyj.util;   
  2.   
  3. import java.io.File;   
  4. import java.io.FileOutputStream;   
  5. import java.io.IOException;   
  6. import java.io.OutputStreamWriter;   
  7. import java.io.Writer;   
  8. import java.sql.Connection;   
  9. import java.util.Map;   
  10.   
  11. import org.apache.log4j.LogManager;   
  12. import org.apache.log4j.Logger;   
  13.   
  14. import com.szxhdz.util.Factory;   
  15.   
  16. import freemarker.template.Configuration;   
  17. import freemarker.template.DefaultObjectWrapper;   
  18. import freemarker.template.Template;   
  19. import freemarker.template.TemplateException;   
  20.   
  21. public abstract class AbstractEngine {   
  22.   
  23.  protected static Logger logger = LogManager.getLogger(AbstractEngine.class);   
  24.   
  25.  protected static Connection con;   
  26.   
  27.  protected static Configuration cfg;   
  28.   
  29.  public AbstractEngine(String templatePath) {   
  30.   con = Factory.getConnection();   
  31.   try {   
  32.    cfg = new Configuration();   
  33.    cfg.setDefaultEncoding("gb2312");   
  34.    cfg.setDirectoryForTemplateLoading(new File(templatePath));   
  35.    cfg.setObjectWrapper(new DefaultObjectWrapper());   
  36.   } catch (Exception e) {   
  37.    e.printStackTrace();   
  38.   }   
  39.  }   
  40.   
  41.  public static void create(String ftlTemplate, Map contents, String savePath,   
  42.    String saveFilename) {   
  43.   try {   
  44.    Template temp = cfg.getTemplate(ftlTemplate);   
  45.    logger.info("generate file " + saveFilename + "  in path "  
  46.      + savePath);   
  47.    File file = new File(savePath);   
  48.    if (!file.exists()) {   
  49.     file.mkdirs();   
  50.    }   
  51.    Writer out = new OutputStreamWriter(new FileOutputStream(savePath   
  52.      + "/" + saveFilename), "gb2312");   
  53.    temp.process(contents, out);   
  54.    out.flush();   
  55.   } catch (IOException e) {   
  56.    e.printStackTrace();   
  57.   } catch (TemplateException e1) {   
  58.    e1.printStackTrace();   
  59.   }   
  60.  }   
  61. }  
package com.hyj.util;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.sql.Connection;
import java.util.Map;

import org.apache.log4j.LogManager;
import org.apache.log4j.Logger;

import com.szxhdz.util.Factory;

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

public abstract class AbstractEngine {

 protected static Logger logger = LogManager.getLogger(AbstractEngine.class);

 protected static Connection con;

 protected static Configuration cfg;

 public AbstractEngine(String templatePath) {
  con = Factory.getConnection();
  try {
   cfg = new Configuration();
   cfg.setDefaultEncoding("gb2312");
   cfg.setDirectoryForTemplateLoading(new File(templatePath));
   cfg.setObjectWrapper(new DefaultObjectWrapper());
  } catch (Exception e) {
   e.printStackTrace();
  }
 }

 public static void create(String ftlTemplate, Map contents, String savePath,
   String saveFilename) {
  try {
   Template temp = cfg.getTemplate(ftlTemplate);
   logger.info("generate file " + saveFilename + "  in path "
     + savePath);
   File file = new File(savePath);
   if (!file.exists()) {
    file.mkdirs();
   }
   Writer out = new OutputStreamWriter(new FileOutputStream(savePath
     + "/" + saveFilename), "gb2312");
   temp.process(contents, out);
   out.flush();
  } catch (IOException e) {
   e.printStackTrace();
  } catch (TemplateException e1) {
   e1.printStackTrace();
  }
 }
}

 

2。生成代码的实现类

 

Java代码 复制代码 收藏代码
  1. package com.hyj.util;   
  2.   
  3. import java.sql.SQLException;   
  4. import java.util.ArrayList;   
  5. import java.util.HashMap;   
  6. import java.util.Iterator;   
  7. import java.util.List;   
  8. import java.util.Map;   
  9.   
  10. import com.hyj.util.JDBCUtil.Column;   
  11. import com.szxhdz.util.SystemConstant;   
  12.   
  13. public class CodeGenerator extends AbstractEngine implements Creater{   
  14.     private static final String BASE_PATH = SystemConstant.getProperty("code.projectpath");   
  15.   
  16.     private static final String PACKAGE_PATH = "/src/com/hyj/";   
  17.   
  18.     private static final String JAVA_CODE_GENERATOR_PATH = BASE_PATH   
  19.             + PACKAGE_PATH;   
  20.   
  21.     private static final String TEMPLATE_PATH = BASE_PATH   
  22.             + "/freemarker/template";   
  23.   
  24.     public CodeGenerator() {   
  25.         super(TEMPLATE_PATH);   
  26.     }   
  27.   
  28.     public CodeGenerator(String templatePath) {   
  29.         super(templatePath);   
  30.     }   
  31.   
  32.     public void generatorJSP(String table) {   
  33.         try {   
  34.             List priList = JDBCUtil   
  35.             .getPrimaryKeyColumns(con, null, null, table);   
  36.             Column pk = (Column) priList.get(0);   
  37.             Map params = new HashMap();   
  38.             params.put("table", table);   
  39.             params.put("columnList", JDBCUtil.getTableColumns(con, null, null,   
  40.                     table));   
  41.             params.put("pk", pk);   
  42.             create("add.ftl", params, BASE_PATH + "/freemarker/jsp", table   
  43.                     .toLowerCase().replaceAll("_", "")   
  44.                     + "_add.jsp");   
  45.   
  46.         } catch (Exception e) {   
  47.             e.printStackTrace();   
  48.         }   
  49.     }   
  50.   
  51.     public void generaorAll() {   
  52.         List tbList = null;   
  53.         try {   
  54.             tbList = JDBCUtil.getTables(con, null, "FTFJ_WEB", null);   
  55.             for (Iterator iter = tbList.iterator(); iter.hasNext();) {   
  56.                 String table = (String) iter.next();   
  57.                 generatorJSP(table);   
  58.             }   
  59.         } catch (SQLException e) {   
  60.             e.printStackTrace();   
  61.         }   
  62.     }   
  63.   
  64.     public void generateModel(String table) {   
  65.         try {   
  66.             Map params = new HashMap();   
  67.             params.put("table", table);   
  68.             params.put("columnList", JDBCUtil.getTableColumns(con, null, null,   
  69.                     table));   
  70.             create("bean.ftl", params, JAVA_CODE_GENERATOR_PATH + "model",   
  71.                     table.substring(0, 1).toUpperCase()   
  72.                             + table.toLowerCase().replaceAll("_", "")   
  73.                                     .substring(1) + ".java");   
  74.         } catch (Exception e) {   
  75.             e.printStackTrace();   
  76.         }   
  77.     }   
  78.   
  79.     public void generatorModels() {   
  80.         List tbList = null;   
  81.         try {   
  82.             tbList = JDBCUtil.getTables(con, null, "FTFJ_WEB", null);   
  83.             for (Iterator iter = tbList.iterator(); iter.hasNext();) {   
  84.                 String table = (String) iter.next();   
  85.                 generateModel(table);   
  86.             }   
  87.         } catch (SQLException e) {   
  88.             e.printStackTrace();   
  89.         }   
  90.     }   
  91.   
  92.     public void generateDao(String tableName) {   
  93.         try {   
  94.             List priList = JDBCUtil   
  95.                     .getPrimaryKeyColumns(con, null, null, tableName);   
  96.             Column pk = (Column) priList.get(0);   
  97.             Map params = new HashMap();   
  98.             params.put("tableName", tableName);   
  99.             params.put("columnList", JDBCUtil.getTableColumns(con, null, null,   
  100.                     tableName));   
  101.             params.put("pk", pk);   
  102.             create("dao.ftl", params, BASE_PATH +"/src/"+ SystemConstant.getProperty("code.dao").replace(".","/"),"I"+ tableName   
  103.                     .substring(0, 1).toUpperCase()   
  104.                     + tableName.substring(1,tableName.length()).toLowerCase()   
  105.                             .replaceAll("_", "") + "Dao.java");   
  106.         } catch (Exception e) {   
  107.             e.printStackTrace();   
  108.         }   
  109.     }   
  110.     public void generateDaoImpl(String table) {   
  111.         try {   
  112.             List priList = JDBCUtil   
  113.                     .getPrimaryKeyColumns(con, null, null, table);   
  114.             Column pk = (Column) priList.get(0);   
  115.             Map params = new HashMap();   
  116.             params.put("table", table);   
  117.             params.put("columnList", JDBCUtil.getTableColumns(con, null, null,   
  118.                     table));   
  119.             params.put("pk", pk);   
  120.             create("daoimpl.ftl", params, JAVA_CODE_GENERATOR_PATH + "dao/jdbc", table   
  121.                     .substring(0, 1).toUpperCase()   
  122.                     + table.substring(1, table.length()).toLowerCase()   
  123.                             .replaceAll("_", "") + "Dao.java");   
  124.         } catch (Exception e) {   
  125.             e.printStackTrace();   
  126.         }   
  127.     }   
  128.   
  129.     public void generateSql() {   
  130.   
  131.         List tbList = null;   
  132.         List ls = new ArrayList();   
  133.         try {   
  134.             tbList = JDBCUtil.getTables(con, null, null, null);   
  135.             Map params = new HashMap();   
  136.   
  137.             for (Iterator iter = tbList.iterator(); iter.hasNext();) {   
  138.                 String table = (String) iter.next();   
  139.                 List columnlist = JDBCUtil.getTableColumns(con, null, null,   
  140.                         table);   
  141.                 Column pk = (Column) JDBCUtil.getPrimaryKeyColumns(con, null,   
  142.                         null, table).get(0);   
  143.                 SqlBean sqlbean = new SqlBean();   
  144.                 sqlbean.setColumnList(columnlist);   
  145.                 sqlbean.setPk(pk);   
  146.                 sqlbean.setTable(table);   
  147.                 ls.add(sqlbean);   
  148.             }   
  149.             params.put("ls", ls);   
  150.             create("db.ftl", params, JAVA_CODE_GENERATOR_PATH + "sql", "db.sql");   
  151.         } catch (SQLException e) {   
  152.             e.printStackTrace();   
  153.         }   
  154.     }   
  155.   
  156.     public void generateAction(String table) {   
  157.         try {   
  158.   
  159.             Map params = new HashMap();   
  160.   
  161.             List columnlist = JDBCUtil.getTableColumns(con, null, null, table);   
  162.             params.put("columnlist", columnlist);   
  163.             params.put("table", table);   
  164.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,   
  165.                     table).get(0));   
  166.             create("action.ftl", params, JAVA_CODE_GENERATOR_PATH + "action",   
  167.                     table.substring(0, 1).toUpperCase()   
  168.                             + table.substring(1).toLowerCase().replaceAll("_",   
  169.                                     "") + "Action.java");   
  170.         } catch (Exception e) {   
  171.             e.printStackTrace();   
  172.         }   
  173.     }   
  174.   
  175.     public void generateList(String table) {   
  176.         try {   
  177.   
  178.             Map params = new HashMap();   
  179.   
  180.             List columnlist = JDBCUtil.getTableColumns(con, null, null, table);   
  181.             params.put("columnlist", columnlist);   
  182.             params.put("table", table);   
  183.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,   
  184.                     table).get(0));   
  185.             create("list.ftl", params, BASE_PATH + "/freemarker/jsp", table   
  186.                     .toLowerCase().replaceAll("_", "")   
  187.                     + "_list.jsp");   
  188.         } catch (Exception e) {   
  189.             e.printStackTrace();   
  190.         }   
  191.     }   
  192.   
  193.     public void generateXwork(String table) {   
  194.         try {   
  195.   
  196.             Map params = new HashMap();   
  197.   
  198.             List columnlist = JDBCUtil.getTableColumns(con, null, null, table);   
  199.             params.put("columnlist", columnlist);   
  200.             params.put("table", table);   
  201.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,   
  202.                     table).get(0));   
  203.             create("xwork.ftl", params, JAVA_CODE_GENERATOR_PATH, "xwork-"  
  204.                     + table.toLowerCase().replaceAll("_", "") + ".xml");   
  205.         } catch (Exception e) {   
  206.             e.printStackTrace();   
  207.         }   
  208.     }   
  209.   
  210.     public void generateApplicationContext() {   
  211.         List tbList = null;   
  212.         List tableList = new ArrayList();   
  213.         try {   
  214.             tbList = JDBCUtil.getTables(con, null, null, null);   
  215.             for (Iterator iter = tbList.iterator(); iter.hasNext();) {   
  216.                 String table = (String) iter.next();   
  217.                 tableList.add(table);   
  218.             }   
  219.             Map params = new HashMap();   
  220.             params.put("tableList", tableList);   
  221.             create("applicationContext.ftl", params, JAVA_CODE_GENERATOR_PATH,   
  222.                     "applicationContext.xml");   
  223.         } catch (SQLException e) {   
  224.             e.printStackTrace();   
  225.         } catch (Exception e) {   
  226.             e.printStackTrace();   
  227.         }   
  228.     }   
  229.   
  230.     public void generateValidation(String table) {   
  231.         try {   
  232.             Map params = new HashMap();   
  233.             List columnlist = JDBCUtil.getTableColumns(con, null, null, table);   
  234.             params.put("columnlist", columnlist);   
  235.             params.put("table", table);   
  236.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,table).get(0));   
  237.             create("validation.ftl", params, JAVA_CODE_GENERATOR_PATH + "action",table.substring(0,1).toUpperCase()+table.substring(1).toLowerCase().replaceAll("_", "") + "Action-validation.xml");   
  238.         } catch (Exception e) {   
  239.             e.printStackTrace();   
  240.         }   
  241.     }   
  242.     public void generateValidation_properties(String table) {   
  243.         try {   
  244.             Map params = new HashMap();   
  245.             List columnlist = JDBCUtil.getTableColumns(con, null, null, table);   
  246.             params.put("columnlist", columnlist);   
  247.             params.put("table", table);   
  248.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,table).get(0));   
  249.             create("validation_properties.ftl", params, JAVA_CODE_GENERATOR_PATH + "action",table.substring(0,1).toUpperCase()+table.substring(1).toLowerCase().replaceAll("_", "") + "Action.properties");   
  250.         } catch (Exception e) {   
  251.             e.printStackTrace();   
  252.         }   
  253.     }   
  254.   
  255.   
  256.     public void generateService(String tableName) {   
  257.         try {   
  258.             Map params = new HashMap();   
  259.             List columnlist = JDBCUtil.getTableColumns(con, null, null, tableName);   
  260.             params.put("columnlist", columnlist);   
  261.             params.put("tableName", tableName);   
  262.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,tableName).get(0));   
  263.             create("service.ftl", params, BASE_PATH+"/src/" + SystemConstant.getProperty("code.service").replace(".","/"),"I"+tableName.substring(0,1).toUpperCase()+tableName.substring(1).toLowerCase().replaceAll("_", "") + "Service.java");   
  264.         } catch (Exception e) {   
  265.             e.printStackTrace();   
  266.         }   
  267.            
  268.     }   
  269.   
  270.     public void generateServiceImpl(String tableName) {   
  271.         try {   
  272.             Map params = new HashMap();   
  273.             List columnlist = JDBCUtil.getTableColumns(con, null, null, tableName);   
  274.             params.put("columnlist", columnlist);   
  275.             params.put("tableName", tableName);   
  276.             params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,tableName).get(0));   
  277.             create("serviceimpl.ftl", params, BASE_PATH+"/src/" + SystemConstant.getProperty("code.serviceimpl").replace(".","/"),tableName.substring(0,1).toUpperCase()+tableName.substring(1).toLowerCase().replaceAll("_", "") + "Service.java");   
  278.         } catch (Exception e) {   
  279.             e.printStackTrace();   
  280.         }   
  281.            
  282.     }   
  283.     public void generate(String tableName){   
  284.         generateAction(tableName);   
  285.         generateApplicationContext();   
  286.         generateDao(tableName);   
  287.         generateDaoImpl(tableName);   
  288.         generateList(tableName);   
  289.         generateModel(tableName);   
  290.         generateXwork(tableName);   
  291.         generatorJSP(tableName);   
  292.         generateValidation(tableName);   
  293.         generateValidation_properties(tableName);   
  294.         generateService(tableName);   
  295.         generateServiceImpl(tableName);   
  296.     }   
  297.        
  298.     public static void main(String[] args) {   
  299.         CodeGenerator generator = new CodeGenerator();   
  300.     //  generator.generateDao("ADV_POSITION");   
  301.         //generator.generateAction("ADV_POSITION");   
  302.         generator.generatorJSP("ADV_POSITION");   
  303.     }   
  304.   
  305.        
  306.   
  307. }  
package com.hyj.util;

import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import com.hyj.util.JDBCUtil.Column;
import com.szxhdz.util.SystemConstant;

public class CodeGenerator extends AbstractEngine implements Creater{
	private static final String BASE_PATH = SystemConstant.getProperty("code.projectpath");

	private static final String PACKAGE_PATH = "/src/com/hyj/";

	private static final String JAVA_CODE_GENERATOR_PATH = BASE_PATH
			+ PACKAGE_PATH;

	private static final String TEMPLATE_PATH = BASE_PATH
			+ "/freemarker/template";

	public CodeGenerator() {
		super(TEMPLATE_PATH);
	}

	public CodeGenerator(String templatePath) {
		super(templatePath);
	}

	public void generatorJSP(String table) {
		try {
			List priList = JDBCUtil
			.getPrimaryKeyColumns(con, null, null, table);
			Column pk = (Column) priList.get(0);
			Map params = new HashMap();
			params.put("table", table);
			params.put("columnList", JDBCUtil.getTableColumns(con, null, null,
					table));
			params.put("pk", pk);
			create("add.ftl", params, BASE_PATH + "/freemarker/jsp", table
					.toLowerCase().replaceAll("_", "")
					+ "_add.jsp");

		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generaorAll() {
		List tbList = null;
		try {
			tbList = JDBCUtil.getTables(con, null, "FTFJ_WEB", null);
			for (Iterator iter = tbList.iterator(); iter.hasNext();) {
				String table = (String) iter.next();
				generatorJSP(table);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public void generateModel(String table) {
		try {
			Map params = new HashMap();
			params.put("table", table);
			params.put("columnList", JDBCUtil.getTableColumns(con, null, null,
					table));
			create("bean.ftl", params, JAVA_CODE_GENERATOR_PATH + "model",
					table.substring(0, 1).toUpperCase()
							+ table.toLowerCase().replaceAll("_", "")
									.substring(1) + ".java");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generatorModels() {
		List tbList = null;
		try {
			tbList = JDBCUtil.getTables(con, null, "FTFJ_WEB", null);
			for (Iterator iter = tbList.iterator(); iter.hasNext();) {
				String table = (String) iter.next();
				generateModel(table);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public void generateDao(String tableName) {
		try {
			List priList = JDBCUtil
					.getPrimaryKeyColumns(con, null, null, tableName);
			Column pk = (Column) priList.get(0);
			Map params = new HashMap();
			params.put("tableName", tableName);
			params.put("columnList", JDBCUtil.getTableColumns(con, null, null,
					tableName));
			params.put("pk", pk);
			create("dao.ftl", params, BASE_PATH +"/src/"+ SystemConstant.getProperty("code.dao").replace(".","/"),"I"+ tableName
					.substring(0, 1).toUpperCase()
					+ tableName.substring(1,tableName.length()).toLowerCase()
							.replaceAll("_", "") + "Dao.java");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	public void generateDaoImpl(String table) {
		try {
			List priList = JDBCUtil
					.getPrimaryKeyColumns(con, null, null, table);
			Column pk = (Column) priList.get(0);
			Map params = new HashMap();
			params.put("table", table);
			params.put("columnList", JDBCUtil.getTableColumns(con, null, null,
					table));
			params.put("pk", pk);
			create("daoimpl.ftl", params, JAVA_CODE_GENERATOR_PATH + "dao/jdbc", table
					.substring(0, 1).toUpperCase()
					+ table.substring(1, table.length()).toLowerCase()
							.replaceAll("_", "") + "Dao.java");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generateSql() {

		List tbList = null;
		List ls = new ArrayList();
		try {
			tbList = JDBCUtil.getTables(con, null, null, null);
			Map params = new HashMap();

			for (Iterator iter = tbList.iterator(); iter.hasNext();) {
				String table = (String) iter.next();
				List columnlist = JDBCUtil.getTableColumns(con, null, null,
						table);
				Column pk = (Column) JDBCUtil.getPrimaryKeyColumns(con, null,
						null, table).get(0);
				SqlBean sqlbean = new SqlBean();
				sqlbean.setColumnList(columnlist);
				sqlbean.setPk(pk);
				sqlbean.setTable(table);
				ls.add(sqlbean);
			}
			params.put("ls", ls);
			create("db.ftl", params, JAVA_CODE_GENERATOR_PATH + "sql", "db.sql");
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public void generateAction(String table) {
		try {

			Map params = new HashMap();

			List columnlist = JDBCUtil.getTableColumns(con, null, null, table);
			params.put("columnlist", columnlist);
			params.put("table", table);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,
					table).get(0));
			create("action.ftl", params, JAVA_CODE_GENERATOR_PATH + "action",
					table.substring(0, 1).toUpperCase()
							+ table.substring(1).toLowerCase().replaceAll("_",
									"") + "Action.java");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generateList(String table) {
		try {

			Map params = new HashMap();

			List columnlist = JDBCUtil.getTableColumns(con, null, null, table);
			params.put("columnlist", columnlist);
			params.put("table", table);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,
					table).get(0));
			create("list.ftl", params, BASE_PATH + "/freemarker/jsp", table
					.toLowerCase().replaceAll("_", "")
					+ "_list.jsp");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generateXwork(String table) {
		try {

			Map params = new HashMap();

			List columnlist = JDBCUtil.getTableColumns(con, null, null, table);
			params.put("columnlist", columnlist);
			params.put("table", table);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,
					table).get(0));
			create("xwork.ftl", params, JAVA_CODE_GENERATOR_PATH, "xwork-"
					+ table.toLowerCase().replaceAll("_", "") + ".xml");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generateApplicationContext() {
		List tbList = null;
		List tableList = new ArrayList();
		try {
			tbList = JDBCUtil.getTables(con, null, null, null);
			for (Iterator iter = tbList.iterator(); iter.hasNext();) {
				String table = (String) iter.next();
				tableList.add(table);
			}
			Map params = new HashMap();
			params.put("tableList", tableList);
			create("applicationContext.ftl", params, JAVA_CODE_GENERATOR_PATH,
					"applicationContext.xml");
		} catch (SQLException e) {
			e.printStackTrace();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public void generateValidation(String table) {
		try {
			Map params = new HashMap();
			List columnlist = JDBCUtil.getTableColumns(con, null, null, table);
			params.put("columnlist", columnlist);
			params.put("table", table);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,table).get(0));
			create("validation.ftl", params, JAVA_CODE_GENERATOR_PATH + "action",table.substring(0,1).toUpperCase()+table.substring(1).toLowerCase().replaceAll("_", "") + "Action-validation.xml");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	public void generateValidation_properties(String table) {
		try {
			Map params = new HashMap();
			List columnlist = JDBCUtil.getTableColumns(con, null, null, table);
			params.put("columnlist", columnlist);
			params.put("table", table);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,table).get(0));
			create("validation_properties.ftl", params, JAVA_CODE_GENERATOR_PATH + "action",table.substring(0,1).toUpperCase()+table.substring(1).toLowerCase().replaceAll("_", "") + "Action.properties");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}


	public void generateService(String tableName) {
		try {
			Map params = new HashMap();
			List columnlist = JDBCUtil.getTableColumns(con, null, null, tableName);
			params.put("columnlist", columnlist);
			params.put("tableName", tableName);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,tableName).get(0));
			create("service.ftl", params, BASE_PATH+"/src/" + SystemConstant.getProperty("code.service").replace(".","/"),"I"+tableName.substring(0,1).toUpperCase()+tableName.substring(1).toLowerCase().replaceAll("_", "") + "Service.java");
		} catch (Exception e) {
			e.printStackTrace();
		}
		
	}

	public void generateServiceImpl(String tableName) {
		try {
			Map params = new HashMap();
			List columnlist = JDBCUtil.getTableColumns(con, null, null, tableName);
			params.put("columnlist", columnlist);
			params.put("tableName", tableName);
			params.put("pk", JDBCUtil.getPrimaryKeyColumns(con, null, null,tableName).get(0));
			create("serviceimpl.ftl", params, BASE_PATH+"/src/" + SystemConstant.getProperty("code.serviceimpl").replace(".","/"),tableName.substring(0,1).toUpperCase()+tableName.substring(1).toLowerCase().replaceAll("_", "") + "Service.java");
		} catch (Exception e) {
			e.printStackTrace();
		}
		
	}
	public void generate(String tableName){
		generateAction(tableName);
		generateApplicationContext();
		generateDao(tableName);
	    generateDaoImpl(tableName);
		generateList(tableName);
		generateModel(tableName);
		generateXwork(tableName);
		generatorJSP(tableName);
		generateValidation(tableName);
		generateValidation_properties(tableName);
		generateService(tableName);
		generateServiceImpl(tableName);
	}
	
	public static void main(String[] args) {
		CodeGenerator generator = new CodeGenerator();
	//	generator.generateDao("ADV_POSITION");
		//generator.generateAction("ADV_POSITION");
		generator.generatorJSP("ADV_POSITION");
	}

	

}

 

 3。下面是jdbc底层操作类

Java代码 复制代码 收藏代码
  1. /*  
  2.  * JDBCUtil.java  
  3.  *  
  4.  * Created on November 9, 2002, 4:27 PM  
  5.  */  
  6.   
  7. package com.hyj.util;   
  8. import java.sql.Connection;   
  9. import java.sql.DatabaseMetaData;   
  10. import java.sql.ResultSet;   
  11. import java.sql.SQLException;   
  12. import java.sql.Types;   
  13. import java.util.HashMap;   
  14. import java.util.HashSet;   
  15. import java.util.LinkedList;   
  16. import java.util.List;   
  17. import java.util.Map;   
  18. import java.util.Set;   
  19.   
  20. import net.sf.hibernate.Hibernate;   
  21. import net.sf.hibernate.type.NullableType;   
  22.   
  23. import org.apache.commons.logging.Log;   
  24. import org.apache.commons.logging.LogFactory;   
  25.   
  26. /**  
  27.  *  
  28.  * @author  Administrator  
  29.  */  
  30. public class JDBCUtil {   
  31.     private static final Log logger = LogFactory.getLog(JDBCUtil.class);   
  32.     public static class Column {   
  33.         public String name;   
  34.         public int sqlType;   
  35.         public int sqlColumnLength;   
  36.         public int sqlDecimalLength;   
  37.         public boolean sqlNotNull;   
  38.         public boolean sqlReadOnly;   
  39.         public NullableType hibernateType;   
  40.         public Class javaType;   
  41.         public String sqlTypeName;   
  42.         public String defalutValue;   
  43.         public boolean equals(Object o) {   
  44.             boolean rv = false;   
  45.             if (o != null && o instanceof JDBCUtil.Column) {   
  46.                 rv = (name.equals(((JDBCUtil.Column)o).name));   
  47.             }   
  48.             return rv;   
  49.         }   
  50.         public int hashCode() {   
  51.             return (name != null) ? name.hashCode() : 0;   
  52.         }   
  53.         public String getName() {   
  54.             return name;   
  55.         }   
  56.         public void setName(String name) {   
  57.             this.name = name;   
  58.         }   
  59.         public NullableType getHibernateType() {   
  60.             return hibernateType;   
  61.         }   
  62.         public void setHibernateType(NullableType hibernateType) {   
  63.             this.hibernateType = hibernateType;   
  64.         }   
  65.         public Class getJavaType() {   
  66.             return javaType;   
  67.         }   
  68.         public void setJavaType(Class javaType) {   
  69.             this.javaType = javaType;   
  70.         }   
  71.         public int getSqlColumnLength() {   
  72.             return sqlColumnLength;   
  73.         }   
  74.         public void setSqlColumnLength(int sqlColumnLength) {   
  75.             this.sqlColumnLength = sqlColumnLength;   
  76.         }   
  77.         public int getSqlDecimalLength() {   
  78.             return sqlDecimalLength;   
  79.         }   
  80.         public void setSqlDecimalLength(int sqlDecimalLength) {   
  81.             this.sqlDecimalLength = sqlDecimalLength;   
  82.         }   
  83.         public boolean isSqlNotNull() {   
  84.             return sqlNotNull;   
  85.         }   
  86.         public void setSqlNotNull(boolean sqlNotNull) {   
  87.             this.sqlNotNull = sqlNotNull;   
  88.         }   
  89.         public boolean isSqlReadOnly() {   
  90.             return sqlReadOnly;   
  91.         }   
  92.         public void setSqlReadOnly(boolean sqlReadOnly) {   
  93.             this.sqlReadOnly = sqlReadOnly;   
  94.         }   
  95.         public int getSqlType() {   
  96.             return sqlType;   
  97.         }   
  98.         public void setSqlType(int sqlType) {   
  99.             this.sqlType = sqlType;   
  100.         }   
  101.         public String getSqlTypeName() {   
  102.             return sqlTypeName;   
  103.         }   
  104.         public void setSqlTypeName(String sqlTypeName) {   
  105.             this.sqlTypeName = sqlTypeName;   
  106.         }   
  107.         public String getDefalutValue() {   
  108.             return defalutValue;   
  109.         }   
  110.         public void setDefalutValue(String defalutValue) {   
  111.             this.defalutValue = defalutValue;   
  112.         }   
  113.            
  114.            
  115.     };   
  116.        
  117.     public static List getCatalogs(Connection c) throws SQLException {   
  118.         DatabaseMetaData dmd = c.getMetaData();   
  119.         ResultSet rs = null;   
  120.         try {   
  121.             rs = dmd.getCatalogs();   
  122.             List l = new LinkedList();   
  123.             while (rs.next()) {   
  124.                 l.add(rs.getString(1));   
  125.             }   
  126.             return l;   
  127.         }   
  128.         finally {   
  129.             if (rs != null) rs.close();   
  130.         }   
  131.     }   
  132.        
  133.     public static Map getSchemas(Connection c) throws SQLException {   
  134.         DatabaseMetaData dmd = c.getMetaData();   
  135.         ResultSet rs = null;   
  136.         try {   
  137.             rs = dmd.getSchemas();   
  138.             Map map = new HashMap();   
  139.             List l;   
  140.             while (rs.next()) {   
  141.                 String schema = rs.getString(1);   
  142.                 String catalog = null;   
  143.                 if (rs.getMetaData().getColumnCount() > 1) {   
  144.                     catalog = rs.getString(2);   
  145.                 };   
  146.                 l = (List)map.get(catalog);   
  147.                 if (l == null) {   
  148.                     l = new LinkedList();   
  149.                     map.put(catalog, l);   
  150.                 }   
  151.                 l.add(schema);   
  152.             }   
  153.             return map;   
  154.         }   
  155.         finally {   
  156.             if (rs != null) rs.close();   
  157.         }   
  158.     }   
  159.        
  160.     public static List getTables(Connection c, String catalog, String schema, String tablePattern) throws SQLException {   
  161.         logger.debug("catalog='" + catalog + "'");   
  162.         logger.debug("schema='" + schema + "'");   
  163.         logger.debug("table='" + tablePattern + "'");   
  164.         DatabaseMetaData dmd = c.getMetaData();   
  165.         ResultSet rs = null;   
  166.         try {   
  167.             rs = dmd.getTables(catalog, schema, tablePattern, new String[] {"TABLE", "VIEW", "SYNONYM", "ALIAS"} );   
  168.             List l= new LinkedList();   
  169.             while (rs.next()) {   
  170.                 l.add(rs.getString(3));   
  171.             }   
  172.             return l;   
  173.         }   
  174.         finally {   
  175.             if (rs != null) rs.close();   
  176.         }   
  177.     }   
  178.        
  179.     public static Set getForeignKeyColumns(Connection c, String catalog, String schema, String table) throws SQLException {   
  180.         logger.debug("catalog='" + catalog + "'");   
  181.         logger.debug("schema='" + schema + "'");   
  182.         logger.debug("table='" + table + "'");   
  183.         DatabaseMetaData dmd = c.getMetaData();   
  184.         ResultSet rs = null;   
  185.         try {   
  186.             rs = dmd.getImportedKeys(catalog, schema, table);   
  187.             HashSet columns = new HashSet();   
  188.             while (rs.next()) {   
  189.                 columns.add(rs.getString(8));   
  190.             }   
  191.             return columns;   
  192.         }   
  193.         finally {   
  194.             if (rs != null) rs.close();   
  195.         }   
  196.     }   
  197.        
  198.     public static List getPrimaryKeyColumns(Connection c, String catalog, String schema, String table) throws SQLException {   
  199.         logger.debug("catalog='" + catalog + "'");   
  200.         logger.debug("schema='" + schema + "'");   
  201.         logger.debug("table='" + table + "'");   
  202.         DatabaseMetaData dmd = c.getMetaData();   
  203.         ResultSet rs = null;   
  204.         try {   
  205.             rs = dmd.getPrimaryKeys(catalog, schema, table);   
  206.                
  207.             List pkColumns = new LinkedList();;   
  208.             while (rs.next()) {   
  209.                 List tmp = getTableColumns(c, catalog, schema, table, rs.getString(4));   
  210.                 Column pkColumn = (Column)tmp.get(0);   
  211.                 pkColumns.add(pkColumn);   
  212.             }   
  213.             return pkColumns;   
  214.         }   
  215.         finally {   
  216.             if (rs != null) rs.close();   
  217.         }   
  218.     }   
  219.        
  220.     public static List getTableColumns(Connection c, String catalog, String schema, String table) throws SQLException  {   
  221.         return getTableColumns(c, catalog, schema, table, null);   
  222.     }   
  223.        
  224.     public static List getTableColumns(Connection c, String catalog, String schema, String table, String columnPattern) throws SQLException {   
  225.         logger.debug("catalog='" + catalog + "'");   
  226.         logger.debug("schema='" + schema + "'");   
  227.         logger.debug("table='" + table + "'");   
  228.         logger.debug("column='" + columnPattern+ "'");   
  229.         DatabaseMetaData dmd = c.getMetaData();   
  230.         ResultSet rs = null;   
  231.         try {   
  232.             rs = dmd.getColumns(catalog, schema, table, columnPattern);   
  233.             List columns = new LinkedList();   
  234.             while (rs.next()) {   
  235.                 JDBCUtil.Column aCol = new JDBCUtil.Column();   
  236.                 aCol.sqlTypeName = rs.getString(6);   
  237.                 aCol.defalutValue = rs.getString(13);   
  238.                 aCol.name = rs.getString(4);   
  239.                 aCol.sqlType = rs.getShort(5);   
  240.                 aCol.sqlColumnLength=rs.getInt(7);   
  241.                 aCol.sqlDecimalLength=rs.getInt(9);   
  242.                 aCol.sqlNotNull = ("NO".equals(rs.getString(18)));   
  243.                 aCol.hibernateType = getHibernateType(   
  244.                     aCol.sqlType,   
  245.                     aCol.sqlColumnLength,   
  246.                     aCol.sqlDecimalLength   
  247.                 );   
  248.                 aCol.javaType = getJavaType(   
  249.                     aCol.sqlType,   
  250.                     aCol.sqlColumnLength,   
  251.                     aCol.sqlDecimalLength   
  252.                 );   
  253.                 columns.add(aCol);   
  254.             }   
  255.             return columns;   
  256.         }   
  257.         finally {   
  258.             if (rs != null) rs.close();   
  259.         }   
  260.            
  261.     }   
  262.        
  263.     public static NullableType getHibernateType(int sqlType, int columnSize, int decimalDigits) {   
  264.         logger.debug("sqlType=" + sqlType);   
  265.         logger.debug("columnSize=" + columnSize);   
  266.         logger.debug("decimalDigits=" + decimalDigits);   
  267.         NullableType rv=Hibernate.SERIALIZABLE;   
  268.         if (sqlType == Types.CHAR || sqlType == Types.VARCHAR) {   
  269.             rv = Hibernate.STRING;   
  270.         }   
  271.         else if (sqlType == Types.FLOAT || sqlType == Types.REAL) {   
  272.             rv = Hibernate.FLOAT;   
  273.         }   
  274.         else if (sqlType == Types.INTEGER) {   
  275.             rv = Hibernate.INTEGER;   
  276.         }   
  277.         else if (sqlType == Types.DOUBLE) {   
  278.             rv = Hibernate.DOUBLE;   
  279.         }   
  280.         else if (sqlType == Types.DATE) {   
  281.             rv = Hibernate.DATE;   
  282.         }   
  283.         else if (sqlType == Types.TIMESTAMP) {   
  284.             rv = Hibernate.TIMESTAMP;   
  285.         }   
  286.         else if (sqlType == Types.TIME) {   
  287.             rv = Hibernate.TIME;   
  288.         }   
  289.         // commented to support JDK version < 1.4   
  290.         /*      else if (sqlType == Types.BOOLEAN) {  
  291.             rv = Hibernate.BOOLEAN;  
  292.         } */  
  293.         else if (sqlType == Types.SMALLINT) {   
  294.             rv = Hibernate.SHORT;   
  295.         }   
  296.         else if (sqlType == Types.BIT) {   
  297.             rv = Hibernate.BYTE;   
  298.         }   
  299.         else if (sqlType == Types.BIGINT) {   
  300.             rv = Hibernate.LONG;   
  301.         }   
  302.         else if (sqlType == Types.NUMERIC || sqlType == Types.DECIMAL) {   
  303.             if (decimalDigits == 0) {   
  304.                 if (columnSize == 1) {   
  305.                     rv = Hibernate.BYTE;   
  306.                 }   
  307.                 else if (columnSize < 5) {   
  308.                     rv = Hibernate.SHORT;   
  309.                 }   
  310.                 else if (columnSize < 10) {   
  311.                     rv = Hibernate.INTEGER;   
  312.                 }   
  313.                 else {   
  314.                     rv = Hibernate.LONG;   
  315.                 }   
  316.             }   
  317.             else {   
  318.                 if (columnSize < 9) {   
  319.                     rv = Hibernate.FLOAT;   
  320.                 }   
  321.                 else {   
  322.                     rv = Hibernate.DOUBLE;   
  323.                 }   
  324.             }   
  325.         }   
  326.         return rv;   
  327.     }   
  328.        
  329.     public static Class getJavaType(int sqlType, int columnSize, int decimalDigits) {   
  330.         logger.debug("sqlType=" + sqlType);   
  331.         logger.debug("columnSize=" + columnSize);   
  332.         logger.debug("decimalDigits=" + decimalDigits);   
  333.         Class rv=String.class;   
  334.         if (sqlType == Types.CHAR || sqlType == Types.VARCHAR) {   
  335.             rv = String.class;   
  336.         }   
  337.         else if (sqlType == Types.FLOAT || sqlType == Types.REAL) {   
  338.             rv = Float.class;   
  339.         }   
  340.         else if (sqlType == Types.INTEGER) {   
  341.             rv = Integer.class;   
  342.         }   
  343.         else if (sqlType == Types.DOUBLE) {   
  344.             rv = Double.class;   
  345.         }   
  346.         else if (sqlType == Types.DATE) {   
  347.             //rv = java.util.Date.class;   
  348.             rv = String.class;   
  349.         }   
  350.         else if (sqlType == Types.TIMESTAMP) {   
  351.             //rv = java.util.Date.class;   
  352.             rv = String.class;   
  353.         }   
  354.         else if (sqlType == Types.TIME) {   
  355.             //rv = java.util.Date.class;   
  356.             rv = String.class;   
  357.         }   
  358.         // commented to support JDK version < 1.4   
  359.         /*      else if (sqlType == Types.BOOLEAN) {  
  360.             rv = Boolean.class;  
  361.         } */  
  362.         else if (sqlType == Types.SMALLINT) {   
  363.             rv = Short.class;   
  364.         }   
  365.         else if (sqlType == Types.BIT) {   
  366. //          rv = Byte.class;   
  367.             rv = Integer.class;   
  368.         }   
  369.         else if (sqlType == Types.BIGINT) {   
  370.             rv = Long.class;   
  371.         }   
  372.         else if (sqlType == Types.NUMERIC || sqlType == Types.DECIMAL) {   
  373.             if (decimalDigits == 0) {   
  374.                 if (columnSize == 1) {   
  375. //                  rv = Byte.class;   
  376.                     rv=  Integer.class;   
  377.                 }   
  378.                 else if (columnSize < 5) {   
  379.                     rv = Short.class;   
  380.                 }   
  381.                 else if (columnSize < 10) {   
  382.                     rv = Integer.class;   
  383.                 }   
  384.                 else {   
  385.                     rv = Long.class;   
  386.                 }   
  387.             }   
  388.             else {   
  389.                 if (columnSize < 9) {   
  390.                     rv = Float.class;   
  391.                 }   
  392.                 else {   
  393.                     rv = Double.class;   
  394.                 }   
  395.             }   
  396.         }   
  397.         return rv;   
  398.     }   
  399.   
  400. }  
/*
 * JDBCUtil.java
 *
 * Created on November 9, 2002, 4:27 PM
 */

package com.hyj.util;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import net.sf.hibernate.Hibernate;
import net.sf.hibernate.type.NullableType;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 *
 * @author  Administrator
 */
public class JDBCUtil {
	private static final Log logger = LogFactory.getLog(JDBCUtil.class);
	public static class Column {
		public String name;
		public int sqlType;
		public int sqlColumnLength;
		public int sqlDecimalLength;
		public boolean sqlNotNull;
		public boolean sqlReadOnly;
		public NullableType hibernateType;
		public Class javaType;
		public String sqlTypeName;
		public String defalutValue;
		public boolean equals(Object o) {
			boolean rv = false;
			if (o != null && o instanceof JDBCUtil.Column) {
				rv = (name.equals(((JDBCUtil.Column)o).name));
			}
			return rv;
		}
		public int hashCode() {
			return (name != null) ? name.hashCode() : 0;
		}
		public String getName() {
			return name;
		}
		public void setName(String name) {
			this.name = name;
		}
		public NullableType getHibernateType() {
			return hibernateType;
		}
		public void setHibernateType(NullableType hibernateType) {
			this.hibernateType = hibernateType;
		}
		public Class getJavaType() {
			return javaType;
		}
		public void setJavaType(Class javaType) {
			this.javaType = javaType;
		}
		public int getSqlColumnLength() {
			return sqlColumnLength;
		}
		public void setSqlColumnLength(int sqlColumnLength) {
			this.sqlColumnLength = sqlColumnLength;
		}
		public int getSqlDecimalLength() {
			return sqlDecimalLength;
		}
		public void setSqlDecimalLength(int sqlDecimalLength) {
			this.sqlDecimalLength = sqlDecimalLength;
		}
		public boolean isSqlNotNull() {
			return sqlNotNull;
		}
		public void setSqlNotNull(boolean sqlNotNull) {
			this.sqlNotNull = sqlNotNull;
		}
		public boolean isSqlReadOnly() {
			return sqlReadOnly;
		}
		public void setSqlReadOnly(boolean sqlReadOnly) {
			this.sqlReadOnly = sqlReadOnly;
		}
		public int getSqlType() {
			return sqlType;
		}
		public void setSqlType(int sqlType) {
			this.sqlType = sqlType;
		}
		public String getSqlTypeName() {
			return sqlTypeName;
		}
		public void setSqlTypeName(String sqlTypeName) {
			this.sqlTypeName = sqlTypeName;
		}
		public String getDefalutValue() {
			return defalutValue;
		}
		public void setDefalutValue(String defalutValue) {
			this.defalutValue = defalutValue;
		}
		
		
	};
	
	public static List getCatalogs(Connection c) throws SQLException {
		DatabaseMetaData dmd = c.getMetaData();
		ResultSet rs = null;
		try {
			rs = dmd.getCatalogs();
			List l = new LinkedList();
			while (rs.next()) {
				l.add(rs.getString(1));
			}
			return l;
		}
		finally {
			if (rs != null) rs.close();
		}
	}
	
	public static Map getSchemas(Connection c) throws SQLException {
		DatabaseMetaData dmd = c.getMetaData();
		ResultSet rs = null;
		try {
			rs = dmd.getSchemas();
			Map map = new HashMap();
			List l;
			while (rs.next()) {
				String schema = rs.getString(1);
				String catalog = null;
				if (rs.getMetaData().getColumnCount() > 1) {
					catalog = rs.getString(2);
				};
				l = (List)map.get(catalog);
				if (l == null) {
					l = new LinkedList();
					map.put(catalog, l);
				}
				l.add(schema);
			}
			return map;
		}
		finally {
			if (rs != null) rs.close();
		}
	}
	
	public static List getTables(Connection c, String catalog, String schema, String tablePattern) throws SQLException {
		logger.debug("catalog='" + catalog + "'");
		logger.debug("schema='" + schema + "'");
		logger.debug("table='" + tablePattern + "'");
		DatabaseMetaData dmd = c.getMetaData();
		ResultSet rs = null;
		try {
			rs = dmd.getTables(catalog, schema, tablePattern, new String[] {"TABLE", "VIEW", "SYNONYM", "ALIAS"} );
			List l= new LinkedList();
			while (rs.next()) {
				l.add(rs.getString(3));
			}
			return l;
		}
		finally {
			if (rs != null) rs.close();
		}
	}
	
	public static Set getForeignKeyColumns(Connection c, String catalog, String schema, String table) throws SQLException {
		logger.debug("catalog='" + catalog + "'");
		logger.debug("schema='" + schema + "'");
		logger.debug("table='" + table + "'");
		DatabaseMetaData dmd = c.getMetaData();
		ResultSet rs = null;
		try {
			rs = dmd.getImportedKeys(catalog, schema, table);
			HashSet columns = new HashSet();
			while (rs.next()) {
				columns.add(rs.getString(8));
			}
			return columns;
		}
		finally {
			if (rs != null) rs.close();
		}
	}
	
	public static List getPrimaryKeyColumns(Connection c, String catalog, String schema, String table) throws SQLException {
		logger.debug("catalog='" + catalog + "'");
		logger.debug("schema='" + schema + "'");
		logger.debug("table='" + table + "'");
		DatabaseMetaData dmd = c.getMetaData();
		ResultSet rs = null;
		try {
			rs = dmd.getPrimaryKeys(catalog, schema, table);
			
			List pkColumns = new LinkedList();;
			while (rs.next()) {
				List tmp = getTableColumns(c, catalog, schema, table, rs.getString(4));
				Column pkColumn = (Column)tmp.get(0);
				pkColumns.add(pkColumn);
			}
			return pkColumns;
		}
		finally {
			if (rs != null) rs.close();
		}
	}
	
	public static List getTableColumns(Connection c, String catalog, String schema, String table) throws SQLException  {
		return getTableColumns(c, catalog, schema, table, null);
	}
	
	public static List getTableColumns(Connection c, String catalog, String schema, String table, String columnPattern) throws SQLException {
		logger.debug("catalog='" + catalog + "'");
		logger.debug("schema='" + schema + "'");
		logger.debug("table='" + table + "'");
		logger.debug("column='" + columnPattern+ "'");
		DatabaseMetaData dmd = c.getMetaData();
		ResultSet rs = null;
		try {
			rs = dmd.getColumns(catalog, schema, table, columnPattern);
			List columns = new LinkedList();
			while (rs.next()) {
				JDBCUtil.Column aCol = new JDBCUtil.Column();
				aCol.sqlTypeName = rs.getString(6);
				aCol.defalutValue = rs.getString(13);
				aCol.name = rs.getString(4);
				aCol.sqlType = rs.getShort(5);
				aCol.sqlColumnLength=rs.getInt(7);
				aCol.sqlDecimalLength=rs.getInt(9);
				aCol.sqlNotNull = ("NO".equals(rs.getString(18)));
				aCol.hibernateType = getHibernateType(
					aCol.sqlType,
					aCol.sqlColumnLength,
					aCol.sqlDecimalLength
				);
				aCol.javaType = getJavaType(
					aCol.sqlType,
					aCol.sqlColumnLength,
					aCol.sqlDecimalLength
				);
				columns.add(aCol);
			}
			return columns;
		}
		finally {
			if (rs != null) rs.close();
		}
		
	}
	
	public static NullableType getHibernateType(int sqlType, int columnSize, int decimalDigits) {
		logger.debug("sqlType=" + sqlType);
		logger.debug("columnSize=" + columnSize);
		logger.debug("decimalDigits=" + decimalDigits);
		NullableType rv=Hibernate.SERIALIZABLE;
		if (sqlType == Types.CHAR || sqlType == Types.VARCHAR) {
			rv = Hibernate.STRING;
		}
		else if (sqlType == Types.FLOAT || sqlType == Types.REAL) {
			rv = Hibernate.FLOAT;
		}
		else if (sqlType == Types.INTEGER) {
			rv = Hibernate.INTEGER;
		}
		else if (sqlType == Types.DOUBLE) {
			rv = Hibernate.DOUBLE;
		}
		else if (sqlType == Types.DATE) {
			rv = Hibernate.DATE;
		}
		else if (sqlType == Types.TIMESTAMP) {
			rv = Hibernate.TIMESTAMP;
		}
		else if (sqlType == Types.TIME) {
			rv = Hibernate.TIME;
		}
		// commented to support JDK version < 1.4
		/*      else if (sqlType == Types.BOOLEAN) {
			rv = Hibernate.BOOLEAN;
		} */
		else if (sqlType == Types.SMALLINT) {
			rv = Hibernate.SHORT;
		}
		else if (sqlType == Types.BIT) {
			rv = Hibernate.BYTE;
		}
		else if (sqlType == Types.BIGINT) {
			rv = Hibernate.LONG;
		}
		else if (sqlType == Types.NUMERIC || sqlType == Types.DECIMAL) {
			if (decimalDigits == 0) {
				if (columnSize == 1) {
					rv = Hibernate.BYTE;
				}
				else if (columnSize < 5) {
					rv = Hibernate.SHORT;
				}
				else if (columnSize < 10) {
					rv = Hibernate.INTEGER;
				}
				else {
					rv = Hibernate.LONG;
				}
			}
			else {
				if (columnSize < 9) {
					rv = Hibernate.FLOAT;
				}
				else {
					rv = Hibernate.DOUBLE;
				}
			}
		}
		return rv;
	}
	
	public static Class getJavaType(int sqlType, int columnSize, int decimalDigits) {
		logger.debug("sqlType=" + sqlType);
		logger.debug("columnSize=" + columnSize);
		logger.debug("decimalDigits=" + decimalDigits);
		Class rv=String.class;
		if (sqlType == Types.CHAR || sqlType == Types.VARCHAR) {
			rv = String.class;
		}
		else if (sqlType == Types.FLOAT || sqlType == Types.REAL) {
			rv = Float.class;
		}
		else if (sqlType == Types.INTEGER) {
			rv = Integer.class;
		}
		else if (sqlType == Types.DOUBLE) {
			rv = Double.class;
		}
		else if (sqlType == Types.DATE) {
			//rv = java.util.Date.class;
			rv = String.class;
		}
		else if (sqlType == Types.TIMESTAMP) {
			//rv = java.util.Date.class;
			rv = String.class;
		}
		else if (sqlType == Types.TIME) {
			//rv = java.util.Date.class;
			rv = String.class;
		}
		// commented to support JDK version < 1.4
		/*      else if (sqlType == Types.BOOLEAN) {
			rv = Boolean.class;
		} */
		else if (sqlType == Types.SMALLINT) {
			rv = Short.class;
		}
		else if (sqlType == Types.BIT) {
//			rv = Byte.class;
			rv = Integer.class;
		}
		else if (sqlType == Types.BIGINT) {
			rv = Long.class;
		}
		else if (sqlType == Types.NUMERIC || sqlType == Types.DECIMAL) {
			if (decimalDigits == 0) {
				if (columnSize == 1) {
//					rv = Byte.class;
					rv=  Integer.class;
				}
				else if (columnSize < 5) {
					rv = Short.class;
				}
				else if (columnSize < 10) {
					rv = Integer.class;
				}
				else {
					rv = Long.class;
				}
			}
			else {
				if (columnSize < 9) {
					rv = Float.class;
				}
				else {
					rv = Double.class;
				}
			}
		}
		return rv;
	}

}

 

 

 

分享到：
| CXF 2.3 集成Spring3.0入门 HelloWorld
评论
1 楼 smilease 2016-06-09  
帮大忙了，非常感谢
发表评论

您还没有登录,请您登录后再发表评论

相关推荐

    使用freemarker模版引擎自动生成java项目源代码.zip

    java项目源码

    FreeMarker模板开发手册-用于代码生成器.pdf

    FreeMarker 是一款模板引擎：一种基于...格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包或者说 是类库。它不是面向最终用户，而是为程序员提供的可以嵌入他们开发产品的一款应用程序。

    中文FreeMarker教程《全》.pdf（英文手册的翻译中文版本）

    FreeMarker 是一款模板引擎：即一种基于模板、用来生成输出文本(任何来自于 HTML 格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包，或者 说是一个类库。它不是面向最终用户的，而是为...

    FreeMarker 2.3.19 中文手册

    格式的文本用来自动生成源代码)的通用工具。它是为Java 程序员提供的一个开发包，或者 说是一个类库。它不是面向最终用户的，而是为程序员提供的一款可以嵌入他们所开发产品 的应用程序。 版权说明： FreeMarker ...

    FreeMarker模板引擎资料大全

    FreeMarker 是一款模板引擎：即一种基于模板、用来生成输出文本(任何来自于 HTML格式，pdf格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包，本教程绝对详细，叫你从精通，玩转它

    springMVC+freemarker框架

    FreeMarker 是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于 HTML 格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包或者说是类库。它不是面向最终用户，而是为程序员...

    FreeMarker中文版（详细教程）

    FreeMarker 是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于 HTML 格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包或者说是类库。它不是面向最终用户，而是为程序员...

    学成在线-freemarker v1.1.pdf

    freemarker开发文档...格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包，或者 说是一个类库。它不是面向最终用户的，而是为程序员提供的一款可以嵌入他们所开发产品 的应用程序。

    FreeMarker中文手册

    格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包，或者 说是一个类库。 它不是面向最终用户的，而是为程序员提供的一款可以嵌入他们所开发产品 的应用程序。 FreeMarker 实际上是被...

    FreeMarker 2.3.16手册

    FreeMarker是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于HTML格式的文本用来自动生成源代码)的通用工具。

    FreeMarker+SpringBoot模板引擎demo

    FreeMarker 是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于 HTML 格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包或者说是类库。它不是面向最终用户，而是为程序员...

    FreeMarker_Manual

    FreeMarker是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于HTML格式的文本用来自动生成源代码)的通用工具。本文教你怎么使用Freemarker。

    freeMarker，中文文档，插件

    包含了中文文档，eclipse的编辑插件 和最新FreeMarker压缩包。 FreeMarker是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于HTML格式的文本用来自动生成源代码)的通用工具。

    p12swagger+freemarker.pdf

    Sagger2 是一个可以根据接口定义自动生成接口 API 文档的框架 Freemarker freemarker 是一款比较优秀的模板引擎 5.1.简介 参考：http://freemarker.foofun.cn/ FreeMarker 是一款 模板引擎： 即一种基于模板和要...

    JAVA上百实例源码以及开源项目源代码

    日历表格面板 [ConfigLine.java] 控制条类 [RoundBox.java] 限定选择控件 [MonthMaker.java] 月份表算法类 [Pallet.java] 调色板，统一配色类 Java扫雷源码 Java生成自定义控件源代码 2个目标文件 Java实现HTTP连接...

    FreeMarker手册

    FreeMarker手册 FreeMarker 是一款模板...格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包或者说 是类库。它不是面向最终用户，而是为程序员提供的可以嵌入他们开发产品的一款应用程序

    jeecg-framework 源代码

    • 代码生成器：自动生成美观大方的页面及后台代码 • 查询条件生成器： 动态拼SQL,追加查询条件 • 页面校验器：EasyUI 页面检验机制 • 完整Spring用户权限 • 报表整合：Excel简易导出工具类+Highcharts图形...

    好用的代码生成源码

    易于做二次开发,整个生成器本身就是java源代码,源代码核心十分精简,并且鼓励你修改代码,也可以作为任何语言的代码生成器 配置简单,只有一个配置文件generator.properties 生成器的运行 在eclipse中运行 配置...

    SpringBoot+FreeMarker模板引擎demo

    FreeMarker 是一款模板引擎：一种基于模板的、用来生成输出文本(任何来自于 HTML 格式的文本用来自动生成源代码)的通用工具。它是为 Java 程序员提供的一个开发包或者说是类库。它不是面向最终用户，而是为程序员...

Magicbox
Global site tag (gtag.js) - Google Analytics