java - 如何使用jooq codegen自动添加自定义字段以构建pojos？

我使用jooq codegen生成这样的实体类：

public class TCoactivitiesPinan extends BaseEntity implements Serializable {
private static final long serialVersionUID = 2007524284;
private Integer   id;
private String    openid;
private String    tel;
private Timestamp createdtime;
...}

但是，我希望它自动确定是否为时间类型，它会自动添加两个字段。像这样

public class TCoactivitiesPinan extends BaseEntity implements Serializable {
private static final long serialVersionUID = 2007524284;
private Integer   id;
private String    openid;
private String    tel;
private Timestamp createdtime;
private String createdtime_start; //  创建时间_开始时间
private String createdtime_end; //  创建时间_结束时间
private Integer   checkstate;
...
}

有什么办法解决吗？ 。

最佳答案

您可以实现自己的代码生成器，并扩展POJO的“自定义代码部分”，如此处所述：

https://www.jooq.org/doc/latest/manual/code-generation/codegen-custom-code

本质上，只需编写这样的类：

public class MyGenerator extends JavaGenerator {

    @Override
    protected void generatePojoClassFooter(TableDefinition table, JavaWriter out) {
        for (ColumnDefinition column : table.getColumns()) {
            if (column.getType().getType().equals("TIMESTAMP")) {
                out.tab(1).println("private String %s_start; //  创建时间_开始时间",
                    getStrategy().getJavaMemberName(column, Mode.POJO));
                out.tab(1).println("private String %s_end; //  创建时间_结束时间",
                    getStrategy().getJavaMemberName(column, Mode.POJO));
            }
        }
    }
}

然后，将其配置如下：

<configuration>
    ...
    <generator>
        <name>com.example.MyGenerator</name>
        ...
    </generator>
</configuration>

当然，上述解决方案仅会生成所需的成员，当您使用生成的POJO类型时，它不会生成任何逻辑来填充这些成员。