gbnvNt0Y

1require 'rubygems'
2require 'right_aws'
3require 'nkf'
4require 'net/smtp'
5
6# このスクリプトでジョブの結果をメールで通知する場合はローカル環境にsmtpサーバが必要です
7# 動作確認に使用した環境は以下の通りです
8# * ruby 1.8.7
9# * postfix 2.3.3
10# スクリプトを使用する場合は以下の順番で引数を指定して下さい
11# 1. AWSリージョン
12# 2. 保存世代
13# 3. スナップショットを取得するvolume-id
14# 4. スナップショットのdescriptionに使用する名前(英語)
15# 5. 転送先e-mailアドレス
16# スクリプトを使用する場合は以下の文字列を指定して下さい
17# * ACCESS_KEY
18# * SECRET_KEY
19# * from_mail_addr
20
21# 定数設定
22ACCESS_KEY = 'SET UP YOUR ACCESS KEY'
23SECRET_KEY = 'SET UP YOUR SECRET KEY'
24
25# 引数チェック
26unless ARGV.size == 5 then
27  puts '使い方: create_snapshot.rb <aws_region> <num> <volume_id> <snapshot_memo> <email_address>'
28  exit 0
29end
30
31# 引数からスナップショット保存の設定をする
32aws_region = ARGV[0]
33history = ARGV[1].to_i                                                        # EBSスナップショットを何世代残しておくか
34vol_id = ARGV[2]                                                              # EBSスナップショットを取得するEBSボリュームID
35snap_name = ARGV[3]                                                           # EBSスナップショットのdescriptionにつける名前
36to_mail_addr = ARGV[4]                                                        # バックアップ完了通知先アドレス
37from_mail_addr = 'noreply@from_addr.com'                                      # バックアップ完了通知元アドレス
38subject = NKF.nkf("-wjm0", 'スナップショット通知メール')                      # スナップショット通知メールタイトル
39failure_message = NKF.nkf("-wjm0", 'スナップショットの作成に失敗しました。')  # バックアップ失敗通知メールの本文
40success_message = NKF.nkf("-wjm0", '完了しました。スナップショットのリストは以下の通りです。')  # バックアップ成功通知メールの本文
41
42# スナップショットのdescriptionを作成
43t = Time.now
44snapshot_description = "#{snap_name} #{t.strftime('%Y/%m/%d %H:%M')}"
45
46# EC2のインターフェースを作成するための認証をします
47ec2 = RightAws::Ec2.new(ACCESS_KEY, SECRET_KEY, :region => "#{aws_region}")
48
49# スナップショットバックアップの処理
50begin
51    # 対象のEBSボリュームからスナップショットを作成します
52    ec2.create_snapshot(vol_id, snapshot_description)
53# スナップショット処理に失敗が発生した場合の処理
54rescue => ex
55# 失敗時のメッセージを作成する
56data = <<EOF
57Subject: #{subject}
58
59#{failure_message}
60EOF
61    # バックアップ失敗通知メールを送信する
62    Net::SMTP.start('localhost', 25){ |smtp| smtp.sendmail data, from_mail_addr, to_mail_addr }
63    exit
64end
65# 保存しているスナップショットで、目的のIDを持つものをリスト化する
66snapshots_list = []
67ec2.describe_snapshots.each do |snapshot|
68    if vol_id == snapshot[:aws_volume_id]
69        snapshots_list << snapshot
70    end
71end
72    
73# 履歴分を残して古いスナップショットを削除
74snapshots_list = snapshots_list.sort_by {|snapshot| snapshot[:aws_started_at] }.reverse
75snapshots_list.each_with_index do |snapshot, i|
76    unless i < history
77        ec2.delete_snapshot(snapshot[:aws_id])
78    end
79end
80    
81# ジョブ完了後、保存しているスナップショットで、目的のIDを持つものをリスト化する
82new_snapshots_list = []
83ec2.describe_snapshots.each do |snapshot|
84    if vol_id == snapshot[:aws_volume_id]
85       new_snapshots_list << snapshot[:aws_description]
86       new_snapshots_list << snapshot[:aws_id]
87       new_snapshots_list << snapshot[:aws_status]
88       new_snapshots_list << "\n"
89    end
90end
91    
92# バックアップ完了通知メールの本文を作成
93data = <<EOF
94Subject: #{subject}
95
96#{success_message}
97#{new_snapshots_list}
98EOF
99# 現在のスナップショットリストを本文にした完了通知メールを送信する
100Net::SMTP.start('localhost', 25){|smtp| smtp .sendmail data, from_mail_addr, to_mail_addr }
101exit