DynamicEnumUtil 动态添加枚举类的枚举值

时间:2021-10-18 10:29:47
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 <T> 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 <T extends Enum<?>> void addEnum(Class<T> 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<T> values = new ArrayList<T>(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<TestEnum> 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;
} }
}