study/Spring-Cloud-Alibaba
1
0
Fork 0
mirror of https://gitee.com/mirrors/Spring-Cloud-Alibaba.git synced 2021-06-26 13:25:11 +08:00
Code Issues Releases Wiki Activity

extend AbstractAutoServiceRegistration instead of implements AutoServiceRegistration , Fixes #18

Browse Source
This commit is contained in:
flystar32 2018-09-17 23:28:39 +08:00
parent bd814e8697
commit 625e85b040
2 changed files with 62 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spring-cloud-alibaba-nacos-discovery/src/main/java/org/springframework/cloud/alibaba/nacos/NacosDiscoveryAutoConfiguration.java
View File

@ -27,7 +27,6 @@ import org.springframework.cloud.alibaba.nacos.registry.NacosRegistration;
import org.springframework.cloud.alibaba.nacos.registry.NacosServiceRegistry; import org.springframework.cloud.alibaba.nacos.registry.NacosServiceRegistry;
import org.springframework.cloud.client.serviceregistry.AutoServiceRegistrationAutoConfiguration; import org.springframework.cloud.client.serviceregistry.AutoServiceRegistrationAutoConfiguration;
import org.springframework.cloud.client.serviceregistry.AutoServiceRegistrationProperties; import org.springframework.cloud.client.serviceregistry.AutoServiceRegistrationProperties;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
@ -64,7 +63,10 @@ public class NacosDiscoveryAutoConfiguration {
@Bean @Bean
@ConditionalOnBean(AutoServiceRegistrationProperties.class) @ConditionalOnBean(AutoServiceRegistrationProperties.class)
public NacosAutoServiceRegistration nacosAutoServiceRegistration( public NacosAutoServiceRegistration nacosAutoServiceRegistration(
ApplicationContext context, NacosServiceRegistry registry, NacosRegistration registration) { NacosServiceRegistry registry,
return new NacosAutoServiceRegistration(context, registry, registration); AutoServiceRegistrationProperties autoServiceRegistrationProperties,
NacosRegistration registration) {
return new NacosAutoServiceRegistration(registry,
autoServiceRegistrationProperties, registration);
} }
} }

118
spring-cloud-alibaba-nacos-discovery/src/main/java/org/springframework/cloud/alibaba/nacos/registry/NacosAutoServiceRegistration.java
View File

@ -16,112 +16,90 @@
package org.springframework.cloud.alibaba.nacos.registry; package org.springframework.cloud.alibaba.nacos.registry;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.web.context.ConfigurableWebServerApplicationContext; import org.springframework.cloud.client.serviceregistry.AbstractAutoServiceRegistration;
import org.springframework.boot.web.context.WebServerInitializedEvent; import org.springframework.cloud.client.serviceregistry.AutoServiceRegistrationProperties;
import org.springframework.boot.web.servlet.context.ServletWebServerInitializedEvent;
import org.springframework.cloud.client.discovery.event.InstanceRegisteredEvent;
import org.springframework.cloud.client.serviceregistry.AutoServiceRegistration;
import org.springframework.cloud.client.serviceregistry.ServiceRegistry; import org.springframework.cloud.client.serviceregistry.ServiceRegistry;
import org.springframework.context.ApplicationContext; import org.springframework.util.Assert;
import org.springframework.context.SmartLifecycle; import org.springframework.util.StringUtils;
import org.springframework.context.event.ContextClosedEvent;
import org.springframework.context.event.EventListener;
import org.springframework.core.Ordered;
/** /**
* @author xiaojing * @author xiaojing
*/ */
public class NacosAutoServiceRegistration public class NacosAutoServiceRegistration extends AbstractAutoServiceRegistration<NacosRegistration> {
implements AutoServiceRegistration, SmartLifecycle, Ordered { private static final Logger LOGGER = LoggerFactory.getLogger(NacosAutoServiceRegistration.class);
private static final Logger logger = LoggerFactory.getLogger(NacosAutoServiceRegistration.class);
@Autowired @Autowired
private NacosRegistration registration; private NacosRegistration registration;
private int order = 0;
private AtomicBoolean running = new AtomicBoolean(false);
private AtomicInteger port = new AtomicInteger(0);
private ApplicationContext context;
private ServiceRegistry serviceRegistry;
public NacosAutoServiceRegistration(ApplicationContext context, public NacosAutoServiceRegistration(ServiceRegistry<NacosRegistration> serviceRegistry,
ServiceRegistry<NacosRegistration> serviceRegistry, AutoServiceRegistrationProperties autoServiceRegistrationProperties,
NacosRegistration registration) { NacosRegistration registration) {
this.context = context; super(serviceRegistry, autoServiceRegistrationProperties);
this.serviceRegistry = serviceRegistry;
this.registration = registration; this.registration = registration;
} }
@Override @Deprecated
public void start() { public void setPort(int port) {
if (this.port.get() != 0) { getPort().set(port);
this.registration.setPort(port.get());
}
if (!this.running.get() && this.registration.getPort() > 0) {
this.serviceRegistry.register(this.registration);
this.context
.publishEvent(new InstanceRegisteredEvent(this, this.registration));
this.running.set(true);
}
} }
@Override @Override
public void stop() { protected NacosRegistration getRegistration() {
this.serviceRegistry.deregister(this.registration); if (this.registration.getPort() < 0 && this.getPort().get() > 0) {
this.running.set(false); this.registration.setPort(this.getPort().get());
}
Assert.isTrue(this.registration.getPort() > 0, "service.port has not been set");
return this.registration;
} }
@Override @Override
public boolean isRunning() { protected NacosRegistration getManagementRegistration() {
return this.running.get(); return null;
} }
@Override @Override
public int getPhase() { protected void register() {
return 0; if (!this.registration.getNacosDiscoveryProperties().isRegisterEnabled()) {
LOGGER.debug("Registration disabled.");
return;
}
if (this.registration.getPort() < 0) {
this.registration.setPort(getPort().get());
}
super.register();
} }
@Override @Override
public boolean isAutoStartup() { protected void registerManagement() {
return true; if (!this.registration.getNacosDiscoveryProperties().isRegisterEnabled()) {
return;
}
super.registerManagement();
}
@Override
protected Object getConfiguration() {
return this.registration.getNacosDiscoveryProperties();
} }
@Override @Override
public void stop(Runnable callback) { protected boolean isEnabled() {
this.stop(); return this.registration.getNacosDiscoveryProperties().isRegisterEnabled();
callback.run();
} }
@Override @Override
public int getOrder() { @SuppressWarnings("deprecation")
return this.order; protected String getAppName() {
} String appName = registration.getNacosDiscoveryProperties().getService();
return StringUtils.isEmpty(appName) ? super.getAppName() : appName;
@EventListener(WebServerInitializedEvent.class)
public void onApplicationEvent(WebServerInitializedEvent event) {
int localPort = event.getWebServer().getPort();
ApplicationContext context = event.getApplicationContext();
if(!(context instanceof ConfigurableWebServerApplicationContext) || !"management".equals(((ConfigurableWebServerApplicationContext)context).getServerNamespace())) {
logger.info("Updating port to {}", localPort);
this.port.compareAndSet(0, event.getWebServer().getPort());
this.start();
}
}
@EventListener({ ContextClosedEvent.class })
public void onApplicationEvent(ContextClosedEvent event) {
if (event.getApplicationContext() == this.context) {
this.stop();
}
} }
} }