JDBC之使用策略模式和模板方法模式优化代码

在处理ResultSet时，不同的方法处理的方式是不一样的，但在继承的时候，rowMapper方法只能以一种方式实现，因此可以将对ResultSet处理的部分单独提取成一个抽象的部分，在不同的方法需要处理的时候，各自以自己的方式进行定义。

1、定义一个策略接口RowMapper

package cn.itcast.jdbc.dao.refactor;

import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * 策略接口
 */
public interface RowMapper {
	
	/**
	 * 策略方法
	 * @param rs
	 * @return
	 * @throws SQLException
	 */
	public Object mapRow(ResultSet rs) throws SQLException;
	
}

2、定义Dao模板类

package cn.itcast.jdbc.dao.refactor;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import cn.itcast.jdbc.JdbcUtils;
import cn.itcast.jdbc.dao.DaoException;

/**
 * 自定义Dao处理结果集的模板方法
 */
public class MyDaoTemplate {
	
	/**
	 * 执行查询操作的模板方法
	 * @param sql
	 * @param args
	 * @param rowMapper
	 * @return
	 */
	public Object find(String sql, Object[] args, RowMapper rowMapper) {
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		
		try {
			conn = JdbcUtils.getConnection();
			ps = conn.prepareStatement(sql);
			for(int i = 0; i < args.length; i ++) {
				ps.setObject(i + 1, args[i]);
			}
			rs = ps.executeQuery();
			Object obj = null;
			if(rs.next()) {
				//这里的执行交给具体的执行方法去处理
				obj = rowMapper.mapRow(rs);
			}
			return obj;
		} catch (SQLException e) {
			throw new DaoException(e.getMessage(), e);
		} finally {
			JdbcUtils.free(rs, ps, conn);
		}
		
	}
}

3、编写Dao的实现类

package cn.itcast.jdbc.dao.refactor;

import java.sql.ResultSet;
import java.sql.SQLException;

import cn.itcast.jdbc.domain.User;

/**
 * 使用模板方法模式和策略模式进行优化代码
 */
public class UserDaoImpl2 {
	
	private MyDaoTemplate template = new MyDaoTemplate();
	
	/**
	 * 根据姓名查询用户
	 * @param loginName
	 * @return
	 */
	public User findUser(String loginName) {
		String sql = "select id, name, money, birthday"
				+ " from user where name = ?";
		Object[] args = new Object[]{loginName};
		//显式创建一个行映射器
		RowMapper mapper = new UserRowMapper();
		Object user =this.template.find(sql, args, mapper);
		return (User) user;
	}
	
	/**
	 * 根据id查询用户姓名
	 * @param id
	 * @return
	 */
	public String findUserName(int id) {
		String sql = "select name from user where id = ?";
		Object[] args = new Object[]{id};
		//通过匿名内部类的方式创建一个行映射器
		Object name = this.template.find(sql, args, new RowMapper(){

			public Object mapRow(ResultSet rs) throws SQLException {
				return rs.getString("name");
			}
			
		});
		return (String) name;
	}
	
}

class UserRowMapper implements RowMapper {

	/**
	 * 具体的方法
	 */
	public Object mapRow(ResultSet rs) throws SQLException {
		User user = new User();
		user.setId(rs.getInt("id"));
		user.setName(rs.getString("name"));
		user.setMoney(rs.getFloat("money"));
		user.setBirthday(rs.getDate("birthday"));
		return user;
	}
	
}

4、编写测试类进行测试

package cn.itcast.jdbc.dao.refactor;

import org.junit.Test;

import cn.itcast.jdbc.domain.User;

public class TestUserDaoImpl2 {
	
	private UserDaoImpl2 impl2 = new UserDaoImpl2();
	
	/**
	 * 测试根据姓名查询用户
	 */
	@Test
	public void testFindUser() {
		User user = impl2.findUser("lisi");
		System.out.println(user);
	}
	
	/**
	 * 测试根据id查询用户姓名
	 */
	@Test
	public void testFindUserName() {
		String username = impl2.findUserName(2);
		System.out.println(username);
	}
}