如何通過擴展 jOOQ 的 DefaultGeneratorStrategy 更改表 class 名稱？

Question

我正在使用 jooq codgen gradle 插件更改生成表的命名約定，以在 class 名稱末尾包含Table 。 但是，我不確定如何在生成器中區分表和模式。

我的命名覆蓋是：

@Override
public String getJavaClassName(Definition definition, Mode mode) {
    String result = super.getJavaClassName(definition, mode);
    if (mode == Mode.DEFAULT) {
        result += "Table";
    }
    return result;
}

有沒有辦法確定當前的 object 是否擴展TableImpl或者我是否需要采取不同的方法？

Answer 1

只需對您的definition使用instanceof檢查，如下所示：

@Override
public String getJavaClassName(Definition definition, Mode mode) {
    String result = super.getJavaClassName(definition, mode);
    if (mode == Mode.DEFAULT && definition instanceof TableDefinition) {
        result += "Table";
    }
    return result;
}