Spring Batch 集成 MyBatis,durid 连接池兼容MyBatisCursorItemReader 存在bug, 选用其他连接池


import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.batch.MyBatisCursorItemReader;
import org.springframework.batch.core.Job;
import org.springframework.batch.core.JobParameter;
import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.annotation.JobBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.core.configuration.annotation.StepScope;
import org.springframework.batch.core.step.tasklet.TaskletStep;
import org.springframework.batch.item.ItemReader;
import org.springframework.batch.item.ItemStreamReader;
import org.springframework.batch.item.ItemWriter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Scope;
import org.springframework.context.annotation.ScopedProxyMode;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

/**
 * @ClassName readByMybatisJob
 * @Description: 读取数据来自mybtatis from reader
 * @Author 
 * @Date 2020/5/29 9:54
 * @Version V1.0
 **/
@Slf4j
@Component
public class ReadByMybatisJob {

    @Autowired
    private SqlSessionFactory sqlSessionFactory;

    @Autowired
    protected JobBuilderFactory jobBuilderFactory;

    @Autowired
    protected StepBuilderFactory stepBuilderFactory;

    @Bean
    @StepScope
    public ItemStreamReader<BatchMessageField> myBatisCursorItemDemoReader(@Value("#{jobParameters[operationTypeId]}") String operationTypeId, @Value("#{jobParameters[messageType]}") String messageType) {
        MyBatisCursorItemReader reader = new MyBatisCursorItemReader();
        Map<String, Object> parameterValues = new HashMap<>();
        parameterValues.put("operationTypeId", operationTypeId);
        parameterValues.put("messageType", messageType);

        reader.setSqlSessionFactory(sqlSessionFactory);
        reader.setParameterValues(parameterValues);
        reader.setQueryId("*.mapper.BatchMessageFieldMapper.getBatchMessageFields");
        return reader;
    }


    /***
     * 公共写demo
     * @methodName myBatisCursorItemWriter
     * @param
     * @return org.springframework.batch.item.ItemWriter
     * @author 
     * @date 2020/5/29 9:38
     * @throws
     */
    @Bean("myBatisCursorItemWriter")
    public ItemWriter myBatisCursorItemWriter() {
        CommonFileItemWriter commonFileItemWriter = new CommonFileItemWriter(BatchMessageField.class);
        return commonFileItemWriter;
    }


    /**
     * 定义Job
     *
     * @param jobBuilderFactory
     * @param myBatisCursorStep 步骤
     * @return
     */
    @Bean("myBatisCursorDemoJob")
    public Job readJsonFileDemoJob(JobBuilderFactory jobBuilderFactory, Step myBatisCursorStep) {
        return jobBuilderFactory.get("myBatisCursorDemoJob")
                .incrementer(parameters -> {
                    Map<String, JobParameter> parameterMap = parameters.getParameters();
                    parameterMap.put("key", new JobParameter(UUID.randomUUID().toString()));
                    return parameters;
                }).start(myBatisCursorStep).build();

    }


    /**
     * step 配置
     *
     * @param
     * @param myBatisCursorItemDemoReader 读bean
     * @param myBatisCursorItemWriter     写bean
     * @return
     */
    @Bean("myBatisCursorStep")
    public Step myBatisCursorStep( ItemReader myBatisCursorItemDemoReader, @Qualifier("myBatisCursorItemWriter") ItemWriter myBatisCursorItemWriter) {
       return  stepBuilderFactory.get("myBatisCursorStep")
                .chunk(1)
                .reader(myBatisCursorItemDemoReader)
                // .processor(processor)
                .writer(myBatisCursorItemWriter)
                .build();

    }


}

 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值