From 687ab22a6ae93f4dab678f7182c04b2d1ce6b208 Mon Sep 17 00:00:00 2001 From: Valentyn Kahamlyk Date: Mon, 20 May 2024 19:34:35 -0700 Subject: [PATCH] extend ConnectionHelper --- .../gremlin/console/jsr223/ConnectionHelper.java | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/gremlin-console/src/main/java/org/apache/tinkerpop/gremlin/console/jsr223/ConnectionHelper.java b/gremlin-console/src/main/java/org/apache/tinkerpop/gremlin/console/jsr223/ConnectionHelper.java index 99234f0a71..9b15884e9f 100644 --- a/gremlin-console/src/main/java/org/apache/tinkerpop/gremlin/console/jsr223/ConnectionHelper.java +++ b/gremlin-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)); + } }