Example 16 with Result
use of com.baidu.hugegraph.structure.gremlin.Result in project incubator-hugegraph-toolchain by apache.
the class HugeClientUtil method tryConnect.
public static HugeClient tryConnect(GraphConnection connection) {
String graph = connection.getGraph();
String host = connection.getHost();
Integer port = connection.getPort();
String username = connection.getUsername();
String password = connection.getPassword();
int timeout = connection.getTimeout();
String protocol = connection.getProtocol() == null ? DEFAULT_PROTOCOL : connection.getProtocol();
String trustStoreFile = connection.getTrustStoreFile();
String trustStorePassword = connection.getTrustStorePassword();
String url = UriComponentsBuilder.newInstance().scheme(protocol).host(host).port(port).toUriString();
if (username == null) {
username = "";
password = "";
}
HugeClient client;
try {
client = HugeClient.builder(url, graph).configUser(username, password).configTimeout(timeout).configSSL(trustStoreFile, trustStorePassword).build();
} catch (IllegalStateException e) {
String message = e.getMessage();
if (message != null && message.startsWith("The version")) {
throw new ExternalException("client-server.version.unmatched", e);
}
if (message != null && (message.startsWith("Error loading trust store from") || message.startsWith("Cannot find trust store file"))) {
throw new ExternalException("https.load.truststore.error", e);
}
throw e;
} catch (ServerException e) {
String message = e.getMessage();
if (Constant.STATUS_UNAUTHORIZED == e.status() || (message != null && message.startsWith("Authentication"))) {
throw new ExternalException("graph-connection.username-or-password.incorrect", e);
}
if (message != null && message.contains("Invalid syntax for " + "username and password")) {
throw new ExternalException("graph-connection.missing-username-password", e);
}
throw e;
} catch (ClientException e) {
Throwable cause = e.getCause();
if (cause == null || cause.getMessage() == null) {
throw e;
}
String message = cause.getMessage();
if (message.contains("Connection refused")) {
throw new ExternalException("service.unavailable", e, host, port);
} else if (message.contains("java.net.UnknownHostException") || message.contains("Host name may not be null")) {
throw new ExternalException("service.unknown-host", e, host);
} else if (message.contains("<!doctype html>")) {
throw new ExternalException("service.suspected-web", e, host, port);
}
throw e;
}
try {
ResultSet rs = client.gremlin().gremlin("g.V().limit(1)").execute();
rs.iterator().forEachRemaining(Result::getObject);
} catch (ServerException e) {
if (Constant.STATUS_UNAUTHORIZED == e.status()) {
throw new ExternalException("graph-connection.username-or-password.incorrect", e);
}
String message = e.message();
if (message != null && message.contains("Could not rebind [g]")) {
throw new ExternalException("graph-connection.graph.unexist", e, graph, host, port);
}
if (!isAcceptable(message)) {
throw e;
}
} catch (Exception e) {
client.close();
throw e;
}
return client;
}
Also used :
HugeClient(com.baidu.hugegraph.driver.HugeClient)
ServerException(com.baidu.hugegraph.exception.ServerException)
ResultSet(com.baidu.hugegraph.structure.gremlin.ResultSet)
ClientException(com.baidu.hugegraph.rest.ClientException)
ExternalException(com.baidu.hugegraph.exception.ExternalException)
ServerException(com.baidu.hugegraph.exception.ServerException)
ExternalException(com.baidu.hugegraph.exception.ExternalException)
ClientException(com.baidu.hugegraph.rest.ClientException)
Result(com.baidu.hugegraph.structure.gremlin.Result)
Aggregations
Result (com.baidu.hugegraph.structure.gremlin.Result)16
Vertex (com.baidu.hugegraph.structure.graph.Vertex)10
ResultSet (com.baidu.hugegraph.structure.gremlin.ResultSet)9
Test (org.junit.Test)9
Edge (com.baidu.hugegraph.structure.graph.Edge)8
RestResult (com.baidu.hugegraph.rest.RestResult)6
Path (com.baidu.hugegraph.structure.graph.Path)6
Response (com.baidu.hugegraph.structure.gremlin.Response)6
ArrayList (java.util.ArrayList)5
HugeClient (com.baidu.hugegraph.driver.HugeClient)4
GremlinResult (com.baidu.hugegraph.entity.query.GremlinResult)4
TypedResult (com.baidu.hugegraph.entity.query.TypedResult)4
GremlinRequest (com.baidu.hugegraph.api.gremlin.GremlinRequest)3
HashMap (java.util.HashMap)3
SchemaManager (com.baidu.hugegraph.driver.SchemaManager)2
GraphView (com.baidu.hugegraph.entity.query.GraphView)2
Type (com.baidu.hugegraph.entity.query.GremlinResult.Type)2
ExternalException (com.baidu.hugegraph.exception.ExternalException)2
ServerException (com.baidu.hugegraph.exception.ServerException)2
ImmutableMap (com.google.common.collect.ImmutableMap)2
