jOOQ 自定义 Pojo 和 DAO 生成

Posted 2023-03-28

【中文标题】jOOQ 自定义 Pojo 和 DAO 生成

【英文标题】：jOOQ Custom Pojo & DAO Generation

【发布时间】：2018-09-04 22:13:45

【问题描述】：

问题

在代码生成过程中，我在配置到自定义 Pojo 的映射时遇到了一些问题。

问题

我已经实现了RecordMapperProvider，但我想知道如何注册它以在代码生成阶段使用，或者即使可以这样做？

更多上下文

我喜欢生成 Pojo 和 DAO 的事实，但我想自己定义 Pojo，而不需要太多配置代码。我正在使用ModelMapper 从类型映射到目标：

@Override
public <R extends Record, E> RecordMapper<R, E> provide(RecordType<R> recordType,
                                                        Class<? extends E> type) 

    if (mapping.containsKey(type)) 
        return record -> modelMapper.map(mapping.get(type), type);
    

    return new DefaultRecordMapper<>(recordType, type);

如果有帮助，我将使用 DefaultConfiguration 对象（它是一个 bean）来配置 jOOQ：

@Bean
public DefaultConfiguration configuration() 
    DefaultConfiguration jooqConfiguration = new DefaultConfiguration();
    jooqConfiguration.setConnectionProvider(dataSourceConnectionProvider());
    jooqConfiguration.setExecuteListenerProvider(new DefaultExecuteListenerProvider(
            jooqToSpringExceptionTranslator()));
    jooqConfiguration.setSQLDialect(
            SQLDialect.valueOf(env.getRequiredProperty("jooq.sql.dialect")));
    jooqConfiguration.setRecordMapperProvider(new JooqRecordMapperFactory(modelMapper()));

    return jooqConfiguration;

然后对于代码生成，我在 gradle 中对其进行配置：

jooq 
version = '3.10.5'
edition = 'OSS'

myDb(sourceSets.getByName("main")) 
    jdbc 
        driver = dbDriver
        url = dbUrl
        user = dbUsername
    
    generator 
        name = 'org.jooq.util.JavaGenerator'
        strategy 
            name = 'org.jooq.util.DefaultGeneratorStrategy'
        
        database 
            name = 'org.jooq.util.postgres.PostgresDatabase'
            inputSchema = dbSchema
        
        generate 
            relations = true
            deprecated = false
            records = true
            immutablePojos = true
            fluentSetters = true
            daos = true
        
        target 
            packageName = 'com.textiq.quinn.common.dao.model.generated'
        
    

我确信这两种配置之间存在脱节，但我无法从文档中收集到我是如何同步这些配置的。理想情况下，我希望 jOOQ 生成 Pojos（基于 ModelMapper 在我的 RecordMapperProvider 实现中提供的映射）并且还让 jOOQ 为这些 Pojos 提供 DAO。这可能吗？文档指出：

如果您使用 jOOQ 的代码生成器，您可以将其配置为为您生成 POJO，但您不需要使用这些生成的 POJO。你可以使用你自己的。请参阅手册中关于带有自定义 RecordMappers 的 POJO 的部分，以了解如何修改 jOOQ 的标准 POJO 映射行为。

来源：https://www.jooq.org/doc/3.9/manual/sql-execution/fetching/pojos/

这对我来说表明了这种可能性，但只会让我实现RecordMapperProvider，之后什么都没有。

【参考方案1】：

我已经实现了 RecordMapperProvider，但想知道如何注册它以在代码生成阶段使用，或者即使可以这样？

不，这是不可能的，开箱即用。

我喜欢 Pojos 和 DAO 生成的事实，但我想自己定义 Pojo 而无需太多配置代码

然后，我建议关闭 POJO 和 DAO 的生成并自己滚动。要么创建 DAO 的手动实现，要么扩展 JavaGenerator 以执行此操作。

【参考方案2】：

我迟到了几年，但实际上我找到了一个非常简单的方法来做到这一点。我承认它有点脆弱，但您可以进一步完善它以满足您的需求。

在您的 gradle 项目中创建一个新模块，例如名为 jooq-generator 将 jooq-codegen 作为 compileOnly 依赖添加到模块 在模块中新建一个类：

public class Generator extends JavaGenerator 
    @Override
    public boolean generatePojos() 
        return false;
    

在模块中新建一个类：

public class MyGeneratorStrategy extends DefaultGeneratorStrategy 
   @Override
   public String getJavaPackageName(Definition definition, Mode mode) 
       if (mode != Mode.POJO) 
           return super.getJavaPackageName(definition, mode);
       
       return "com.example.my.model.package.prefix";
   

将模块作为依赖添加到 jooqGenerator jooqGenerator project(":jooq-generator") 将新类添加到 jooq 配置中

jooq 
    configurations 
        main 
            generationTool 
                generator 
                    name = 'com.example.my.package.name.Generator'
                    strategy 
                        name = 'com.example.my.package.name.MyGeneratorStrategy'
                    
                
            
        
    

现在将使用 POJO 的包前缀而不是生成的包名称来生成 daos。