org.springframework.web.server.session.WebSessionManager Java Examples

@Test
void enableSpringWebSessionConfiguresThings() {

	this.context = new AnnotationConfigApplicationContext();
	this.context.register(GoodConfig.class);
	this.context.refresh();

	WebSessionManager webSessionManagerFoundByType = this.context.getBean(WebSessionManager.class);
	Object webSessionManagerFoundByName = this.context.getBean(WebHttpHandlerBuilder.WEB_SESSION_MANAGER_BEAN_NAME);

	assertThat(webSessionManagerFoundByType).isNotNull();
	assertThat(webSessionManagerFoundByName).isNotNull();
	assertThat(webSessionManagerFoundByType).isEqualTo(webSessionManagerFoundByName);

	assertThat(this.context.getBean(ReactiveSessionRepository.class)).isNotNull();
}
 

DefaultServerWebExchange(ServerHttpRequest request, ServerHttpResponse response,
		WebSessionManager sessionManager, ServerCodecConfigurer codecConfigurer,
		LocaleContextResolver localeContextResolver, @Nullable ApplicationContext applicationContext) {

	Assert.notNull(request, "'request' is required");
	Assert.notNull(response, "'response' is required");
	Assert.notNull(sessionManager, "'sessionManager' is required");
	Assert.notNull(codecConfigurer, "'codecConfigurer' is required");
	Assert.notNull(localeContextResolver, "'localeContextResolver' is required");

	// Initialize before first call to getLogPrefix()
	this.attributes.put(ServerWebExchange.LOG_ID_ATTRIBUTE, request.getId());

	this.request = request;
	this.response = response;
	this.sessionMono = sessionManager.getSession(this).cache();
	this.localeContextResolver = localeContextResolver;
	this.formDataMono = initFormData(request, codecConfigurer, getLogPrefix());
	this.multipartDataMono = initMultipartData(request, codecConfigurer, getLogPrefix());
	this.applicationContext = applicationContext;
}
 

DefaultServerWebExchange(ServerHttpRequest request, ServerHttpResponse response,
		WebSessionManager sessionManager, ServerCodecConfigurer codecConfigurer,
		LocaleContextResolver localeContextResolver, @Nullable ApplicationContext applicationContext) {

	Assert.notNull(request, "'request' is required");
	Assert.notNull(response, "'response' is required");
	Assert.notNull(sessionManager, "'sessionManager' is required");
	Assert.notNull(codecConfigurer, "'codecConfigurer' is required");
	Assert.notNull(localeContextResolver, "'localeContextResolver' is required");

	// Initialize before first call to getLogPrefix()
	this.attributes.put(ServerWebExchange.LOG_ID_ATTRIBUTE, request.getId());

	this.request = request;
	this.response = response;
	this.sessionMono = sessionManager.getSession(this).cache();
	this.localeContextResolver = localeContextResolver;
	this.formDataMono = initFormData(request, codecConfigurer, getLogPrefix());
	this.multipartDataMono = initMultipartData(request, codecConfigurer, getLogPrefix());
	this.applicationContext = applicationContext;
}
 

/**
 * Configure a {@link WebSessionManager} using a provided
 * {@link ReactiveSessionRepository}.
 * @param repository a bean that implements {@link ReactiveSessionRepository}.
 * @return a configured {@link WebSessionManager} registered with a preconfigured
 * name.
 */
@Bean(WebHttpHandlerBuilder.WEB_SESSION_MANAGER_BEAN_NAME)
public WebSessionManager webSessionManager(ReactiveSessionRepository<? extends Session> repository) {
	SpringSessionWebSessionStore<? extends Session> sessionStore = new SpringSessionWebSessionStore<>(repository);
	DefaultWebSessionManager manager = new DefaultWebSessionManager();
	manager.setSessionStore(sessionStore);

	if (this.webSessionIdResolver != null) {
		manager.setSessionIdResolver(this.webSessionIdResolver);
	}

	return manager;
}
 

@Override
public <T extends B> T webSessionManager(WebSessionManager sessionManager) {
	this.sessionManager = sessionManager;
	return self();
}
 

private MockServerWebExchange(MockServerHttpRequest request, WebSessionManager sessionManager) {
	super(request, new MockServerHttpResponse(), sessionManager,
			ServerCodecConfigurer.create(), new AcceptHeaderLocaleContextResolver());
}
 

private MockServerWebExchange(MockServerHttpRequest request, WebSessionManager sessionManager) {
	super(request, new MockServerHttpResponse(), sessionManager,
			ServerCodecConfigurer.create(), new AcceptHeaderLocaleContextResolver());
}
 

public DefaultServerWebExchange(ServerHttpRequest request, ServerHttpResponse response,
		WebSessionManager sessionManager, ServerCodecConfigurer codecConfigurer,
		LocaleContextResolver localeContextResolver) {

	this(request, response, sessionManager, codecConfigurer, localeContextResolver, null);
}
 

/**
 * Return the configured {@link WebSessionManager}.
 */
public WebSessionManager getSessionManager() {
	return this.sessionManager;
}
 

@Bean
public WebSessionManager webSessionManager() {
	MockWebSession session = new MockWebSession();
	return exchange -> Mono.just(session);
}
 

@Bean
WebSessionManager webSessionManager() {
	return exchange -> Mono.just(mockWebSession);
}
 

@Bean
public WebSessionManager webSessionManager() {
	MockWebSession session = new MockWebSession();
	return exchange -> Mono.just(session);
}
 

@Override
public <T extends B> T webSessionManager(WebSessionManager sessionManager) {
	this.sessionManager = sessionManager;
	return self();
}
 

private MockServerWebExchange(MockServerHttpRequest request, WebSessionManager sessionManager) {
	super(request, new MockServerHttpResponse(), sessionManager,
			ServerCodecConfigurer.create(), new AcceptHeaderLocaleContextResolver());
}
 

private MockServerWebExchange(MockServerHttpRequest request, WebSessionManager sessionManager) {
	super(request, new MockServerHttpResponse(), sessionManager,
			ServerCodecConfigurer.create(), new AcceptHeaderLocaleContextResolver());
}
 

public DefaultServerWebExchange(ServerHttpRequest request, ServerHttpResponse response,
		WebSessionManager sessionManager, ServerCodecConfigurer codecConfigurer,
		LocaleContextResolver localeContextResolver) {

	this(request, response, sessionManager, codecConfigurer, localeContextResolver, null);
}
 

/**
 * Return the configured {@link WebSessionManager}.
 */
public WebSessionManager getSessionManager() {
	return this.sessionManager;
}
 

/**
 * Configure a custom {@link WebSessionManager} to use for managing web
 * sessions. The provided instance is set on each created
 * {@link DefaultServerWebExchange}.
 * <p>By default this is set to {@link DefaultWebSessionManager}.
 * @param sessionManager the session manager to use
 */
public void setSessionManager(WebSessionManager sessionManager) {
	Assert.notNull(sessionManager, "WebSessionManager must not be null");
	this.sessionManager = sessionManager;
}
 

/**
 * Provide a session manager instance for the mock server.
 * <p>By default an instance of
 * {@link org.springframework.web.server.session.DefaultWebSessionManager
 * DefaultWebSessionManager} is used.
 * @param sessionManager the session manager to use
 */
<T extends B> T webSessionManager(WebSessionManager sessionManager);
 

/**
 * Configure the {@link WebSessionManager} to set on the
 * {@link ServerWebExchange WebServerExchange}.
 * <p>By default {@link DefaultWebSessionManager} is used.
 * @param manager the session manager
 * @see HttpWebHandlerAdapter#setSessionManager(WebSessionManager)
 */
public WebHttpHandlerBuilder sessionManager(WebSessionManager manager) {
	this.sessionManager = manager;
	return this;
}
 

/**
 * Configure the {@link WebSessionManager} to set on the
 * {@link ServerWebExchange WebServerExchange}.
 * <p>By default {@link DefaultWebSessionManager} is used.
 * @param manager the session manager
 * @see HttpWebHandlerAdapter#setSessionManager(WebSessionManager)
 */
public WebHttpHandlerBuilder sessionManager(WebSessionManager manager) {
	this.sessionManager = manager;
	return this;
}
 

/**
 * Provide a session manager instance for the mock server.
 * <p>By default an instance of
 * {@link org.springframework.web.server.session.DefaultWebSessionManager
 * DefaultWebSessionManager} is used.
 * @param sessionManager the session manager to use
 */
<T extends B> T webSessionManager(WebSessionManager sessionManager);
 

/**
 * Configure a custom {@link WebSessionManager} to use for managing web
 * sessions. The provided instance is set on each created
 * {@link DefaultServerWebExchange}.
 * <p>By default this is set to {@link DefaultWebSessionManager}.
 * @param sessionManager the session manager to use
 */
public void setSessionManager(WebSessionManager sessionManager) {
	Assert.notNull(sessionManager, "WebSessionManager must not be null");
	this.sessionManager = sessionManager;
}