DynamoDBでランダムに出力するオミクジ的なサイトを作る #lambda

ランダムで表示するサイトを作るのが好きなのだが、前に作ったGoogle SpreadsheetとかCloud Functionsは遅いのでDynamoDBで作り変えてみた。

まず、DynamoDBで適当なテーブル（ここではxxxx）を作成する。プライマリパーティションキーはid（数値）にしている。ソートキーは設定しない。

ローカルでurl_list.txtというファイルから1行ずつ読み込み、DynamoDBに流し込む。

import boto3
import json
from boto3.dynamodb.conditions import Key, Attr

dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table('xxxx')

def dynamo_update(url):
    res = table.scan(
        FilterExpression=Attr('url').eq(url)
    )
    if res["Count"] == 0: # まだ未登録のURLであればput_itemする
        res = table.scan()
        cnt = res["Count"]
        table.put_item(Item={"id":cnt+1,"url":url})

with open("url_list.txt") as f:
    lines = f.readlines()
    for line in lines:
        data = line.rstrip()
        dynamo_update(data)
f.close()

読み出す方はLambdaで作成する。テーブルから件数を取得して、その件数からランダムな数値を取得して、get_itemする。

import boto3
from boto3.dynamodb.conditions import Key, Attr
import random

def lambda_handler(event, context):

    dynamodb = boto3.resource('dynamodb')
    table = dynamodb.Table('xxxx')
    res = table.scan()
    cnt = res["Count"]
    rnd = random.randrange(cnt)

    res = table.get_item(Key={
        "id":rnd
    })

    return {
        'id':rnd,
        'url': res['Item']['url']
    }

これをAPI GatewayからアクセスできるようにしてJavaScriptから呼び出す。今回、このHTMLファイルもS3に置いたのだが、S3でもアクセス権限でCORSRuleを書かなければいけない。

<!DOCTYPE html>
<html>
  <head>
    <meta charset="UTF-8" />
    <title>unknown protocol</title>
  </head>
  <body>
    <script>
    const url = "https://xxxx.execute-api.ap-northeast-1.amazonaws.com/default/getRandomFromNote";
    fetch(url)
        .then(function(response) {
            return response.json();
        })
        .then(function(json) {
            location.href = json["url"];
        });
    </script>
  </body>
</html>

DynamoDBから件数を取得し、ランダムで表示するという仕様だが、DynamoDBから途中のデータを削除するとどうなるんだっけ？といった詰めはできてないので、あくまでサンプルということで。

肝心のパフォーマンス比較。Google Spreadsheetで作った似たような処理が1.5秒ぐらい。

それに対してDynamoDB+Lambda+S3が1秒ぐらい。

htmlファイルが最初にロードされる分、体感的には、もっと早い気がする。