AWS

【AWS】サーバーレスオフラインでDynamoDBに保存したデータを取得する

作成日：2021年12月29日

更新日：2021年12月31日

前回は、サーバーレスオフラインで DynamoDB にデータを保存しました。

【AWS】サーバーレスオフラインでDynamoDBにデータを保存する

今回は、DynamoDB に保存したデータを取得します。

まずは、データを取得するためにハンドラーを作成します。

api フォルダに、get-notes.js を作成します。

前回作成した、create-note.js の内容を流用します。

js
const AWS = require("aws-sdk");
AWS.config.update({ region: "us-west-2" });

const dynamodb = new AWS.DynamoDB.DocumentClient();
const tableName = process.env.NOTES_TABLE;

exports.handler = async (event) => {
  try {
    return {
      statudCode: 200,
      headers: { "Access-Control-Allow-Origin": "*" },
      body: JSON.stringify(item),
    };
  } catch (err) {
    console.log("Error:", err);
    return {
      statusCode: err.statusCode ? err.statusCode : 500,
      headers: { "Access-Control-Allow-Origin": "*" },
      body: JSON.stringify({
        error: err.name ? err.name : "Exception",
        message: err.message ? err.message : "Unknown error",
      }),
    };
  }
};

まずは、eventのquieryStringParametersで、パラメータを取得できるようにします。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;

一度に取得するデータを最大 10 件にします。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;
    let limit = query && query.limit ? parseInt(query.limit) : 10;

headersに指定している、user_idを取得します。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;
    let limit = query && query.limit ? parseInt(query.limit) : 10;
    let user_id = event.headers.user_id;

パラメータにTableNameとKeyConditionExpression、ExpressionAttributeValues、Limit、ScanIndexForwardを設定します。

KeyConditionExpressionは、クエリをじっっこうするキーを指定します。

ExpressionAttributeValuesで、KeyConditionExpressionの値を置換します。

ソートの順番を反転させたいので、ScanIndexForwardを false にします。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;
    let limit = query && query.limit ? parseInt(query.limit) : 10;
    let user_id = event.headers.user_id;

    let params = {
      TableName: tableName,
      KeyConditionExpression: "user_id = :uid",
      ExpressionAttributeValues: {
        ":uid": user_id,
      },
      Limit: limit,
      ScanIndexForward: false,
    };

11 件目からを表示できるように、最後のtimestampをパラメータで取得します。

パラメータがない場合、0 とします。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;
    let limit = query && query.limit ? parseInt(query.limit) : 10;
    let user_id = event.headers.user_id;

    let params = {
      TableName: tableName,
      KeyConditionExpression: "user_id = :uid",
      ExpressionAttributeValues: {
        ":uid": user_id,
      },
      Limit: limit,
      ScanIndexForward: false,
    };

    let timestamp = query && query.start ? parseInt(query.start) : 0;

timestampがある場合、ExclusiveStartKeyを設定します。

ExclusiveStartKeyとは、取得できた最後のパラメータが入っています。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;
    let limit = query && query.limit ? parseInt(query.limit) : 1;
    let user_id = event.headers.user_id;

    let params = {
      TableName: tableName,
      KeyConditionExpression: "user_id = :uid",
      ExpressionAttributeValues: {
        ":uid": user_id,
      },
      Limit: limit,
      ScanIndexForward: false,
    };

    let timestamp = query && query.start ? parseInt(query.start) : 0;

    if (timestamp > 0) {
      params.ExclusiveStartKey = {
        user_id,
        timestamp,
      };
    }

dynamodbのqueryでパラメータを設定します。

js
exports.handler = async (event) => {
  try {
    let query = event.queryStringParameters;
    let limit = query && query.limit ? parseInt(query.limit) : 1;
    let user_id = event.headers.user_id;

    let params = {
      TableName: tableName,
      KeyConditionExpression: "user_id = :uid",
      ExpressionAttributeValues: {
        ":uid": user_id,
      },
      Limit: limit,
      ScanIndexForward: false,
    };

    let timestamp = query && query.start ? parseInt(query.start) : 0;

    if (timestamp > 0) {
      params.ExclusiveStartKey = {
        user_id,
        timestamp,
      };
    }

    let item = await dynamodb.query(params).promise();

ハンドラーが完成したので、serverless.yml の YAML を設定します。

providerのiamRoleStatementsにあるActionに、Queryを設定します。

yaml
provider:
  name: aws
  runtime: nodejs14.x
  region: us-west-2
  stage: dev
  memorySize: 128
  timeout: 5
  endpointType: regional
  lambdaHashingVersion: 20201221
  environment:
    NOTES_TABLE: ${self:service}-${opt:stage, self:provider.stage}
  iamRoleStatements:
    - Effect: Allow
      Action:
        - dynamodb:Query
        - dynamodb:PutItem
      Resource: "arn:aws:dynamodb:${opt:region, self:provider.region}:*:table/${self:provider.environment.NOTES_TABLE}"

functionsにget-notesを設定します。

pathは、notesとします。

methodは、getとします。

yaml
get-notes:
  handler: api/get-notes.handler
  description: GET /notes
  events:
    - http:
        path: notes
        method: get
        cors:
          origin: "*"
          headers: ${self:custom.allowedHeaders}