org.springframework.messaging.handler.invocation.AbstractExceptionHandlerMethodResolver Java Examples

/**
 * Find an exception handling method for the given exception.
 * <p>The default implementation searches methods in the class hierarchy of
 * the HandlerMethod first and if not found, it continues searching for
 * additional handling methods registered via
 * {@link #registerExceptionHandlerAdvice}.
 * @param handlerMethod the method where the exception was raised
 * @param ex the exception raised or signaled
 * @return a method to handle the exception, or {@code null}
 */
@Nullable
public InvocableHandlerMethod initExceptionHandlerMethod(HandlerMethod handlerMethod, Throwable ex) {
	if (logger.isDebugEnabled()) {
		logger.debug("Searching for methods to handle " + ex.getClass().getSimpleName());
	}
	Class<?> beanType = handlerMethod.getBeanType();
	AbstractExceptionHandlerMethodResolver resolver = this.exceptionHandlerCache.get(beanType);
	if (resolver == null) {
		resolver = this.exceptionMethodResolverFactory.apply(beanType);
		this.exceptionHandlerCache.put(beanType, resolver);
	}
	InvocableHandlerMethod exceptionHandlerMethod = null;
	Method method = resolver.resolveMethod(ex);
	if (method != null) {
		exceptionHandlerMethod = new InvocableHandlerMethod(handlerMethod.getBean(), method);
	}
	else {
		for (MessagingAdviceBean advice : this.exceptionHandlerAdviceCache.keySet()) {
			if (advice.isApplicableToBeanType(beanType)) {
				resolver = this.exceptionHandlerAdviceCache.get(advice);
				method = resolver.resolveMethod(ex);
				if (method != null) {
					exceptionHandlerMethod = new InvocableHandlerMethod(advice.resolveBean(), method);
					break;
				}
			}
		}
	}
	if (exceptionHandlerMethod != null) {
		logger.debug("Found exception handler " + exceptionHandlerMethod.getShortLogMessage());
		exceptionHandlerMethod.setArgumentResolvers(this.argumentResolvers.getResolvers());
	}
	else {
		logger.error("No exception handling method", ex);
	}
	return exceptionHandlerMethod;
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionHandlerMethodResolverFor(
    Class<?> beanType) {
  return new AnnotationExceptionHandlerMethodResolver(beanType);
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionHandlerMethodResolverFor(Class<?> beanType) {
	return new AnnotationExceptionHandlerMethodResolver(beanType);
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionMethodResolverFor(Class<?> beanType) {
	return new AnnotationExceptionHandlerMethodResolver(beanType);
}
 

public InvocableHelper(
		Function<Class<?>, AbstractExceptionHandlerMethodResolver> exceptionMethodResolverFactory) {

	this.exceptionMethodResolverFactory = exceptionMethodResolverFactory;
}
 

/**
 * Method to populate the MessagingAdviceBean cache (e.g. to support "global"
 * {@code @MessageExceptionHandler}).
 */
public void registerExceptionHandlerAdvice(
		MessagingAdviceBean bean, AbstractExceptionHandlerMethodResolver resolver) {

	this.exceptionHandlerAdviceCache.put(bean, resolver);
}
 

/**
 * Subclasses can invoke this method to populate the MessagingAdviceBean cache
 * (e.g. to support "global" {@code @MessageExceptionHandler}).
 */
protected void registerExceptionHandlerAdvice(
		MessagingAdviceBean bean, AbstractExceptionHandlerMethodResolver resolver) {

	this.invocableHelper.registerExceptionHandlerAdvice(bean, resolver);
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionMethodResolverFor(Class<?> beanType) {
	return new TestExceptionResolver(beanType);
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionHandlerMethodResolverFor(Class<?> beanType) {
	return new AnnotationExceptionHandlerMethodResolver(beanType);
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionHandlerMethodResolverFor(Class<?> beanType) {
	return new AnnotationExceptionHandlerMethodResolver(beanType);
}
 

@Override
protected AbstractExceptionHandlerMethodResolver createExceptionHandlerMethodResolverFor(
		Class<?> beanType) {
	return new AnnotationExceptionHandlerMethodResolver(beanType);
}
 

/**
 * Create a concrete instance of {@link AbstractExceptionHandlerMethodResolver}
 * that finds exception handling methods based on some criteria, e.g. based
 * on the presence of {@code @MessageExceptionHandler}.
 * @param beanType the class in which an exception occurred during handling
 * @return the resolver to use
 */
protected abstract AbstractExceptionHandlerMethodResolver createExceptionMethodResolverFor(Class<?> beanType);