feat:support setting rule based router failover type in service metadata. (#677)

Signed-off-by: Haotian Zhang <[email protected]>

Author: SkyeBeFreeman

polaris-plugins/polaris-plugins-router/router-metadata/src/main/java/com/tencent/polaris/plugins/router/metadata/MetadataRouter.java

@@ -17,6 +17,7 @@
 
 package com.tencent.polaris.plugins.router.metadata;
 
+import com.tencent.polaris.annonation.JustForTest;
 import com.tencent.polaris.api.config.consumer.ServiceRouterConfig;
 import com.tencent.polaris.api.config.plugin.PluginConfigProvider;
 import com.tencent.polaris.api.config.verify.Verifier;
@@ -78,20 +79,6 @@ public class MetadataRouter extends AbstractServiceRouter implements PluginConfi
 
     @Override
     public RouteResult router(RouteInfo routeInfo, ServiceInstances instances) throws PolarisException {
-        FailOverType failOverType = config.getMetadataFailOverType();
-        Map<String, String> svcMetadata = instances.getMetadata();
-        if (MapUtils.isNotEmpty(svcMetadata)) {
-            if (svcMetadata.containsKey(KEY_METADATA_FAILOVER_TYPE)) {
-                String value = svcMetadata.get(KEY_METADATA_FAILOVER_TYPE);
-                if (valueToFailoverType.containsKey(value)) {
-                    failOverType = valueToFailoverType.get(value);
-                }
-            }
-        }
-        MetadataFailoverType metadataFailoverType = routeInfo.getMetadataFailoverType();
-        if (null != metadataFailoverType) {
-            failOverType = inputToFailoverType.get(metadataFailoverType);
-        }
         boolean availableInsFlag;
         Map<String, String> reqMetadata = getRouterMetadata(routeInfo);
         List<Instance> instanceList = new ArrayList<>();
@@ -113,6 +100,25 @@ public RouteResult router(RouteInfo routeInfo, ServiceInstances instances) throw
         if (!CollectionUtils.isEmpty(instanceList)) {
             return new RouteResult(instanceList, RouteResult.State.Next);
         }
+        MetadataFailoverType metadataFailoverType = routeInfo.getMetadataFailoverType();
+        FailOverType failOverType = null;
+        if (null != metadataFailoverType) {
+            failOverType = inputToFailoverType.get(metadataFailoverType);
+        }
+        if (failOverType == null) {
+            Map<String, String> svcMetadata = instances.getMetadata();
+            if (MapUtils.isNotEmpty(svcMetadata)) {
+                if (svcMetadata.containsKey(KEY_METADATA_FAILOVER_TYPE)) {
+                    String value = svcMetadata.get(KEY_METADATA_FAILOVER_TYPE);
+                    if (valueToFailoverType.containsKey(value)) {
+                        failOverType = valueToFailoverType.get(value);
+                    }
+                }
+            }
+        }
+        if (failOverType == null) {
+            failOverType = config.getMetadataFailOverType();
+        }
         switch (failOverType) {
             case all:
                 return new RouteResult(instances.getInstances(), RouteResult.State.Next);
@@ -164,6 +170,11 @@ public Class<? extends Verifier> getPluginConfigClazz() {
         return MetadataRouterConfig.class;
     }
 
+    @JustForTest
+    void setConfig(MetadataRouterConfig config) {
+        this.config = config;
+    }
+
     public MetadataRouterConfig getConfig() {
         return config;
     }