-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Make rest-client invocation context implement ArcInvocationContext
- Loading branch information
Showing
2 changed files
with
214 additions
and
10 deletions.
There are no files selected for viewing
198 changes: 198 additions & 0 deletions
198
...ent/runtime/src/main/java/io/quarkus/restclient/runtime/QuarkusInvocationContextImpl.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,198 @@ | ||
| package io.quarkus.restclient.runtime; | ||
|
|
||
| import java.lang.annotation.Annotation; | ||
| import java.lang.reflect.Constructor; | ||
| import java.lang.reflect.InvocationTargetException; | ||
| import java.lang.reflect.Method; | ||
| import java.util.ArrayList; | ||
| import java.util.Collections; | ||
| import java.util.HashMap; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.Set; | ||
| import java.util.concurrent.CompletionException; | ||
|
|
||
| import jakarta.enterprise.inject.spi.InterceptionType; | ||
| import jakarta.enterprise.inject.spi.Interceptor; | ||
| import jakarta.interceptor.InvocationContext; | ||
| import jakarta.ws.rs.client.ResponseProcessingException; | ||
|
|
||
| import org.jboss.resteasy.microprofile.client.ExceptionMapping; | ||
|
|
||
| import io.quarkus.arc.ArcInvocationContext; | ||
|
|
||
| /** | ||
| * A Quarkus copy of {@link org.jboss.resteasy.microprofile.client.InvocationContextImpl} which makes it implement | ||
| * {@link ArcInvocationContext} instead so that it's compatible with Quarkus interceptors. | ||
| */ | ||
| public class QuarkusInvocationContextImpl implements ArcInvocationContext { | ||
|
|
||
| private final Object target; | ||
|
|
||
| private final Method method; | ||
|
|
||
| private Object[] args; | ||
|
|
||
| private final int position; | ||
|
|
||
| private final Map<String, Object> contextData; | ||
|
|
||
| private final List<QuarkusInvocationContextImpl.InterceptorInvocation> chain; | ||
|
|
||
| private final Set<Annotation> interceptorBindings; | ||
|
|
||
| public QuarkusInvocationContextImpl(final Object target, final Method method, final Object[] args, | ||
| final List<QuarkusInvocationContextImpl.InterceptorInvocation> chain, Set<Annotation> interceptorBindings) { | ||
| this(target, method, args, chain, 0, interceptorBindings); | ||
| } | ||
|
|
||
| private QuarkusInvocationContextImpl(final Object target, final Method method, final Object[] args, | ||
| final List<QuarkusInvocationContextImpl.InterceptorInvocation> chain, final int position, | ||
| Set<Annotation> interceptorBindings) { | ||
| this.target = target; | ||
| this.method = method; | ||
| this.args = args; | ||
| this.interceptorBindings = interceptorBindings == null ? Collections.emptySet() : interceptorBindings; | ||
| this.contextData = new HashMap<>(); | ||
| // put in bindings under Arc's specific key | ||
| this.contextData.put(ArcInvocationContext.KEY_INTERCEPTOR_BINDINGS, interceptorBindings); | ||
| this.position = position; | ||
| this.chain = chain; | ||
| } | ||
|
|
||
| boolean hasNextInterceptor() { | ||
| return position < chain.size(); | ||
| } | ||
|
|
||
| protected Object invokeNext() throws Exception { | ||
| return chain.get(position).invoke(nextContext()); | ||
| } | ||
|
|
||
| private InvocationContext nextContext() { | ||
| return new QuarkusInvocationContextImpl(target, method, args, chain, position + 1, interceptorBindings); | ||
| } | ||
|
|
||
| protected Object interceptorChainCompleted() throws Exception { | ||
| try { | ||
| return method.invoke(target, args); | ||
| } catch (InvocationTargetException e) { | ||
| Throwable cause = e.getCause(); | ||
| if (cause instanceof CompletionException) { | ||
| cause = cause.getCause(); | ||
| } | ||
| if (cause instanceof ExceptionMapping.HandlerException) { | ||
| ((ExceptionMapping.HandlerException) cause).mapException(method); | ||
| } | ||
| if (cause instanceof ResponseProcessingException) { | ||
| ResponseProcessingException rpe = (ResponseProcessingException) cause; | ||
| // Note that the default client engine leverages a single connection | ||
| // MP FT: we need to close the response otherwise we would not be able to retry if the method returns jakarta.ws.rs.core.Response | ||
| rpe.getResponse().close(); | ||
| cause = rpe.getCause(); | ||
| if (cause instanceof RuntimeException) { | ||
| throw (RuntimeException) cause; | ||
| } | ||
| } | ||
| throw e; | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public Object proceed() throws Exception { | ||
| try { | ||
| if (hasNextInterceptor()) { | ||
| return invokeNext(); | ||
| } else { | ||
| return interceptorChainCompleted(); | ||
| } | ||
| } catch (InvocationTargetException e) { | ||
| Throwable cause = e.getCause(); | ||
| if (cause instanceof Error) { | ||
| throw (Error) cause; | ||
| } | ||
| if (cause instanceof Exception) { | ||
| throw (Exception) cause; | ||
| } | ||
| throw new RuntimeException(cause); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public Object getTarget() { | ||
| return target; | ||
| } | ||
|
|
||
| @Override | ||
| public Method getMethod() { | ||
| return method; | ||
| } | ||
|
|
||
| @Override | ||
| public Constructor<?> getConstructor() { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public Object[] getParameters() throws IllegalStateException { | ||
| return args; | ||
| } | ||
|
|
||
| @Override | ||
| public void setParameters(Object[] params) throws IllegalStateException, IllegalArgumentException { | ||
| this.args = params; | ||
| } | ||
|
|
||
| @Override | ||
| public Map<String, Object> getContextData() { | ||
| return contextData; | ||
| } | ||
|
|
||
| @Override | ||
| public Object getTimer() { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public Set<Annotation> getInterceptorBindings() { | ||
| return interceptorBindings; | ||
| } | ||
|
|
||
| @Override | ||
| public <T extends Annotation> T findIterceptorBinding(Class<T> annotationType) { | ||
| for (Annotation annotation : getInterceptorBindings()) { | ||
| if (annotation.annotationType().equals(annotationType)) { | ||
| return (T) annotation; | ||
| } | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public <T extends Annotation> List<T> findIterceptorBindings(Class<T> annotationType) { | ||
| List<T> found = new ArrayList<>(); | ||
| for (Annotation annotation : getInterceptorBindings()) { | ||
| if (annotation.annotationType().equals(annotationType)) { | ||
| found.add((T) annotation); | ||
| } | ||
| } | ||
| return found; | ||
| } | ||
|
|
||
| public static class InterceptorInvocation { | ||
|
|
||
| @SuppressWarnings("rawtypes") | ||
| private final Interceptor interceptor; | ||
|
|
||
| private final Object interceptorInstance; | ||
|
|
||
| public InterceptorInvocation(final Interceptor<?> interceptor, final Object interceptorInstance) { | ||
| this.interceptor = interceptor; | ||
| this.interceptorInstance = interceptorInstance; | ||
| } | ||
|
|
||
| @SuppressWarnings("unchecked") | ||
| Object invoke(InvocationContext ctx) throws Exception { | ||
| return interceptor.intercept(InterceptionType.AROUND_INVOKE, interceptorInstance, ctx); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters