Allow Customizing Redis Session Mapper

Closes gh-2021

spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/ReactiveRedisSessionRepositoryKeyMissITests.java

@@ -0,0 +1,150 @@
+/*
+ * Copyright 2014-2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.session.data.redis;
+
+import java.time.Instant;
+import java.util.Map;
+import java.util.function.BiFunction;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import reactor.core.publisher.Mono;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.ReactiveHashOperations;
+import org.springframework.data.redis.core.ReactiveRedisOperations;
+import org.springframework.session.MapSession;
+import org.springframework.session.config.ReactiveSessionRepositoryCustomizer;
+import org.springframework.session.data.redis.ReactiveRedisSessionRepository.RedisSession;
+import org.springframework.session.data.redis.config.annotation.web.server.EnableRedisWebSession;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+import org.springframework.test.util.ReflectionTestUtils;
+import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.BDDMockito.given;
+import static org.mockito.BDDMockito.willAnswer;
+import static org.mockito.Mockito.spy;
+
+/**
+ * Key miss error tests for {@link ReactiveRedisSessionRepository}
+ *
+ * @author Marcus da Coregio
+ * @see <a href="https://github.com/spring-projects/spring-session/issues/2021">Related
+ * GitHub Issue</a>
+ */
+@ExtendWith(SpringExtension.class)
+class ReactiveRedisSessionRepositoryKeyMissITests extends AbstractRedisITests {
+
+	private ReactiveRedisSessionRepository sessionRepository;
+
+	private ReactiveRedisOperations<String, Object> spyOperations;
+
+	AnnotationConfigWebApplicationContext context = new AnnotationConfigWebApplicationContext();
+
+	@Test
+	void findByIdWhenSessionDeletedWhileSavingDeltaThenThrowIllegalStateException() {
+		this.context.register(Config.class);
+		refreshAndPrepareFields();
+		RedisSession session = createAndSaveSession(Instant.now());
+		session.setAttribute("new", "value");
+
+		ReactiveHashOperations<String, Object, Object> opsForHash = spy(this.spyOperations.opsForHash());
+		given(this.spyOperations.opsForHash()).willReturn(opsForHash);
+		willAnswer((invocation) -> this.sessionRepository.deleteById(session.getId())
+				.then((Mono<Void>) invocation.callRealMethod())).given(opsForHash).putAll(any(), any());
+
+		this.sessionRepository.save(session).block();
+		assertThatIllegalStateException().isThrownBy(() -> this.sessionRepository.findById(session.getId()).block())
+				.withMessage("creationTime key must not be null");
+	}
+
+	@Test
+	void findByIdWhenSessionDeletedWhileSavingDeltaAndSafeMapperThenSessionIsNull() {
+		this.context.register(RedisSessionMapperConfig.class);
+		refreshAndPrepareFields();
+		RedisSession session = createAndSaveSession(Instant.now());
+		session.setAttribute("new", "value");
+
+		ReactiveHashOperations<String, Object, Object> opsForHash = spy(this.spyOperations.opsForHash());
+		given(this.spyOperations.opsForHash()).willReturn(opsForHash);
+		willAnswer((invocation) -> this.sessionRepository.deleteById(session.getId())
+				.then((Mono<Void>) invocation.callRealMethod())).given(opsForHash).putAll(any(), any());
+
+		this.sessionRepository.save(session).block();
+		assertThat(this.sessionRepository.findById(session.getId()).block()).isNull();
+	}
+
+	@SuppressWarnings("unchecked")
+	private void refreshAndPrepareFields() {
+		this.context.refresh();
+		this.sessionRepository = this.context.getBean(ReactiveRedisSessionRepository.class);
+		ReactiveRedisOperations<String, Object> redisOperations = (ReactiveRedisOperations<String, Object>) ReflectionTestUtils
+				.getField(this.sessionRepository, "sessionRedisOperations");
+		this.spyOperations = spy(redisOperations);
+		ReflectionTestUtils.setField(this.sessionRepository, "sessionRedisOperations", this.spyOperations);
+	}
+
+	private RedisSession createAndSaveSession(Instant lastAccessedTime) {
+		RedisSession session = this.sessionRepository.createSession().block();
+		session.setLastAccessedTime(lastAccessedTime);
+		session.setAttribute("attribute1", "value1");
+		this.sessionRepository.save(session).block();
+		return this.sessionRepository.findById(session.getId()).block();
+	}
+
+	@Configuration
+	@EnableRedisWebSession
+	static class Config extends BaseConfig {
+
+	}
+
+	@Configuration
+	@EnableRedisWebSession
+	static class RedisSessionMapperConfig extends BaseConfig {
+
+		@Bean
+		ReactiveSessionRepositoryCustomizer<ReactiveRedisSessionRepository> redisSessionRepositoryCustomizer() {
+			return (redisSessionRepository) -> redisSessionRepository
+					.setRedisSessionMapper(new SafeRedisSessionMapper(redisSessionRepository));
+		}
+
+	}
+
+	static class SafeRedisSessionMapper implements BiFunction<String, Map<String, Object>, Mono<MapSession>> {
+
+		private final RedisSessionMapper delegate = new RedisSessionMapper();
+
+		private final ReactiveRedisSessionRepository sessionRepository;
+
+		SafeRedisSessionMapper(ReactiveRedisSessionRepository sessionRepository) {
+			this.sessionRepository = sessionRepository;
+		}
+
+		@Override
+		public Mono<MapSession> apply(String sessionId, Map<String, Object> map) {
+			return Mono.fromSupplier(() -> this.delegate.apply(sessionId, map)).onErrorResume(
+					IllegalStateException.class,
+					(ex) -> this.sessionRepository.deleteById(sessionId).then(Mono.empty()));
+		}
+
+	}
+
+}

spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/RedisIndexedSessionRepositoryKeyMissITests.java

@@ -0,0 +1,158 @@
+/*
+ * Copyright 2014-2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.session.data.redis;
+
+import java.time.Instant;
+import java.util.Map;
+import java.util.function.BiFunction;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.BoundHashOperations;
+import org.springframework.data.redis.core.RedisOperations;
+import org.springframework.session.MapSession;
+import org.springframework.session.config.SessionRepositoryCustomizer;
+import org.springframework.session.data.redis.RedisIndexedSessionRepository.RedisSession;
+import org.springframework.session.data.redis.config.annotation.web.http.EnableRedisIndexedHttpSession;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+import org.springframework.test.util.ReflectionTestUtils;
+import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.BDDMockito.given;
+import static org.mockito.BDDMockito.willAnswer;
+import static org.mockito.Mockito.spy;
+
+/**
+ * Key miss error tests for {@link RedisIndexedSessionRepository}
+ *
+ * @author Marcus da Coregio
+ * @see <a href="https://github.com/spring-projects/spring-session/issues/2021">Related
+ * GitHub Issue</a>
+ */
+@ExtendWith(SpringExtension.class)
+class RedisIndexedSessionRepositoryKeyMissITests extends AbstractRedisITests {
+
+	private RedisIndexedSessionRepository sessionRepository;
+
+	private RedisOperations<String, Object> spyOperations;
+
+	AnnotationConfigWebApplicationContext context = new AnnotationConfigWebApplicationContext();
+
+	@Test
+	void findByIdWhenSessionDeletedWhileSavingDeltaThenThrowIllegalStateException() {
+		this.context.register(Config.class);
+		refreshAndPrepareFields();
+		RedisSession session = createAndSaveSession(Instant.now());
+		session.setAttribute("new", "value");
+
+		BoundHashOperations<String, Object, Object> opsForHash = spy(this.spyOperations.boundHashOps(anyString()));
+		given(this.spyOperations.boundHashOps(anyString())).willReturn(opsForHash);
+		willAnswer((invocation) -> {
+			this.sessionRepository.deleteById(session.getId());
+			return invocation.callRealMethod();
+		}).given(opsForHash).putAll(any());
+
+		this.sessionRepository.save(session);
+		assertThatIllegalStateException().isThrownBy(() -> this.sessionRepository.findById(session.getId()))
+				.withMessage("creationTime key must not be null");
+	}
+
+	@Test
+	void findByIdWhenSessionDeletedWhileSavingDeltaAndSafeMapperThenSessionIsNull() {
+		this.context.register(RedisSessionMapperConfig.class);
+		refreshAndPrepareFields();
+		RedisSession session = createAndSaveSession(Instant.now());
+		session.setAttribute("new", "value");
+
+		BoundHashOperations<String, Object, Object> opsForHash = spy(this.spyOperations.boundHashOps(anyString()));
+		given(this.spyOperations.boundHashOps(anyString())).willReturn(opsForHash);
+		willAnswer((invocation) -> {
+			this.sessionRepository.deleteById(session.getId());
+			return invocation.callRealMethod();
+		}).given(opsForHash).putAll(any());
+
+		this.sessionRepository.save(session);
+		assertThat(this.sessionRepository.findById(session.getId())).isNull();
+	}
+
+	@SuppressWarnings("unchecked")
+	private void refreshAndPrepareFields() {
+		this.context.refresh();
+		this.sessionRepository = this.context.getBean(RedisIndexedSessionRepository.class);
+		RedisOperations<String, Object> redisOperations = (RedisOperations<String, Object>) ReflectionTestUtils
+				.getField(this.sessionRepository, "sessionRedisOperations");
+		this.spyOperations = spy(redisOperations);
+		ReflectionTestUtils.setField(this.sessionRepository, "sessionRedisOperations", this.spyOperations);
+	}
+
+	private RedisSession createAndSaveSession(Instant lastAccessedTime) {
+		RedisSession session = this.sessionRepository.createSession();
+		session.setLastAccessedTime(lastAccessedTime);
+		session.setAttribute("attribute1", "value1");
+		this.sessionRepository.save(session);
+		return this.sessionRepository.findById(session.getId());
+	}
+
+	@Configuration
+	@EnableRedisIndexedHttpSession
+	static class Config extends BaseConfig {
+
+	}
+
+	@Configuration
+	@EnableRedisIndexedHttpSession
+	static class RedisSessionMapperConfig extends BaseConfig {
+
+		@Bean
+		SessionRepositoryCustomizer<RedisIndexedSessionRepository> redisSessionRepositoryCustomizer() {
+			return (redisSessionRepository) -> redisSessionRepository.setRedisSessionMapper(
+					new SafeRedisSessionMapper(redisSessionRepository.getSessionRedisOperations()));
+		}
+
+	}
+
+	static class SafeRedisSessionMapper implements BiFunction<String, Map<String, Object>, MapSession> {
+
+		private final RedisSessionMapper delegate = new RedisSessionMapper();
+
+		private final RedisOperations<String, Object> redisOperations;
+
+		SafeRedisSessionMapper(RedisOperations<String, Object> redisOperations) {
+			this.redisOperations = redisOperations;
+		}
+
+		@Override
+		public MapSession apply(String sessionId, Map<String, Object> map) {
+			try {
+				return this.delegate.apply(sessionId, map);
+			}
+			catch (IllegalStateException ex) {
+				this.redisOperations.delete("spring:session:sessions:" + sessionId);
+				return null;
+			}
+		}
+
+	}
+
+}

spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/RedisSessionRepositoryKeyMissITests.java

@@ -0,0 +1,157 @@
+/*
+ * Copyright 2014-2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.session.data.redis;
+
+import java.time.Instant;
+import java.util.Map;
+import java.util.function.BiFunction;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.HashOperations;
+import org.springframework.data.redis.core.RedisOperations;
+import org.springframework.session.MapSession;
+import org.springframework.session.config.SessionRepositoryCustomizer;
+import org.springframework.session.data.redis.RedisSessionRepository.RedisSession;
+import org.springframework.session.data.redis.config.annotation.web.http.EnableRedisHttpSession;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+import org.springframework.test.util.ReflectionTestUtils;
+import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.BDDMockito.given;
+import static org.mockito.BDDMockito.willAnswer;
+import static org.mockito.Mockito.spy;
+
+/**
+ * Key miss error tests for {@link RedisSessionRepository}
+ *
+ * @author Marcus da Coregio
+ * @see <a href="https://github.com/spring-projects/spring-session/issues/2021">Related
+ * GitHub Issue</a>
+ */
+@ExtendWith(SpringExtension.class)
+class RedisSessionRepositoryKeyMissITests extends AbstractRedisITests {
+
+	private RedisSessionRepository sessionRepository;
+
+	private RedisOperations<String, Object> spyOperations;
+
+	AnnotationConfigWebApplicationContext context = new AnnotationConfigWebApplicationContext();
+
+	@Test
+	void findByIdWhenSessionDeletedWhileSavingDeltaThenThrowIllegalStateException() {
+		this.context.register(Config.class);
+		refreshAndPrepareFields();
+		RedisSession session = createAndSaveSession(Instant.now());
+		session.setAttribute("new", "value");
+
+		HashOperations<String, Object, Object> opsForHash = spy(this.spyOperations.opsForHash());
+		given(this.spyOperations.opsForHash()).willReturn(opsForHash);
+		willAnswer((invocation) -> {
+			this.sessionRepository.deleteById(session.getId());
+			return invocation.callRealMethod();
+		}).given(opsForHash).putAll(any(), any());
+
+		this.sessionRepository.save(session);
+		assertThatIllegalStateException().isThrownBy(() -> this.sessionRepository.findById(session.getId()))
+				.withMessage("creationTime key must not be null");
+	}
+
+	@Test
+	void findByIdWhenSessionDeletedWhileSavingDeltaAndSafeMapperThenSessionIsNull() {
+		this.context.register(RedisSessionMapperConfig.class);
+		refreshAndPrepareFields();
+		RedisSession session = createAndSaveSession(Instant.now());
+		session.setAttribute("new", "value");
+
+		HashOperations<String, Object, Object> opsForHash = spy(this.spyOperations.opsForHash());
+		given(this.spyOperations.opsForHash()).willReturn(opsForHash);
+		willAnswer((invocation) -> {
+			this.sessionRepository.deleteById(session.getId());
+			return invocation.callRealMethod();
+		}).given(opsForHash).putAll(any(), any());
+
+		this.sessionRepository.save(session);
+		assertThat(this.sessionRepository.findById(session.getId())).isNull();
+	}
+
+	@SuppressWarnings("unchecked")
+	private void refreshAndPrepareFields() {
+		this.context.refresh();
+		this.sessionRepository = this.context.getBean(RedisSessionRepository.class);
+		RedisOperations<String, Object> redisOperations = (RedisOperations<String, Object>) ReflectionTestUtils
+				.getField(this.sessionRepository, "sessionRedisOperations");
+		this.spyOperations = spy(redisOperations);
+		ReflectionTestUtils.setField(this.sessionRepository, "sessionRedisOperations", this.spyOperations);
+	}
+
+	private RedisSession createAndSaveSession(Instant lastAccessedTime) {
+		RedisSession session = this.sessionRepository.createSession();
+		session.setLastAccessedTime(lastAccessedTime);
+		session.setAttribute("attribute1", "value1");
+		this.sessionRepository.save(session);
+		return this.sessionRepository.findById(session.getId());
+	}
+
+	@Configuration
+	@EnableRedisHttpSession
+	static class Config extends BaseConfig {
+
+	}
+
+	@Configuration
+	@EnableRedisHttpSession
+	static class RedisSessionMapperConfig extends BaseConfig {
+
+		@Bean
+		SessionRepositoryCustomizer<RedisSessionRepository> redisSessionRepositoryCustomizer() {
+			return (redisSessionRepository) -> redisSessionRepository
+					.setRedisSessionMapper(new SafeRedisSessionMapper(redisSessionRepository));
+		}
+
+	}
+
+	static class SafeRedisSessionMapper implements BiFunction<String, Map<String, Object>, MapSession> {
+
+		private final RedisSessionMapper delegate = new RedisSessionMapper();
+
+		private final RedisSessionRepository sessionRepository;
+
+		SafeRedisSessionMapper(RedisSessionRepository sessionRepository) {
+			this.sessionRepository = sessionRepository;
+		}
+
+		@Override
+		public MapSession apply(String sessionId, Map<String, Object> map) {
+			try {
+				return this.delegate.apply(sessionId, map);
+			}
+			catch (IllegalStateException ex) {
+				this.sessionRepository.deleteById(sessionId);
+				return null;
+			}
+		}
+
+	}
+
+}

spring-session-data-redis/src/main/java/org/springframework/session/data/redis/ReactiveRedisSessionRepository.java

@@ -21,6 +21,7 @@ import java.time.Instant;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
+import java.util.function.BiFunction;
 
 import org.reactivestreams.Publisher;
 import reactor.core.publisher.Mono;
@@ -66,6 +67,8 @@ public class ReactiveRedisSessionRepository
 
 	private SessionIdGenerator sessionIdGenerator = UuidSessionIdGenerator.getInstance();
 
+	private BiFunction<String, Map<String, Object>, Mono<MapSession>> redisSessionMapper = new RedisSessionMapperAdapter();
+
 	/**
 	 * Create a new {@link ReactiveRedisSessionRepository} instance.
 	 * @param sessionRedisOperations the {@link ReactiveRedisOperations} to use for
@@ -154,7 +157,7 @@ public class ReactiveRedisSessionRepository
 		return this.sessionRedisOperations.opsForHash().entries(sessionKey)
 				.collectMap((e) -> e.getKey().toString(), Map.Entry::getValue)
 				.filter((map) -> !map.isEmpty())
-				.map(new RedisSessionMapper(id))
+				.flatMap((map) -> this.redisSessionMapper.apply(id, map))
 				.filter((session) -> !session.isExpired())
 				.map((session) -> new RedisSession(session, false))
 				.switchIfEmpty(Mono.defer(() -> deleteById(id).then(Mono.empty())));
@@ -186,6 +189,16 @@ public class ReactiveRedisSessionRepository
 		this.sessionIdGenerator = sessionIdGenerator;
 	}
 
+	/**
+	 * Set the {@link BiFunction} used to convert a {@link Map} to a {@link MapSession}.
+	 * @param redisSessionMapper the mapper to use, cannot be null
+	 * @since 3.2
+	 */
+	public void setRedisSessionMapper(BiFunction<String, Map<String, Object>, Mono<MapSession>> redisSessionMapper) {
+		Assert.notNull(redisSessionMapper, "redisSessionMapper cannot be null");
+		this.redisSessionMapper = redisSessionMapper;
+	}
+
 	/**
 	 * A custom implementation of {@link Session} that uses a {@link MapSession} as the
 	 * basis for its mapping. It keeps track of any attributes that have changed. When
@@ -349,4 +362,16 @@ public class ReactiveRedisSessionRepository
 
 	}
 
+	private static final class RedisSessionMapperAdapter
+			implements BiFunction<String, Map<String, Object>, Mono<MapSession>> {
+
+		private final RedisSessionMapper mapper = new RedisSessionMapper();
+
+		@Override
+		public Mono<MapSession> apply(String sessionId, Map<String, Object> map) {
+			return Mono.fromSupplier(() -> this.mapper.apply(sessionId, map));
+		}
+
+	}
+
 }

spring-session-data-redis/src/main/java/org/springframework/session/data/redis/RedisIndexedSessionRepository.java

@@ -23,6 +23,7 @@ import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
+import java.util.function.BiFunction;
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -326,6 +327,8 @@ public class RedisIndexedSessionRepository
 
 	private SessionIdGenerator sessionIdGenerator = UuidSessionIdGenerator.getInstance();
 
+	private BiFunction<String, Map<String, Object>, MapSession> redisSessionMapper = new RedisSessionMapper();
+
 	/**
 	 * Creates a new instance. For an example, refer to the class level javadoc.
 	 * @param sessionRedisOperations the {@link RedisOperations} to use for managing the
@@ -523,8 +526,8 @@ public class RedisIndexedSessionRepository
 		if ((entries == null) || entries.isEmpty()) {
 			return null;
 		}
-		MapSession loaded = new RedisSessionMapper(id).apply(entries);
-		if (!allowExpired && loaded.isExpired()) {
+		MapSession loaded = this.redisSessionMapper.apply(id, entries);
+		if (loaded == null || (!allowExpired && loaded.isExpired())) {
 			return null;
 		}
 		RedisSession result = new RedisSession(loaded, false);
@@ -568,9 +571,11 @@ public class RedisIndexedSessionRepository
 			String sessionId = channel.substring(channel.lastIndexOf(":") + 1);
 			@SuppressWarnings("unchecked")
 			Map<String, Object> entries = (Map<String, Object>) this.defaultSerializer.deserialize(message.getBody());
-			MapSession loaded = new RedisSessionMapper(sessionId).apply(entries);
-			RedisSession session = new RedisSession(loaded, false);
-			handleCreated(session);
+			MapSession loaded = this.redisSessionMapper.apply(sessionId, entries);
+			if (loaded != null) {
+				RedisSession session = new RedisSession(loaded, false);
+				handleCreated(session);
+			}
 			return;
 		}
 
@@ -730,6 +735,17 @@ public class RedisIndexedSessionRepository
 		this.sessionIdGenerator = sessionIdGenerator;
 	}
 
+	/**
+	 * Set the {@link BiFunction} used to map {@link MapSession} to a
+	 * {@link ReactiveRedisSessionRepository.RedisSession}.
+	 * @param redisSessionMapper the mapper to use, cannot be null
+	 * @since 3.2
+	 */
+	public void setRedisSessionMapper(BiFunction<String, Map<String, Object>, MapSession> redisSessionMapper) {
+		Assert.notNull(redisSessionMapper, "redisSessionMapper cannot be null");
+		this.redisSessionMapper = redisSessionMapper;
+	}
+
 	/**
 	 * A custom implementation of {@link Session} that uses a {@link MapSession} as the
 	 * basis for its mapping. It keeps track of any attributes that have changed. When

spring-session-data-redis/src/main/java/org/springframework/session/data/redis/RedisSessionMapper.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2014-2019 the original author or authors.
+ * Copyright 2014-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -19,6 +19,7 @@ package org.springframework.session.data.redis;
 import java.time.Duration;
 import java.time.Instant;
 import java.util.Map;
+import java.util.function.BiFunction;
 import java.util.function.Function;
 
 import org.springframework.session.MapSession;
@@ -30,9 +31,10 @@ import org.springframework.util.Assert;
  * {@link MapSession}.
  *
  * @author Vedran Pavic
+ * @author Marcus da Coregio
  * @since 2.2.0
  */
-final class RedisSessionMapper implements Function<Map<String, Object>, MapSession> {
+public final class RedisSessionMapper implements BiFunction<String, Map<String, Object>, MapSession> {
 
 	/**
 	 * The key in the hash representing {@link Session#getCreationTime()}.
@@ -56,17 +58,15 @@ final class RedisSessionMapper implements Function<Map<String, Object>, MapSessi
 	 */
 	static final String ATTRIBUTE_PREFIX = "sessionAttr:";
 
-	private final String sessionId;
-
-	RedisSessionMapper(String sessionId) {
-		Assert.hasText(sessionId, "sessionId must not be empty");
-		this.sessionId = sessionId;
+	private static void handleMissingKey(String key) {
+		throw new IllegalStateException(key + " key must not be null");
 	}
 
 	@Override
-	public MapSession apply(Map<String, Object> map) {
+	public MapSession apply(String sessionId, Map<String, Object> map) {
+		Assert.hasText(sessionId, "sessionId must not be empty");
 		Assert.notEmpty(map, "map must not be empty");
-		MapSession session = new MapSession(this.sessionId);
+		MapSession session = new MapSession(sessionId);
 		Long creationTime = (Long) map.get(CREATION_TIME_KEY);
 		if (creationTime == null) {
 			handleMissingKey(CREATION_TIME_KEY);
@@ -90,8 +90,4 @@ final class RedisSessionMapper implements Function<Map<String, Object>, MapSessi
 		return session;
 	}
 
-	private static void handleMissingKey(String key) {
-		throw new IllegalStateException(key + " key must not be null");
-	}
-
 }

spring-session-data-redis/src/main/java/org/springframework/session/data/redis/RedisSessionRepository.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2014-2022 the original author or authors.
+ * Copyright 2014-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -21,6 +21,7 @@ import java.time.Instant;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
+import java.util.function.BiFunction;
 
 import org.springframework.data.redis.core.RedisOperations;
 import org.springframework.session.FlushMode;
@@ -60,6 +61,8 @@ public class RedisSessionRepository implements SessionRepository<RedisSessionRep
 
 	private SessionIdGenerator sessionIdGenerator = UuidSessionIdGenerator.getInstance();
 
+	private BiFunction<String, Map<String, Object>, MapSession> redisSessionMapper = new RedisSessionMapper();
+
 	/**
 	 * Create a new {@link RedisSessionRepository} instance.
 	 * @param sessionRedisOperations the {@link RedisOperations} to use for managing
@@ -136,8 +139,8 @@ public class RedisSessionRepository implements SessionRepository<RedisSessionRep
 		if (entries.isEmpty()) {
 			return null;
 		}
-		MapSession session = new RedisSessionMapper(sessionId).apply(entries);
-		if (session.isExpired()) {
+		MapSession session = this.redisSessionMapper.apply(sessionId, entries);
+		if (session == null || session.isExpired()) {
 			deleteById(sessionId);
 			return null;
 		}
@@ -176,6 +179,17 @@ public class RedisSessionRepository implements SessionRepository<RedisSessionRep
 		this.sessionIdGenerator = sessionIdGenerator;
 	}
 
+	/**
+	 * Set the {@link BiFunction} used to map {@link MapSession} to a
+	 * {@link ReactiveRedisSessionRepository.RedisSession}.
+	 * @param redisSessionMapper the mapper to use, cannot be null
+	 * @since 3.2
+	 */
+	public void setRedisSessionMapper(BiFunction<String, Map<String, Object>, MapSession> redisSessionMapper) {
+		Assert.notNull(redisSessionMapper, "redisSessionMapper cannot be null");
+		this.redisSessionMapper = redisSessionMapper;
+	}
+
 	/**
 	 * An internal {@link Session} implementation used by this {@link SessionRepository}.
 	 */

spring-session-data-redis/src/test/java/org/springframework/session/data/redis/RedisSessionMapperTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2014-2019 the original author or authors.
+ * Copyright 2014-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -22,7 +22,6 @@ import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 
-import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
 import org.springframework.session.MapSession;
@@ -38,34 +37,29 @@ import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
  */
 class RedisSessionMapperTests {
 
-	private RedisSessionMapper mapper;
-
-	@BeforeEach
-	void setUp() {
-		this.mapper = new RedisSessionMapper("id");
-	}
+	private RedisSessionMapper mapper = new RedisSessionMapper();
 
 	@Test
-	void constructor_NullId_ShouldThrowException() {
-		assertThatIllegalArgumentException().isThrownBy(() -> new RedisSessionMapper(null))
+	void apply_NullId_ShouldThrowException() {
+		assertThatIllegalArgumentException().isThrownBy(() -> this.mapper.apply(null, Collections.emptyMap()))
 				.withMessage("sessionId must not be empty");
 	}
 
 	@Test
-	void constructor_EmptyId_ShouldThrowException() {
-		assertThatIllegalArgumentException().isThrownBy(() -> new RedisSessionMapper(" "))
+	void apply_EmptyId_ShouldThrowException() {
+		assertThatIllegalArgumentException().isThrownBy(() -> this.mapper.apply(" ", Collections.emptyMap()))
 				.withMessage("sessionId must not be empty");
 	}
 
 	@Test
 	void apply_NullMap_ShouldThrowException() {
-		assertThatIllegalArgumentException().isThrownBy(() -> this.mapper.apply(null))
+		assertThatIllegalArgumentException().isThrownBy(() -> this.mapper.apply("1234", null))
 				.withMessage("map must not be empty");
 	}
 
 	@Test
 	void apply_EmptyMap_ShouldThrowException() {
-		assertThatIllegalArgumentException().isThrownBy(() -> this.mapper.apply(Collections.emptyMap()))
+		assertThatIllegalArgumentException().isThrownBy(() -> this.mapper.apply("1234", Collections.emptyMap()))
 				.withMessage("map must not be empty");
 	}
 
@@ -74,7 +68,7 @@ class RedisSessionMapperTests {
 		Map<String, Object> sessionMap = new HashMap<>();
 		sessionMap.put(RedisSessionMapper.LAST_ACCESSED_TIME_KEY, 0L);
 		sessionMap.put(RedisSessionMapper.MAX_INACTIVE_INTERVAL_KEY, 1800);
-		assertThatIllegalStateException().isThrownBy(() -> this.mapper.apply(sessionMap))
+		assertThatIllegalStateException().isThrownBy(() -> this.mapper.apply("id", sessionMap))
 				.withMessage(RedisSessionMapper.CREATION_TIME_KEY + " key must not be null");
 	}
 
@@ -83,7 +77,7 @@ class RedisSessionMapperTests {
 		Map<String, Object> sessionMap = new HashMap<>();
 		sessionMap.put(RedisSessionMapper.CREATION_TIME_KEY, 0L);
 		sessionMap.put(RedisSessionMapper.MAX_INACTIVE_INTERVAL_KEY, 1800);
-		assertThatIllegalStateException().isThrownBy(() -> this.mapper.apply(sessionMap))
+		assertThatIllegalStateException().isThrownBy(() -> this.mapper.apply("id", sessionMap))
 				.withMessage(RedisSessionMapper.LAST_ACCESSED_TIME_KEY + " key must not be null");
 	}
 
@@ -92,7 +86,7 @@ class RedisSessionMapperTests {
 		Map<String, Object> sessionMap = new HashMap<>();
 		sessionMap.put(RedisSessionMapper.CREATION_TIME_KEY, 0L);
 		sessionMap.put(RedisSessionMapper.LAST_ACCESSED_TIME_KEY, 0L);
-		assertThatIllegalStateException().isThrownBy(() -> this.mapper.apply(sessionMap))
+		assertThatIllegalStateException().isThrownBy(() -> this.mapper.apply("id", sessionMap))
 				.withMessage(RedisSessionMapper.MAX_INACTIVE_INTERVAL_KEY + " key must not be null");
 	}
 
@@ -104,7 +98,7 @@ class RedisSessionMapperTests {
 		sessionMap.put(RedisSessionMapper.MAX_INACTIVE_INTERVAL_KEY, 1800);
 		sessionMap.put(RedisSessionMapper.ATTRIBUTE_PREFIX + "existing", "value");
 		sessionMap.put(RedisSessionMapper.ATTRIBUTE_PREFIX + "missing", null);
-		MapSession session = this.mapper.apply(sessionMap);
+		MapSession session = this.mapper.apply("id", sessionMap);
 		assertThat(session.getId()).isEqualTo("id");
 		assertThat(session.getCreationTime()).isEqualTo(Instant.ofEpochMilli(0));
 		assertThat(session.getLastAccessedTime()).isEqualTo(Instant.ofEpochMilli(0));