@Override
public void onException(Message message, Exception exception) {
context.getThreadAwareContext().setCurrentThread(true);
try {
wrapped.onException(message, exception);
} finally {
context.getThreadAwareContext().clearCurrentThread(true);
}
}
@Override
public void onCompletion(Message message) {
context.setCurrentThread(true);
try {
wrapped.onCompletion(message);
} finally {
context.clearCurrentThread(true);
}
}