修改DbUtils支持表名下划线映射

Commons DbUtils很一个很好用JDBC工具类，地球人都知道，就不介绍了，不知道的看这里：http://commons.apache.org/dbutils/

 

在进行查询自动类映射的时候domain类的属性必须和数据库中名称一样，但是如果数据库的表名使用带下划线的设计，domain类中的属性也要使用下划线，看起来很不雅观，而且使用hibernate工具自动生成的domain类会自动去掉下划线，下划线后面的用大写字母，比如user_name转换后为userName，因此有必要让DbUtils支持这种设计。

 

好，目标已经明确，下来就是开工了!

 

1、下载DbUtils的源代码

2、找到commons-dbutils-1.2-src\src\java\org\apache\commons\dbutils下的BeanProcessor.java，增加一个函数，代码如下：

private String getPropertyName(String columnName){		
	StringBuilder sb = new StringBuilder();
	boolean match = false;
	for (int i=0; i<columnName.length(); i++){
		char ch = columnName.charAt(i);
		if (match && ch>=97 && ch<=122)
			ch -= 32;
		if (ch!='_'){
			match = false;
			sb.append(ch);	
		}else{				
			match = true;	
		}			
	}
	return sb.toString();
}

 3、修改mapColumnsToProperties函数，红色部分是修改的地方。

protected int[] mapColumnsToProperties(ResultSetMetaData rsmd,
            PropertyDescriptor[] props) throws SQLException {

    int cols = rsmd.getColumnCount();
    int columnToProperty[] = new int[cols + 1];
    Arrays.fill(columnToProperty, PROPERTY_NOT_FOUND);

    for (int col = 1; col <= cols; col++) {
        String columnName = getPropertyName(rsmd.getColumnName(col));
        for (int i = 0; i < props.length; i++) {

            if (columnName.equalsIgnoreCase(props[i].getName())) {
                columnToProperty[col] = i;
                break;
            }
        }
    }

    return columnToProperty;
}

 4、mvn package编译，搞定！

 

懒得编译的同学，直接下载附件吧

评论

3 楼 zhaoyou 2010-03-22  

按照你的意思，是列名的问题吧，用查询的时候用 as 关键字可以解决吧，然后修改一下源码把getColumnName变成getLableName可以吗？

2 楼 dean_liu 2010-02-06  

hack源码好像不是最好的方式.
看下我的方式:
http://dean-liu.iteye.com/blog/588753

1 楼 jzinfo 2009-12-11  

DButils的 1.3版本似乎已经修正了这个bug了吧，sql中可以使用别名来避免。

1.3中 映射方法中代码如下：

 protected int[] mapColumnsToProperties(ResultSetMetaData rsmd,
            PropertyDescriptor[] props) throws SQLException {

        int cols = rsmd.getColumnCount();
        int columnToProperty[] = new int[cols + 1];
        Arrays.fill(columnToProperty, PROPERTY_NOT_FOUND);

        for (int col = 1; col <= cols; col++) {
            
            //已经改过来为getColumnLabel了
           
            String columnName = rsmd.getColumnLabel(col);
            if (null == columnName || 0 == columnName.length()) {
              columnName = rsmd.getColumnName(col);
            }
            for (int i = 0; i < props.length; i++) {

                if (columnName.equalsIgnoreCase(props[i].getName())) {
                    columnToProperty[col] = i;
                    break;
                }
            }
        }

        return columnToProperty;
    }