AWS RDS auto start/stop ruby script

rightaws_runstop_dbinstance.rb

# -*- coding: utf-8 -*-
require 'rubygems'
require 'right_aws'

### 使用例
# ruby rightaws_runstop_dbinstance.rb rds_instance_name db.m1.small security-group parameter_group

### 事前に指定するパラメータ
# * RDSのインスタンス名
# * RDSのインスタンスタイプ
# * RDSのセキュリティグループ
rds_db_instance = ARGV[0]
rds_instance_type = ARGV[1]
rds_security_group = ARGV[2]
rds_parameter_group = ARGV[3]

### スクリプト内に埋め込む設定
# * AWSのリージョン設定
rds_end_point_url = 'ap-north-1.rds.amazonaws.com'

## 引数が正しく設定されているかチェックします
unless ARGV.size == 4
  puts "Usage: #{$0} <rds_db_instance> <rds_instance_type> <rds_security_group> <rds_parameter_group>"
  exit 0
end

## アクセスIDとシークレットアクセスキーを指定します
ACCESS_KEY = 'set your access key'
SECRET_KEY = 'set your secret key'

## RDSのインスターフェースを作成するために認証をします
rds = RightAws::RdsInterface.new(ACCESS_KEY, SECRET_KEY, :default_endpoint_url => rds_end_point_url)

## DB削除時に作成するファイナルスナップショットの名前を作成します
today = Time.now
final_db_snapshot = "#{rds_db_instance}-#{today.strftime('%Y-%m-%d-%H-%M')}"

## DBリストアの元となる最新のスナップショットのリストを作成します
db_snapshots_list = []
rds.describe_db_snapshots.each do |dbsnapshot|
  if rds_db_instance == dbsnapshot[:instance_aws_id]
    db_snapshots_list << dbsnapshot
  end
end

## リストア候補のスナップショットを作成日時でソートします
sorted_db_snapshots_list = db_snapshots_list.sort do |a, b|
  b[:create_time] <=> a[:create_time]
end

## ソートされたスナップショットの候補から最新のスナップショットを特定します
restore_db_snapshot = sorted_db_snapshots_list[0]

## 指定した名前のDBインスタンスが起動している場合はリストに入れときます
check_db_instance = []
rds.describe_db_instances.each do |dbinstance|
  if rds_db_instance == dbinstance[:aws_id]
    check_db_instance << dbinstance
  end
end

## 指定した名前のDBインスタンスが起動していなければ最新のスナップショットからDBをリストアします
## 指定した名前のDBインスタンスが起動していればファイナルスナップショットを作成してDBインスタンスを削除します
if check_db_instance.size == 0
  rds.restore_db_instance_from_db_snapshot("#{restore_db_snapshot[:aws_id]}", rds_db_instance, params={:instance_class => rds_instance_type})
  sleep 500
  rds.modify_db_instance(rds_db_instance, params={:db_security_groups => rds_security_group, :db_parameter_group => rds_parameter_group, :apply_immediately => true})
elsif check_db_instance.size != 0
  rds.delete_db_instance(rds_db_instance, params={:snapshot_aws_id => final_db_snapshot})
end