mysql-pause 0.1.2 → 0.1.3

data/lib/mysql-pause/constants.rb

@@ -1,2 +1,2 @@
 APP_NAME = 'mysql-pause'
-Version = '0.1.2'
+Version = '0.1.3'

data/lib/mysql-pause/proxy.rb

@@ -2,14 +2,25 @@ require 'eventmachine'
 require 'mysql-pause/backend'
 require 'mysql-pause/error'
 require 'logger'
+require 'socket'
 
 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)
+
+      begin
+        ping(be_host, be_port)
+        @backend = EM.connect(be_host, be_port, MysqlPause::Backend, self)
+        raise('connect error') if @backend.error?
+      rescue => e
+        @logger.error("#{e.class.name}: #{e.message}")
+        close_backend_connection
+        sleep(@options[:interval])
+        retry
+      end
     end
 
     def receive_data(data)
@@ -22,9 +33,12 @@ module MysqlPause
 
         if @backend.error?
           @logger.info("backend error: #{data.inspect}")
+          close_backend_connection
+
           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)
@@ -41,5 +55,16 @@ module MysqlPause
       [payload_length, sequence_id]
     end
 
+    def close_backend_connection
+      @backend.close_connection if @backend
+    rescue Exception => e
+      @logger.warn("#{e.class.name}: #{e.message}")
+    end
+
+    def ping(addr, port)
+      sock = TCPSocket.open(addr, port)
+      sock.close
+    end
+
   end # Proxy
 end # MysqlPause

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mysql-pause
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
   prerelease: 
 platform: ruby
 authors: