java 动态添加枚举值_DynamicEnumUtil 动态添加枚举类的枚举值

import java.lang.reflect.AccessibleObject;

import java.lang.reflect.Array;

import java.lang.reflect.Field;

import java.lang.reflect.Modifier;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import sun.reflect.ConstructorAccessor;

import sun.reflect.FieldAccessor;

import sun.reflect.ReflectionFactory;

public class DynamicEnumUtil {

private static ReflectionFactory reflectionFactory = ReflectionFactory.getReflectionFactory();

private static void setFailsafeFieldValue(Field field, Object target, Object value) throws NoSuchFieldException,

IllegalAccessException {

// let's make the field accessible

field.setAccessible(true);

// next we change the modifier in the Field instance to

// not be final anymore, thus tricking reflection into

// letting us modify the static final field

Field modifiersField = Field.class.getDeclaredField("modifiers");

modifiersField.setAccessible(true);

int modifiers = modifiersField.getInt(field);

// blank out the final bit in the modifiers int

modifiers &= ~Modifier.FINAL;

modifiersField.setInt(field, modifiers);

FieldAccessor fa = reflectionFactory.newFieldAccessor(field, false);

fa.set(target, value);

}

private static void blankField(Class> enumClass, String fieldName) throws NoSuchFieldException,

IllegalAccessException {

for (Field field : Class.class.getDeclaredFields()) {

if (field.getName().contains(fieldName)) {

AccessibleObject.setAccessible(new Field[] { field }, true);

setFailsafeFieldValue(field, enumClass, null);

break;

}

}

}

private static void cleanEnumCache(Class> enumClass) throws NoSuchFieldException, IllegalAccessException {

blankField(enumClass, "enumConstantDirectory"); // Sun (Oracle?!?) JDK 1.5/6

blankField(enumClass, "enumConstants"); // IBM JDK

}

private static ConstructorAccessor getConstructorAccessor(Class> enumClass, Class>[] additionalParameterTypes)

throws NoSuchMethodException {

Class>[] parameterTypes = new Class[additionalParameterTypes.length + 2];

parameterTypes[0] = String.class;

parameterTypes[1] = int.class;

System.arraycopy(additionalParameterTypes, 0, parameterTypes, 2, additionalParameterTypes.length);

return reflectionFactory.newConstructorAccessor(enumClass.getDeclaredConstructor(parameterTypes));

}

private static Object makeEnum(Class> enumClass, String value, int ordinal, Class>[] additionalTypes,

Object[] additionalValues) throws Exception {

Object[] parms = new Object[additionalValues.length + 2];

parms[0] = value;

parms[1] = Integer.valueOf(ordinal);

System.arraycopy(additionalValues, 0, parms, 2, additionalValues.length);

return enumClass.cast(getConstructorAccessor(enumClass, additionalTypes).newInstance(parms));

}

/**

* Add an enum instance to the enum class given as argument

*

* @param the type of the enum (implicit)

* @param enumType the class of the enum to be modified

* @param enumName the name of the new enum instance to be added to the class.

*/

@SuppressWarnings("unchecked")

public static > void addEnum(Class enumType, String enumName, Class>[] additionalTypes, Object[] additionalValues) {

// 0. Sanity checks

if (!Enum.class.isAssignableFrom(enumType)) {

throw new RuntimeException("class " + enumType + " is not an instance of Enum");

}

// 1. Lookup "$VALUES" holder in enum class and get previous enum instances

Field valuesField = null;

Field[] fields = enumType.getDeclaredFields();

for (Field field : fields) {

if (field.getName().contains("$VALUES")) {

valuesField = field;

break;

}

}

AccessibleObject.setAccessible(new Field[] { valuesField }, true);

try {

// 2. Copy it

T[] previousValues = (T[]) valuesField.get(enumType);

List values = new ArrayList(Arrays.asList(previousValues));

// 3. build new enum

T newValue = (T) makeEnum(enumType, enumName, values.size(), additionalTypes, additionalValues);

// 4. add new value

values.add(newValue);

// 5. Set new values field

setFailsafeFieldValue(valuesField, null, values.toArray((T[]) Array.newInstance(enumType, 0)));

// 6. Clean enum cache

cleanEnumCache(enumType);

} catch (Exception e) {

e.printStackTrace();

throw new RuntimeException(e.getMessage(), e);

}

}

}

测试类

import java.lang.reflect.Field;

public class MainTest {

private static enum TestEnum {

a(111,"AAA"),

b(222,"BBB"),

c(333,"CCC");

private int value;

private String text;

TestEnum(int value,String text){

this.value = value;

this.text = text;

}

public int getValue() {

return value;

}

public void setValue(int value) {

this.value = value;

}

public String getText() {

return text;

}

public void setText(String text) {

this.text = text;

}

@Override

public String toString() {

return "TestEnum{" +

"value=" + value +

", text='" + text + '\'' +

'}';

}

};

private static void addTestEnum(String enumName,int value,String text){

DynamicEnumUtil.addEnum(TestEnum.class,enumName,new Class>[] {int.class,String.class},new Object[] {value,text});

}

public static void main(String[] args) {

addTestEnum("d",444,"DDD");

addTestEnum("e",555,"EEE");

addTestEnum("f",666,"FFF");

addTestEnum("g",777,"GGG");

for (TestEnum testEnum:TestEnum.values()){

System.out.println(testEnum.toString());

}

Class ccc = TestEnum.class;

Field[] fff = ccc.getDeclaredFields();

TestEnum testEnum = TestEnum.b;

switch (testEnum){

case a:{

System.out.println("22222reefwsfsds");

break;

}

case b:

System.out.println("bbb");

break;

}

}

}