fizzgate/fizz-gateway-node
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Shutdown registry center client after removing registry center

Browse Source
This commit is contained in:
lancer.hong
2022-05-06 10:55:02 +08:00
parent 5ae4919216
commit 80a4e444a4
4 changed files with 35 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
fizz-core/src/main/java/we/service_registry/FizzServiceRegistration.java
View File

@@ -50,7 +50,7 @@ public abstract class FizzServiceRegistration {
UP, DOWN, STARTING, OUT_OF_SERVICE, UNKNOWN;
}
private String id;
protected String id;
private Type type;
@@ -91,12 +91,22 @@ public abstract class FizzServiceRegistration {
public void register() {
serviceRegistry.register(registration);
LOGGER.info("register to {} {}", type, id);
}
public void deregister() {
serviceRegistry.deregister(registration);
LOGGER.info("deregister to {} {}", type, id);
}
public void close() {
serviceRegistry.close();
shutdownClient();
LOGGER.info("close {} {}", type, id);
}
protected abstract void shutdownClient();
public abstract ServerStatus getServerStatus();
public abstract List<String> getServices();

25
fizz-core/src/main/java/we/service_registry/RegistryCenterService.java
View File

@@ -32,7 +32,6 @@ import we.util.JacksonUtils;
import we.util.Result;
import we.util.ThreadContext;
import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.util.Collections;
import java.util.HashMap;
@@ -45,7 +44,7 @@ import java.util.Map;
@Service
public class RegistryCenterService implements ApplicationListener<ContextRefreshedEvent> {
private static final Logger log = LoggerFactory.getLogger(RegistryCenterService.class);
private static final Logger LOGGER = LoggerFactory.getLogger(RegistryCenterService.class);
private Map<String, RegistryCenter> registryCenterMap = new HashMap<>();
@@ -81,7 +80,7 @@ public class RegistryCenterService implements ApplicationListener<ContextRefresh
json = (String) e.getValue();
RegistryCenter rc = JacksonUtils.readValue(json, RegistryCenter.class);
registryCenterMap.put(rc.name, rc);
log.info("init registry center {}", rc.name);
LOGGER.info("init registry center {}", rc);
rc.initFizzServiceRegistration(applicationContext);
rc.getFizzServiceRegistration().register();
}
@@ -91,7 +90,7 @@ public class RegistryCenterService implements ApplicationListener<ContextRefresh
result.t = t;
}
} else {
log.info("no registry center");
LOGGER.info("no registry center");
}
return Mono.empty();
}
@@ -118,12 +117,12 @@ public class RegistryCenterService implements ApplicationListener<ContextRefresh
result.code = Result.FAIL;
result.msg = "lsn error, channel: " + channel;
result.t = t;
log.error("lsn channel {} error", channel, t);
LOGGER.error("lsn channel {} error", channel, t);
}
)
.doOnSubscribe(
s -> {
log.info("success to lsn on {}", channel);
LOGGER.info("success to lsn on {}", channel);
}
)
.doOnNext(
@@ -134,19 +133,23 @@ public class RegistryCenterService implements ApplicationListener<ContextRefresh
RegistryCenter prev = null;
if (rc.isDeleted) {
prev = registryCenterMap.remove(rc.name);
log.info("remove registry center {}", rc.name);
prev.getFizzServiceRegistration().deregister();
LOGGER.info("remove registry center {}", prev);
FizzServiceRegistration fizzServiceRegistration = prev.getFizzServiceRegistration();
fizzServiceRegistration.deregister();
fizzServiceRegistration.close();
} else {
prev = registryCenterMap.put(rc.name, rc);
log.info("update registry center {}", rc.name);
LOGGER.info("update registry center {} with {}", prev, rc);
if (prev != null) {
prev.getFizzServiceRegistration().deregister();
FizzServiceRegistration fizzServiceRegistration = prev.getFizzServiceRegistration();
fizzServiceRegistration.deregister();
fizzServiceRegistration.close();
}
rc.initFizzServiceRegistration(applicationContext);
rc.getFizzServiceRegistration().register();
}
} catch (Throwable t) {
log.error("update registry center error, {}", message, t);
LOGGER.error("update registry center error, {}", message, t);
}
}
)

6
fizz-core/src/main/java/we/service_registry/eureka/FizzEurekaServiceRegistration.java
View File

@@ -55,6 +55,12 @@ public class FizzEurekaServiceRegistration extends FizzServiceRegistration {
return client;
}
@Override
protected void shutdownClient() {
client.shutdown();
LOGGER.info("shutdown {} client", id);
}
@Override
public ServerStatus getServerStatus() {
EurekaClientConfig eurekaClientConfig = client.getEurekaClientConfig();

4
fizz-core/src/main/java/we/service_registry/nacos/FizzNacosServiceRegistration.java
View File

@@ -67,6 +67,10 @@ public class FizzNacosServiceRegistration extends FizzServiceRegistration {
return namingService;
}
@Override
protected void shutdownClient() {
}
@Override
public ServerStatus getServerStatus() {
String status = namingService.getServerStatus();