如何使用静态成员对Java枚举进行泛化?

时间:2010-03-10 17:36:09

标签: java generics static enums

我正在重构遗留应用程序的一部分,该应用程序处理从Excel表格导出和导入数据库表格。我们为每个表都有一个Formatter子类,以提供该表的定义:它有多少列,以及每列的名称,格式和验证器。然后,通过模板方法调用提供此数据的getter,该方法导出/导入表。我已将列数据提取到枚举中,这大大简化了代码。格式化程序现在看起来像这样(为简洁起见省略了一些细节):

public class DamageChargeFormatter extends BaseFormatter {
    public static final int NUM_COLUMNS = 7;

    public enum Column {
        VEHICLE_GROUP(0, "Vehicle Group", /* more params */),
        NAME_OF_PART(1, "Name of Part", /* more params */),
        //...
        LOSS_OF_USE(6, "Loss of Use", /* more params */);

        private static final Map<Integer, Column> intToColumn = new HashMap<Integer, Column>();

        static {
            for (Column type : values()) {
                intToColumn.put(type.getIndex(), type);
            }
        }

        public static TableColumn valueOf(int index) {
            return intToColumn.get(index);
        }

        private int index;
        private String name;

        Column(int index, String name, /* more params */) {
            this.index = index;
            this.name = name;
            //...
        }

        public int getIndex() { return index; }

        public String getName() { return name; }

        // more members and getters...
    }

    protected String getSheetName() {
        return "Damage Charges";
    }

    public String getColumnName(int columnNumber) {
        TableColumn column = Column.valueOf(columnNumber);

        if (column != null) {
            return column.getName();
        }
        return null;
    }

    // more getters...

    protected int getNumColumns() {
        return NUM_COLUMNS;
    }

    protected boolean isVariableColumnCount() {
        return false;
    }
}

现在,我有十几个这样的类,每个类都包含完全相同的代码,但NUM_COLUMNS除外,Column的枚举值不同。有什么方法可以以某种方式对此进行泛化吗?对此的主要障碍是静态Column.valueOf()方法和静态常量NUM_COLUMNS。后者的另一个问题是它确实属于一个更高级别的抽象,即表格,而不是单个列 - 以某种方式将其合并到通用解决方案中会很好。

从技术上讲,我可以通过基本接口(下面是TableColumn)和反射解决这个问题,但我真的不喜欢这样,因为除了将编译时错误交易到运行时错误之外,它还会使代码变得丑陋(至我):

public class GenericFormatter<E extends TableColumn> extends BaseFormatter {
    private Method valueOfMethod;

    public GenericFormatter(Class<E> columnClass) {
        try {
            valueOfMethod = columnClass.getDeclaredMethod("valueOf", Integer.class);
        } catch (NoSuchMethodException e) {
            throw new RuntimeException(e);
        }
    }

    public String getColumnName(int columnNumber) {
        try {
            @SuppressWarnings("unchecked")
            E elem = (E) valueOfMethod.invoke(columnNumber);

            if (elem != null) {
                return elem.getName();
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
        return null;
    }

    //...
}

请注意,此代码纯粹是实验性的,尚未经过测试......

有更好,更清洁,更安全的方式吗?

1 个答案:

答案 0 :(得分:1)

可能是这样的:

public class TableMetadata<E extends Enum & TableColumn> {
    private Map<Integer, TableColumn> columns = new HashMap<Integer, TableColumn>();

    public TableMetadata(Class<E> c) {
        for (E e: c.getEnumConstants()) {
            columns.put(e.getIndex(), e);
        }
    }

    public String getColumnName(int index) {
        return columns.get(index).getName();
    }
}

public class GenericFormatter<E extends TableColumn> extends BaseFormatter {  
    private TableMetadata<E> m;  

    public GenericFormatter(TableMetadata<E> m) {  
         this.m = m;
    }  

    public String getColumnName(int columnNumber) {  
        return m.getColumnName(index);
    }  

    //...  
}

编辑: Enum添加到type参数以提高编译时的安全性