theworkinggroup-mysql-replication-helper 0.1.0 → 0.2.0

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
 :major: 0
-:minor: 1
+:minor: 2
 :patch: 0

data/lib/mysql_replication_helper.rb

@@ -1,6 +1,10 @@
+require 'rubygems'
+require 'mysql'
+
 module MysqlReplicationHelper
   # == Autoloads ============================================================
   
+  autoload(:Agent, 'mysql_replication_helper/agent')
   autoload(:ErrorHandler, 'mysql_replication_helper/error_handler')
   autoload(:Daemon, 'mysql_replication_helper/daemon')
 end

data/lib/mysql_replication_helper/agent.rb

@@ -0,0 +1,51 @@
+module MysqlReplicationHelper
+  class Agent
+    autoload(:Master, 'mysql_replication_helper/agent/master')
+    autoload(:Slave, 'mysql_replication_helper/agent/slave')
+
+    DEFAULT_OPTIONS = {
+      :host => 'localhost',
+      :user => 'root',
+      :master_socket => '/local/db/mysql.sock',
+      :slave_socket => '/ebs/db/mysql.sock'
+    }
+    
+    def initialize(options)
+      @options = with_default_options(options)
+    end
+    
+    def connection
+      @connection ||=
+        Mysql.real_connect(
+          @options[:host],
+          user_name,
+          @options[:password],
+          @options[:db],
+          @options[:port],
+          socket_name
+        )
+    end
+    
+    def user_name
+      @options[:user]
+    end
+    
+    def socket_name
+      @options[:socket]
+    end
+    
+    def with_default_options(options)
+      DEFAULT_OPTIONS.merge(options)
+    end
+    
+    def execute(statement)
+      STDERR.puts(statement)
+      connection.real_query(statement)
+    end
+    
+    def query(statement)
+      STDERR.puts(statement)
+      connection.query(statement)
+    end
+  end
+end

data/lib/mysql_replication_helper/agent/master.rb

@@ -0,0 +1,28 @@
+module MysqlReplicationHelper
+  class Agent
+    class Master < Agent
+      def poll!
+        # Nothing yet
+      end
+      
+      def master_status
+        row = query("SHOW MASTER STATUS").fetch_row
+        
+        return unless (row)
+        
+        {
+          :master_log_file => row[0],
+          :master_log_position => row[1].to_i
+        }
+      end
+      
+      def user_name
+        @options[:master_user] or super
+      end
+
+      def socket_name
+        @options[:master_socket] or super
+      end
+    end
+  end
+end

data/lib/mysql_replication_helper/agent/slave.rb

@@ -0,0 +1,68 @@
+module MysqlReplicationHelper
+  class Agent
+    class Slave < Agent
+      include MysqlReplicationHelper::ErrorHandler
+
+      def poll!
+        if (configured?)
+          if (error_message = slave_error)
+            if (statements = sql_to_recover_from(error_message))
+              statements.each do |sql|
+                connection.real_query(sql)
+              end
+            else
+              # Unrecoverable error?
+            end
+          end
+        else
+          assign_master(@options[:master])
+          start!
+        end
+      end
+      
+      def configured?
+        !!query("SHOW SLAVE STATUS").fetch_row
+      end
+      
+      def slave_error
+        row = query("SHOW SLAVE STATUS").fetch_row
+        
+        row and row[19]
+      end
+      
+      def assign_master(master)
+        master_status = master.master_status
+        
+        master_options =
+          {
+            'MASTER_HOST' => 'localhost',
+            'MASTER_USER' => master.user_name,
+            'MASTER_PORT' => 3306,
+            'MASTER_LOG_FILE' => master_status[:master_log_file],
+            'MASTER_LOG_POS' => master_status[:master_log_position]
+          }.collect do |k, v|
+            case (v)
+            when String:
+              "#{k}='#{Mysql.quote(v)}'"
+            else
+              "#{k}=#{v}"
+            end
+          end
+        
+        execute("CHANGE MASTER TO #{master_options * ', '}")
+      end
+      
+      def start!
+        execute("START SLAVE")
+      end
+      
+      def user_name
+        @options[:slave_user] or super
+      end
+
+      def socket_name
+        @options[:slave_socket] or super
+      end
+    end
+  end
+end

data/lib/mysql_replication_helper/daemon.rb

@@ -0,0 +1,18 @@
+module MysqlReplicationHelper
+  class Daemon
+    def initialize(options)
+      @options = options
+      
+      @options[:master] = @master = Agent::Master.new(options)
+      @options[:slave] = @slave = Agent::Slave.new(options)
+    end
+    
+    def run!
+      while (true)
+        @master.poll!
+        @slave.poll!
+        sleep(10)
+      end
+    end
+  end
+end

data/lib/mysql_replication_helper/daemon_launcher.rb

@@ -0,0 +1,59 @@
+#!/usr/bin/env ruby
+#
+# Daemonize module
+#
+
+require 'rubygems'
+require 'yaml'
+require 'optparse'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..'))
+
+require 'mysql_replication_helper'
+
+# == Constants ==============================================================
+
+CONFIG_FILE_LOCATIONS = [
+  "/etc/replication-helper.conf",
+  "/etc/replication-helper/config",
+  "~/.replication-helper/config"
+].collect { |p| File.expand_path(p) }.freeze
+
+DEFAULT_CONFIG = {
+}
+
+# == Options ================================================================
+
+op = OptionParser.new
+options = { }
+config = { }
+config_file = nil
+
+op.on("--master-socket=s") { |socket| options[:master_socket] = socket }
+op.on("--master-data=s") { |dir| options[:master_data] = dir }
+op.on("--master-user=s") { |name| options[:master_user] = name }
+
+op.on("--slave-socket=s") { |socket| options[:slave_socket] = socket }
+op.on("--slave-data=s") { |dir| options[:slave_data] = dir }
+op.on("--slave-user=s") { |name| options[:slave_user] = name }
+
+op.on("-c", "--config=s") { |path| config_file = path }
+op.on("-v", "--verbose") { options[:verbose] = true }
+op.on("-d", "--daemon") { options[:daemon] = true }
+op.on("-h", "--help") { show_help }
+
+args = op.parse(*ARGV)
+
+# == Configuration ==========================================================
+
+if (config_file)
+  if (File.exist?(config_file))
+    config = YAML.load(open(config_file))
+  end
+end
+
+config = DEFAULT_CONFIG.merge(config).merge(options)
+
+# == Main ===================================================================
+
+MysqlReplicationHelper::Daemon.new(config).run!

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: theworkinggroup-mysql-replication-helper
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Scott Tadman
@@ -9,10 +9,19 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-04-24 00:00:00 -07:00
+date: 2009-04-26 00:00:00 -07:00
 default_executable: replication-helper
-dependencies: []
-
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: daemons
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
 description: 
 email: github@tadman.ca
 executables: 
@@ -27,6 +36,11 @@ files:
 - VERSION.yml
 - bin/replication-helper
 - lib/mysql_replication_helper.rb
+- lib/mysql_replication_helper/agent.rb
+- lib/mysql_replication_helper/agent/master.rb
+- lib/mysql_replication_helper/agent/slave.rb
+- lib/mysql_replication_helper/daemon.rb
+- lib/mysql_replication_helper/daemon_launcher.rb
 - lib/mysql_replication_helper/error_handler.rb
 - test/mysql_replication_helper/error_handler_test.rb
 - test/mysql_replication_helper_test.rb
@@ -55,7 +69,7 @@ requirements: []
 rubyforge_project: 
 rubygems_version: 1.2.0
 signing_key: 
-specification_version: 3
+specification_version: 2
 summary: MySQL Replication Helper
 test_files: 
 - test/mysql_replication_helper/error_handler_test.rb