Fix handling of root stub reconnect when declared as AutoDispose.

cojen · Feb 1, 2025 · f52e12c · f52e12c
1 parent d1c9d30
commit f52e12c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/org/cojen/dirmi/core/ClientSession.java b/src/main/java/org/cojen/dirmi/core/ClientSession.java
@@ -161,7 +161,7 @@ private boolean reconnectAttempt(Object result) {
 
         mEngine.changeIdentity(this, newSession.id);
 
-        var newRoot = (StubInvoker) newSession.mRoot;
+        var newRoot = ((Stub) newSession.mRoot).invoker();
         Object removed = newSession.mStubs.remove(newRoot);
         assert newRoot == removed;
         assert newSession.mStubs.size() == 0;
@@ -176,7 +176,7 @@ private boolean reconnectAttempt(Object result) {
 
         cServerSessionIdHandle.setRelease(this, newSession.mServerSessionId);
 
-        var root = (StubInvoker) mRoot;
+        var root = ((Stub) mRoot).invoker();
         mStubs.changeIdentity(root, newRoot.id);
 
         Map<String, RemoteInfo> typeMap;
@@ -219,7 +219,7 @@ private boolean reconnectAttempt(Object result) {
             return false;
         }
 
-        StubInvoker.cSupportHandle.setRelease(mRoot, newSupport);
+        StubInvoker.cSupportHandle.setRelease(root, newSupport);
 
         // For all restorable stubs, update the MethodIdWriter and set a support object that
         // allows them to restore on demand.