More than 1 year has passed since last update.

AWS LambdaでPostgreSQLのRDSプロキシを使用する方法 - Node.js

Posted at 2022-07-25

はいさい！ちゅらデータぬオースティンやいびーん。

概要

RDSデータベースへの接続を管理してくれるRDSプロキシを介して、AWSのLambda関数でPostgreSQLのデータベースに接続する方法を紹介します。

今回は、Node.js用のLambda関数を使います。

背景

Lambda関数でコネクションを張ると、パフォーマンスの問題が大きく関わってくるので、直接に接続するのは避けた方がいいです。

そこで、RDSプロキシを使用すると、Lambda関数の中で効率よくデータベースにアクセスできます。

前回の記事で、PostgreSQLのRDSプロキシの作り方を紹介しましたが、それを元に今回の記事を書いていきます。

Node.jsのLambda関数のソースコードを書く

ローカルで作成したLambda関数のソースコードを後でアップロードしますが、まずは必要なファイルを用意しておきます。

`pg`をインストールする

今回のプロジェクトでは、pgというnpmパッケージが必要なので、まずは、任意のダイレクトリに入って、yarn initでpackage.jsonを作ります。

cd my-project
yarn init

そして、pgをインストールする

yarn add pg

`index.js`を追加し、ロジックを書く

Lambda関数のnode.jsのテンプレートを元にpgを使ったソースコードを書きます。

index.js

const pg = require("pg");
const AWS = require("aws-sdk");

exports.handler = async (event) => {
  const signer = new AWS.RDS.Signer({
    region: "ap-northeast-1", // RDSプロキシを作ったRegionに合わせる
    hostname: process.env.HOST, // 後々環境変数で設定する
    port: 5432,
    username: "master", // 前回の記事で作成したSecretのusername
  });

  const token = signer.getAuthToken({
    username: "master",
  });

  const dbConfig = {
    user: "master",
    password: token,
    port: 5432,
    database: "dev", // データベース作成時に指定したもの
    host: process.env.HOST,
    ssl: true,
  };

  const client = new pg.Client(dbConfig);

  client.connect();

  const res = await client.query("SELECT NOW();");

  await client.end();

  const response = {
    statusCode: 200,
    body: JSON.stringify(res.rows[0]),
  };
  return response;
};