複数のフラグからマッチするEnumオブジェクトを取得する（ビット演算とEnumSet）

やりたいこと

こんなかんじのフラグのセットを受け取って

boolean useSaba = true;
boolean useTori = false;
boolean useNira = false;
boolean isOomori = true; 

マッチするEnumオブジェクトを取得したい。
（上のフラグの場合は LunchSet.SABA_DAI ）

enum LunchSet {
    GOHAN("ご飯セット"),
    SABA("サバ味噌定食"),
    TORI("トリ唐揚定食"),
    NIRA("ニラ炒め定食"),
    GOHAN_DAI("ご飯セット大盛り"),
    SABA_DAI("サバ味噌定食大盛り"),
    TORI_DAI("トリ唐揚定食大盛り"),
    NIRA_DAI("ニラ炒め定食大盛り");

    private final String name;
    private LunchSet(String name) {
        this.name = name;
    }
    public String getName() {
        return name;
    }
}

方法1 ビット演算

• Enumのコンストラクタにビット演算用の数値を追加する。
• 数値を元にマッチするEnumオブジェクトを返すメソッドを定義する。

enum LunchSet {
    GOHAN("ご飯セット", 0),
    SABA("サバ味噌定食", 1),
    TORI("トリ唐揚定食", 2),
    NIRA("ニラ炒め定食", 4),
    GOHAN_DAI("ご飯セット大盛り", 8),
    SABA_DAI("サバ味噌定食大盛り", 9),
    TORI_DAI("トリ唐揚定食大盛り", 10),
    NIRA_DAI("ニラ炒め定食大盛り", 12);

    private final String name;
    private final int flag;
    private LunchSet(String name, int flag) {
        this.name = name;
        this.flag = flag;
    }
    public String getName() {
        return name;
    }
    public int getFlag() {
        return flag;
    }
    public static LunchSet getByFlag(int flag) {
        return Arrays.stream(values())
                .filter(s -> s.getFlag() == flag)
                .findFirst()
                .orElse(LunchSet.GOHAN);
    }
}

こんなふうにフラグからビットの論理和を求めて取得できる。

        int flag = 0;
        if (useSaba) { flag += 1; }
        if (useTori) { flag += 2; }
        if (useNira) { flag += 4; }
        if (isOomori) { flag += 8; }
        LunchSet lunchSet = LunchSet.getByFlag(flag);

でもビット演算はちょっとわかりづらい。

方法2 EnumSet

Effective Javaで有名なビット演算の置き換え。

enum Menu {
    SABA,
    TORI,
    NIRA,
    OOMORI;
}

enum LunchSet {
    GOHAN("ご飯セット", EnumSet.noneOf(Menu.class)),
    SABA("サバ味噌定食", EnumSet.of(Menu.SABA)),
    TORI("トリ唐揚定食", EnumSet.of(Menu.TORI)),
    NIRA("ニラ炒め定食", EnumSet.of(Menu.NIRA)),
    GOHAN_DAI("ご飯セット大盛り", EnumSet.of(Menu.OOMORI)),
    SABA_DAI("サバ味噌定食大盛り", EnumSet.of(Menu.SABA, Menu.OOMORI)),
    TORI_DAI("トリ唐揚定食大盛り", EnumSet.of(Menu.TORI, Menu.OOMORI)),
    NIRA_DAI("ニラ炒め定食大盛り", EnumSet.of(Menu.NIRA, Menu.OOMORI));

    private final String name;
    private final EnumSet<Menu> flag;
    private LunchSet(String name, EnumSet<Menu> flag) {
        this.name = name;
        this.flag = flag;
    }
    public String getName() {
        return name;
    }
    public EnumSet<Menu> getFlag() {
        return flag;
    }
    public static LunchSet getByFlag(EnumSet<Menu> flag) {
        return Arrays.stream(values())
                .filter(s -> s.getFlag().equals(flag))
                .findFirst()
                .orElse(LunchSet.GOHAN);
    }
}

こんなふうにEnumSetのフラグから取得できる。

        EnumSet<Menu> flag = EnumSet.noneOf(Menu.class);
        if (useSaba) { flag.add(Menu.SABA); }
        if (useTori) { flag.add(Menu.TORI); }
        if (useNira) { flag.add(Menu.NIRA); }
        if (isOomori) { flag.add(Menu.OOMORI); }
        LunchSet lunchSet = LunchSet.getByFlag(flag);

コード量が増えるのが難点だけど、ビット演算よりは何をしているのかわかるようになり、Enumからフラグのセットを求める逆の操作もやりやすい。