From 7e4fc8075768ed5b7a9bd110d9a1f9601445ab31 Mon Sep 17 00:00:00 2001
From: Dave Syer <dsyer@vmware.com>
Date: Tue, 5 Apr 2011 11:54:47 +0100
Subject: [PATCH] BATCH-1725: fix SubclassClassifier

---
 .../batch/classify/SubclassClassifier.java                | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/spring-batch-infrastructure/src/main/java/org/springframework/batch/classify/SubclassClassifier.java b/spring-batch-infrastructure/src/main/java/org/springframework/batch/classify/SubclassClassifier.java
index a8d76800a..ef2910d44 100644
--- a/spring-batch-infrastructure/src/main/java/org/springframework/batch/classify/SubclassClassifier.java
+++ b/spring-batch-infrastructure/src/main/java/org/springframework/batch/classify/SubclassClassifier.java
@@ -21,6 +21,8 @@ import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
 import java.util.TreeSet;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
 
 /**
  * A {@link Classifier} for a parameterised object type based on a map.
@@ -35,7 +37,7 @@ import java.util.TreeSet;
  */
 public class SubclassClassifier<T, C> implements Classifier<T, C> {
 
-	private Map<Class<? extends T>, C> classified = new HashMap<Class<? extends T>, C>();
+	private ConcurrentMap<Class<? extends T>, C> classified = new ConcurrentHashMap<Class<? extends T>, C>();
 
 	private C defaultValue = null;
 
@@ -63,7 +65,7 @@ public class SubclassClassifier<T, C> implements Classifier<T, C> {
 	 */
 	public SubclassClassifier(Map<Class<? extends T>, C> typeMap, C defaultValue) {
 		super();
-		this.classified = new HashMap<Class<? extends T>, C>(typeMap);
+		this.classified = new ConcurrentHashMap<Class<? extends T>, C>(typeMap);
 		this.defaultValue = defaultValue;
 	}
 
@@ -85,7 +87,7 @@ public class SubclassClassifier<T, C> implements Classifier<T, C> {
 	 * @param map a map from type to class
 	 */
 	public void setTypeMap(Map<Class<? extends T>, C> map) {
-		this.classified = new HashMap<Class<? extends T>, C>(map);
+		this.classified = new ConcurrentHashMap<Class<? extends T>, C>(map);
 	}
 
 	/**