Skip close lock if acquired by other thread already

Closes gh-32445
This commit is contained in:
Juergen Hoeller 2024-04-23 16:06:48 +02:00
parent 16d8d69cf9
commit d151931f86
1 changed files with 22 additions and 21 deletions

View File

@ -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) { if (!CloseStatus.SESSION_NOT_RELIABLE.equals(status)) {
// Ignore try {
} checkSessionLimits();
if (this.limitExceeded) { }
if (logger.isDebugEnabled()) { catch (SessionLimitExceededException ex) {
logger.debug("Changing close status " + status + " to SESSION_NOT_RELIABLE."); // 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();
} }
} }