Uso de AWS Lambda con Amazon Aurora DSQL

Inicie sesión en la AWS Management Console y abra la consola AWS Lambda en https://console.aws.amazon.com/lambda/.

Seleccione Creación de función.

Proporcione un nombre, como dsql-sample.

No edite la configuración predeterminada para asegurarse de que Lambda crea un nuevo rol con los permisos de Lambda básicos.

Seleccione Creación de función.

En la función de Lambda, elija Configuración > Permisos.

Elija el nombre del rol para abrir el rol de ejecución en la consola de IAM.

Elija Agregar permisos > Crear política insertada y utilice el editor JSON.

En Acción pegue la siguiente acción para autorizar a la identidad de IAM para que se conecte mediante el rol de base de datos administrador.

"Action": ["dsql:DbConnectAdmin"],

En Recurso, agregue el nombre de recurso de Amazon (ARN) del clúster. También puede utilizar un carácter comodín.

"Resource": ["*"]

Elija Siguiente.

Introduzca un Nombre para la política, como dsql-sample-dbconnect.

Seleccione Crear política.

Cree una carpeta denominada myfunction.

En la carpeta, cree un nuevo archivo llamado package.json con el siguiente contenido.

{
  "dependencies": {
    "@aws-sdk/dsql-signer": "^3.705.0",
    "assert": "2.1.0",
    "pg": "^8.13.1"
  }
}

En la carpeta, cree un archivo denominado index.mjs en el directorio con el siguiente contenido.

import { DsqlSigner } from "@aws-sdk/dsql-signer";
import pg from "pg";
import assert from "node:assert";
const { Client } = pg;


async function dsql_sample(clusterEndpoint, region) {
  let client;
  try {
    // The token expiration time is optional, and the default value 900 seconds
    const signer = new DsqlSigner({
      hostname: clusterEndpoint,
      region,
    });
    const token = await signer.getDbConnectAdminAuthToken();
    // <https://node-postgres.com/apis/client>
    // By default `rejectUnauthorized` is true in TLS options
    // <https://nodejs.org/api/tls.html#tls_tls_connect_options_callback>
    // The config does not offer any specific parameter to set sslmode to verify-full
    // Settings are controlled either via connection string or by setting
    // rejectUnauthorized to false in ssl options
    client = new Client({
      host: clusterEndpoint,
      user: "admin",
      password: token,
      database: "postgres",
      port: 5432,
      // <https://node-postgres.com/announcements> for version 8.0
      ssl: true,
      rejectUnauthorized: false
    });

    // Connect
    await client.connect();

    // Create a new table
    await client.query(`CREATE TABLE IF NOT EXISTS owner (
      id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
      name VARCHAR(30) NOT NULL,
      city VARCHAR(80) NOT NULL,
      telephone VARCHAR(20)
    )`);

    // Insert some data
    await client.query("INSERT INTO owner(name, city, telephone) VALUES($1, $2, $3)", 
      ["John Doe", "Anytown", "555-555-1900"]
    );

    // Check that data is inserted by reading it back
    const result = await client.query("SELECT id, city FROM owner where name='John Doe'");
    assert.deepEqual(result.rows[0].city, "Anytown")
    assert.notEqual(result.rows[0].id, null)

    await client.query("DELETE FROM owner where name='John Doe'");

  } catch (error) {
    console.error(error);
    throw new Error("Failed to connect to the database");
  } finally {
    client?.end();
  }
  Promise.resolve();
}


// https://docs.aws.amazon.com/lambda/latest/dg/nodejs-handler.html
export const handler = async (event) => {
  const endpoint = event.endpoint;
  const region = event.region;
  const responseCode = await dsql_sample(endpoint, region);
  
  const response = {
    statusCode: responseCode,
    endpoint: endpoint,
  };
  return response;
};

Utilice los siguientes comandos para crear un paquete.

npm install
zip -r pkg.zip .

En la pestaña Código de la función de Lambda, elija Cargar desde > Archivo .zip

Cargue el archivo pkg.zip que ha creado. Para obtener más información, consulte Implementación de funciones de Lambda de Node.js con archivos .zip.

En la pestaña Prueba de la función de Lambda, pegue la siguiente carga útil JSON y modifíquela para utilizar el ID de clúster.

En la pestaña Prueba de la función de Lambda, utilice el siguiente Evento JSON modificado para especificar el punto de conexión del clúster.

{"endpoint": "replace_with_your_cluster_endpoint"}

Escriba un nombre de evento, como dsql-sample-test. Seleccione Save.

Seleccione Test (Probar).

Elija Detalles para ampliar la respuesta de ejecución y registrar la salida.

Si se ha realizado correctamente, la respuesta de ejecución de la función de Lambda debe devolver un código de estado 200:

{statusCode": 200, "endpoint": "your_cluster_endpoint"}

Si la base de datos devuelve un error o si no se realiza la conexión con la base de datos, la respuesta de ejecución de la función de Lambda devuelve un código de estado 500.

{"statusCode": 500,"endpoint": "your_cluster_endpoint"}