实现自己的JDBC框架
使用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); } } }
