Sha256: 611e2a4a1cf3a030ba54e7480479cfd2b04637205b7c4a8e00c68e7545807081

Contents?: true

Size: 1.3 KB

Versions: 40

Compression:

Stored size: 1.3 KB

Contents

# frozen_string_literal: true

module ActiveRecord
  module ConnectionAdapters
    class StatementPool # :nodoc:
      include Enumerable

      DEFAULT_STATEMENT_LIMIT = 1000

      def initialize(statement_limit = nil)
        @cache = Hash.new { |h, pid| h[pid] = {} }
        @statement_limit = statement_limit || DEFAULT_STATEMENT_LIMIT
      end

      def each(&block)
        cache.each(&block)
      end

      def key?(key)
        cache.key?(key)
      end

      def [](key)
        cache[key]
      end

      def length
        cache.length
      end

      def []=(sql, stmt)
        while @statement_limit <= cache.size
          dealloc(cache.shift.last)
        end
        cache[sql] = stmt
      end

      def clear
        cache.each_value do |stmt|
          dealloc stmt
        end
        cache.clear
      end

      # Clear the pool without deallocating; this is only safe when we
      # know the server has independently deallocated all statements
      # (e.g. due to a reconnect, or a DISCARD ALL)
      def reset
        cache.clear
      end

      def delete(key)
        dealloc cache[key]
        cache.delete(key)
      end

      private
        def cache
          @cache[Process.pid]
        end

        def dealloc(stmt)
          raise NotImplementedError
        end
    end
  end
end

Version data entries

40 entries across 40 versions & 5 rubygems

Version Path
minato_ruby_api_client-0.2.2 vendor/bundle/ruby/3.2.0/gems/activerecord-7.1.3.4/lib/active_record/connection_adapters/statement_pool.rb
activerecord-8.0.0.1 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.2.2.1 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.1.5.1 lib/active_record/connection_adapters/statement_pool.rb
activerecord-8.0.0 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.2.2 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.1.5 lib/active_record/connection_adapters/statement_pool.rb
activerecord-8.0.0.rc2 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.2.1.2 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.1.4.2 lib/active_record/connection_adapters/statement_pool.rb
activerecord-8.0.0.rc1 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.2.1.1 lib/active_record/connection_adapters/statement_pool.rb
activerecord-7.1.4.1 lib/active_record/connection_adapters/statement_pool.rb
activerecord-8.0.0.beta1 lib/active_record/connection_adapters/statement_pool.rb
omg-activerecord-8.0.0.alpha9 lib/active_record/connection_adapters/statement_pool.rb
omg-activerecord-8.0.0.alpha8 lib/active_record/connection_adapters/statement_pool.rb
omg-activerecord-8.0.0.alpha7 lib/active_record/connection_adapters/statement_pool.rb
omg-activerecord-8.0.0.alpha4 lib/active_record/connection_adapters/statement_pool.rb
omg-activerecord-8.0.0.alpha3 lib/active_record/connection_adapters/statement_pool.rb
omg-activerecord-8.0.0.alpha2 lib/active_record/connection_adapters/statement_pool.rb