mysql-pause 0.1.0

data/README

@@ -0,0 +1,54 @@
+= mysql-pause
+
+== Description
+
+mysql-pause is a proxy server to pause a query to MySQL.
+
+== Source Code
+
+https://bitbucket.org/winebarrel/mysql-pause
+
+== Dependency
+
+* EventMachine
+* RExec
+
+== Install
+
+gem install mysql-pause
+
+== Example
+=== Start server
+
+  shell> mysql-pause start
+  Starting daemon...
+  Waiting for daemon to start...
+  Daemon status: running pid=56921
+
+  shell> mysql -h 127.0.0.1 -P 13306 -u scott -ptiger
+  mysql> select 1;
+  +---+
+  | 1 |
+  +---+
+  | 1 |
+  +---+
+  1 row in set (0.00 sec)
+
+=== Pause
+
+  shell> mpctl pause
+
+  mysql> select 1;
+  (...no response...)
+
+=== Resume
+
+  shell> mpctl resume
+
+  (...resume response...)
+  +---+
+  | 1 |
+  +---+
+  | 1 |
+  +---+
+  1 row in set (18.01 sec)

data/bin/mpctl

@@ -0,0 +1,43 @@
+#!/usr/bin/env ruby
+$: << File.expand_path(File.dirname(__FILE__) + '/../lib')
+
+require 'drb/drb'
+require 'optparse'
+require 'mysql-pause/constants'
+
+COMMANDS = {
+  'pause'  => true,
+  'resume' => false,
+}
+
+options = nil
+
+ARGV.options do |parser|
+  cmds = COMMANDS.keys
+
+  options = {
+    :socket => "/var/tmp/#{APP_NAME}.sock",
+  }
+
+  parser.on('-i', '--inerval=N'       ) {|v| options[:interval]     = v.to_i }
+  parser.on(''  , '--socket=SOCK_FILE') {|v| options[:socket]       = v      }
+
+  help_and_exit = lambda do |v|
+    $stderr.puts parser.help.sub(APP_NAME, "#{APP_NAME} {#{cmds.join('|')}}")
+    exit 1
+  end
+
+  parser.on('-h', '--help', &help_and_exit)
+
+  parser.parse!
+
+  unless (ARGV.length == 0 and options.length > 1) or (ARGV.length == 1 and cmds.include?(ARGV[0]))
+    help_and_exit.call(true)
+  end
+end # parse options
+
+cmd = ARGV[0]
+server_options = DRbObject.new_with_uri("drbunix:#{options[:socket]}")
+
+server_options[:pause] = COMMANDS[cmd] if cmd
+server_options[:interval] = options[:interval] if options[:interval]

data/bin/mysql-pause

@@ -0,0 +1,46 @@
+#!/usr/bin/env ruby
+$: << File.expand_path(File.dirname(__FILE__) + '/../lib')
+
+require 'optparse'
+require 'mysql-pause/constants'
+require 'mysql-pause/server'
+
+options = nil
+
+ARGV.options do |parser|
+  daemon_commands = %w(start stop restart status)
+
+  options = {
+    :addr => '0.0.0.0',
+    :port => 13306,
+    :backend_addr => '127.0.0.1',
+    :backend_port => 3306,
+    :interval => 3,
+    :working_dir => '/var',
+    :socket => "/var/tmp/#{APP_NAME}.sock",
+  }
+
+  parser.on('-a', '--addr=ADDR'        ) {|v| options[:addr]         = v      }
+  parser.on('-p', '--port=PORT'        ) {|v| options[:port]         = v.to_i }
+  parser.on('-A', '--backend-addr=ADDR') {|v| options[:backend_addr] = v      }
+  parser.on('-P', '--backend-port=PORT') {|v| options[:backend_port] = v.to_i }
+  parser.on('-i', '--inerval=N'        ) {|v| options[:interval]     = v.to_i }
+  parser.on(''  , '--working-dir=DIR'  ) {|v| options[:working_dir]  = v      }
+  parser.on(''  , '--socket=SOCK_FILE' ) {|v| options[:socket]       = v      }
+
+  help_and_exit = lambda do |v|
+    $stderr.puts parser.help.sub(APP_NAME, "#{APP_NAME} {#{daemon_commands.join('|')}}")
+    exit 1
+  end
+
+  parser.on('-h', '--help', &help_and_exit)
+
+  parser.parse!
+
+  unless ARGV.length == 1 and daemon_commands.include?(ARGV[0])
+    help_and_exit.call(true)
+  end
+end # parse options
+
+MysqlPause::Server.options = options
+MysqlPause::Server.daemonize

data/lib/mysql-pause/abstract_mysql_adapter.rb

@@ -0,0 +1,26 @@
+require 'active_record'
+require 'active_record/connection_adapters/abstract_mysql_adapter'
+require 'mysql-pause/error'
+
+module ActiveRecord
+  module ConnectionAdapters
+    class AbstractMysqlAdapter
+
+      alias mysql_pause_execute_orig execute
+
+      def execute(sql, name = nil)
+        begin
+          mysql_pause_execute_orig(sql, name)
+        rescue => e
+          raise(e) unless MysqlPause::Error.mysql_pause_error?(e)
+
+          message = "#{e.class.name}: #{e.message}: #{sql}"
+          @logger.warn message if @logger
+          reconnect!
+          retry
+        end
+      end
+
+    end # AbstractMysqlAdapter
+  end # ConnectionAdapters
+end # ActiveRecord

data/lib/mysql-pause/backend.rb

@@ -0,0 +1,13 @@
+require 'eventmachine'
+
+module MysqlPause
+  class Backend < EM::Connection
+    def initialize(proxy)
+      @proxy = proxy
+    end
+
+    def receive_data(data)
+      @proxy.send_data(data)
+    end
+  end
+end

data/lib/mysql-pause/constants.rb

@@ -0,0 +1,2 @@
+APP_NAME = 'mysql-pause'
+Version = '0.1.0'

data/lib/mysql-pause/error.rb

@@ -0,0 +1,45 @@
+module MysqlPause
+  class Error
+
+    ERROR_HEADER = '%MYSQL_PAUSE%'
+    ERROR_SEPARATOR = ';;'
+    ERROR_LIST = {}
+
+    class << self
+      def create_error_message(error_code, sequence_id)
+        # fetch error info
+        mysql_error_code, sql_state, message = ERROR_LIST[error_code]
+
+        # create mysql error message
+        message = [ERROR_HEADER, error_code, message].join(ERROR_SEPARATOR)
+
+        # create header
+        mysql_error_code = convert_to_chars(mysql_error_code, 2)
+        payload = ["\xFF", mysql_error_code, '#', sql_state, message].join
+        payload_length = convert_to_chars(payload.length, 3)
+        sequence_id = convert_to_chars(sequence_id, 1)
+
+        [payload_length, sequence_id, payload].join
+      end
+
+      def mysql_pause_error?(e)
+        /\A#{Regexp.escape(MYSQL_PAUSE)}/ =~ e.message
+      end
+
+      private
+
+      def convert_to_chars(number, length)
+        (0...length).map {|i| (number >> (8 * i)) & 0xFF }.pack("C*")
+      end
+
+      def define_error(name, code, values)
+        const_set(name, code)
+        ERROR_LIST[code] = values
+      end
+    end # self
+
+    # error list
+    define_error :ABORTED_BACKEND_CONNECTION, 1000, [2013, '08S01', 'Aborted backend connection']
+
+  end # Errot
+end # MysqlPause

data/lib/mysql-pause/proxy.rb

@@ -0,0 +1,45 @@
+require 'eventmachine'
+require 'mysql-pause/backend'
+require 'mysql-pause/error'
+require 'logger'
+
+module MysqlPause
+  class Proxy < EM::Connection
+
+    def initialize(be_host, be_port, options)
+      @backend = EM.connect(be_host, be_port, MysqlPause::Backend, self)
+      @options = options
+      @logger = Logger.new($stdout)
+    end
+
+    def receive_data(data)
+      EM.defer {
+        if @options[:pause]
+          @logger.info("pause: #{data.inspect}")
+          sleep(@options[:interval]) while @options[:pause]
+          @logger.info("resume: #{data.inspect}")
+        end
+
+        if @backend.error?
+          @logger.info("backend error: #{data.inspect}")
+          payload_length, sequence_id = parse_mysql_packet(data)
+          error_message = MysqlPause::Error.create_error_message(MysqlPause::Error::ABORTED_BACKEND_CONNECTION, sequence_id + 1)
+          send_data(error_message)
+          close_connection_after_writing
+        else
+          @backend.send_data(data)
+        end
+      }
+    end
+
+    private
+
+    def parse_mysql_packet(data)
+      header = data.slice(0, 4).unpack("C4")
+      payload_length = (0..2).map {|i| header[i] << (8 * i) }.inject(0) {|r, i| r + i }
+      sequence_id = header[3]
+      [payload_length, sequence_id]
+    end
+
+  end # Proxy
+end # MysqlPause

data/lib/mysql-pause/server.rb

@@ -0,0 +1,45 @@
+require 'drb/drb'
+require 'fileutils'
+require 'eventmachine'
+require 'rexec'
+require 'rexec/daemon'
+
+require 'mysql-pause/proxy'
+
+module RExec; module Daemon; class Base
+  def self.daemon_name; APP_NAME; end
+end; end; end
+
+module MysqlPause
+  class Server < RExec::Daemon::Base
+
+    class << self
+      def options=(options)
+        @@options = options
+        @@base_directory = options[:working_dir]
+      end
+
+      def run
+        @@control_options = {
+          :pause    => false,
+          :interval => @@options[:interval],
+        }
+
+        # start DRb
+        FileUtils.rm_f(@@options[:socket])
+        DRb.start_service("drbunix:#{@@options[:socket]}", @@control_options)
+        File.chmod(0700, @@options[:socket])
+        at_exit { FileUtils.rm_f(@@options[:socket]) }
+
+        EM.epoll
+
+        EM.run {
+          EM.start_server(
+            @@options[:addr], @@options[:port], MysqlPause::Proxy,
+            @@options[:backend_addr], @@options[:backend_port], @@control_options)
+        }
+      end
+    end # self
+
+  end # Server
+end # MysqlPause

metadata

@@ -0,0 +1,87 @@
+--- !ruby/object:Gem::Specification
+name: mysql-pause
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- winebarrel
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.3
+- !ruby/object:Gem::Dependency
+  name: rexec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.5.1
+description: 
+email: sgwr_dts@yahoo.co.jp
+executables:
+- mysql-pause
+- mpctl
+extensions: []
+extra_rdoc_files: []
+files:
+- README
+- bin/mpctl
+- bin/mysql-pause
+- lib/mysql-pause/abstract_mysql_adapter.rb
+- lib/mysql-pause/backend.rb
+- lib/mysql-pause/constants.rb
+- lib/mysql-pause/error.rb
+- lib/mysql-pause/proxy.rb
+- lib/mysql-pause/server.rb
+homepage: https://bitbucket.org/winebarrel/mysql-pause
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: mysql-pause is a proxy server to pause a query to MySQL.
+test_files: []