estimate_count 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4820f4b135cde08c8d827ebe9232a0b7fba84d33deefb0b866b904b14cddc754
-  data.tar.gz: 303e049e4e60d36dfbe53b3c58fdbbe3514b9c40c628d79bf42bc779ce30fc15
+  metadata.gz: a183b9931539602b197f560cb1e2f3c1a23735e0ac88f3a1bdfe1e7d3055420a
+  data.tar.gz: 1399ce07b08369ab9a30e2a92f70598d8e502da7c7505bc1970c331a461d9ff8
 SHA512:
-  metadata.gz: 027fe306e6a234b4b3cfd0d810f44990186e5e9e5d59a3e8e5f2ead9acd3e14817729577aad748bb4baccaaafc673cef8fe21d79fbd1cf4ad32e99912c618623
-  data.tar.gz: 89dd325d30d773c160681a4e48660cc0239113ee9b41cef257292b93f75595644af8afab1677beb8ecc108650b8198275b6ecdf829ccdee164aa03de36d91741
+  metadata.gz: 202a74c4860aede412634c0dd9962b24c31663db14970fedd1dde06ff1de9e5fc12fede2d59ca5ca9e54210094721f06f37a22e4a60b45fc3ef3eb0282917e8f
+  data.tar.gz: e825b7766d5eb08d63fbd59039f995bb4fb15c82602de9b42f1f0db1d4ffe6327be7b864e47b5ec575f081ad1ac5f354cc36bae9aa5d1d15824c8649d093379d

data/Gemfile

@@ -8,3 +8,5 @@ gemspec
 gem "rake", "~> 13.0"
 
 gem "rspec", "~> 3.0"
+
+gem "pry"

data/Gemfile.lock

@@ -3,27 +3,32 @@ PATH
   specs:
     estimate_count (0.1.0)
       activerecord
-      pg
 
 GEM
   remote: https://rubygems.org/
   specs:
-    activemodel (7.0.3.1)
-      activesupport (= 7.0.3.1)
-    activerecord (7.0.3.1)
-      activemodel (= 7.0.3.1)
-      activesupport (= 7.0.3.1)
-    activesupport (7.0.3.1)
+    activemodel (7.0.4.3)
+      activesupport (= 7.0.4.3)
+    activerecord (7.0.4.3)
+      activemodel (= 7.0.4.3)
+      activesupport (= 7.0.4.3)
+    activesupport (7.0.4.3)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 1.6, < 2)
       minitest (>= 5.1)
       tzinfo (~> 2.0)
-    concurrent-ruby (1.1.10)
+    coderay (1.1.3)
+    concurrent-ruby (1.2.2)
     diff-lcs (1.5.0)
-    i18n (1.12.0)
+    i18n (1.13.0)
       concurrent-ruby (~> 1.0)
-    minitest (5.16.2)
+    method_source (1.0.0)
+    minitest (5.18.0)
+    mysql2 (0.5.5)
     pg (1.4.1)
+    pry (0.14.2)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
     rake (13.0.6)
     rspec (3.11.0)
       rspec-core (~> 3.11.0)
@@ -38,17 +43,22 @@ GEM
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.11.0)
     rspec-support (3.11.0)
-    tzinfo (2.0.4)
+    tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
 
 PLATFORMS
+  arm64-darwin-21
   arm64-darwin-22
+  ruby
+  x86_64-linux
 
 DEPENDENCIES
-  activerecord
   estimate_count!
+  mysql2
+  pg
+  pry
   rake (~> 13.0)
   rspec (~> 3.0)
 
 BUNDLED WITH
-   2.3.26
+   2.3.10

data/docker-compose.yml

@@ -0,0 +1,27 @@
+version: "3"
+
+services:
+  mysql:
+      image: mysql:8.0
+      environment:
+          MYSQL_ROOT_PASSWORD: root
+          MYSQL_DATABASE: test
+          MYSQL_USER: test
+          MYSQL_PASSWORD: test
+      ports:
+          - 3306:3306
+      volumes:
+          - ./db/mysql:/var/lib/mysql
+      healthcheck:
+          test: [ "CMD", "mysql", "-u", "root", "-e", "USE test;" ]
+          retries: 1
+  postgresql:
+      image: postgres:15-alpine
+      environment:
+          POSTGRES_USER: test
+          POSTGRES_PASSWORD: test
+          POSTGRES_DB: test
+      ports:
+          - 5432:5432
+      volumes:
+          - ./db/postgresql:/var/lib/postgresql/data

data/lib/estimate_count/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module EstimateCount
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/estimate_count.rb

@@ -1,18 +1,36 @@
 # frozen_string_literal: true
 
 require_relative "estimate_count/version"
+require "active_record"
 
 module ActiveRecord
   class Base
     def self.estimate_count(threshold: 1000)
-      full_scope = current_scope.limit(nil).offset(nil)
+      full_scope = current_scope&.limit(nil)&.offset(nil) || all
       rows = estimate_rows(full_scope.to_sql)
       rows = full_scope.count if threshold.is_a?(Integer) && rows < threshold
       rows
     end
 
     private_class_method def self.estimate_rows(query)
+      case connection.adapter_name
+      when "PostgreSQL"
+        estimate_rows_postgresql(query)
+      when "MySQL", "Mysql2"
+        estimate_rows_mysql(query)
+      else
+        raise "Unsupported database"
+      end
+    end
+
+    private_class_method def self.estimate_rows_postgresql(query)
       connection.execute("EXPLAIN #{query}").to_a.first["QUERY PLAN"].match(/rows=(\d+)/)[1].to_i
     end
+
+    private_class_method def self.estimate_rows_mysql(query)
+      result = connection.execute("EXPLAIN FORMAT=TRADITIONAL #{query}")
+      index = result.fields.index("rows")
+      result.first[index].to_i
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: estimate_count
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Chris Hasiński
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-04 00:00:00.000000000 Z
+date: 2023-05-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -31,7 +31,35 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-  type: :runtime
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  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: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
@@ -52,6 +80,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- docker-compose.yml
 - lib/estimate_count.rb
 - lib/estimate_count/version.rb
 - sig/estimate_count.rbs
@@ -77,7 +106,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.26
+rubygems_version: 3.2.32
 signing_key:
 specification_version: 4
 summary: Adds a method to get an estimate count for an ActiveRecord::Relation