AWS Lambda+Python3でEC2起動・停止を自動化

aws lambda + python

リンクバル技術部の川畑です。以前こちらの記事(AWS Lambdaを使ってEC2インスタンスを起動・停止する)をご紹介させて頂きました。それから約1年ぐらいが経過しましたが、その間にAWSのLambdaがpython3に対応したり、環境変数に対応したりと便利なアップデートがありスクリプトを修正しましたので、こちらでもご紹介させて頂きます。

目次

AWS Lambdaのスケジュール機能の良いところ

以前の記事でも述べさせて頂きましたが、AWS Lambdaのスケジュール機能には以下のメリットがあると思います。

  • Linux上のcrontabを設定する場合、1台サーバを常時起動しなければならないがAWS Lambdaではサーバ不要(Server Less)
  • 常時起動しているサーバが単一障害点になるが、AWS LambdaではマルチAZで構成され可用性が高い(High Availability)
  • スケジュールがAWSコンソールで確認できる(Standardization)

スクリプト説明

起動・停止の機能は理解のしやすさを考慮し、ファイルを別々にしてあります。スクリプトからAWSを操作するために「botocore」「boto3」、Lambdaに設定した環境変数を読み込むため「os」パッケージをimportします。Lambdaで環境変数「TAG_NAME」を設定し、タグを設定すると一致したEC2インスタンスが処理対象となり、起動・停止処理を実行する仕組みとなります。

使用言語

  • python3.6

EC2起動スクリプト

EC2停止スクリプト

GitHubリポジトリはこちら

AWS Lambda設定手順

次にAWS Lambdaの設定に移ります。AWS Lambdaに関数を登録したり、上記で準備したファイルをアップロードしたりする必要があります。以下はEC2をAWS Lambdaから自動起動する手順となります。

1.スクリプトの準備

  • AWS Lambdaにアップロードするファイルの準備
[bash] git clone https://github.com/makaaso/lambda-ec2-operation.git
cd lambda-ec2-operation
pip install botocore -t ./ –upgrade
pip install boto3 -t ./ –upgrade
zip -r ../lambda-ec2-operation.zip .
[/bash]

2.AWSコンソールにログイン

  • AWSコンソールのLambda画面で「Functions」を選択し、「Create function」を選択

AWS Lambda

3.python3ファンクションの検索

  • Blueprints下のフィルタに「python3」を入力して検索

AWS Lambda

4.python3ファンクションの選択

  • 検索結果から「hello-world-python3」を選択

Lambda Python3

5.functionの作成

  • Nameは適宜入力
  • Roleは作成済みであることを想定しております(※lambdaロール未作成の場合は、適切な権限を付与して作成)
  • Existing roleで適切なLambdaロールを選択
  • すべて入力したら「Create function」を選択

Lambda Python3

6.スクリプトのアップロードと設定

  • Code entry typeで「Upload a .ZIP file」を選択して、zipファイルをアップロード
  • HandlerはEC2起動の場合は「lambda-ec2-start.lambda_handler」、EC2停止の場合は「lambda-ec2-stop.lambda_handler」と入力
  • Environment variablesで、key「TAG_NAME」、valueにタグ名を入力

lambda

7.トリガの設定

  • 「Trigger」を選択して「Add Trigger」を選択

8.トリガの追加

  • 「CloudWatch Event」を選択

lambda trigger

9.CloudWatch Eventの設定

  • Rule nameは適宜設定
  • Rule typeは「Schedule expression」を設定
  • 「Schedule expression」の値はcronフォーマットで設定(※時間はUTC)

cloudwatch event

10.functionの保存

  • 「Save」を選択して登録完了

AWS Lambdaで環境変数が使用できるようになり、環境ごとにスクリプトを修正することなく設定できるのは大変助かります。最近はRDSも停止できるようになったので、RDSの起動・停止スケジュールについても挑戦していきたいと思います。

リンクバルに興味がある方は こちら ご覧ください。