com.github.developframework.mybatis.boot.MybatisExtensionAutoConfiguration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-extension-spring-boot-starter Show documentation

The newest version!

package com.github.developframework.mybatis.boot;

import com.github.developframework.mybatis.extension.core.autoinject.*;
import com.github.developframework.mybatis.extension.core.idgenerator.AutoIncrementIdGenerator;
import com.github.developframework.mybatis.extension.core.idgenerator.IdGenerator;
import com.github.developframework.mybatis.extension.core.idgenerator.IdGeneratorRegistry;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;

/**
 * @author qiushui on 2023-10-07.
 */
@Configuration
@ComponentScan("com.github.developframework.mybatis.boot")
public class MybatisExtensionAutoConfiguration {

    @Bean
    public AutoInjectProviderRegistry autoInjectProviderRegistry(ObjectProvider autoInjectProviders) {
        final AutoInjectProviderRegistry registry = new AutoInjectProviderRegistry();
        autoInjectProviders.forEach(provider -> registry.put(provider.getClass(), provider));
        return registry;
    }

    @Bean
    public IdGeneratorAutoInjectProvider idGeneratorAutoInjectProvider(ObjectProvider idGenerators) {
        IdGeneratorRegistry idGeneratorRegistry = new IdGeneratorRegistry();
        idGeneratorRegistry.register(new AutoIncrementIdGenerator());
        for (IdGenerator idGenerator : idGenerators) {
            idGeneratorRegistry.register(idGenerator);
        }
        return new IdGeneratorAutoInjectProvider(idGeneratorRegistry);
    }

    @Bean
    public AuditCreateTimeAutoInjectProvider auditCreateDateTimeAutoInjectProvider() {
        return new AuditCreateTimeAutoInjectProvider();
    }

    @Bean
    public AuditModifyTimeAutoInjectProvider auditModifyTimeAutoInjectProvider() {
        return new AuditModifyTimeAutoInjectProvider();
    }
}