从两种类型的枚举之一实例化EnumSet

时间:2016-10-05 02:17:18

标签: java generics enums enumset

我将这两个枚举声明为:

public enum EnumOne implements EnumInterface {
    SomethingHere,
    SomethingThere;
    public void someMethod () { }
}

public enum EnumTwo implements EnumInterface {
    SomethingOverYonder;
    public void someMethod () { }
}

如此处所示,它们都实现了此接口。

public Interface EnumInterface {
     someMethod();
}

我在这里定义了一个包含EnumSet的类,它将从Enum1或Enum2中获取所有元素。这将通过一个以索引作为参数的方法来定义。像这样:

public class SomeClass {
    private EnumSet someEnum;
    public void setEnumType (int index) {
        switch (index) {
        case 1:
            someEnum = EnumSet.allOf(EnumOne.class);
            break;
        case 2:
            someEnum = EnumTwo.allOf(EnumTwo.class);
            break;
        }
    }
}

现在,我知道我必须使用泛型以某种方式完成此操作,但我不知道如何完成它,因为我对它不够熟悉而且我希望你能够启发我一点。

提前感谢您的回答。

2 个答案:

答案 0 :(得分:1)

目前还不清楚您在寻找什么,但如果您只是希望someEnum不能使用原始类型进行声明,则可行:

public class SomeClass {
    private EnumSet<? extends EnumInterface> someEnum;
    public void setEnumType (int index) {
        switch (index) {
        case 1:
            someEnum = EnumSet.allOf(EnumOne.class);
            break;
        case 2:
            someEnum = EnumSet.allOf(EnumTwo.class);
            break;
        }
    }
}

答案 1 :(得分:0)

我就这样做了:

public class SomeClass {                                                                                                                                                  

    private EnumSet<? extends EnumInterface> someEnum;                                                                                                                    

    // I control it, so it's checked                                                                                                                                      
    @SuppressWarnings("unchecked")                                                                                                                                        
    public <E extends Enum<E> & EnumInterface> void setEnumType(int index) {                                                                                              
        switch (index) {                                                                                                                                                  
        case 1:                                                                                                                                                           
            someEnum = (EnumSet<E>) EnumSet.allOf(Enum1.class);                                                                                                           
            break;                                                                                                                                                        
        case 2:                                                                                                                                                           
            someEnum = (EnumSet<E>) EnumSet.allOf(Enum2.class);                                                                                                           
            break;                                                                                                                                                        
        default:                                                                                                                                                          
        }                                                                                                                                                                 

    }  


    public EnumSet<? extends EnumInterface> getEnum() {                                                                                                                   
        return someEnum;                                                                                                                                                  
    }     

}

测试:

public static void main(String[] args) {                                                                                                                              
    SomeClass c = new SomeClass();                                                                                                                                    
    c.setEnumType(2);                                                                                                                                                 
    EnumSet<? extends EnumInterface> e = c.getEnum();                                                                                                                 
    System.out.println(e);                                                                                                                                            

    for (EnumInterface ei : e)                                                                                                                                        
        ei.someMethod();                                                                                                                                              

    }                                                                                                                                                                     

}