update websocket integration docs

spring-projects · Jul 18, 2024 · 1583c36 · 1583c36
1 parent c736e07
commit 1583c36
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/docs/modules/ROOT/pages/servlet/integrations/websocket.adoc b/docs/modules/ROOT/pages/servlet/integrations/websocket.adoc
@@ -365,16 +365,24 @@ Java::
 @Configuration
 public class WebSocketSecurityConfig implements WebSocketMessageBrokerConfigurer {
 
+    private final ApplicationContext applicationContext;
+
+    private final AuthorizationManager<Message<?>> authorizationManager;
+    
+    public WebSocketSecurityConfig(ApplicationContext applicationContext, AuthorizationManager<Message<?>> authorizationManager) {
+        this.applicationContext = applicationContext;
+        this.authorizationManager = authorizationManager;
+    }
+
     @Override
     public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
         argumentResolvers.add(new AuthenticationPrincipalArgumentResolver());
     }
 
     @Override
     public void configureClientInboundChannel(ChannelRegistration registration) {
-        AuthorizationManager<Message<?>> myAuthorizationRules = AuthenticatedAuthorizationManager.authenticated();
-        AuthorizationChannelInterceptor authz = new AuthorizationChannelInterceptor(myAuthorizationRules);
-        AuthorizationEventPublisher publisher = new SpringAuthorizationEventPublisher(this.context);
+        AuthorizationChannelInterceptor authz = new AuthorizationChannelInterceptor(authorizationManager);
+        AuthorizationEventPublisher publisher = new SpringAuthorizationEventPublisher(applicationContext);
         authz.setAuthorizationEventPublisher(publisher);
         registration.interceptors(new SecurityContextChannelInterceptor(), authz);
     }
@@ -386,17 +394,16 @@ Kotlin::
 [source,kotlin,role="secondary"]
 ----
 @Configuration
-open class WebSocketSecurityConfig : WebSocketMessageBrokerConfigurer {
+open class WebSocketSecurityConfig(val applicationContext: ApplicationContext, val authorizationManager: AuthorizationManager<Message<*>>) : WebSocketMessageBrokerConfigurer {
     @Override
     override fun addArgumentResolvers(argumentResolvers: List<HandlerMethodArgumentResolver>) {
         argumentResolvers.add(AuthenticationPrincipalArgumentResolver())
     }
 
     @Override
     override fun configureClientInboundChannel(registration: ChannelRegistration) {
-        var myAuthorizationRules: AuthorizationManager<Message<*>> = AuthenticatedAuthorizationManager.authenticated()
-        var authz: AuthorizationChannelInterceptor = AuthorizationChannelInterceptor(myAuthorizationRules)
-        var publisher: AuthorizationEventPublisher = SpringAuthorizationEventPublisher(this.context)
+        var authz: AuthorizationChannelInterceptor = AuthorizationChannelInterceptor(authorizationManager)
+        var publisher: AuthorizationEventPublisher = SpringAuthorizationEventPublisher(applicationContext)
         authz.setAuthorizationEventPublisher(publisher)
         registration.interceptors(SecurityContextChannelInterceptor(), authz)
     }