问题

以下是翻译好的代码部分：

我有如下批处理作业代码

    import kr.test.testApp.core.domain.user.UserRepository
    import mu.KotlinLogging
    import org.springframework.batch.core.Job
    import org.springframework.batch.core.Step
    import org.springframework.batch.core.configuration.annotation.StepScope
    import org.springframework.batch.core.job.builder.JobBuilder
    import org.springframework.batch.core.repository.JobRepository
    import org.springframework.batch.core.step.builder.StepBuilder
    import org.springframework.batch.core.step.tasklet.Tasklet
    import org.springframework.batch.repeat.RepeatStatus
    import org.springframework.context.annotation.Bean
    import org.springframework.context.annotation.Configuration
    import org.springframework.transaction.PlatformTransactionManager
    
    private val log = KotlinLogging.logger {  }
    
    @Configuration
    class BatchConfig(
        val jobRepository: JobRepository,
        val transactionManager: PlatformTransactionManager,
        val userRepository: UserRepository,
    ){
        @Bean
        fun myJob(): Job {
            return JobBuilder("taskletJob", jobRepository)
                .start(myStep())
                .build()
        }
    
        @Bean
        fun myStep(): Step {
            return StepBuilder("taskletStep", jobRepository)
                .allowStartIfComplete(true)
                .tasklet(myTask(), transactionManager)
                .build()
        }
    
        @Bean
        @StepScope
        fun myTask(): Tasklet {
            return Tasklet { _, _ ->
                log.info { "测试作业" } // 我想要输出用户ID
                RepeatStatus.FINISHED
            }
        }
    }

使用计划表在设置的时间运行批处理，如何在运行批处理时从数据库加载特定用户并将此数据发送到作业，并在作业中使用此用户？

    @Scheduled(cron = "30 * * * * *")
    fun batchGo(){
        val user = userRepository.findByIdOrNull(1) ?: throw Exception("未找到用户")

        val jobParameters = JobParametersBuilder()
            .toJobParameters()

        val jobExecution: JobExecution = jobLauncher.run(batchConfig.myJob(),jobParameters)
        if (jobExecution.status == BatchStatus.COMPLETED) {
            println("批处理成功")
        } else {
            println("否")
        }
    }

因此，在从数据库中获取用户后，我们将其传递给批处理运行，我希望在myTask()日志部分中打印传递给它的用户的ID，您能给我一些建议吗？

请注意，我已经将代码中的HTML实体编码（如"）还原为正常的引号和字符，以使代码更易于阅读。

import kr.test.testApp.core.domain.user.UserRepository
import mu.KotlinLogging
import org.springframework.batch.core.Job
import org.springframework.batch.core.Step
import org.springframework.batch.core.configuration.annotation.StepScope
import org.springframework.batch.core.job.builder.JobBuilder
import org.springframework.batch.core.repository.JobRepository
import org.springframework.batch.core.step.builder.StepBuilder
import org.springframework.batch.core.step.tasklet.Tasklet
import org.springframework.batch.repeat.RepeatStatus
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Configuration
import org.springframework.transaction.PlatformTransactionManager
private val log = KotlinLogging.logger {  }
@Configuration
class BatchConfig(
val jobRepository: JobRepository,
val transactionManager: PlatformTransactionManager,
val userRepository: UserRepository,
){
@Bean
fun myJob(): Job {
return JobBuilder("taskletJob", jobRepository)
.start(myStep())
.build()
}
@Bean
fun myStep(): Step {
return StepBuilder("taskletStep", jobRepository)
.allowStartIfComplete(true)
.tasklet(myTask(), transactionManager)
.build()
}
@Bean
@StepScope
fun myTask(): Tasklet {
return Tasklet { _, _ ->
log.info { " test job " } // i want output userid
RepeatStatus.FINISHED
}
}
}

@Scheduled(cron = "30 * * * * *")
fun batchGo(){
val user = userRepository.findByIdOrNull(1) ?: throw Exception("user not found")
val jobParameters = JobParametersBuilder()
.toJobParameters()
val jobExecution: JobExecution = jobLauncher.run(batchConfig.myJob(),jobParameters)
if (jobExecution.status == BatchStatus.COMPLETED) {
println("batch success")
} else {
println("no")
}
}

答案1

得分: 1

你可以将用户ID作为作业参数传递：

val userId = ...
val jobParameters = JobParametersBuilder()
        .addString("userId", userId)
        .toJobParameters()

然后通过StepContribution#getStepExecution#getJobParameters或ChunkContext#getStepContext#getJobParameters在tasklet中获取它。

@Bean
@StepScope
fun myTask(): Tasklet {
   return Tasklet { contribution, chunkContext ->
            val userId = contribution.stepExecution.jobParameters.getString("userId")
            log.info { " test job $userId" } // 我想要输出用户ID
            RepeatStatus.FINISHED
   }
}

String userId = ...
val jobParameters = JobParametersBuilder()
.addString("userId", userId)
.toJobParameters()

@Bean
@StepScope
fun myTask(): Tasklet {
return Tasklet { (contribution, chunkContext) ->
String userId = contribution.getStepExecution().getJobParameters().getString("userId");
log.info { " test job " } // i want output userid
RepeatStatus.FINISHED
}
}