sr-scripts 0.1.15 → 0.1.16

data/bin/sr-backup-mysql

@@ -6,10 +6,10 @@ require 'fog'
 require 'mysql'
 require 'sr-scripts'
 
-connection = SrScripts::Compute.get
-
 current_instance_id = `curl -s http://169.254.169.254/latest/meta-data/instance-id`
 
+connection = SrScripts::Compute.find_connection(current_instance_id)
+
 server = connection.servers.get(current_instance_id)
 
 disks = server.tags["mysql_disks"].split(":")

data/bin/sr-detect-slaves

@@ -49,7 +49,7 @@ end
 p "Found #{all_servers.length} Running Servers: " + all_servers.map {|s| s.id}.join(", ")
 
 def print_server (server, padding)
-  printf "%#{padding}s: %90s\n", server.id, server.created_at
+  printf "%#{padding}s: %60s %s\n", server.id, server.created_at, server.dns_name
 end
 
 def print_all_slaves (server, all_servers, padding)

data/bin/sr-ec2-consistent-snapshot

@@ -5,8 +5,10 @@
 # A bunch of his features aren't ported over yet... just the stuff I needed.
 
 require 'rubygems'
+$:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
 require 'optparse'
 require 'fog'
+require 'sr-scripts'
 
 $opts = {
   :aws_access_key => ENV["AWS_ACCESS_KEY_ID"],
@@ -49,6 +51,8 @@ if $opts[:aws_access_key].nil? || $opts[:aws_secret_access_key].nil?
   exit 1
 end
 
+current_instance_id = `curl -s http://169.254.169.254/latest/meta-data/instance-id`
+
 if ARGV.empty?
   puts "You must provide at least one volume id to snapshot"
   exit 1
@@ -118,7 +122,7 @@ end
 snapshots = []
 master_info = nil
 
-connection = Fog::Compute.new(:provider => "AWS", :aws_access_key_id => $opts[:aws_access_key], :aws_secret_access_key => $opts[:aws_secret_access_key], :region => $opts[:aws_region])
+connection = SrScripts::Compute.find_connection(current_instance_id)
 sdb = Fog::AWS::SimpleDB.new(:aws_access_key_id => $opts[:aws_access_key], :aws_secret_access_key => $opts[:aws_secret_access_key])
 begin
   mysql_locked() do |info|
@@ -136,7 +140,7 @@ end
 
 random_id = rand(36**8).to_s(36)
 attributes = Hash.new
-attributes["instance_id"] = `curl -s http://169.254.169.254/latest/meta-data/instance-id`
+attributes["instance_id"] = current_instance_id
 attributes["snapshots"] = snapshots.join(",")
 attributes["timestamp"] = Time.now.to_i
 attributes["master_log_file"] = master_info["MASTER_LOG_FILE"]

data/bin/sr-promote-to-master

@@ -1,6 +1,7 @@
 #!/usr/bin/env ruby
 
 require 'rubygems'
+$:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
 require 'fog'
 require 'mysql'
 require 'optparse'
@@ -22,6 +23,9 @@ optparse = OptionParser.new do|opts|
   opts.on( '--mysql-pass PASSWORD', 'Mysql Password') do |o|
     options[:mysql_password] = o
   end
+  opts.on( '--sns-topic-id TOPIC_ID', 'SNS Topic Id To Post To') do |o|
+    options[:sns_topic_id] = o
+  end
   opts.on( '--skip-master', 'Use This Flag To Skip Connecting To --current-master-id') do |o|
     options[:skip_master] = true
   end
@@ -58,18 +62,21 @@ def get_public_ip(fog, instance_id)
 	return fog.servers.get(instance_id).public_ip_address
 end
 
+fog_current_master = connection.servers.get(options[:current_master_id])
+fog_new_master = connection.servers.get(options[:new_master_id])
+
 mysql_user = options[:mysql_user]
 mysql_pass = options[:mysql_password]
 
 current_master_ip = get_public_ip(connection, options[:current_master_id])
 new_master_ip = get_public_ip(connection, options[:new_master_id])
 
-current_master, new_master = nil
+current_master_mysql, new_master_mysql = nil
 
 unless options[:skip_master]
   begin
-    current_master = Mysql.new(current_master_ip, mysql_user, mysql_pass)
-    p current_master.query("SELECT version();")
+    current_master_mysql = Mysql.new(current_master_ip, mysql_user, mysql_pass)
+    p current_master_mysql.query("SELECT version();")
   rescue 
     puts "Error Connecting to --master-instance-id.  Try running with --skip-master"
     exit 1
@@ -77,18 +84,32 @@ unless options[:skip_master]
 end
 
 begin
-  new_master = Mysql.new(new_master_ip, mysql_user, mysql_pass)
+  new_master_mysql = Mysql.new(new_master_ip, mysql_user, mysql_pass)
 rescue
   puts "Error connecting to new master"
   exit 1
 end
 
 unless options[:skip_master]
-  current_master.query("SET GLOBAL read_only = 1")
+  current_master_mysql.query("SET GLOBAL read_only = 1")
   sleep 1 # HACKY CRAP
-  kill_mysql_connections current_master 
+  kill_mysql_connections current_master_mysql 
 end
-new_master.query("STOP SLAVE;")
-new_master.query("SET GLOBAL read_only = 0")
+new_master_mysql.query("STOP SLAVE;")
+new_master_mysql.query("SET GLOBAL read_only = 0")
 
-puts "Script Complete"
+puts "New Master Set"
+private_dns_alias = fog_new_master.tags["master_alias"]
+if private_dns_alias
+  puts "Updating DNS Entries"
+  puts "Creating entry: #{private_dns_alias}  #{fog_new_master.private_ip_address}"
+  SrScripts::HostsManager.new.update_dns(private_dns_alias, fog_new_master.private_ip_address)
+  public_dns_alias = "#{private_dns_alias}.external"
+  puts "Creating entry: #{public_dns_alias}  #{fog_new_master.public_ip_address}"
+  SrScripts::HostsManager.new.update_dns(public_dns_alias, fog_new_master.public_ip_address)
+  if options[:sns_topic_id]
+    puts "POSTING MESSAGE TO SNS TOPIC: #{options[:sns_topic_id]}"
+    sns = SrScripts::SNS.get
+    sns.publish(options[:sns_topic_id], "ReloadHostsFile")
+  end
+end

data/bin/sr-start-slave

@@ -7,22 +7,21 @@ require 'mysql'
 require 'ostruct'
 require 'sr-scripts'
 
-connection = SrScripts::Compute.get
+instance_id = ARGV[0] 
+if ARGV.length != 1
+	p "Must specify an instance id"
+	exit
+end
+
+connection = SrScripts::Compute.find_connection(instance_id)
 sdb = SrScripts::SimpleDB.get
 
 #STUPID HACK because the RPM install starts the server
 `kill -9 $(ps opid= -C mysqld)`
 
 # somewhere up here check to see if the devices are already attached
-
 p connection.snapshots.all.first
 
-if ARGV.length != 1
-	p "Must specify an instance id"
-	exit
-end
-
-instance_id = ARGV[0] 
 
 records = sdb.select("SELECT * FROM db_recovery_info WHERE instance_id = '#{instance_id}' AND timestamp > '1' ORDER BY timestamp DESC LIMIT 5").body["Items"]
 
@@ -63,6 +62,10 @@ end
 
 #latest_snapshots now equals the snapshot ids
 p latest_snapshots
+if latest_snapshots == nil
+  puts "Exiting:  Couldn't Find Any Available Snapshots To Use"
+  exit
+end
 
 def get_latest_snapshot(snapshots, instance_id)
         filtered_snap = snapshots.find_all { |s| s.tags["instance_id"] == instance_id }

data/lib/sr-scripts/version.rb

@@ -1,5 +1,5 @@
 module Sr
   module Scripts
-    VERSION = "0.1.15"
+    VERSION = "0.1.16"
   end
 end

data/lib/sr-scripts.rb

@@ -20,12 +20,26 @@ module SrScripts
     end
   end
 	class Compute 
-		def self.get 
+		def self.get region=nil
+      @region = region || 'us-west-1'
       yml = ConfigFile.get
 			@aws_access_key = yml["aws_access_key"]
 			@aws_secret_key = yml["aws_secret_key"]	
-			return Fog::Compute.new(:provider => "AWS", :aws_access_key_id => @aws_access_key, :aws_secret_access_key => @aws_secret_key, :region => "us-west-1")
+			return Fog::Compute.new(:provider => "AWS", :aws_access_key_id => @aws_access_key, :aws_secret_access_key => @aws_secret_key, :region => @region)
 		end
+    def self.find_connection(instance_id)
+      self.get_regions.each do |region|
+        conn = self.get(region)
+        server = conn.servers.get(instance_id)
+        if server
+          return conn
+        end
+      end
+      puts "Couldn't Find Server With Instance Id: #{instance_id}"
+    end
+    def self.get_regions
+      return ['us-west-1', 'us-west-2', 'us-east-1']
+    end
 	end
 	class SimpleDB
 		def self.get
@@ -43,6 +57,14 @@ module SrScripts
 			return Fog::AWS::SES.new(:aws_access_key_id => @aws_access_key, :aws_secret_access_key => @aws_secret_key)
 		end
 	end
+	class SNS
+		def self.get
+      yml = ConfigFile.get
+			@aws_access_key = yml["aws_access_key"]
+			@aws_secret_key = yml["aws_secret_key"]	
+			return Fog::AWS::SNS.new(:aws_access_key_id => @aws_access_key, :aws_secret_access_key => @aws_secret_key, :region => 'us-east-1')
+		end
+	end
 	class Log
 		def self.get 
 			@log = Logger.new(STDOUT)

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 1
-  - 15
-  version: 0.1.15
+  - 16
+  version: 0.1.16
 platform: ruby
 authors: 
 - Davy Campano
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-03-26 00:00:00 -04:00
+date: 2012-03-30 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency