Regard permission denied as legal when create graph connection (#251)

hugegraph · Nov 11, 2020 · eb21cec · eb21cec
1 parent 1b112c2
commit eb21cec
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/hubble-be/src/main/java/com/baidu/hugegraph/util/HugeClientUtil.java b/hubble-be/src/main/java/com/baidu/hugegraph/util/HugeClientUtil.java
@@ -19,6 +19,8 @@
 
 package com.baidu.hugegraph.util;
 
+import java.util.Set;
+
 import org.springframework.web.util.UriComponentsBuilder;
 
 import com.baidu.hugegraph.common.Constant;
@@ -29,11 +31,16 @@
 import com.baidu.hugegraph.rest.ClientException;
 import com.baidu.hugegraph.structure.gremlin.Result;
 import com.baidu.hugegraph.structure.gremlin.ResultSet;
+import com.google.common.collect.ImmutableSet;
 
 public final class HugeClientUtil {
 
     private static final String DEFAULT_PROTOCOL = "http";
 
+    private static final Set<String> ACCEPTABLE_EXCEPTIONS = ImmutableSet.of(
+            "Permission denied: execute Resource"
+    );
+
     public static HugeClient tryConnect(GraphConnection connection) {
         String graph = connection.getGraph();
         String host = connection.getHost();
@@ -105,11 +112,25 @@ public static HugeClient tryConnect(GraphConnection connection) {
                 throw new ExternalException("graph-connection.graph.unexist", e,
                                             graph, host, port);
             }
-            throw e;
+            if (!isAcceptable(message)) {
+                throw e;
+            }
         } catch (Exception e) {
             client.close();
             throw e;
         }
         return client;
     }
+
+    private static boolean isAcceptable(String message) {
+        if (message == null) {
+            return false;
+        }
+        for (String acceptableMessage : ACCEPTABLE_EXCEPTIONS) {
+            if (message.contains(acceptableMessage)) {
+                return true;
+            }
+        }
+        return false;
+    }
 }