Enhanced for Nacos weight configuration

sofastack · Mar 26, 2024 · 1842b8f · 1842b8f
1 parent 70f16f3
commit 1842b8f
Show file tree

Hide file tree

Showing 2 changed files with 65 additions and 1 deletion.
diff --git a/.../registry-nacos/src/main/java/com/alipay/sofa/rpc/registry/nacos/NacosRegistryHelper.java b/.../registry-nacos/src/main/java/com/alipay/sofa/rpc/registry/nacos/NacosRegistryHelper.java
@@ -123,6 +123,13 @@ static List<ProviderInfo> convertInstancesToProviders(List<Instance> allInstance
         for (Instance instance : allInstances) {
             String url = convertInstanceToUrl(instance);
             ProviderInfo providerInfo = ProviderHelper.toProviderInfo(url);
+
+            // Nacos的默认权重为1.0
+            // 当Nacos默认权重传入1.0，根据代码逻辑计算结果为100，与sofa-rpc默认权重一致
+            // 因为是接口级别，如果不同的服务具有不同的权重也不会出现被覆盖或者冲突的情况
+            long weight = Math.round(providerInfo.getWeight() * instance.getWeight());
+            providerInfo.setWeight(Math.round(weight));
+
             providerInfos.add(providerInfo);
         }
         return providerInfos;

diff --git a/...istry-nacos/src/test/java/com/alipay/sofa/rpc/registry/nacos/NacosRegistryHelperTest.java b/...istry-nacos/src/test/java/com/alipay/sofa/rpc/registry/nacos/NacosRegistryHelperTest.java
@@ -136,4 +136,61 @@ public void buildServiceName() {
         serviceName = NacosRegistryHelper.buildServiceName(provider, RpcConstants.PROTOCOL_TYPE_REST);
         assertEquals(serviceName, "com.alipay.xxx.TestService:nacos-test:" + RpcConstants.PROTOCOL_TYPE_REST);
     }
-}
+
+    @Test
+    public void testNacosWeight() {
+        Instance instance = new Instance();
+        instance.setClusterName(NacosRegistryHelper.DEFAULT_CLUSTER);
+        instance.setIp("1.1.1.1");
+        instance.setPort(12200);
+        instance.setServiceName("com.alipay.xxx.TestService");
+        instance.setWeight(0.99D);
+
+        List<ProviderInfo> providerInfos = NacosRegistryHelper
+            .convertInstancesToProviders(Lists.newArrayList(instance));
+        assertNotNull(providerInfos);
+        assertEquals(1, providerInfos.size());
+
+        ProviderInfo providerInfo = providerInfos.get(0);
+        assertNotNull(providerInfo);
+        assertEquals(99, providerInfo.getWeight());
+    }
+
+    @Test
+    public void testNacosWeightLessThan0() {
+        Instance instance = new Instance();
+        instance.setClusterName(NacosRegistryHelper.DEFAULT_CLUSTER);
+        instance.setIp("1.1.1.1");
+        instance.setPort(12200);
+        instance.setServiceName("com.alipay.xxx.TestService");
+        instance.setWeight(-0.1D);
+
+        List<ProviderInfo> providerInfos = NacosRegistryHelper
+            .convertInstancesToProviders(Lists.newArrayList(instance));
+        assertNotNull(providerInfos);
+        assertEquals(1, providerInfos.size());
+
+        ProviderInfo providerInfo = providerInfos.get(0);
+        assertNotNull(providerInfo);
+        assertEquals(-10, providerInfo.getWeight());
+    }
+
+    @Test
+    public void testNacosWeightWith0() {
+        Instance instance = new Instance();
+        instance.setClusterName(NacosRegistryHelper.DEFAULT_CLUSTER);
+        instance.setIp("1.1.1.1");
+        instance.setPort(12200);
+        instance.setServiceName("com.alipay.xxx.TestService");
+        instance.setWeight(0.0D);
+
+        List<ProviderInfo> providerInfos = NacosRegistryHelper
+            .convertInstancesToProviders(Lists.newArrayList(instance));
+        assertNotNull(providerInfos);
+        assertEquals(1, providerInfos.size());
+
+        ProviderInfo providerInfo = providerInfos.get(0);
+        assertNotNull(providerInfo);
+        assertEquals(0, providerInfo.getWeight());
+    }
+}