fix classlaoder setting

sofastack · May 30, 2024 · a22ee85 · a22ee85
1 parent f6dd49e
commit a22ee85
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/main/java/com/alipay/remoting/rpc/RpcInvokeCallbackListener.java b/src/main/java/com/alipay/remoting/rpc/RpcInvokeCallbackListener.java
@@ -126,10 +126,9 @@ public void run() {
                         oldClassLoader = Thread.currentThread().getContextClassLoader();
                         Thread.currentThread().setContextClassLoader(future.getAppClassLoader());
                     }
-
+                    response.setInvokeContext(future.getInvokeContext());
                     Object responseObj = RpcResponseResolver.resolveResponseObject(response,
                         this.remoteAddress);
-                    response.setInvokeContext(future.getInvokeContext());
                     try {
                         callback.onResponse(responseObj);
                     } catch (Throwable e) {