postgresql_cursor 0.6.4 → 0.6.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3f0d7815c65eeeeb83cc41344d0e4a76906236c43392f837e9e2ba7bb14277a7
-  data.tar.gz: 66b9dbcb6a17046a2d1338297f0b9e4233b46467bd0efb76a663db5d6cfcb4be
+  metadata.gz: c85acb2e98683cf55a34048f3b4b20ee78b210741f22dcb634bd5c4a2fe53395
+  data.tar.gz: 2c9c6bef23d6a18d6e878b8bf0cd5f85ec94c4fe013238ca95fa1a95f9af8ada
 SHA512:
-  metadata.gz: cc42f9f219afb24cbbe15abf0829de57b8f3c8002a3658e931bfbc97a065d080706e500ba86b3a3bacf8778ca824032556497f520ee1153348f8f9057fc98b6b
-  data.tar.gz: ec58f29782cfac48985b1c8d491726cfb23ebc1cefae89766045bff36989b238e01a10fa0eb6f436efdc0446b73395b6fdb32f13cfd239166b3713736c4d8f20
+  metadata.gz: 707d3922478b915d36b503a9e3ef4973e2064d7e1b0388b2b31191e91e4bef1e118ac3d08eb8aaeae4a7b94e6c099c522116d7651aa23f0645f16541e8b68793
+  data.tar.gz: 7bda9582106b1d8472a7c09d3ce9db8e66a519f2f22f2903c9e29b5f9814083f81ffa3df3c2aa171b5e2f8445c2bf05f0fac5025176c77ef74ae7c7733c0dec6

data/.gitignore

@@ -15,6 +15,7 @@ tmtags
 
 ## IntelliJ/Rubymine
 .idea
+*.iml
 
 ## PROJECT::GENERAL
 coverage
@@ -22,4 +23,4 @@ rdoc
 pkg
 
 ## PROJECT::SPECIFIC
-Gemfile.lock
+Gemfile.lock

data/.travis.yml

@@ -0,0 +1,27 @@
+language: ruby
+rvm:
+  - 2.6.5
+  - 2.7.1
+before_install:
+  - sudo apt-get update
+  - sudo apt-get --yes remove postgresql\*
+  - sudo apt-get install -y postgresql-12 postgresql-client-12
+  - sudo cp /etc/postgresql/{9.6,12}/main/pg_hba.conf
+  - sudo service postgresql restart 12
+gemfile:
+  - gemfiles/activerecord_4.gemfile
+  - gemfiles/activerecord_5.gemfile
+  - gemfiles/activerecord_6.gemfile
+matrix:
+  exclude:
+    - rvm: 2.7.1
+      gemfile: gemfiles/activerecord_4.gemfile
+services:
+  - postgresql
+before_script:
+  - psql -c 'create database postgresql_cursor_test;' -U postgres
+  - psql -c 'CREATE ROLE travis SUPERUSER LOGIN CREATEDB;' -U postgres
+  - psql -c 'create table products ( id serial primary key, data varchar);' -U postgres -d postgresql_cursor_test
+  - psql -c 'create table prices ( id serial primary key, data varchar, product_id integer);' -U postgres -d postgresql_cursor_test
+addons:
+  postgresql: '12.3'

data/Appraisals

@@ -0,0 +1,12 @@
+appraise "activerecord-4" do
+  gem "activerecord", "4.2.11.1"
+  gem "pg", "~> 0.15"
+end
+
+appraise "activerecord-5" do
+  gem "activerecord", "5.2.3"
+end
+
+appraise "activerecord-6" do
+  gem "activerecord", "6.0.0"
+end

data/Rakefile

@@ -21,4 +21,5 @@ desc "Setup testing database and table"
 task :setup do
   sh %q(createdb postgresql_cursor_test)
   sh %Q<echo "create table products ( id serial primary key, data varchar);" | psql postgresql_cursor_test>
+  sh %Q<echo "create table prices ( id serial primary key, data varchar, product_id integer);" | psql postgresql_cursor_test>
 end

data/gemfiles/activerecord_4.gemfile

@@ -0,0 +1,8 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "activerecord", "4.2.11.1"
+gem "pg", "~> 0.15"
+
+gemspec path: "../"

data/gemfiles/activerecord_5.gemfile

@@ -0,0 +1,7 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "activerecord", "5.2.3"
+
+gemspec path: "../"

data/gemfiles/activerecord_6.gemfile

@@ -0,0 +1,7 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "activerecord", "6.0.0"
+
+gemspec path: "../"

data/lib/postgresql_cursor/cursor.rb

@@ -41,29 +41,32 @@ module PostgreSQLCursor
     #   PostgreSQLCursor::Cursor.new("select ....")
     #
     # Returns the cursor object when called with new.
-    def initialize(sql, options={})
-      @sql        = sql
-      @options    = options
+    def initialize(sql, options = {})
+      @sql = sql
+      @options = options
       @connection = @options.fetch(:connection) { ::ActiveRecord::Base.connection }
-      @count      = 0
-      @iterate    = options[:instances] ? :each_instance : :each_row
-      @batched    = false
+      @count = 0
+      @iterate = options[:instances] ? :each_instance : :each_row
+      @batched = false
     end
 
-    # Specify the type to instantiate, or reset to return a Hash
-    def iterate_type(type=nil)
-      if type.nil? || type == Hash
+    # Specify the type to instantiate, or reset to return a Hash.
+    #
+    # Explicitly check for type class to prevent calling equality
+    # operator on active record relation, which will load it.
+    def iterate_type(type = nil)
+      if type.nil? || (type.instance_of?(Class) && type == Hash)
         @iterate = :each_row
-      elsif type == Array
+      elsif type.instance_of?(Class) && type == Array
         @iterate = :each_array
       else
         @iterate = :each_instance
-        @type    = type
+        @type = type
       end
       self
     end
 
-    def iterate_batched(batched=true)
+    def iterate_batched(batched = true)
       @batched = batched
       self
     end
@@ -76,16 +79,16 @@ module PostgreSQLCursor
     # Returns the count of rows processed
     def each(&block)
       if @iterate == :each_row
-        @batched ? self.each_row_batch(&block) : self.each_row(&block)
+        @batched ? each_row_batch(&block) : each_row(&block)
       elsif @iterate == :each_array
-        @batched ? self.each_array_batch(&block) : self.each_array(&block)
+        @batched ? each_array_batch(&block) : each_array(&block)
       else
-        @batched ? self.each_instance_batch(@type, &block) : self.each_instance(@type, &block)
+        @batched ? each_instance_batch(@type, &block) : each_instance(@type, &block)
       end
     end
 
     def each_row(&block)
-      self.each_tuple do |row|
+      each_tuple do |row|
         row = row.symbolize_keys if @options[:symbolize_keys]
         block.call(row)
       end
@@ -95,7 +98,7 @@ module PostgreSQLCursor
       old_iterate = @iterate
       @iterate = :each_array
       begin
-        rv = self.each_tuple do |row|
+        rv = each_tuple do |row|
           block.call(row)
         end
       ensure
@@ -104,11 +107,11 @@ module PostgreSQLCursor
       rv
     end
 
-    def each_instance(klass=nil, &block)
+    def each_instance(klass = nil, &block)
       klass ||= @type
-      self.each_tuple do |row|
+      each_tuple do |row|
         if ::ActiveRecord::VERSION::MAJOR < 4
-          model = klass.send(:instantiate,row)
+          model = klass.send(:instantiate, row)
         else
           @column_types ||= column_types
           model = klass.send(:instantiate, row, @column_types)
@@ -118,7 +121,7 @@ module PostgreSQLCursor
     end
 
     def each_row_batch(&block)
-      self.each_batch do |batch|
+      each_batch do |batch|
         batch.map!(&:symbolize_keys) if @options[:symbolize_keys]
         block.call(batch)
       end
@@ -128,7 +131,7 @@ module PostgreSQLCursor
       old_iterate = @iterate
       @iterate = :each_array
       begin
-        rv = self.each_batch do |batch|
+        rv = each_batch do |batch|
           block.call(batch)
         end
       ensure
@@ -137,15 +140,15 @@ module PostgreSQLCursor
       rv
     end
 
-    def each_instance_batch(klass=nil, &block)
+    def each_instance_batch(klass = nil, &block)
       klass ||= @type
-      self.each_batch do |batch|
+      each_batch do |batch|
         models = batch.map do |row|
           if ::ActiveRecord::VERSION::MAJOR < 4
-            model = klass.send(:instantiate, row)
+            klass.send(:instantiate, row)
           else
             @column_types ||= column_types
-            model = klass.send(:instantiate, row, @column_types)
+            klass.send(:instantiate, row, @column_types)
           end
         end
         block.call(models)
@@ -160,11 +163,11 @@ module PostgreSQLCursor
       options = cols.last.is_a?(Hash) ? cols.pop : {}
       @options.merge!(options)
       @options[:symbolize_keys] = true
-      self.iterate_type(options[:class]) if options[:class]
-      cols    = cols.map {|c| c.to_sym }
-      result  = []
+      iterate_type(options[:class]) if options[:class]
+      cols = cols.map { |c| c.to_sym }
+      result = []
 
-      self.each() do |row|
+      each do |row|
         row = row.symbolize_keys if row.is_a?(Hash)
         result << cols.map { |c| row[c] }
       end
@@ -173,48 +176,44 @@ module PostgreSQLCursor
       result
     end
 
-    def each_tuple(&block) #:nodoc:
-      has_do_until  = @options.has_key?(:until)
-      has_do_while  = @options.has_key?(:while)
-      @count        = 0
+    def each_tuple(&block) # :nodoc:
+      has_do_until = @options.has_key?(:until)
+      has_do_while = @options.has_key?(:while)
+      @count = 0
       @column_types = nil
       with_optional_transaction do
-        begin
-          open
-          while (row = fetch) do
-            break if row.size==0
-            @count += 1
-            rc = block.call(row)
-            break if has_do_until && rc == @options[:until]
-            break if has_do_while && rc != @options[:while]
-          end
-        rescue Exception => e
-          raise e
-        ensure
-          close if @block
+        open
+        while (row = fetch)
+          break if row.size == 0
+          @count += 1
+          rc = block.call(row)
+          break if has_do_until && rc == @options[:until]
+          break if has_do_while && rc != @options[:while]
         end
+      rescue => e
+        raise e
+      ensure
+        close if @block && connection.active?
       end
       @count
     end
 
-    def each_batch(&block) #:nodoc:
+    def each_batch(&block) # :nodoc:
       has_do_until = @options.key?(:until)
       has_do_while = @options.key?(:while)
       @count = 0
       @column_types = nil
       with_optional_transaction do
-        begin
-          open
-          while (batch = fetch_block)
-            break if batch.empty?
-            @count += 1
-            rc = block.call(batch)
-            break if has_do_until && rc == @options[:until]
-            break if has_do_while && rc != @options[:while]
-          end
-        ensure
-          close if @block
+        open
+        while (batch = fetch_block)
+          break if batch.empty?
+          @count += 1
+          rc = block.call(batch)
+          break if has_do_until && rc == @options[:until]
+          break if has_do_while && rc != @options[:while]
         end
+      ensure
+        close if @block && connection.active?
       end
       @count
     end
@@ -230,16 +229,14 @@ module PostgreSQLCursor
       types = {}
       fields = @result.fields
       fields.each_with_index do |fname, i|
-        ftype = @result.ftype i
-        fmod  = @result.fmod i
-        types[fname] = @connection.get_type_map.fetch(ftype, fmod) do |oid, mod|
-          warn "unknown OID: #{fname}(#{oid}) (#{sql})"
-          if ::ActiveRecord::VERSION::MAJOR <= 4
-            ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter::OID::Identity.new
-          else
-            ::ActiveRecord::Type::Value.new
-          end
-        end
+        ftype = @result.ftype(i)
+        fmod = @result.fmod(i)
+
+        # From @netrusov 2023-01-18. This is the same call used in the PostgreSQL Adapter
+        types[fname] = @connection.send(:get_oid_type, ftype, fmod, fname)
+
+        # # From @simi 2023-01-18 (Works as well, used old calling method)
+        # types[fname] = @connection.get_type_map.fetch(ftype)
       end
 
       @column_types = types
@@ -248,8 +245,8 @@ module PostgreSQLCursor
     # Public: Opens (actually, "declares") the cursor. Call this before fetching
     def open
       set_cursor_tuple_fraction
-      @cursor = @options[:cursor_name] || ("cursor_" + SecureRandom.uuid.gsub("-",""))
-      hold = @options[:with_hold] ? 'with hold ' : ''
+      @cursor = @options[:cursor_name] || ("cursor_" + SecureRandom.uuid.delete("-"))
+      hold = @options[:with_hold] ? "with hold " : ""
       @result = @connection.execute("declare #{@cursor} no scroll cursor #{hold}for #{@sql}")
       @block = []
     end
@@ -257,23 +254,23 @@ module PostgreSQLCursor
     # Public: Returns the next row from the cursor, or empty hash if end of results
     #
     # Returns a row as a hash of {'colname'=>value,...}
-    def fetch(options={})
+    def fetch(options = {})
       open unless @block
-      fetch_block if @block.size==0
+      fetch_block if @block.size == 0
       row = @block.shift
       row = row.symbolize_keys if row && options[:symbolize_keys]
       row
     end
 
     # Private: Fetches the next block of rows into @block
-    def fetch_block(block_size=nil)
-      block_size ||= @block_size ||= @options.fetch(:block_size) { 1000 }
+    def fetch_block(block_size = nil)
+      block_size ||= @block_size ||= @options.fetch(:block_size, 1000)
       @result = @connection.execute("fetch #{block_size} from #{@cursor}")
 
-      if @iterate == :each_array
-        @block = @result.each_row.collect {|row| row }
+      @block = if @iterate == :each_array
+        @result.each_row.collect { |row| row }
       else
-        @block = @result.collect {|row| row }
+        @result.collect { |row| row }
       end
     end
 
@@ -295,8 +292,8 @@ module PostgreSQLCursor
     # This is a value between 0.1 and 1.0 (PostgreSQL defaults to 0.1, this library defaults to 1.0)
     # used to determine the expected fraction (percent) of result rows returned the the caller.
     # This value determines the access path by the query planner.
-    def set_cursor_tuple_fraction(frac=1.0)
-      @cursor_tuple_fraction ||= @options.fetch(:fraction) { 1.0 }
+    def set_cursor_tuple_fraction(frac = 1.0)
+      @cursor_tuple_fraction ||= @options.fetch(:fraction, 1.0)
       return @cursor_tuple_fraction if frac == @cursor_tuple_fraction
       @cursor_tuple_fraction = frac
       @result = @connection.execute("set cursor_tuple_fraction to  #{frac}")

data/lib/postgresql_cursor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PostgresqlCursor
-  VERSION = '0.6.4'
+  VERSION = "0.6.8"
 end

data/postgresql_cursor.gemspec

@@ -1,15 +1,15 @@
 # frozen_string_literal: true
 
-lib = File.expand_path('lib', __dir__)
+lib = File.expand_path("lib", __dir__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
-require 'postgresql_cursor/version'
+require "postgresql_cursor/version"
 
 Gem::Specification.new do |spec|
-  spec.name          = 'postgresql_cursor'
-  spec.version       = PostgresqlCursor::VERSION
-  spec.authors       = ['Allen Fair']
-  spec.email         = ['allen.fair@gmail.com']
-  spec.summary       = <<-SUMMARY
+  spec.name = "postgresql_cursor"
+  spec.version = PostgresqlCursor::VERSION
+  spec.authors = ["Allen Fair"]
+  spec.email = ["allen.fair@gmail.com"]
+  spec.summary = <<-SUMMARY
   ActiveRecord PostgreSQL Adapter extension for using a cursor to return a
   large result set
   SUMMARY
@@ -20,25 +20,21 @@ Gem::Specification.new do |spec|
   rows in 'chunks' (default of 1_000 rows), buffers them, and returns the rows
   one at a time.
   DESCRIPTION
-  spec.homepage      = 'http://github.com/afair/postgresql_cursor'
-  spec.license       = 'MIT'
+  spec.homepage = "http://github.com/afair/postgresql_cursor"
+  spec.license = "MIT"
 
-  spec.files         = `git ls-files -z`.split("\x0")
-  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
-  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
-  spec.require_paths = ['lib']
+  spec.files = `git ls-files -z`.split("\x0")
+  spec.executables = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
 
-  # Remove this for jruby which should use 'activerecord-jdbcpostgresql-adapter'
-  # spec.add_dependency 'pg'
+  spec.add_dependency "activerecord", ">= 6.0"
+  # spec.add_development_dependency "activerecord", "~> 6.1"
+  # spec.add_development_dependency "activerecord", "~> 7.0"
 
-  spec.add_dependency 'activerecord', '>= 3.1.0'
-  # spec.add_dependency 'activerecord', '~> 3.1.0'
-  # spec.add_dependency 'activerecord', '~> 4.1.0'
-  # spec.add_dependency 'activerecord', '~> 5.0.0'
-  # spec.add_dependency 'activerecord', '~> 6.0.0'
-
-  spec.add_development_dependency 'irb'
-  spec.add_development_dependency 'minitest'
-  spec.add_development_dependency 'pg'
-  spec.add_development_dependency 'rake'
+  # PG dependency held back for Jruby. See pg_jruby or jdbc-postgres-adapter gems.
+  spec.add_development_dependency "pg" unless RUBY_PLATFORM == "java"
+  spec.add_development_dependency "irb"
+  spec.add_development_dependency "minitest"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "appraisal"
 end

data/test/helper.rb

@@ -10,6 +10,8 @@ ActiveRecord::Base.establish_connection(adapter: 'postgresql',
   username: ENV['TEST_USER']     || ENV['USER'] || 'postgresql_cursor')
 
 class Product < ActiveRecord::Base
+  has_many :prices
+
   # create table records (id serial primary key);
   def self.generate(max=1_000)
     max.times do |i|
@@ -18,5 +20,9 @@ class Product < ActiveRecord::Base
   end
 end
 
+class Price < ActiveRecord::Base
+  belongs_to :product
+end
+
 Product.destroy_all
 Product.generate(1000)

data/test/test_postgresql_cursor.rb

@@ -3,16 +3,15 @@
 #     rake setup
 # or create the database manually if your environment doesn't permit
 ################################################################################
-require_relative 'helper'
-require 'minitest/autorun'
-require 'minitest/pride'
+require_relative "helper"
+require "minitest/autorun"
+require "minitest/pride"
 
 class TestPostgresqlCursor < Minitest::Test
-
   def test_each
     c = PostgreSQLCursor::Cursor.new("select * from products order by 1")
     nn = 0
-    n = c.each { nn += 1}
+    n = c.each { nn += 1 }
     assert_equal nn, n
   end
 
@@ -29,22 +28,22 @@ class TestPostgresqlCursor < Minitest::Test
   end
 
   def test_each_while_until
-    c = PostgreSQLCursor::Cursor.new("select * from products order by 1", until:true)
-    n = c.each { |r| r['id'].to_i > 100 }
+    c = PostgreSQLCursor::Cursor.new("select * from products order by 1", until: true)
+    n = c.each { |r| r["id"].to_i > 100 }
     assert_equal 101, n
 
-    c = PostgreSQLCursor::Cursor.new("select * from products order by 1", while:true)
-    n = c.each { |r| r['id'].to_i < 100 }
+    c = PostgreSQLCursor::Cursor.new("select * from products order by 1", while: true)
+    n = c.each { |r| r["id"].to_i < 100 }
     assert_equal 100, n
   end
 
   def test_each_batch_while_until
     c = PostgreSQLCursor::Cursor.new("select * from products order by id asc", until: true, block_size: 50)
-    n = c.each_batch { |b| b.last['id'].to_i > 100 }
+    n = c.each_batch { |b| b.last["id"].to_i > 100 }
     assert_equal 3, n
 
     c = PostgreSQLCursor::Cursor.new("select * from products order by id asc", while: true, block_size: 50)
-    n = c.each_batch { |b| b.last['id'].to_i < 100 }
+    n = c.each_batch { |b| b.last["id"].to_i < 100 }
     assert_equal 2, n
   end
 
@@ -68,20 +67,26 @@ class TestPostgresqlCursor < Minitest::Test
 
   def test_relation
     nn = 0
-    Product.where("id>0").each_row {|r| nn += 1 }
+    Product.where("id>0").each_row { |r| nn += 1 }
     assert_equal 1000, nn
   end
 
   def test_relation_batch
     nn = 0
     row = nil
-    Product.where("id>0").each_row_batch(block_size: 100) { |b| row = b.last; nn += 1 }
+    Product.where("id>0").each_row_batch(block_size: 100) { |b|
+      row = b.last
+      nn += 1
+    }
     assert_equal 10, nn
     assert_equal Hash, row.class
 
     nn = 0
     row = nil
-    Product.where("id>0").each_instance_batch(block_size: 100) { |b| row = b.last; nn += 1 }
+    Product.where("id>0").each_instance_batch(block_size: 100) { |b|
+      row = b.last
+      nn += 1
+    }
     assert_equal 10, nn
     assert_equal Product, row.class
   end
@@ -89,12 +94,18 @@ class TestPostgresqlCursor < Minitest::Test
   def test_activerecord
     nn = 0
     row = nil
-    Product.each_row_by_sql("select * from products") {|r| row = r; nn += 1 }
+    Product.each_row_by_sql("select * from products") { |r|
+      row = r
+      nn += 1
+    }
     assert_equal 1000, nn
     assert_equal Hash, row.class
 
     nn = 0
-    Product.each_instance_by_sql("select * from products") {|r| row = r; nn += 1 }
+    Product.each_instance_by_sql("select * from products") { |r|
+      row = r
+      nn += 1
+    }
     assert_equal 1000, nn
     assert_equal Product, row.class
   end
@@ -102,32 +113,52 @@ class TestPostgresqlCursor < Minitest::Test
   def test_activerecord_batch
     nn = 0
     row = nil
-    Product.each_row_batch_by_sql("select * from products", block_size: 100) { |b| row = b.last; nn += 1 }
+    Product.each_row_batch_by_sql("select * from products", block_size: 100) { |b|
+      row = b.last
+      nn += 1
+    }
     assert_equal 10, nn
     assert_equal Hash, row.class
 
     nn = 0
-    Product.each_instance_batch_by_sql("select * from products", block_size: 100) { |b| row = b.last; nn += 1 }
+    Product.each_instance_batch_by_sql("select * from products", block_size: 100) { |b|
+      row = b.last
+      nn += 1
+    }
     assert_equal 10, nn
     assert_equal Product, row.class
   end
 
   def test_exception
-    begin
-      Product.each_row_by_sql("select * from products") do |r|
-        raise "Oops"
-      end
-    rescue Exception => e
-      assert_equal e.message, 'Oops'
+    Product.each_row_by_sql("select * from products") do |r|
+      raise "Oops"
     end
+  rescue => e
+    assert_equal e.message, "Oops"
   end
 
   def test_batch_exception
     Product.each_row_batch_by_sql("select * from products") do |r|
-      raise 'Oops'
+      raise "Oops"
+    end
+  rescue => e
+    assert_equal e.message, "Oops"
+  end
+
+  def test_exception_in_failed_transaction
+    Product.each_row_by_sql("select * from products") do |r|
+      Product.connection.execute("select kaboom")
+    end
+  rescue => e
+    assert_match(/PG::(InFailedSqlTransaction|UndefinedColumn)/, e.message)
+  end
+
+  def test_batch_exception_in_failed_transaction
+    Product.each_row_batch_by_sql("select * from products") do |r|
+      Product.connection.execute("select kaboom")
     end
   rescue => e
-    assert_equal e.message, 'Oops'
+    assert_match(/PG::(InFailedSqlTransaction|UndefinedColumn)/, e.message)
   end
 
   def test_cursor
@@ -144,11 +175,11 @@ class TestPostgresqlCursor < Minitest::Test
   def test_batched_cursor
     cursor = Product.all.each_row_batch(block_size: 100)
     assert cursor.respond_to?(:each)
-    b = cursor.map { |batch| batch.map { |r| r['id'] } }
+    b = cursor.map { |batch| batch.map { |r| r["id"] } }
     assert_equal 10, b.size
     cursor = Product.each_row_batch_by_sql("select * from products", block_size: 100)
     assert cursor.respond_to?(:each)
-    b = cursor.map { |batch| batch.map { |r| r['id'] } }
+    b = cursor.map { |batch| batch.map { |r| r["id"] } }
     assert_equal 10, b.size
   end
 
@@ -162,9 +193,9 @@ class TestPostgresqlCursor < Minitest::Test
 
   def test_with_hold
     items = 0
-    Product.where("id < 4") .each_instance(with_hold: true, block_size:1) do |row|
+    Product.where("id < 4").each_instance(with_hold: true, block_size: 1) do |row|
       Product.transaction do
-        row.update(data:Time.now.to_f.to_s)
+        row.update(data: Time.now.to_f.to_s)
         items += 1
       end
     end
@@ -173,22 +204,25 @@ class TestPostgresqlCursor < Minitest::Test
 
   def test_fetch_symbolize_keys
     Product.transaction do
-      #cursor = PostgreSQLCursor::Cursor.new("select * from products order by 1")
+      # cursor = PostgreSQLCursor::Cursor.new("select * from products order by 1")
       cursor = Product.all.each_row
       r = cursor.fetch
       assert r.has_key?("id")
-      r = cursor.fetch(symbolize_keys:true)
+      r = cursor.fetch(symbolize_keys: true)
       assert r.has_key?(:id)
       cursor.close
     end
   end
 
   def test_bad_sql
-    begin
-      ActiveRecord::Base.each_row_by_sql('select * from bad_table') { }
-      raise "Did Not Raise Expected Exception"
-    rescue Exception => e
-      assert_match(/bad_table/, e.message)
-    end
+    ActiveRecord::Base.each_row_by_sql("select * from bad_table") {}
+    raise "Did Not Raise Expected Exception"
+  rescue => e
+    assert_match(/bad_table/, e.message)
+  end
+
+  def test_relation_association_is_not_loaded
+    cursor = Product.first.prices.each_instance
+    refute cursor.instance_variable_get(:@type).loaded?
   end
 end

data/test-app/Gemfile

@@ -1,15 +1,8 @@
 # A sample Gemfile
 source "https://rubygems.org"
 
-gem 'activerecord', '~> 5.2.0'
-#gem 'activerecord', '~> 3.2.0'
-#gem 'activerecord', '~> 4.0.0'
-#gem 'activerecord', '~> 4.1.0'
-#gem 'activerecord', '4.1.2.rc1'
+gem "activerecord", "~> 7.0.4.1"
+# gem 'activerecord', '~> 6.1.7.1'
 
-# For testing against Edge Rails
-#gem 'activerecord', github: 'rails/rails', branch: 'master'
-#gem 'arel', github: 'rails/arel', branch: 'master'
-
-gem 'pg'
-gem 'postgresql_cursor', path:"../"
+gem "pg"
+gem "postgresql_cursor", path: "../"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: postgresql_cursor
 version: !ruby/object:Gem::Version
-  version: 0.6.4
+  version: 0.6.8
 platform: ruby
 authors:
 - Allen Fair
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-10-03 00:00:00.000000000 Z
+date: 2023-01-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -16,14 +16,28 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.1.0
+        version: '6.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 3.1.0
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: irb
   requirement: !ruby/object:Gem::Requirement
@@ -53,7 +67,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: pg
+  name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -67,7 +81,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: appraisal
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -94,11 +108,15 @@ extra_rdoc_files: []
 files:
 - ".document"
 - ".gitignore"
+- ".travis.yml"
+- Appraisals
 - Gemfile
-- Gemfile.lock
 - LICENSE
 - README.md
 - Rakefile
+- gemfiles/activerecord_4.gemfile
+- gemfiles/activerecord_5.gemfile
+- gemfiles/activerecord_6.gemfile
 - lib/postgresql_cursor.rb
 - lib/postgresql_cursor/active_record/connection_adapters/postgresql_type_map.rb
 - lib/postgresql_cursor/active_record/relation/cursor_iterators.rb
@@ -116,7 +134,7 @@ homepage: http://github.com/afair/postgresql_cursor
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -131,11 +149,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.6
-signing_key: 
+rubygems_version: 3.3.7
+signing_key:
 specification_version: 4
 summary: ActiveRecord PostgreSQL Adapter extension for using a cursor to return a
   large result set
-test_files:
-- test/helper.rb
-- test/test_postgresql_cursor.rb
+test_files: []

data/Gemfile.lock

@@ -1,44 +0,0 @@
-PATH
-  remote: .
-  specs:
-    postgresql_cursor (0.6.4)
-      activerecord (>= 3.1.0)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    activemodel (6.0.0)
-      activesupport (= 6.0.0)
-    activerecord (6.0.0)
-      activemodel (= 6.0.0)
-      activesupport (= 6.0.0)
-    activesupport (6.0.0)
-      concurrent-ruby (~> 1.0, >= 1.0.2)
-      i18n (>= 0.7, < 2)
-      minitest (~> 5.1)
-      tzinfo (~> 1.1)
-      zeitwerk (~> 2.1, >= 2.1.8)
-    concurrent-ruby (1.1.5)
-    i18n (1.6.0)
-      concurrent-ruby (~> 1.0)
-    irb (1.0.0)
-    minitest (5.12.0)
-    pg (1.1.4)
-    rake (13.0.0)
-    thread_safe (0.3.6)
-    tzinfo (1.2.5)
-      thread_safe (~> 0.1)
-    zeitwerk (2.1.10)
-
-PLATFORMS
-  ruby
-
-DEPENDENCIES
-  irb
-  minitest
-  pg
-  postgresql_cursor!
-  rake
-
-BUNDLED WITH
-   1.17.3