消除if-else之为Enum添加行为实现策略模式

多味葵花籽

已于 2022-02-16 16:58:42 修改

阅读量213

点赞数

CC 4.0 BY-SA版权

分类专栏： java clean code 文章标签：策略模式

于 2022-02-16 16:52:21 首次发布

本文链接：https://blog.youkuaiyun.com/wenpeng12/article/details/122967234

java 同时被 2 个专栏收录

23 篇文章

订阅专栏

clean code

1 篇文章

订阅专栏

本文介绍了如何使用Java的枚举类和方法重载，实现根据不同文件类型自动执行写入操作，避免if-else条件判断，提高代码可读性和维护性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

场景描述

Java提供枚举类给了开发者更可读的代码实现，我们可以将很多字段作为枚举类型。并且赋予枚举类行为，可以省略掉根据枚举类判断而实现不同行为的众多if…else…

如下是根据参数而下载不同类型的文件的枚举代码，给每个枚举类型添加了对应的代码

public enum DownloadFileType {
    CSV {
        @Override
        public void writeToLocal() {
        	//写入csv格式实现
            exportUtil.writeCsv();
        }
    },
    EXCEL {
        @Override
        public void writeToLocal() {
        	//写入excel格式实现
            exportUtil.writeExcel();
        }
    };

    public abstract void writeToLocal();

    public static DownloadFileType of(String fileType) {
        return Arrays.stream(DownloadFileType.values())
                .filter(type -> type.toString().equals(fileType.toUpperCase(Locale.ROOT)))
                .findFirst()
                .orElse(EXCEL);
    }
}

调用方可以免除掉if…else 只需要获取到下载的类型来执行不同的写入。

//write to local tmp dir
//fileType: "excel" or "csv"
DownloadFileType.of(fileType).writeToLocal();