Просмотр исходного кода

Add ServerHttpSecurity.exceptionHandling()

Fixes gh-4750
Rob Winch 7 лет назад
Родитель
Сommit
4fa9b4dd15

+ 26 - 0
config/src/main/java/org/springframework/security/config/web/server/ServerHttpSecurity.java

@@ -96,6 +96,8 @@ public class ServerHttpSecurity {
 
 	private CsrfBuilder csrf = new CsrfBuilder();
 
+	private ExceptionHandlingBuilder exceptionHandling = new ExceptionHandlingBuilder();
+
 	private HttpBasicBuilder httpBasic;
 
 	private FormLoginBuilder formLogin;
@@ -173,6 +175,13 @@ public class ServerHttpSecurity {
 		return this.headers;
 	}
 
+	public ExceptionHandlingBuilder exceptionHandling() {
+		if(this.exceptionHandling == null) {
+			this.exceptionHandling = new ExceptionHandlingBuilder();
+		}
+		return this.exceptionHandling;
+	}
+
 	public AuthorizeExchangeBuilder authorizeExchange() {
 		if(this.authorizeExchangeBuilder == null) {
 			this.authorizeExchangeBuilder = new AuthorizeExchangeBuilder();
@@ -404,6 +413,23 @@ public class ServerHttpSecurity {
 		private CsrfBuilder() {}
 	}
 
+	/**
+	 * @author Rob Winch
+	 * @since 5.0
+	 */
+	public class ExceptionHandlingBuilder {
+		public ExceptionHandlingBuilder serverAuthenticationEntryPoint(ServerAuthenticationEntryPoint authenticationEntryPoint) {
+			ServerHttpSecurity.this.serverAuthenticationEntryPoint = authenticationEntryPoint;
+			return this;
+		}
+
+		public ServerHttpSecurity and() {
+			return ServerHttpSecurity.this;
+		}
+
+		private ExceptionHandlingBuilder() {}
+	}
+
 	/**
 	 * @author Rob Winch
 	 * @since 5.0