Skip to content

Commit b4f8fc8

Browse files
committed
Use static accessors in DefaultSimpUserRegistry
Closes gh-26010
1 parent 8c3cdc6 commit b4f8fc8

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

spring-websocket/src/main/java/org/springframework/web/socket/messaging/DefaultSimpUserRegistry.java

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,14 @@
2727
import org.springframework.core.Ordered;
2828
import org.springframework.lang.Nullable;
2929
import org.springframework.messaging.Message;
30+
import org.springframework.messaging.MessageHeaders;
3031
import org.springframework.messaging.simp.SimpMessageHeaderAccessor;
3132
import org.springframework.messaging.simp.user.DestinationUserNameProvider;
3233
import org.springframework.messaging.simp.user.SimpSession;
3334
import org.springframework.messaging.simp.user.SimpSubscription;
3435
import org.springframework.messaging.simp.user.SimpSubscriptionMatcher;
3536
import org.springframework.messaging.simp.user.SimpUser;
3637
import org.springframework.messaging.simp.user.SimpUserRegistry;
37-
import org.springframework.messaging.support.MessageHeaderAccessor;
3838
import org.springframework.util.Assert;
3939

4040
/**
@@ -84,19 +84,16 @@ public boolean supportsEventType(Class<? extends ApplicationEvent> eventType) {
8484
public void onApplicationEvent(ApplicationEvent event) {
8585
AbstractSubProtocolEvent subProtocolEvent = (AbstractSubProtocolEvent) event;
8686
Message<?> message = subProtocolEvent.getMessage();
87+
MessageHeaders headers = message.getHeaders();
8788

88-
SimpMessageHeaderAccessor accessor =
89-
MessageHeaderAccessor.getAccessor(message, SimpMessageHeaderAccessor.class);
90-
Assert.state(accessor != null, "No SimpMessageHeaderAccessor");
91-
92-
String sessionId = accessor.getSessionId();
89+
String sessionId = SimpMessageHeaderAccessor.getSessionId(headers);
9390
Assert.state(sessionId != null, "No session id");
9491

9592
if (event instanceof SessionSubscribeEvent) {
9693
LocalSimpSession session = this.sessions.get(sessionId);
9794
if (session != null) {
98-
String id = accessor.getSubscriptionId();
99-
String destination = accessor.getDestination();
95+
String id = SimpMessageHeaderAccessor.getSubscriptionId(headers);
96+
String destination = SimpMessageHeaderAccessor.getDestination(headers);
10097
if (id != null && destination != null) {
10198
session.addSubscription(id, destination);
10299
}
@@ -137,7 +134,7 @@ else if (event instanceof SessionDisconnectEvent) {
137134
else if (event instanceof SessionUnsubscribeEvent) {
138135
LocalSimpSession session = this.sessions.get(sessionId);
139136
if (session != null) {
140-
String subscriptionId = accessor.getSubscriptionId();
137+
String subscriptionId = SimpMessageHeaderAccessor.getSubscriptionId(headers);
141138
if (subscriptionId != null) {
142139
session.removeSubscription(subscriptionId);
143140
}

0 commit comments

Comments
 (0)