active_record_postgresql_retry 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: f79de3930cd1788519e1bf452cb4be1d504fa918
+  data.tar.gz: 9ced0fbd898596e837d02f2d0092beaa4d78e601
+SHA512:
+  metadata.gz: 3ff6d5014c101a483588d7d391af24c8cf6036382883d1e993c3261d5700bfcd7539b26a97b521523e51dd10df7c3e761ea1eb59f2d5094f82379c44f2ca9431
+  data.tar.gz: c99c357c216ace4ca55045ca1a4ed9cee7e5e00f1cc7d1da0021b4453303568c0376fc34c156e0b415d20a4ef1a4ef246f7e72b85c87f911a237b5bd74022763

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2017 Alexandru Szasz
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.md

@@ -0,0 +1,10 @@
+# active_record_postgresql_retry
+This gem provides a patch that once included will retry the PostgreSQL operation in case the server has gone away
+
+1. add it to your Gemfile
+2. declare the following environment variables
+- PG_SLEEP_RETRY, default 0.33 seconds
+- PG_QUERY_RETRY, default 0
+- PG_RECONNECT_RETRY, default 0
+- PG_CONNECT_RETRY, default 0
+

data/lib/active_record_postgresql_retry.rb

@@ -0,0 +1,80 @@
+require 'active_record/connection_adapters/postgresql_adapter'
+
+module ActiveRecord
+  module ConnectionAdapters
+    class PostgreSQLAdapter < AbstractAdapter
+      alias_method :old_execute, :execute
+      alias_method :old_reconnect!, :reconnect!
+      alias_method :old_connect, :connect
+
+      PG_SLEEP_RETRY = ENV['PG_SLEEP_RETRY'].to_f || 0.33
+      PG_QUERY_RETRY = ENV['PG_QUERY_RETRY'].to_i || 0
+      PG_RECONNECT_RETRY = ENV['PG_RECONNECT_RETRY'].to_i || 0
+      PG_CONNECT_RETRY = ENV['PG_CONNECT_RETRY'].to_i || 0
+
+      def execute(sql, name=nil)
+        begin
+          result = old_execute(sql, name)
+          @retry = nil
+          result
+        rescue PG::ConnectionBad, PG::UnableToSend => ex
+          @retry ||= 1
+          $log.warn "PostgreSQL statement invalid, retrying #{sql}: #{ex}"
+          sleep PG_SLEEP_RETRY
+          @retry += 1
+          raise if @retry > PG_QUERY_RETRY
+        end
+
+        if @retry.nil?
+          result
+        else
+          execute(sql, name)
+        end
+
+      end
+
+      def reconnect!
+        begin
+          result = old_reconnect!
+          @retry = nil
+          result
+        rescue PG::Error => ex
+          @retry ||= 1
+          $log.warn "PostgreSQL reconnect not possible, retry ##{@retry}: #{ex}"
+          sleep PG_SLEEP_RETRY
+          @retry += 1
+          raise if @retry > PG_RECONNECT_RETRY
+        end
+
+        if @retry.nil?
+          result
+        else
+          reconnect!
+        end
+
+      end
+
+      def connect
+
+        begin
+          result = old_connect
+          @retry = nil
+          result
+        rescue PG::Error => ex
+          @retry ||= 1
+          $log.warn "PostgreSQL connect not possible, retry ##{@retry}: #{ex}"
+          sleep PG_SLEEP_RETRY
+          @retry += 1
+          raise if @retry > PG_CONNECT_RETRY
+        end
+
+        if @retry.nil?
+          result
+        else
+          connect
+        end
+
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: active_record_postgresql_retry
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Alexandru Szasz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-01-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: This gem provides a patch that once included will retry the PostgreSQL
+  operation in case the server has gone away
+email: alexxed@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- lib/active_record_postgresql_retry.rb
+homepage: https://github.com/alexxed/active_record_postgresql_retry
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: This gem provides a patch that once included will retry the PostgreSQL operation
+  in case the server has gone away
+test_files: []