Browse Source

Skip close lock if acquired by other thread already

Closes gh-32445

(cherry picked from commit d151931f86)
pull/32754/head
Juergen Hoeller 2 years ago
parent
commit
6b6beec4ee
  1. 43
      spring-websocket/src/main/java/org/springframework/web/socket/handler/ConcurrentWebSocketSessionDecorator.java

43
spring-websocket/src/main/java/org/springframework/web/socket/handler/ConcurrentWebSocketSessionDecorator.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2022 the original author or authors. * Copyright 2002-2024 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -248,30 +248,31 @@ public class ConcurrentWebSocketSessionDecorator extends WebSocketSessionDecorat
@Override @Override
public void close(CloseStatus status) throws IOException { public void close(CloseStatus status) throws IOException {
this.closeLock.lock(); if (this.closeLock.tryLock()) {
try { try {
if (this.closeInProgress) { if (this.closeInProgress) {
return; return;
}
if (!CloseStatus.SESSION_NOT_RELIABLE.equals(status)) {
try {
checkSessionLimits();
}
catch (SessionLimitExceededException ex) {
// Ignore
} }
if (this.limitExceeded) { if (!CloseStatus.SESSION_NOT_RELIABLE.equals(status)) {
if (logger.isDebugEnabled()) { try {
logger.debug("Changing close status " + status + " to SESSION_NOT_RELIABLE."); checkSessionLimits();
}
catch (SessionLimitExceededException ex) {
// Ignore
}
if (this.limitExceeded) {
if (logger.isDebugEnabled()) {
logger.debug("Changing close status " + status + " to SESSION_NOT_RELIABLE.");
}
status = CloseStatus.SESSION_NOT_RELIABLE;
} }
status = CloseStatus.SESSION_NOT_RELIABLE;
} }
this.closeInProgress = true;
super.close(status);
}
finally {
this.closeLock.unlock();
} }
this.closeInProgress = true;
super.close(status);
}
finally {
this.closeLock.unlock();
} }
} }

Loading…
Cancel
Save