From 3c6b15ee9dbd262bd1628916f91e2c48c90ed9e7 Mon Sep 17 00:00:00 2001
From: redkale <redkale@qq.com>
Date: Thu, 2 Feb 2023 17:42:33 +0800
Subject: [PATCH] Context.updateServerAddress

---
 src/main/java/org/redkale/net/Context.java       |  4 ++++
 src/main/java/org/redkale/net/Server.java        |  4 ++--
 .../java/org/redkale/net/sncp/SncpContext.java   | 16 +++++++++++++++-
 3 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/main/java/org/redkale/net/Context.java b/src/main/java/org/redkale/net/Context.java
index 1cac4bf03..16a23b2db 100644
--- a/src/main/java/org/redkale/net/Context.java
+++ b/src/main/java/org/redkale/net/Context.java
@@ -167,6 +167,10 @@ public class Context {
         conn.updateWriteIOThread(ioWriteThread);
     }
 
+    protected void updateServerAddress(InetSocketAddress addr) {
+        this.serverAddress = addr;
+    }
+
     public ResourceFactory getResourceFactory() {
         return resourceFactory;
     }
diff --git a/src/main/java/org/redkale/net/Server.java b/src/main/java/org/redkale/net/Server.java
index 245a0b4dd..9bbd2a9f8 100644
--- a/src/main/java/org/redkale/net/Server.java
+++ b/src/main/java/org/redkale/net/Server.java
@@ -329,7 +329,7 @@ public abstract class Server<K extends Serializable, C extends Context, R extend
         Objects.requireNonNull(addr);
         final InetSocketAddress oldAddress = context.serverAddress;
         final ProtocolServer oldServerChannel = this.serverChannel;
-        context.serverAddress = addr;
+        context.updateServerAddress(addr);
         ProtocolServer newServerChannel = null;
         try {
             newServerChannel = ProtocolServer.create(this.netprotocol, context, this.serverClassLoader);
@@ -337,7 +337,7 @@ public abstract class Server<K extends Serializable, C extends Context, R extend
             newServerChannel.bind(addr, backlog);
             newServerChannel.accept(application, this);
         } catch (IOException e) {
-            context.serverAddress = oldAddress;
+            context.updateServerAddress(oldAddress);
             throw e;
         }
         this.address = context.serverAddress;
diff --git a/src/main/java/org/redkale/net/sncp/SncpContext.java b/src/main/java/org/redkale/net/sncp/SncpContext.java
index 5a58b914e..c8298e5d6 100644
--- a/src/main/java/org/redkale/net/sncp/SncpContext.java
+++ b/src/main/java/org/redkale/net/sncp/SncpContext.java
@@ -5,7 +5,8 @@
  */
 package org.redkale.net.sncp;
 
-import org.redkale.net.*;
+import java.net.InetSocketAddress;
+import org.redkale.net.Context;
 
 /**
  * <p>
@@ -15,8 +16,21 @@ import org.redkale.net.*;
  */
 public class SncpContext extends Context {
 
+    protected byte[] serverAddressBytes;
+
+    protected int serverAddressPort;
+
     public SncpContext(SncpContextConfig config) {
         super(config);
+        this.serverAddressBytes = serverAddress.getAddress().getAddress();
+        this.serverAddressPort = serverAddress.getPort();
+    }
+
+    @Override
+    protected void updateServerAddress(InetSocketAddress addr) {
+        super.updateServerAddress(addr);
+        this.serverAddressBytes = addr.getAddress().getAddress();
+        this.serverAddressPort = addr.getPort();
     }
 
     public static class SncpContextConfig extends ContextConfig {