Eclipse
/
jetty.project
mirror of https://github.com/jetty/jetty.project.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add suppressed failures in Callback failed (#11435) 

If an exception is thrown during failure handling, then record the original failure as a suppressed Throwable on the thrown exception
This commit is contained in:
Greg Wilkins 2024-02-29 20:06:50 +01:00 committed by GitHub
parent 56e05a973f
commit 4155e7bc25
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 193 additions and 88 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
jetty-core/jetty-server/src/main/java/org/eclipse/jetty/server/internal/HttpChannelState.java
View File

@ -1114,7 +1114,7 @@ public class HttpChannelState implements HttpChannel, Components
}
if (writeCallback == null)
return null;
return () -> writeCallback.failed(x);
return () -> HttpChannelState.failed(writeCallback, x);
}
public long getContentBytesWritten()
@ -1229,7 +1229,7 @@ public class HttpChannelState implements HttpChannel, Components
if (writeFailure != null)
{
Throwable failure = writeFailure;
httpChannelState._writeInvoker.run(() -> callback.failed(failure));
httpChannelState._writeInvoker.run(() -> HttpChannelState.failed(callback, failure));
return;
}
@ -1297,7 +1297,7 @@ public class HttpChannelState implements HttpChannel, Components
httpChannel.lockedStreamSendCompleted(false);
}
if (callback != null)
httpChannel._writeInvoker.run(() -> callback.failed(x));
httpChannel._writeInvoker.run(() -> HttpChannelState.failed(callback, x));
}
@Override
@ -1512,6 +1512,8 @@ public class HttpChannelState implements HttpChannel, Components
*/
@Override
public void failed(Throwable failure)
{
try
{
// Called when the request/response cycle is completing with a failure.
HttpStream stream;
@ -1549,6 +1551,12 @@ public class HttpChannelState implements HttpChannel, Components
else
_request.getHttpChannelState()._handlerInvoker.failed(failure);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, failure);
throw t;
}
}
private boolean lockedCompleteCallback()
{
@ -1683,7 +1691,7 @@ public class HttpChannelState implements HttpChannel, Components
}
else
{
httpChannelState._handlerInvoker.failed(failure);
HttpChannelState.failed(httpChannelState._handlerInvoker, failure);
}
}
@ -1705,9 +1713,8 @@ public class HttpChannelState implements HttpChannel, Components
httpChannelState = _request.lockedGetHttpChannelState();
httpChannelState._response._status = _errorResponse._status;
}
if (ExceptionUtil.areNotAssociated(failure, x))
failure.addSuppressed(x);
httpChannelState._handlerInvoker.failed(failure);
ExceptionUtil.addSuppressedIfNotAssociated(failure, x);
HttpChannelState.failed(httpChannelState._handlerInvoker, failure);
}
@Override
@ -1735,10 +1742,18 @@ public class HttpChannelState implements HttpChannel, Components
@Override
public void failed(Throwable x)
{
try
{
completing();
super.failed(x);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, x);
throw t;
}
}
private void completing()
{
@ -1868,4 +1883,25 @@ public class HttpChannelState implements HttpChannel, Components
}
}
}
/**
* Invoke a callback failure, handling any {@link Throwable} thrown
* by adding the passed {@code failure} as a suppressed with
* {@link ExceptionUtil#addSuppressedIfNotAssociated(Throwable, Throwable)}.
* @param callback The callback to fail
* @param failure The failure
* @throws RuntimeException If thrown, will have the {@code failure} added as a suppressed.
*/
private static void failed(Callback callback, Throwable failure)
{
try
{
callback.failed(failure);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, failure);
throw t;
}
}
}

144
jetty-core/jetty-util/src/main/java/org/eclipse/jetty/util/Callback.java
View File

@ -120,9 +120,17 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
try
{
completable.completeExceptionally(x);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, x);
throw t;
}
}
@Override
public InvocationType getInvocationType()
@ -164,9 +172,17 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
try
{
failure.accept(x);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, x);
throw t;
}
}
@Override
public InvocationType getInvocationType()
@ -272,14 +288,7 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
try
{
callback.failed(x);
}
finally
{
completed.accept(x);
}
Callback.failed(callback::failed, completed, x);
}
};
}
@ -307,22 +316,19 @@ public interface Callback extends Invocable
}
catch (Throwable t)
{
callback.failed(t);
Callback.failed(callback, t);
}
}
private void completed(Throwable ignored)
{
completed.run();
}
@Override
public void failed(Throwable x)
{
try
{
completed.run();
}
catch (Throwable t)
{
x.addSuppressed(t);
}
callback.failed(x);
Callback.failed(this::completed, callback::failed, x);
}
};
}
@ -348,8 +354,8 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
cause.addSuppressed(x);
callback.failed(cause);
ExceptionUtil.addSuppressedIfNotAssociated(cause, x);
Callback.failed(callback, cause);
}
};
}
@ -380,9 +386,17 @@ public interface Callback extends Invocable
@Override
default void failed(Throwable x)
{
try
{
completed();
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, x);
throw t;
}
}
}
/**
@ -408,6 +422,11 @@ public interface Callback extends Invocable
{
}
private void completed(Throwable ignored)
{
completed();
}
@Override
public void succeeded()
{
@ -424,14 +443,7 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
try
{
callback.failed(x);
}
finally
{
completed();
}
Callback.failed(callback::failed, this::completed, x);
}
@Override
@ -472,18 +484,7 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
try
{
cb1.failed(x);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(x, t);
}
finally
{
cb2.failed(x);
}
Callback.failed(cb1::failed, cb2::failed, x);
}
@Override
@ -533,8 +534,7 @@ public interface Callback extends Invocable
@Override
public void failed(Throwable x)
{
callback.failed(x);
super.failed(x);
Callback.failed(callback::failed, super::failed, x);
}
};
}
@ -592,4 +592,64 @@ public interface Callback extends Invocable
return completable;
}
}
/**
* Invoke a callback failure, handling any {@link Throwable} thrown
* by adding the passed {@code failure} as a suppressed with
* {@link ExceptionUtil#addSuppressedIfNotAssociated(Throwable, Throwable)}.
* @param callback The callback to fail
* @param failure The failure
* @throws RuntimeException If thrown, will have the {@code failure} added as a suppressed.
*/
private static void failed(Callback callback, Throwable failure)
{
try
{
callback.failed(failure);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, failure);
throw t;
}
}
/**
* Invoke two consumers of a failure, handling any {@link Throwable} thrown
* by adding the passed {@code failure} as a suppressed with
* {@link ExceptionUtil#addSuppressedIfNotAssociated(Throwable, Throwable)}.
* @param first The first consumer of a failure
* @param second The first consumer of a failure
* @param failure The failure
* @throws RuntimeException If thrown, will have the {@code failure} added as a suppressed.
*/
private static void failed(Consumer<Throwable> first, Consumer<Throwable> second, Throwable failure)
{
// This is an improved version of:
// try
// {
// first.accept(failure);
// }
// finally
// {
// second.accept(failure);
// }
try
{
first.accept(failure);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(failure, t);
}
try
{
second.accept(failure);
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, failure);
throw t;
}
}
}

9
jetty-ee10/jetty-ee10-servlet/src/main/java/org/eclipse/jetty/ee10/servlet/ServletChannel.java
View File

@ -499,6 +499,9 @@ public class ServletChannel
ExceptionUtil.addSuppressedIfNotAssociated(cause, x);
if (LOG.isDebugEnabled())
LOG.debug("Could not perform error handling, aborting", cause);
try
{
if (_state.isResponseCommitted())
{
// Perform the same behavior as when the callback is failed.
@ -510,6 +513,12 @@ public class ServletChannel
sendErrorResponseAndComplete();
}
}
catch (Throwable t)
{
ExceptionUtil.addSuppressedIfNotAssociated(t, cause);
throw t;
}
}
finally
{
// clean up the context that was set in Response.sendError