使用JDBC操作数据库时，dao层的增删改查有很多重复的代码，比如下面的
public int getTotal() {
Connection conn = null;//通用代码
PreparedStatement ps = null;//通用代码
ResultSet rs = null;//通用代码
try {
conn = ...//获取连接//通用代码
ps = conn.prepareStatement("select count(*) from blog");//只有sql语句不同
rs = ps.executeQuery();
if (rs.next()) {
return rs.getInt(1);
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
rs.close();//通用代码
ps.close();//通用代码
conn.close();//通用代码
} catch (SQLException e) {
e.printStackTrace();
}
}
return 0;
}
如果做一个简单的增删改查，会发现很多代码都是通用的，如果重复使用无疑会让代码显得很冗余
如果使用框架开发，框架会帮我们省去这些麻烦，或者有很多开源的工具类也可以比如apache的DBUtil工具类
不过看了黑马的教学视频，感觉完全可以自定义一个JDBC框架来帮我们省去很多麻烦（虽然我是造*，不过学习初期造*还是很有帮助的）
自定义JDBC框架
对于增删改里面重复的代码最多，且不包含结果集（查询包含的结果集需要封装，所以单独定义），用update方法统一代表增删改，query方法代表查询
对于过程，我们发现用JDBC增删改查里面只有sql语句不一样，sql语句里的占位符又会不一样，所以考虑自定义方法里传入参数包含sql语句和sql语句中需要传入的参数（用数组表示），对于查询由于需要封装结果集到Bean中，但是Bean类实现不知道，所以无法直接封装，考虑使用策略模式
下面直接上代码，具体解释留在注释中
由于使用了DBCP数据源，所以用的DBCP数据源获取数据库连接，这里就不上DBCP的代码了
主类DBAssist
public class DBAssist {
//设置连接从数据源获取
//也可以不使用数据源，自己随便定义个Connection连接
private DataSource ds;
public DBAssist(DataSource ds){
this.ds=ds;
}
//增删改
public void update(String sql,Object[] params){
Connection conn=null;
PreparedStatement ps=null;
ResultSet rs=null;
try{
conn=ds.getConnection();//得到链接
ps=conn.prepareStatement(sql);
//设置参数,
//ParameterMetaData方法获取参数信息
ParameterMetaData pmd=ps.getParameterMetaData();
//获取sql语句中的占位符？个数
int parameterCount=pmd.getParameterCount();
//获取的参数个数与传入的参数个数比较
if(parameterCount>0){
if(params==null||params.length<1)
throw new IllegalArgumentException("the parameter is wrong");
if(params.length!=parameterCount)
throw new IllegalArgumentException("the parameter is wrong");
for(int i=0;i<parameterCount;i++){
ps.setObject(i+1, params[i]);
}
}
ps.executeUpdate();
}catch(Exception e){
throw new DBAssistException(e);
}finally{
release(conn,ps,rs);
}
}
//查询,由于不知道具体的Bean类，考虑使用策略模式
//ResultSetHandler为抽象出的策略接口，策略类需继承该类以完成封装
//策略类框架会给出实现
public Object query(String sql,Object[] params,ResultSetHandler rsh){
Connection conn=null;
PreparedStatement ps=null;
ResultSet rs=null;
try{
conn=ds.getConnection();
ps=conn.prepareStatement(sql);
ParameterMetaData pmd=ps.getParameterMetaData();
int parameterCount=pmd.getParameterCount();//sql语句中的占位符个数
if(parameterCount>0){
if(params==null||params.length<1)
throw new IllegalArgumentException("the parameter is wrong");
if(params.length!=parameterCount)
throw new IllegalArgumentException("the parameter is wrong");
for(int i=0;i<parameterCount;i++){
ps.setObject(i+1, params[i]);
}
}
//得到查询结果集
rs=ps.executeQuery();
//封装结果集到对象中去,这里使用策略模式,最好提供实现类（类似于注册器）
return rsh.handler(rs);
}catch(Exception e){
throw new DBAssistException(e);
}finally{
release(conn,ps,rs);
}
}
//关闭连接
private void release(Connection conn,Statement stmt,ResultSet rs){
if(rs!=null){
try {
rs.close();
} catch (SQLException e) {
e.printStackTrace();
}
rs=null;
}
if(stmt!=null){
try {
stmt.close();
} catch (SQLException e) {
e.printStackTrace();
}
stmt=null;
}
if(conn!=null){
try {
conn.close();
} catch (SQLException e) {
e.printStackTrace();
}
conn=null;
}
}
}
此时已经可以用update方法进行增删改的功能了
对于查询，query方法中的策略接口代码如下
public interface ResultSetHandler {
public Object handler(ResultSet rs);
}
具体策略类有两个，分别用来查询单行和查询多行（利用反射填充Bean中的字段）
public class BeanHandler implements ResultSetHandler {
private Class<? extends Object> clazz;
public BeanListHandler(Class<? extends Object> clazz) {
this.clazz = clazz;
}
public List<Object> handler(ResultSet rs) {
try {
List<Object> list = new ArrayList<Object>();
while (rs.next()) {
Object bean = clazz.newInstance();
//查询数据库中的元数据信息
ResultSetMetaData rsmd = rs.getMetaData();
int count = rsmd.getColumnCount();
for (int i = 0; i < count; i++) {
String columnName = rsmd.getColumnName(i + 1);
Object columnValue = rs.getObject(i + 1);
Field field = clazz.getDeclaredField(columnName);
field.setAccessible(true);
field.set(bean, columnValue);
}
list.add(bean);
}
return list;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
}
）
//bean中字段名必须与数据库中字段名一致
public class BeanListHandler implements ResultSetHandler {
private Class<? extends Object> clazz;
public BeanHandler(Class<? extends Object> clazz) {
this.clazz=clazz;
}
//查询单记录
public Object handler(ResultSet rs) {
try {
Object bean=clazz.newInstance();
if(rs.next()){
ResultSetMetaData rsmd=rs.getMetaData();
int count=rsmd.getColumnCount();
for(int i=0;i<count;i++){
String columnName=rsmd.getColumnName(i+1);
Object columnValue=rs.getObject(i+1);
Field field=clazz.getDeclaredField(columnName);
field.setAccessible(true);
field.set(bean,columnValue);
}
return bean;
}
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
}
Bean类
public class Blog {
private int id;
private int category_id;
private String title;
private String content;
private Date created_time;
...
//getter和setter方法
@Override
public String toString() {
return "Blog [id=" + id + ", category_id=" + category_id + ", title=" + title + ", content=" + content
+ ", created_time=" + created_time + "]";
}
}
自定义异常类
public class DBAssistException extends RuntimeException {
public DBAssistException() {
}
public DBAssistException(String message) {
super(message);
}
public DBAssistException(Throwable cause) {
super(cause);
}
public DBAssistException(String message, Throwable cause) {
super(message, cause);
}
public DBAssistException(String message, Throwable cause, boolean enableSuppression, boolean writableStackTrace) {
super(message, cause, enableSuppression, writableStackTrace);
}
}
最后测试下
public void update(){
String sql="update blog set created_time=? where id=?";
dba.update(sql, new Object[]{new Date(System.currentTimeMillis()),1});
}
//查询单行记录测试
public void query(){
String sql="select * from blog";
Blog blog=(Blog) dba.query(sql, null,new BeanHandler(Blog.class));
System.out.println(blog.getTitle().length());
}
//查询多行记录测试
@SuppressWarnings("unchecked")
public List<Blog> query2(){
String sql="select * from blog";
List<Blog> list=(List<Blog>) dba.query(sql, null,new BeanListHandler(Blog.class));
return list;
}
public static void main(String args[]) {
BlogDao test = new BlogDao();System.out.println(test.update());
test.query();
List<Blog> list=test.query2();
for(Blog l:list){
System.out.println(l);
}
}
}