Conexión a bases de datos de Amazon Neptune mediante IAM con Java de Gremlin

Uso de TinkerPop 3.4.11 o posterior para conectarse a Neptune con la firma Sig4

A continuación, se ofrece un ejemplo de cómo conectarse a Neptune usando la API de Java de Gremlin con firma Sig4 cuando se usa TinkerPop 3.4.11 o posterior (se supone que se tienen conocimientos generales sobre el uso de Maven). En este ejemplo, se utiliza la biblioteca Amazon Neptune SigV4 Signer para facilitar la firma de solicitudes. Primero, defina las dependencias como parte del archivo pom.xml:

<dependency>
  <groupId>com.amazonaws</groupId>
  <artifactId>amazon-neptune-sigv4-signer</artifactId>
  <version>3.1.0</version>
</dependency>

Amazon Neptune SigV4 Signer admite el uso de las versiones 1.x y 2.x del AWS SDK para Java. En el siguiente ejemplo, se utiliza la versión 2.x, en la que DefaultCredentialsProvider es una software.amazon.awssdk.auth.credentials.AwsCredentialsProvider instancia, pero también se podría utilizar la versión 1.x con cualquier com.amazonaws.auth.AWSCredentialsProvider. Si va a actualizar de la versión 1.x a la 2.x, puede obtener más información sobre los cambios entre ambas versiones en Cambios en el proveedor de credenciales de la documentación del AWS SDK para Java 2.x.

import com.amazonaws.auth.DefaultAWSCredentialsProviderChain;
import com.amazonaws.neptune.auth.NeptuneNettyHttpSigV4Signer;
import com.amazonaws.neptune.auth.NeptuneSigV4SignerException;

 ...

System.setProperty("aws.accessKeyId","your-access-key");
System.setProperty("aws.secretKey","your-secret-key");

 ...

Cluster cluster = Cluster.build((your cluster))
                 .enableSsl(true)
                 .requestInterceptor( r ->
                  {
                    try {
                      NeptuneNettyHttpSigV4Signer sigV4Signer =
                        new NeptuneNettyHttpSigV4Signer("(your region)", DefaultCredentialsProvider.create());
                      sigV4Signer.signRequest(r);
                    } catch (NeptuneSigV4SignerException e) {
                      throw new RuntimeException("Exception occurred while signing the request", e);
                    }
                    return r;
                  }
                 ).create();
try {
  Client client = cluster.connect();
  client.submit("g.V().has('code','IAD')").all().get();
} catch (Exception e) {
  throw new RuntimeException("Exception occurred while connecting to cluster", e);
}

Autenticación de IAM entre cuentas

Amazon Neptune admite la autenticación de IAM entre cuentas mediante el uso de la asunción de roles, también conocida como encadenamiento de roles. Para proporcionar acceso a un clúster de Neptune desde una aplicación alojada en una cuenta de AWS diferente:

Uso de una versión de TinkerPop anterior a la 3.4.11 para conectarse a Neptune con la firma Sig4

Las versiones de TinkerPop anteriores a la 3.4.11 no eran compatibles con la configuración de requestInterceptor() que se muestra en la sección anterior y, por lo tanto, deben confiar en el paquete amazon-neptune-gremlin-java-sigv4. Esta es una biblioteca de Neptune que contiene la clase SigV4WebSocketChannelizer, que reemplaza al canalizador de TinkerPop estándar por uno que puede inyectar automáticamente una firma SigV4. Siempre que sea posible, actualice a TinkerPop 3.4.11 o una versión posterior, ya que la biblioteca amazon-neptune-gremlin-java-sigv4 está en desuso.

A continuación, se ofrece un ejemplo de cómo conectarse a Neptune mediante la API de Java de Gremlin con firma Sig4 cuando se utilizan versiones de TinkerPop anteriores a la 3.4.11 (se supone que se tienen conocimientos generales sobre cómo usar Maven).

Primero, defina las dependencias como parte del archivo pom.xml:

<dependency>
  <groupId>com.amazonaws</groupId>
  <artifactId>amazon-neptune-gremlin-java-sigv4</artifactId>
  <version>2.4.0</version>
</dependency>

La dependencia anterior incluirá la versión del controlador de Gremlin 3.4.10. Aunque es posible utilizar versiones más recientes del controlador de Gremlin (hasta la 3.4.13), la actualización del controlador a partir de la versión 3.4.10 debería incluir un cambio para utilizar el modelo de requestInterceptor() descrito anteriormente.

A continuación, el objeto de clúster gremlin-driver debe configurarse de la siguiente manera en el código Java:

import org.apache.tinkerpop.gremlin.driver.SigV4WebSocketChannelizer;

 ...

Cluster cluster = Cluster.build(your cluster)
                       .enableSsl(true)
                       .channelizer(SigV4WebSocketChannelizer.class)
                       .create();
Client client = cluster.connect();
client.submit("g.V().has('code','IAD')").all().get();