From 4c86f33da977fc48edadb79d526aecbd220f6585 Mon Sep 17 00:00:00 2001
From: Mahmoud Ben Hassine <mahmoud.ben-hassine@broadcom.com>
Date: Thu, 22 May 2025 09:55:26 +0200
Subject: [PATCH] Add support to configure JobKeyGenerator in
 MongoJobRepository

---
 .../AbstractJobRepositoryFactoryBean.java     | 17 ++++++++++
 .../support/JdbcJobRepositoryFactoryBean.java | 31 ++++++-------------
 .../support/JobRepositoryFactoryBean.java     | 16 ----------
 .../MongoJobRepositoryFactoryBean.java        |  4 ++-
 4 files changed, 30 insertions(+), 38 deletions(-)

diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/AbstractJobRepositoryFactoryBean.java b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/AbstractJobRepositoryFactoryBean.java
index f8fd15860..3c23bf616 100644
--- a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/AbstractJobRepositoryFactoryBean.java
+++ b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/AbstractJobRepositoryFactoryBean.java
@@ -22,6 +22,8 @@ import org.aopalliance.intercept.MethodInterceptor;
 import org.springframework.aop.framework.ProxyFactory;
 import org.springframework.aop.support.DefaultPointcutAdvisor;
 import org.springframework.aop.support.NameMatchMethodPointcut;
+import org.springframework.batch.core.DefaultJobKeyGenerator;
+import org.springframework.batch.core.JobKeyGenerator;
 import org.springframework.batch.core.repository.JobRepository;
 import org.springframework.batch.core.repository.dao.ExecutionContextDao;
 import org.springframework.batch.core.repository.dao.JobExecutionDao;
@@ -71,6 +73,8 @@ public abstract class AbstractJobRepositoryFactoryBean implements FactoryBean<Jo
 	 */
 	private static final String DEFAULT_ISOLATION_LEVEL = TRANSACTION_ISOLATION_LEVEL_PREFIX + "SERIALIZABLE";
 
+	protected JobKeyGenerator jobKeyGenerator;
+
 	/**
 	 * @return fully configured {@link JobInstanceDao} implementation.
 	 * @throws Exception thrown if error occurs creating JobInstanceDao.
@@ -176,9 +180,22 @@ public abstract class AbstractJobRepositoryFactoryBean implements FactoryBean<Jo
 		this.transactionAttributeSource = transactionAttributeSource;
 	}
 
+	/**
+	 * * Sets the generator for creating the key used in identifying unique {link
+	 * JobInstance} objects
+	 * @param jobKeyGenerator a {@link JobKeyGenerator}
+	 * @since 5.1
+	 */
+	public void setJobKeyGenerator(JobKeyGenerator jobKeyGenerator) {
+		this.jobKeyGenerator = jobKeyGenerator;
+	}
+
 	@Override
 	public void afterPropertiesSet() throws Exception {
 		Assert.state(transactionManager != null, "TransactionManager must not be null.");
+		if (jobKeyGenerator == null) {
+			jobKeyGenerator = new DefaultJobKeyGenerator();
+		}
 		if (this.transactionAttributeSource == null) {
 			Properties transactionAttributes = new Properties();
 			transactionAttributes.setProperty("create*",
diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JdbcJobRepositoryFactoryBean.java b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JdbcJobRepositoryFactoryBean.java
index eece3efd8..1a60b42aa 100644
--- a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JdbcJobRepositoryFactoryBean.java
+++ b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JdbcJobRepositoryFactoryBean.java
@@ -15,7 +15,6 @@
  */
 package org.springframework.batch.core.repository.support;
 
-import org.springframework.batch.core.JobKeyGenerator;
 import org.springframework.batch.core.repository.ExecutionContextSerializer;
 import org.springframework.batch.core.repository.dao.AbstractJdbcBatchMetadataDao;
 import org.springframework.batch.core.repository.dao.DefaultExecutionContextSerializer;
@@ -147,26 +146,16 @@ public class JdbcJobRepositoryFactoryBean extends JobRepositoryFactoryBean {
         super.setIncrementerFactory(incrementerFactory);
     }
 
-    /**
-     * * Sets the generator for creating the key used in identifying unique {link
-     * JobInstance} objects
-     * @param jobKeyGenerator a {@link JobKeyGenerator}
-     * @since 5.1
-     */
-    public void setJobKeyGenerator(JobKeyGenerator jobKeyGenerator) {
-        super.setJobKeyGenerator(jobKeyGenerator);
-    }
-
-    /**
-     * Set the {@link Charset} to use when serializing/deserializing the execution
-     * context. Defaults to "UTF-8". Must not be {@code null}.
-     * @param charset to use when serializing/deserializing the execution context.
-     * @see JdbcExecutionContextDao#setCharset(Charset)
-     * @since 5.0
-     */
-    public void setCharset(@NonNull Charset charset) {
-        super.setCharset(charset);
-    }
+	/**
+	 * Set the {@link Charset} to use when serializing/deserializing the execution
+	 * context. Defaults to "UTF-8". Must not be {@code null}.
+	 * @param charset to use when serializing/deserializing the execution context.
+	 * @see JdbcExecutionContextDao#setCharset(Charset)
+	 * @since 5.0
+	 */
+	public void setCharset(@NonNull Charset charset) {
+		super.setCharset(charset);
+	}
 
     /**
      * Set the conversion service to use in the job repository. This service is used to
diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JobRepositoryFactoryBean.java b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JobRepositoryFactoryBean.java
index b8226838e..a3f4ecc9e 100644
--- a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JobRepositoryFactoryBean.java
+++ b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/JobRepositoryFactoryBean.java
@@ -90,8 +90,6 @@ public class JobRepositoryFactoryBean extends AbstractJobRepositoryFactoryBean i
 
 	protected DataFieldMaxValueIncrementerFactory incrementerFactory;
 
-	protected JobKeyGenerator jobKeyGenerator;
-
 	protected int maxVarCharLengthForExitMessage = AbstractJdbcBatchMetadataDao.DEFAULT_EXIT_MESSAGE_LENGTH;
 
 	protected int maxVarCharLengthForShortContext = AbstractJdbcBatchMetadataDao.DEFAULT_SHORT_CONTEXT_LENGTH;
@@ -205,16 +203,6 @@ public class JobRepositoryFactoryBean extends AbstractJobRepositoryFactoryBean i
 		this.incrementerFactory = incrementerFactory;
 	}
 
-	/**
-	 * * Sets the generator for creating the key used in identifying unique {link
-	 * JobInstance} objects
-	 * @param jobKeyGenerator a {@link JobKeyGenerator}
-	 * @since 5.1
-	 */
-	public void setJobKeyGenerator(JobKeyGenerator jobKeyGenerator) {
-		this.jobKeyGenerator = jobKeyGenerator;
-	}
-
 	/**
 	 * Set the {@link Charset} to use when serializing/deserializing the execution
 	 * context. Defaults to "UTF-8". Must not be {@code null}.
@@ -251,10 +239,6 @@ public class JobRepositoryFactoryBean extends AbstractJobRepositoryFactoryBean i
 			incrementerFactory = new DefaultDataFieldMaxValueIncrementerFactory(dataSource);
 		}
 
-		if (jobKeyGenerator == null) {
-			jobKeyGenerator = new DefaultJobKeyGenerator();
-		}
-
 		if (databaseType == null) {
 			databaseType = DatabaseType.fromMetaData(dataSource).name();
 			if (logger.isInfoEnabled()) {
diff --git a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/MongoJobRepositoryFactoryBean.java b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/MongoJobRepositoryFactoryBean.java
index 721272dde..f679d50a3 100644
--- a/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/MongoJobRepositoryFactoryBean.java
+++ b/spring-batch-core/src/main/java/org/springframework/batch/core/repository/support/MongoJobRepositoryFactoryBean.java
@@ -49,7 +49,9 @@ public class MongoJobRepositoryFactoryBean extends AbstractJobRepositoryFactoryB
 
 	@Override
 	protected JobInstanceDao createJobInstanceDao() {
-		return new MongoJobInstanceDao(this.mongoOperations);
+		MongoJobInstanceDao mongoJobInstanceDao = new MongoJobInstanceDao(this.mongoOperations);
+		mongoJobInstanceDao.setJobKeyGenerator(this.jobKeyGenerator);
+		return mongoJobInstanceDao;
 	}
 
 	@Override