Contents

module Backend; class BackendDB
    def initialize conf
        @query = conf.delete :query
        @conf = conf
        setup_db
    end

    def find dom_name, tries = 1
        return if tries > 3

        res = nil
    
        begin
            # http://jakeyesbeck.com/2016/02/14/ruby-threads-and-active-record-connections/
            ActiveRecord::Base.connection_pool.with_connection do
                sql = build_query dom_name
                res = ActiveRecord::Base.connection.execute sql
            end

        rescue ActiveRecord::StatementInvalid => e
            Log.e "Query error. Trying to reconnect. Details:\n#{e.desc}"
            # http://geoff.evason.name/2015/01/18/postgres-ssl-connection-has-been-closed-unexpectedly
            ActiveRecord::Base.connection.reconnect! 
            find dom_name, (tries+1)

        rescue => e
            Log.exc e
        end

        first_record = res&.first
        record_values = first_record&.values
        record_values&.first
    end

    def allow_cache
        true
    end

    private

    def build_query dom_name
        @query.sub '$domain', dom_name
    end

    def setup_db
        ActiveRecord::Base.logger = Log.logger

        gem_name = { 
            'postgresql' => 'pg',
            'mysql'      => 'mysql',
            'mysql2'     => 'mysql2'
        }[ @conf[:adapter] ]
          
        unless gem_name
            Util.die "Database adapter '#{@conf[:adapter]}' not supported. Aborting."
        end

        ['active_record', gem_name].each do |req_name|
            begin
                require req_name
            rescue => e
                Util.die "Error on 'require #{req_name}', install with 'gem install #{req_name}'.\nError Details: #{e.desc}"
            end
        end

        ActiveRecord::Base.establish_connection @conf
    end
  
end; end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
dns_one-0.4.20	lib/dns_one/backend/db.rb
dns_one-0.4.19	lib/dns_one/backend/db.rb
dns_one-0.4.18	lib/dns_one/backend/db.rb
dns_one-0.4.17	lib/dns_one/backend/db.rb
dns_one-0.4.16	lib/dns_one/backend/db.rb
dns_one-0.4.15	lib/dns_one/backend/db.rb
dns_one-0.4.14	lib/dns_one/backend/db.rb