Spring/spring-integration
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix AbstractStompSessionManager race condition

Browse Source 
https://build.spring.io/browse/INT-MJATS41-1150

When we add `addHandler()` to the `CompositeStompSessionHandler`,
there is no guarantee that we will have `session` atomically during
this method invocation or after it relying on the later call to the
`this.delegates`.
In other words the session may be populated in between and, therefore,
our `delegate` loses `afterConnected()` event.
We need to synchronize on the barrier and block the concurrent
`afterConnected()` or wait for it.
This way we atomically ensure that our `delegate` is added to the
existing `session` or will be performed afterwards in the `afterConnected()`

**Cherry-pick to 4.3.x**

(cherry picked from commit df10941)
This commit is contained in:
Artem Bilan
2017-10-23 13:45:05 -04:00
parent 618da48512
commit c9ca87b87e
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
spring-integration-stomp/src/main/java/org/springframework/integration/stomp/AbstractStompSessionManager.java
View File

@@ -377,10 +377,12 @@ public abstract class AbstractStompSessionManager implements StompSessionManager
private volatile StompSession session;
void addHandler(StompSessionHandler delegate) {
if (this.session != null) {
delegate.afterConnected(this.session, getConnectHeaders());
synchronized (this.delegates) {
if (this.session != null) {
delegate.afterConnected(this.session, getConnectHeaders());
}
this.delegates.add(delegate);
}
this.delegates.add(delegate);
}
void removeHandler(StompSessionHandler delegate) {
@@ -389,8 +391,8 @@ public abstract class AbstractStompSessionManager implements StompSessionManager
@Override
public void afterConnected(StompSession session, StompHeaders connectedHeaders) {
this.session = session;
synchronized (this.delegates) {
this.session = session;
for (StompSessionHandler delegate : this.delegates) {
delegate.afterConnected(session, connectedHeaders);
}