support nacos registry (#91)

Author: chakra-coder

pom.xml

@@ -20,7 +20,7 @@
 	<groupId>io.github.ningyu</groupId>
 	<artifactId>jmeter-plugins-dubbo</artifactId>
 	<version>${revision}</version>
-	
+
 	<properties>
 		<revision>2.7.3</revision>
 		<java.version>1.8</java.version>
@@ -32,6 +32,7 @@
 		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
 
         <dubbo.version>${revision}</dubbo.version>
+		<nacos.version>1.1.3</nacos.version>
         <!--<dubbo.version>2.7.2</dubbo.version>-->
 		<jorphan.version>3.0</jorphan.version>
 		<avalon.framework.version>4.1.4</avalon.framework.version>
@@ -101,6 +102,22 @@
             <groupId>de.ruedigermoeller</groupId>
             <artifactId>fst</artifactId>
         </dependency>
+		<!-- nacos -->
+		<dependency>
+			<groupId>org.apache.dubbo</groupId>
+			<artifactId>dubbo-registry-nacos</artifactId>
+			<version>${dubbo.version}</version>
+		</dependency>
+		<dependency>
+			<groupId>com.alibaba.nacos</groupId>
+			<artifactId>nacos-api</artifactId>
+			<version>${nacos.version}</version>
+		</dependency>
+		<dependency>
+			<groupId>com.alibaba.nacos</groupId>
+			<artifactId>nacos-client</artifactId>
+			<version>${nacos.version}</version>
+		</dependency>
 		<!-- slf4j -->
 		<dependency>
 			<groupId>org.slf4j</groupId>
@@ -166,7 +183,7 @@
 			</plugin>
 		</plugins>
 	</build>
-	
+
 	<developers>
 		<developer>
 			<name>ningyu</name>
@@ -175,4 +192,4 @@
 		</developer>
 	</developers>
 	<url></url>
-</project>
+</project>

src/main/java/io/github/ningyu/jmeter/plugin/dubbo/gui/DubboCommonPanel.java

@@ -80,7 +80,7 @@ public JPanel drawRegistrySettingsPanel() {
         //Protocol
         JPanel ph = new HorizontalPanel();
         JLabel protocolLable = new JLabel("Protocol:", SwingConstants.RIGHT);
-        registryProtocolText = new JComboBox<String>(new String[]{"","none", "zookeeper", "multicast", "redis", "simple"});
+        registryProtocolText = new JComboBox<String>(new String[]{"","none", "zookeeper", "nacos", "multicast", "redis", "simple"});
         registryProtocolText.setToolTipText("\"none\" is direct connection");
         protocolLable.setLabelFor(registryProtocolText);
         ph.add(protocolLable);

src/main/java/io/github/ningyu/jmeter/plugin/dubbo/sample/DubboSample.java

@@ -45,10 +45,10 @@
  * DubboSample
  */
 public class DubboSample extends AbstractSampler implements Interruptible {
-    
+
     private static final Logger log = LoggingManager.getLoggerForClass();
     private static final long serialVersionUID = -6794913295411458705L;
-    
+
 
     public static ApplicationConfig application = new ApplicationConfig("DubboSample");
 
@@ -94,7 +94,7 @@ private String getSampleData() {
         sb.append("Attachment Args: ").append(Constants.getAttachmentArgs(this).toString());
         return sb.toString();
     }
-    
+
     @SuppressWarnings({"unchecked", "rawtypes"})
     private Object callDubbo(SampleResult res) {
         // This instance is heavy, encapsulating the connection to the registry and the connection to the provider,
@@ -124,6 +124,14 @@ private Object callDubbo(SampleResult res) {
 			reference.setRegistry(registry);
 			reference.setProtocol(rpcProtocol);
 			break;
+		case Constants.REGISTRY_NACOS:
+		    registry = new RegistryConfig();
+		    registry.setProtocol(Constants.REGISTRY_NACOS);
+		    registry.setGroup(registryGroup);
+		    registry.setAddress(address);
+		    reference.setRegistry(registry);
+		    reference.setProtocol(rpcProtocol);
+		    break;
 		case Constants.REGISTRY_MULTICAST:
 			registry = new RegistryConfig();
 			registry.setProtocol(Constants.REGISTRY_MULTICAST);
@@ -249,7 +257,7 @@ private Object callDubbo(SampleResult res) {
                 res.setSuccessful(false);
                 return ErrorCode.MISS_METHOD.getMessage();
             }
-            
+
             // The registry's address is to generate the ReferenceConfigCache key
             ReferenceConfigCache cache = ReferenceConfigCache.getCache(Constants.getAddress(this), new ReferenceConfigCache.KeyGenerator() {
                 @Override

src/main/java/io/github/ningyu/jmeter/plugin/dubbo/sample/ProviderService.java

@@ -59,12 +59,12 @@ public Map<String, URL> findByService(String serviceName) {
     }
 
     public List<String> getProviders(String protocol, String address, String group) throws RuntimeException {
-        if (protocol.equals("zookeeper") || protocol.equals("redis")){
+        if (protocol.equals("zookeeper") || protocol.equals("nacos") || protocol.equals("redis")){
             return executeRegistry(protocol, address, group);
 //        } else if (protocol.equals("none")) {
 //            return executeTelnet();
         } else {
-            throw new RuntimeException("Registry Protocol please use zookeeper or redis!");
+            throw new RuntimeException("Registry Protocol please use zookeeper or nacos or redis!");
         }
     }
 
@@ -85,6 +85,13 @@ private List<String> executeRegistry(String protocol, String address, String gro
                 registry.setAddress(address);
                 reference.setRegistry(registry);
                 break;
+            case Constants.REGISTRY_NACOS:
+                registry = new RegistryConfig();
+                registry.setProtocol(Constants.REGISTRY_NACOS);
+                registry.setGroup(group);
+                registry.setAddress(address);
+                reference.setRegistry(registry);
+                break;
             case Constants.REGISTRY_REDIS:
                 registry = new RegistryConfig();
                 registry.setProtocol(Constants.REGISTRY_REDIS);

src/main/java/io/github/ningyu/jmeter/plugin/util/Constants.java

@@ -28,14 +28,15 @@
  * Constants
  */
 public class Constants {
-	
+
 	//Registry Protocol
 	public static final String REGISTRY_NONE = "none";
 	public static final String REGISTRY_ZOOKEEPER = "zookeeper";
+    public static final String REGISTRY_NACOS = "nacos";
 	public static final String REGISTRY_MULTICAST = "multicast";
 	public static final String REGISTRY_REDIS = "redis";
 	public static final String REGISTRY_SIMPLE = "simple";
-	
+
 	//RPC Protocol
 	public static final String RPC_PROTOCOL_DUBBO = "dubbo";
 	public static final String RPC_PROTOCOL_RMI = "rmi";
@@ -45,10 +46,10 @@ public class Constants {
 	public static final String RPC_PROTOCOL_THRIFT = "thrift";
 	public static final String RPC_PROTOCOL_MEMCACHED = "memcached";
 	public static final String RPC_PROTOCOL_REDIS = "redis";
-	
+
 	public static final String ASYNC = "async";
 	public static final String SYMBOL = "://";
-	
+
 	public static final int INT_DEFAULT = 0;
 	public static final double DOUBLE_DEFAULT = 0.0d;
 	public static final boolean BOOLEAN_DEFAULT = false;