extend ConnectionHelper

apache · May 21, 2024 · 687ab22 · 687ab22
1 parent 6d984a5
commit 687ab22
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/...n-console/src/main/java/org/apache/tinkerpop/gremlin/console/jsr223/ConnectionHelper.java b/...n-console/src/main/java/org/apache/tinkerpop/gremlin/console/jsr223/ConnectionHelper.java
@@ -19,23 +19,35 @@
 package org.apache.tinkerpop.gremlin.console.jsr223;
 
 import org.apache.tinkerpop.gremlin.driver.Cluster;
+import org.apache.tinkerpop.gremlin.driver.auth.Auth;
 import org.apache.tinkerpop.gremlin.driver.remote.DriverRemoteConnection;
 
 import static org.apache.tinkerpop.gremlin.process.traversal.AnonymousTraversalSource.traversal;
 
 public class ConnectionHelper {
+    private static final String DEFAULT_G = "g";
+    private static final int DEFAULT_PORT = 8182;
+
     private ConnectionHelper() {
     }
 
     public static Object connect(final String host) {
-        return connect(host, "g");
+        return connect(host, DEFAULT_G);
     }
 
     public static Object connect(final String host, final String g) {
-        return connect(host, 8182, g);
+        return connect(host, DEFAULT_PORT, g);
+    }
+
+    public static Object connect(final String host, final Auth auth) {
+        return connect(host, DEFAULT_PORT, DEFAULT_G, auth);
     }
 
     public static Object connect(final String host, final int port, final String g) {
         return traversal().with(DriverRemoteConnection.using(Cluster.build(host).port(port).create(), g));
     }
+
+    public static Object connect(final String host, final int port, final String g, final Auth auth) {
+        return traversal().with(DriverRemoteConnection.using(Cluster.build(host).port(port).auth(auth).create(), g));
+    }
 }