转载请注明出处：http://blog.csdn.net/sinyu890807/article/details/8986285
今天你的leader兴致冲冲地找到你，希望你可以帮他一个小忙，他现在急着要去开会。要帮什么忙呢？你很好奇。
他对你说，当前你们项目的数据库中有一张用户信息表，里面存放了很用户的数据，现在需要完成一个选择性查询用户信息的功能。他说会传递给你一个包含许多用户名的数组，你需要根据这些用户名把他们相应的数据都给查出来。
这个功能很简单的嘛，你爽快地答应了。由于你们项目使用的是MySQL数据库，你很快地写出了如下代码：
- public class QueryUtil {
- public void findUserInfo(String[] usernames) throws Exception {
- Class.forName("com.mysql.jdbc.Driver");
- Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/test", "root",
- "123456");
- Statement stat = conn.createStatement();
- StringBuilder sql = new StringBuilder("select * from user_info where ");
- for (String user : usernames) {
- sql.append("username = '");
- sql.append(user);
- sql.append("' or ");
- }
- System.out.println(sql);
- ResultSet resultSet = stat.executeQuery(sql.toString());
- while (resultSet.next()) {
- // 处理从数据库读出来的数据
- }
- // 后面应将读到的数据组装成对象返回，这里略去。
- }
- }
然后，你写了如下代码来测试这个方法：
- public class Test {
- public static void main(String[] args) throws Exception {
- QueryUtil query = new QueryUtil();
- query.findUserInfo(new String[] { "Tom", "Jim", "Anna" });
- }
- }
- select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna' or
这可怎么办呢？
有了！你灵光一闪，想出了一个解决办法。等SQL语句拼装完成后，把最后一个or删除掉不就好了么。于是你将代码改成如下所示：
- public class QueryUtil {
- public void findUserInfo(String[] usernames) throws Exception {
- Class.forName("com.mysql.jdbc.Driver");
- Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/test", "root",
- "123456");
- Statement stat = conn.createStatement();
- 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());
- System.out.println(sql);
- ResultSet resultSet = stat.executeQuery(sql.toString());
- while (resultSet.next()) {
- // 处理从数据库读出来的数据
- }
- // 后面应将读到的数据组装成对象返回，这里略去。
- }
- }
使用StringBuilder的delete方法，把最后多余的一个or删除掉了，这样再运行测试代码，一切就正常了，打印的SQL语句如下所示：
- select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna'
你的leader开完会后，过来看了下你的成果。总体来说，他还挺满意，但对于你使用的SQL语句拼装算法，他总是感觉有些不对劲，可是又说不上哪里不好。于是他告诉了你另一种拼装SQL语句的算法，让你加入到代码中，但是之前的那种算法也不要删除，先保留着再说，然后他又很忙似的跑开了。于是，你把他刚刚教你的算法加了进去，代码如下所示：
- public class QueryUtil {
- public void findUserInfo(String[] usernames, int strategy) throws Exception {
- Class.forName("com.mysql.jdbc.Driver");
- Connection conn = DriverManager.getConnection("jdbc:mysql://localhost:3306/test", "root",
- "123456");
- Statement stat = conn.createStatement();
- StringBuilder sql = new StringBuilder("select * from user_info where ");
- if (strategy == 1) {
- for (String user : usernames) {
- sql.append("username = '");
- sql.append(user);
- sql.append("' or ");
- }
- sql.delete(sql.length() - " or ".length(), sql.length());
- } else if (strategy == 2) {
- boolean needOr = false;
- for (String user : usernames) {
- if (needOr) {
- sql.append(" or ");
- }
- sql.append("username = '");
- sql.append(user);
- sql.append("'");
- needOr = true;
- }
- }
- System.out.println(sql);
- ResultSet resultSet = stat.executeQuery(sql.toString());
- while (resultSet.next()) {
- // 处理从数据库读出来的数据
- }
- // 后面应将读到的数据组装成对象返回，这里略去。
- }
- }
这样测试代码也需要改成如下方式：
- public class Test {
- public static void main(String[] args) throws Exception {
- QueryUtil query = new QueryUtil();
- query.findUserInfo(new String[] { "Tom", "Jim", "Anna" }, 2);
- }
- }
你立刻把你的leader从百忙之中拖了过来，让他检验一下你当前的成果，可是他还是一如既往的挑剔。
“你这样写的话，findUserInfo这个方法的逻辑就太复杂了，非常不利于阅读，也不利于将来的扩展，如果我还有第三第四种算法想加进去，这个方法还能看吗？” 你的leader指点你，遇到这种情况，就要使用策略模式来解决，策略模式的核心思想就是把算法提取出来放到一个独立的对象中。
为了指点你，他不顾自己的百忙，开始教你如何使用策略模式进行优化。
首先定义一个策略接口：
- public interface Strategy {
- String getSQL(String[] usernames);
- }
- 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();
- }
- }
- 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()) {
- // 处理从数据库读出来的数据
- }
- // 后面应将读到的数据组装成对象返回，这里略去。
- }
- }
- 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());
- }
- }
- 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'
使用策略模式修改之后，代码的可读性和扩展性都有了很大的提高，即使以后还需要添加新的算法，你也是手到擒来了！
策略：它定义了算法家庭，分别封装起来。让它们之间可以互相替换，此模式让算法的变化，不会影响到使用算法的客户。
扫一扫
282
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
