Java代码生成器

来公司一个月了,时间虽然不长,但是我感觉自己还是学到了不少的东西,至少在学校从没有过这样的感觉。并且我也从现在开始写博客了,希望博友能多来浏览,提出批评指正。先来看看我的第一篇博客。事情起源于修改项目的UI(这个过程就不多说了)。我的指导人让我把项目的生成模版修改一下,适应我们的新样式,可是模版是freemarker写的,我完全都不会,没办法,只能自己去摸索了,费了九牛二虎之力终于修改好了。闲
摘要由CSDN通过智能技术生成

来公司一个月了,时间虽然不长,但是我感觉自己还是学到了不少的东西,至少在学校从没有过这样的感觉。并且我也从现在开始写博客了,希望博友能多来浏览,提出批评指正。先来看看我的第一篇博客。

事情起源于修改项目的UI(这个过程就不多说了)。我的指导人让我把项目的生成模版修改一下,适应我们的新样式,可是模版是freemarker写的,我完全都不会哭,没办法,只能自己去摸索了,费了九牛二虎之力终于修改好了。闲来无事,我就想这玩意好像还挺有用的,要不我也自己弄一个,于是就开始了编写我自己的代码生成器了。

好了,“废话”不多说,直接上代码。

自定义表字段结构和表结构

public class MyField {	
	private String field_name;
	private String sql_type;
	private String java_type;
	private String field_comment;
	private boolean is_primary;
}

public class MyTable {		

	//表名
	private String tableName;
	//普通字段集合,一个字段包括了:字段名,SQL类型,Java类型,注释,是否主键标志
	private List<MyField> common_fields;
	//主键字段集合
	private List<MyField> key_fields;
	//字段中是否有日期型的,如果有,那么在生成实体类时,就需要导java.util.Date包了
	public boolean date_flag = false;
	//字段中是否有numeric和decimal类型的,如果有,那么,在生成实体类时就需要导java.math.BigDecimal包了
	public boolean math_flag = false;
	//生成实体类,mapper接口等等时的基名
	public String package_name_base;

}


工具类(里面有一部分是我最开始想纯粹用java代码来生成想要的文件,后来发现这种方法完全是硬编码,所以放弃)

public class DataBaseUtil {
	
	private static String jdbc_url ;
	private static String jdbc_driver ;
	private static String jdbc_user ;
	private static String jdbc_password ;
	private static Integer jdbc_poolsize;	

	private static List<Connection> connectionPool;
	
	
	/**
	 * 静态代码块,用于读取db.properties,初始化和数据库连接相关的信息
	 */
	static{
		Properties properties = new Properties();
		InputStream input = null;
		String path = "db.properties";
		input = Thread.currentThread().getContextClassLoader().getResourceAsStream(path);
		try {
			properties.load(input);
			jdbc_driver = properties.getProperty("jdbc_driver");
			jdbc_url = properties.getProperty("jdbc_url");
			jdbc_user = properties.getProperty("jdbc_user");
			jdbc_password = properties.getProperty("jdbc_password");
			jdbc_poolsize = Integer.parseInt(properties.getProperty("jdbc_poolsize"));
			connectionPool = new ArrayList<Connection>(jdbc_poolsize);			
			Class.forName(jdbc_driver);
			int i;
			for(i = 0 ; i < jdbc_poolsize ; i++){
				Connection connection = DriverManager.getConnection(jdbc_url, jdbc_user, jdbc_password);
				connectionPool.add(connection);
			}			
		} catch (IOException e) {
			System.out.println("加载属性文件失败,请检查db.properties文件是否在src下面");
			System.exit(-1);
		} catch (ClassNotFoundException e) {
			StringBuffer buffer = new StringBuffer();
			buffer.append("加载数据库驱动出错,请检查:\r\n");
			buffer.append("(1)db.properties中的jdbc_driver是否写错?\r\n");
			buffer.append("(2)数据库驱动JAR包是否已导入?");
			System.out.println(buffer.toString());
			System.exit(-1);
		} catch (SQLException e) {
			StringBuffer buffer = new StringBuffer();
			buffer.append("Oh,My God! 数据库连接失败,请检查:\r\n");
			buffer.append("(1)数据库服务是否已开启?\r\n");
			buffer.append("(2)db.properties中的jdbc_url是否写错?\r\n");
			buffer.append("(3)db.properties中的jdbc_user是否写错?\r\n");
			buffer.append("(4)db.properties中的jdbc_password是否写错?\r\n");
			System.out.println(buffer.toString());
			System.exit(-1);
		} finally{
			try {
				if(input != null)
					input.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
	/**
	 * @return 获取数据库连接,可能返回null
	 */
	public static Connection getConnection(){
		if(connectionPool != null){
			if(connectionPool.size() > 0){
				return connectionPool.remove(connectionPool.size() - 1);
			}else{
				Connection connection = null;
				try {
					Class.forName(jdbc_driver);
					connection = DriverManager.getConnection(jdbc_url, jdbc_user, jdbc_password);
					return connection;
				} catch (ClassNotFoundException e) {
					e.printStackTrace();
					return connection;
				} catch (SQLException e) {
					e.printStackTrace();
					return connection;
				}
			}
		}else
			return null;
	}
	
	/**
	 * @return 返回某个数据库下所有的表结构
	 * @throws Exception 
	 */
	public static List<MyTable> getAllTables(){
		List<MyTable> tables = null;
		Connection connection = getConnection();		
		if(connection != null){
			PreparedStatement statement = null;
			ResultSet connectionRS = null , queryRS = null;
			tables = new ArrayList<MyTable>();
			try {
				DatabaseMetaData metaData = connection.getMetaData();
				connectionRS = metaData.getTables(null, null, null, new String[]{"TABLE"});
				while(connectionRS.next()){
					String tableName = connectionRS.getString(3);
					String querySQL = "show full columns from " + tableName;
					statement = connection.prepareStatement(querySQL);
					queryRS = statement.executeQuery();
					
					MyTable myTable = new MyTable();
					List<MyField> key_fields = n
  • 6
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
一. 生成器模板路径可以引用相关变量 如 ${basepackage}/${className}.java,根据该变量生成输出文件 二. 自动搜索某个目录所有模板文件,无需配置 三.代码生成器模板可以引用的相关变量 1. g.generateByTable("table_name") 方法可以引用的变量 table : cn.org.rapid_framework.generator.provider.db.table.model.Table 2. g.generateByClass(UserInfo.class) 方法可以引用的变量 clazz : cn.org.rapid_framework.generator.provider.java.model.JavaClass 3. g.generateBySql("select * from user_info where pwd=:pwd") 方法可以引用的变量 sql : cn.org.rapid_framework.generator.provider.db.sql.model.Sql 4.公共变量 env : 系统环境变量 System.getProperties() : 直接引用,没有前缀 generator.properties 文件中的所有属性,直接引用,没有前缀 gg : 模板控制变量, cn.org.rapid_framework.generator.GeneratorControl 四.每个模板有gg变量可以控制自身的自定义配置 (每一个模板都会创建新的gg实例) 如是否生成,是否覆盖目标文件,甚至是生成其它文件 ${gg.setIgnoreOutput(true)}: 如果为true则不生成输出文件 ${gg.generateFile(outputFile,content)} 在模板中生成其它文件 ${gg.getProperty(key,defaultValue)}: 得到proproty,如果没有找到,则返回默认值 ${gg.getInputProperty(key)}: 会弹出一个输入框,提示用户输入值 具体参考: http://code.google.com/p/rapid-framework/wiki/rapid_generator_gg 五.支持生成(gen)及删除操作(del),即生成的代码也可以很方便的删除 六. 自动删除模板扩展名: .ftl,.vm 举例: 如你有一个模板 SqlMap.xml.ftl 将变为 SqlMap.xml 所以你要生成ftl扩展名的文件,应该将文件名从 list.ftl => list.ftl.ftl 七. 模板自动include所有父目录的:macro.include文件,可以存放公共的macro 示例: 如你的模板为 com/project/UserDao.java, 将自动include: com/project/macro.include, com/macro.include, macro.include 八. generator.xml (或者generator.properties)配置文件 1.类似ant可以变量引用,引用环境变量使用${env.JAVA_HOME}, 引用System.getProperties()直接引用 2.自动替换generator.properties中的句号(.)为反斜杠,设置key为key+"_dir"后缀 示例: pkg=com.company => pkg_dir=com/company 九.自动拷贝二进制文件至输出目录 如模板目录下的 zip,rar,doc文件将会自动拷贝至输出目录,不会破坏文件格式 (通过扩展名自动识别) 十. 数据库表配置,用于自定义生成器模板引用的table变量,配置文件必须存放在classpath: generator_config/table/table_name.xml (该文件生成器可以生成,自己再自定义修改) <!-- <数据库列名 columnAlias="列的别名" javaType="自定义javaType" unique="是否唯一性约束" nullable="是否可以为空" pk="是否主键,在表没有主键的情况下,可以指定一个代理主键" updatable="是否可以更新" insertab

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值