Contents

require "pg"
require "connection_pool"

module GitLab
  module Exporter
    module Database
      # An abstract class for interacting with DB
      #
      # It takes a connection string (e.g. "dbname=test port=5432")
      class Base
        def self.connection_pool
          @connection_pool ||= Hash.new do |h, connection_string|
            h[connection_string] = ConnectionPool.new(size: 3, timeout: 5) do
              PG.connect(connection_string)
            end
          end
        end

        def initialize(args, logger: nil)
          @connection_string = args[:connection_string]
          @logger = logger
        end

        def run
          fail NotImplemented
        end

        def connection_pool
          self.class.connection_pool[@connection_string]
        end

        def with_connection_pool
          connection_pool.with do |conn|
            begin
              yield conn
            rescue PG::UnableToSend => e
              @logger.warn "Error sending to the database: #{e}" if @logger
              conn.reset
              raise e
            end
          end
        rescue PG::Error => e
          @logger.error "Error connecting to the database: #{e}" if @logger
          raise e
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
gitlab-exporter-8.0.0	lib/gitlab_exporter/database/base.rb
gitlab-exporter-7.2.0	lib/gitlab_exporter/database/base.rb
gitlab-exporter-7.1.2	lib/gitlab_exporter/database/base.rb
gitlab-exporter-7.1.1	lib/gitlab_exporter/database/base.rb
gitlab-exporter-7.1.0	lib/gitlab_exporter/database/base.rb