From e0fc9e92baf9b98d344579cb4926ac255d15329b Mon Sep 17 00:00:00 2001
From: Vedran Pavic <vedran.pavic@gmail.com>
Date: Sat, 4 Mar 2017 08:18:16 +0100
Subject: [PATCH] Simplify `Map.get` and conditional calls

---
 .../session/data/SessionEventRegistry.java          | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/spring-session/src/integration-test/java/org/springframework/session/data/SessionEventRegistry.java b/spring-session/src/integration-test/java/org/springframework/session/data/SessionEventRegistry.java
index c9c8c2b0..f9061030 100644
--- a/spring-session/src/integration-test/java/org/springframework/session/data/SessionEventRegistry.java
+++ b/spring-session/src/integration-test/java/org/springframework/session/data/SessionEventRegistry.java
@@ -18,13 +18,15 @@ package org.springframework.session.data;
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
 
 import org.springframework.context.ApplicationListener;
 import org.springframework.session.events.AbstractSessionEvent;
 
 public class SessionEventRegistry implements ApplicationListener<AbstractSessionEvent> {
 	private Map<String, AbstractSessionEvent> events = new HashMap<>();
-	private Map<String, Object> locks = new HashMap<>();
+	private ConcurrentMap<String, Object> locks = new ConcurrentHashMap<>();
 
 	public void onApplicationEvent(AbstractSessionEvent event) {
 		String sessionId = event.getSessionId();
@@ -63,13 +65,6 @@ public class SessionEventRegistry implements ApplicationListener<AbstractSession
 	}
 
 	private Object getLock(String sessionId) {
-		synchronized (this.locks) {
-			Object lock = this.locks.get(sessionId);
-			if (lock == null) {
-				lock = new Object();
-				this.locks.put(sessionId, lock);
-			}
-			return lock;
-		}
+		return this.locks.computeIfAbsent(sessionId, k -> new Object());
 	}
 }