如何用策略模式干掉 if-else

王小明_HIT

发布于 2022-06-14 16:30:40

3020

发布于 2022-06-14 16:30:40

文章被收录于专栏：程序员奇点程序员奇点

如何用策略模式干掉 if-else

日常编码过程中遇到很多的 if-else ，代码扩展性和阅读性会受到影响，代码中常常使用策略模式代替 if-else

代码示例：

  /**
     * 上传文件
     *
     * @param storageType 文件存储方式
     * @param file        文件
     */
    public void uploadFile(String storageType, String file) {
        if (storageType.equals(LOCAL)) {
            System.out.println("文件" + file + "已上传到 本地服务器");
        } else if (storageType.equals(FTP)) {
            System.out.println("文件" + file + "已上传到 ftp服务器");
        } else if (storageType.equals(FASTDFS)) {
            System.out.println("文件" + file + "已上传到 fastdfs服务器");
        } else if (storageType.equals(HDFS)) {
            System.out.println("文件" + file + "已上传到 hdfs服务器");
        } else {
            System.out.println("输入的文件存储类型错误");
        }
    }

第一步：创建基类

如果看到上面代码，要怎么优化，抽象一层，搞个 Base 类：

 public abstract class StorageStrategy{
     public abstract void uploadFile(String file);
 }

第二步:实现基类

让所有业务实现基类接口

public class FtpStorageStrategy extends StorageStrategy {
    @Override
    public void uploadFile(String file) {
        System.out.println("文件" + file + "已上传到 ftp服务器");
    }
}

第三步：查表方式选择基类

既然事查表，当然需要在将创建的处理类，放到一个容器中，这里使用 map,将所有的策略放到一个容器中。

public class Factory {

    private static Map<String,StorageStrategy> strategyMap = new ConcurrentHashMap<>();

    public static void register(String name,StorageStrategy strategy){
        strategyMap.put(name,strategy);
    }

    public static StorageStrategy getInvokeStrategy(String name){
        return strategyMap.get(name);
    }
}

第四步：使用

public class Client {

    public static void main(String[] args) {
        // 获取策略
       StorageStrategy strategy = Factory.strategyMap.get("FTP")
       // 执行策略
       strategy.uploadFile("策略模式.txt");
}