可以看到，你leader教你的拼装算法，使用了一个布尔变量来控制是否需要加个or这个关键字，第一次执行for循环的时候因为该布尔值为false，所以不会加上or，在循环的最后将布尔值赋值为true，这样以后循环每次都会在头部加上一个or关键字，由于使用了头部添加or的方法，所以不用再担心SQL语句的尾部会多出一个or来。然后你为了将两个算法都保留，在findUserInfo方法上加了一个参数，strategy值为1表示使用第一种算法，strategy值为2表示使用第二种算法。
这样测试代码也需要改成如下方式：
public class Test {
public static void main(String[] args) throws Exception {
QueryUtil query = new QueryUtil();
query.findUserInfo(new String[] { "Tom", "Jim", "Anna" }, 2);
}
}
这里你通过参数指明了使用第二种算法来拼装SQL语句，打印的结果和使用第一种算法是完全相同的。
你立刻把你的leader从百忙之中拖了过来，让他检验一下你当前的成果，可是他还是一如既往的挑剔。
“你这样写的话，findUserInfo这个方法的逻辑就太复杂了，非常不利于阅读，也不利于将来的扩展，如果我还有第三第四种算法想加进去，这个方法还能看吗？” 你的leader指点你，遇到这种情况，就要使用策略模式来解决，策略模式的核心思想就是把算法提取出来放到一个独立的对象中。
为了指点你，他不顾自己的百忙，开始教你如何使用策略模式进行优化。
首先定义一个策略接口：
public interface Strategy {
String getSQL(String[] usernames);
}
然后定义两个子类都实现了上述接口，并将两种拼装SQL语句的算法分别加入两个子类中：
public class Strategy1 implements Strategy { @Override public String getSQL(String[] usernames) { StringBuilder sql = new StringBuilder("select * from user_info where "); for (String user : usernames) { sql.append("username = '"); sql.append(user); sql.append("' or "); } sql.delete(sql.length() - " or ".length(), sql.length()); return sql.toString(); } } public class Strategy2 implements Strategy { @Override public String getSQL(String[] usernames) { StringBuilder sql = new StringBuilder("select * from user_info where "); boolean needOr = false; for (String user : usernames) { if (needOr) { sql.append(" or "); } sql.append("username = '"); sql.append(user); sql.append("'"); needOr = true; } return sql.toString(); } } |
然后把QueryUtil中findUserInfo方法的第二个参数改成Strategy对象，这样只需要调用Strategy的getSQL方法就可以获得拼装好的SQL语句，代码如下所示：
public class QueryUtil { public void findUserInfo(String[] usernames, Strategy strategy) throws Exception { Class.forName("com.mysql.jdbc.Driver"); Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/test", "root", "123456"); Statement stat = conn.createStatement(); String sql = strategy.getSQL(usernames); System.out.println(sql); ResultSet resultSet = stat.executeQuery(sql); while (resultSet.next()) { // 处理从数据库读出来的数据 } // 后面应将读到的数据组装成对象返回，这里略去。 } } |
最后，测试代码在调用findUserInfo方法时，只需要显示地指明需要使用哪一个策略对象就可以了：
public class Test {
public static void main(String[] args) throws Exception {
QueryUtil query = new QueryUtil();
query.findUserInfo(new String[] { "Tom", "Jim", "Anna" }, new Strategy1());
query.findUserInfo(new String[] { "Jac", "Joe", "Rose" }, new Strategy2());
}
}
打印出的SQL语句丝毫不出预料，如下所示：
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna'
select * from user_info where username = 'Jac' or username = 'Joe' or username = 'Rose'
使用策略模式修改之后，代码的可读性和扩展性都有了很大的提高，即使以后还需要添加新的算法，你也是手到擒来了！
策略：它定义了算法家庭，分别封装起来。让它们之间可以互相替换，此模式让算法的变化，不会影响到使用算法的客户。
