Amazon ElastiCache(Memcached)をPython3.7から操作 | AWSやシステム・アプリ開発

こんにちは。s.kです。今回はPython 3.6を使ってAmazon ElastiCache(Memcached)への通信を行い、一通りの動作を試したいと思います。 *今回はLambdaからElastiCacheへの通信を行います。

概要

"action"、"cache_key"、"cache_value"三つの引数を渡し、Memcachedを操作するLambdaをPythonで作成します。

標準ライブラリ以外も必要となるため今回はzipアップロードとなります。

また今回使用するバージョンはこの通りです。

Python v3.6
ElastiCache Memcached v1.5.16
elasticache_auto_discovery v1.0.0
pymemcache v2.2.2

事前準備

Python用のMemcachedクライアントであるpymemcacheと、ノードをよしなにしてくれる"elasticache_auto_discovery"を下記サイトからダウンロードします。
pymemcache(v2.2.2)
github：https://github.com/pinterest/pymemcache
pypi：https://pypi.org/project/pymemcache/

elasticache_auto_discovery(v1.0.0)
github：https://github.com/djmitche/elasticache-auto-discovery
次にlambda_function.pyファイルを作成します。

# coding=utf-8

import os
import elasticache_auto_discovery
from pymemcache.client.hash import HashClient

endpoint = os.environ.get['ENDPOINT']

def lambda_handler(event, context):
    
    # ElastiCache settings
    nodes = elasticache_auto_discovery.discover(endpoint)
    nodes = map(lambda x: (x[1], int(x[2])), nodes)
    memcache_client = HashClient(nodes)
    
    if 'set' in event['action']:
        result = memcache_client.set(event['cache_key'], event['cache_value'])
    
    elif 'get' in event['action']:
        result = memcache_client.get(event['cache_key'])
        result = &quot;no exist&quot; if result is None else result.decode()
    
    elif 'delete' in event['action']:
        result = memcache_client.delete(event['cache_key'])
    
    print(result)
    return result

これらを以下の構成で圧縮します。

elasticache.zip
├─elasticache_auto_discovery
| └─__init__.py
├─pymemcache
| ├─client/
| ├─test/
| ├─__init__.py
| ├─exceptions.py
| ├─fallback.py
| ├─pool.py
| └─serde.py
└─lambda_function.py

AWS構築

最初にElastiCacheとLambdaで使用するセキュリティグループを作成します。今回は互いのセキュリティグループをリソースに指定して許可する形にする為、お互いに相手側のセキュリティグループIDを設定する。
・Lambda用セキュリティグループ
インバウンド　[設定なし]
アウトバウンド　[カスタム TCP ルール：TCP：11211：ElastiCache用セキュリティグループID]