Clean cache. (sofastack#714)

* add set method of providerBootStrap * reset server as null
404-not-find · Aug 21, 2019 · 733c473 · 733c473
1 parent 29caf6c
commit 733c473
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/core/api/src/main/java/com/alipay/sofa/rpc/config/ServerConfig.java b/core/api/src/main/java/com/alipay/sofa/rpc/config/ServerConfig.java
@@ -805,6 +805,14 @@ public Server getServer() {
         return server;
     }
 
+    /**
+     * Set server
+     * @param server
+     */
+    public void setServer(Server server) {
+        this.server = server;
+    }
+
     /**
      * Sets bound host.
      *

diff --git a/core/api/src/main/java/com/alipay/sofa/rpc/server/ServerFactory.java b/core/api/src/main/java/com/alipay/sofa/rpc/server/ServerFactory.java
@@ -154,6 +154,7 @@ public static void destroyServer(ServerConfig serverConfig) {
         try {
             Server server = serverConfig.getServer();
             if (server != null) {
+                serverConfig.setServer(null);
                 SERVER_MAP.remove(Integer.toString(serverConfig.getPort()));
                 server.destroy();
             }

diff --git a/core/api/src/test/java/com/alipay/sofa/rpc/server/ServerFactoryTest.java b/core/api/src/test/java/com/alipay/sofa/rpc/server/ServerFactoryTest.java
@@ -54,5 +54,6 @@ public void destroyServer() {
         Assert.assertEquals(1, ServerFactory.getServers().size());
         serverConfig.destroy();
         Assert.assertEquals(0, ServerFactory.getServers().size());
+        Assert.assertNull(serverConfig.getServer());
     }
 }