Skip to content

Commit

Permalink
Qos enhancement (#2153)
Browse files Browse the repository at this point in the history
* Not to accept foreign ip by default.

* Log if fail to start qos server. #2046

* Fix typo. No functional change.

* Remove redundant declaration.

* Sync dubbo.xsd to compact dubbo.xsd.

* Simplify code and add comments.

* Add log message if qos is not enabled.

* Fix UT failure.
  • Loading branch information
ralf0131 authored and jerrick-zhu committed Aug 16, 2018
1 parent 3a267f8 commit 7277f5e
Show file tree
Hide file tree
Showing 6 changed files with 381 additions and 364 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -71,10 +71,13 @@ public class ApplicationConfig extends AbstractConfig {
// directory for saving thread dump
private String dumpDirectory;

// whether to enable qos or not
private Boolean qosEnable;

// the qos port to listen
private Integer qosPort;

// should we accept foreign ip or not?
private Boolean qosAcceptForeignIp;

// customized parameters
Expand Down

Large diffs are not rendered by default.

348 changes: 174 additions & 174 deletions dubbo-config/dubbo-config-spring/src/main/resources/META-INF/dubbo.xsd

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,9 @@

import org.apache.dubbo.common.Constants;
import org.apache.dubbo.common.URL;
import org.apache.dubbo.common.logger.Logger;
import org.apache.dubbo.common.logger.LoggerFactory;
import org.apache.dubbo.qos.common.QosConstants;
import org.apache.dubbo.qos.server.Server;
import org.apache.dubbo.rpc.Exporter;
import org.apache.dubbo.rpc.Invoker;
Expand All @@ -30,7 +33,11 @@
import static org.apache.dubbo.common.Constants.QOS_ENABLE;
import static org.apache.dubbo.common.Constants.QOS_PORT;


public class QosProtocolWrapper implements Protocol {

private final Logger logger = LoggerFactory.getLogger(QosProtocolWrapper.class);

private static AtomicBoolean hasStarted = new AtomicBoolean(false);

private Protocol protocol;
Expand Down Expand Up @@ -76,20 +83,23 @@ private void startQosServer(URL url) {
}

try {
boolean qosEnable = Boolean.parseBoolean(url.getParameter(QOS_ENABLE,"true"));
boolean qosEnable = url.getParameter(QOS_ENABLE,true);
if (!qosEnable) {
logger.info("qos won't be started because it is disabled. " +
"Please check dubbo.application.qos.enable is configured either in system property, " +
"dubbo.properties or XML/spring-boot configuration.");
return;
}

int port = Integer.parseInt(url.getParameter(QOS_PORT,"22222"));
boolean acceptForeignIp = Boolean.parseBoolean(url.getParameter(ACCEPT_FOREIGN_IP,"true"));
int port = url.getParameter(QOS_PORT, QosConstants.DEFAULT_PORT);
boolean acceptForeignIp = Boolean.parseBoolean(url.getParameter(ACCEPT_FOREIGN_IP,"false"));
Server server = Server.getInstance();
server.setPort(port);
server.setAcceptForeignIp(acceptForeignIp);
server.start();

} catch (Throwable throwable) {
//throw new RpcException("fail to start qos server", throwable);
logger.warn("Fail to start qos server: ", throwable);
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -42,9 +42,9 @@ public class QosProtocolWrapperTest {

@Before
public void setUp() throws Exception {
when(url.getParameter(Constants.QOS_ENABLE, "true")).thenReturn("true");
when(url.getParameter(Constants.QOS_PORT, "22222")).thenReturn("12345");
when(url.getParameter(Constants.ACCEPT_FOREIGN_IP, "true")).thenReturn("false");
when(url.getParameter(Constants.QOS_ENABLE, true)).thenReturn(true);
when(url.getParameter(Constants.QOS_PORT, 22222)).thenReturn(12345);
when(url.getParameter(Constants.ACCEPT_FOREIGN_IP, true)).thenReturn(false);
when(invoker.getUrl()).thenReturn(url);
when(url.getProtocol()).thenReturn(Constants.REGISTRY_PROTOCOL);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -136,26 +136,26 @@ public <T> Exporter<T> export(final Invoker<T> originInvoker) throws RpcExceptio

//registry provider
final Registry registry = getRegistry(originInvoker);
final URL registedProviderUrl = getRegistedProviderUrl(originInvoker);
final URL registeredProviderUrl = getRegisteredProviderUrl(originInvoker);

//to judge to delay publish whether or not
boolean register = registedProviderUrl.getParameter("register", true);
boolean register = registeredProviderUrl.getParameter("register", true);

ProviderConsumerRegTable.registerProvider(originInvoker, registryUrl, registedProviderUrl);
ProviderConsumerRegTable.registerProvider(originInvoker, registryUrl, registeredProviderUrl);

if (register) {
register(registryUrl, registedProviderUrl);
register(registryUrl, registeredProviderUrl);
ProviderConsumerRegTable.getProviderWrapper(originInvoker).setReg(true);
}

// Subscribe the override data
// FIXME When the provider subscribes, it will affect the scene : a certain JVM exposes the service and call the same service. Because the subscribed is cached key with the name of the service, it causes the subscription information to cover.
final URL overrideSubscribeUrl = getSubscribedOverrideUrl(registedProviderUrl);
final URL overrideSubscribeUrl = getSubscribedOverrideUrl(registeredProviderUrl);
final OverrideListener overrideSubscribeListener = new OverrideListener(overrideSubscribeUrl, originInvoker);
overrideListeners.put(overrideSubscribeUrl, overrideSubscribeListener);
registry.subscribe(overrideSubscribeUrl, overrideSubscribeListener);
//Ensure that a new exporter instance is returned every time export
return new DestroyableExporter<T>(exporter, originInvoker, overrideSubscribeUrl, registedProviderUrl);
return new DestroyableExporter<T>(exporter, originInvoker, overrideSubscribeUrl, registeredProviderUrl);
}

@SuppressWarnings("unchecked")
Expand Down Expand Up @@ -220,10 +220,10 @@ private URL getRegistryUrl(Invoker<?> originInvoker) {
* @param originInvoker
* @return
*/
private URL getRegistedProviderUrl(final Invoker<?> originInvoker) {
private URL getRegisteredProviderUrl(final Invoker<?> originInvoker) {
URL providerUrl = getProviderUrl(originInvoker);
//The address you see at the registry
final URL registedProviderUrl = providerUrl.removeParameters(getFilteredKeys(providerUrl))
return providerUrl.removeParameters(getFilteredKeys(providerUrl))
.removeParameter(Constants.MONITOR_KEY)
.removeParameter(Constants.BIND_IP_KEY)
.removeParameter(Constants.BIND_PORT_KEY)
Expand All @@ -232,7 +232,6 @@ private URL getRegistedProviderUrl(final Invoker<?> originInvoker) {
.removeParameter(ACCEPT_FOREIGN_IP)
.removeParameter(VALIDATION_KEY)
.removeParameter(INTERFACES);
return registedProviderUrl;
}

private URL getSubscribedOverrideUrl(URL registedProviderUrl) {
Expand Down

0 comments on commit 7277f5e

Please sign in to comment.