after_migrate 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a04f2af70c48d8299c3c1eedce77608a6a57aca044711bbc2c0e8030299d18e3
-  data.tar.gz: d511331553aacbe753d40b2004af8d523b2c823ef4ba425f8491eb8a0336d234
+  metadata.gz: 2602aa9ee3d2ffa2c1ea85ca8cbfa1b101679aa42ed05e0a4cfa4c742f258f59
+  data.tar.gz: d581242202742eb6a62be1cd1799a49c196ca82c99e8b36428ee6390fd12c415
 SHA512:
-  metadata.gz: c3f07535162f76e33ff8a7a399bbdee3b45d534b48a5609c600504a1f8b0e9c8d4d2e027ec653b7106c12e6663f91194370a9617912bd35860cdb77e6c451e6d
-  data.tar.gz: 8c529f7d643720957caf72ae2607055e286dbcafc7c15004a03e3908a6be9eab15c7dcc9c66b83c1b741265c70a2a1476e440846178bea2799a4884857639187
+  metadata.gz: d81dce8555f469b190b3cfd37f7eb2b919aab714e6fcfb1842271c323495b213be2c2e779c8400ad67d28832c11dbc7ad7147e90f93b6950069b6b6b9ea8d558
+  data.tar.gz: 77f23e51b264cf80b5cbf028a0e43106d13498f9f7d109b1eafe26c6380269c08bb3574946ec534d08b8ac175a09c008823e64d0c862c4e4441e66c1df145399

data/README.md

@@ -28,7 +28,6 @@ Stale statistics and fragmentation after schema changes silently hurt query perf
 - No monkey-patching of ActiveRecord core classes
 - Works in development, test, CI, and production
 - Rails 7.0+ / Ruby 3.2+ only
-- Dependency-free
 
 ---
 

data/after_migrate.gemspec

@@ -43,5 +43,6 @@ Gem::Specification.new do |spec|
 
   spec.required_ruby_version = '>= 3.2'
 
+  spec.add_dependency 'pg_query', '>= 6.1'
   spec.add_dependency 'rails', '>= 7.0'
 end

data/lib/after_migrate/adapters/mysql.rb

@@ -2,6 +2,8 @@
 
 module AfterMigrate
   module Mysql
+    extend Sql
+
     module_function
 
     def optimize_tables(connection:, table_names:, **)

data/lib/after_migrate/adapters/postgresql.rb

@@ -1,7 +1,11 @@
 # frozen_string_literal: true
 
+require 'pg_query'
+
 module AfterMigrate
   module Postgresql
+    extend Sql
+
     module_function
 
     def vacuum(table_name, schema: nil, verbose: true)
@@ -64,22 +68,26 @@ module AfterMigrate
       tables = table_names - cleaned_tables
       run_analyze(schema:, tables:)
     end
-  end
 
-  def all_tables(schema: nil)
-    connection = ActiveRecord::Base.connection
-    schema_value = schema ? schema.to_s.strip : 'public'
+    def all_tables(schema: nil)
+      connection = ActiveRecord::Base.connection
+      schema_value = schema ? schema.to_s.strip : 'public'
 
-    query = <<~SQL.squish
-      SELECT c.relname AS table_name
-      FROM pg_class c
-      JOIN pg_namespace n ON n.oid = c.relnamespace
-      WHERE n.nspname = #{connection.quote(schema_value)}
-        AND c.relkind IN ('r', 'p')        -- ordinary tables + partitioned tables
-        AND c.relispartition = FALSE      -- exclude partition child tables
-      ORDER BY table_name
-    SQL
+      query = <<~SQL.squish
+        SELECT c.relname AS table_name
+        FROM pg_class c
+        JOIN pg_namespace n ON n.oid = c.relnamespace
+        WHERE n.nspname = #{connection.quote(schema_value)}
+          AND c.relkind IN ('r', 'p')        -- ordinary tables + partitioned tables
+          AND c.relispartition = FALSE      -- exclude partition child tables
+        ORDER BY table_name
+      SQL
+
+      connection.select_values(query)
+    end
 
-    connection.select_values(query)
+    def parse_tables(sql)
+      PgQuery.parse(sql).tables
+    end
   end
 end

data/lib/after_migrate/adapters/sql.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module AfterMigrate
+  module Sql
+    IDENT = /
+      (?:"[\w]+"|\w+)
+      (?:\.(?:"[\w]+"|\w+))*
+    /x
+
+    PATTERNS = {
+      update: /update\s+(?:only\s+)?(#{IDENT})(?!\s*\()/ix,
+      insert: /insert\s+into\s+(#{IDENT})(?!\s*\()/ix,
+      delete: /delete\s+from\s+(#{IDENT})(?!\s*\()/ix,
+      drop_table: /drop\s+table\s+(?:if\s+exists\s+)?(#{IDENT})(?!\s*\()/ix,
+      alter_table: /alter\s+table\s+(#{IDENT})(?!\s*\()/ix,
+      create_table: /create\s+table\s+(?:if\s+not\s+exists\s+)?(#{IDENT})(?!\s*\()/ix,
+      from_join: /(?:from|join)\s+(#{IDENT})(?!\s*\()/ix
+    }.freeze
+
+    def parse_tables(sql)
+      PATTERNS.flat_map { |_, r| sql.scan(r).flatten }.uniq
+    end
+  end
+end

data/lib/after_migrate/adapters/sqlite.rb

@@ -2,6 +2,8 @@
 
 module AfterMigrate
   module Sqlite
+    extend Sql
+
     module_function
 
     def optimize_tables(connection:, **)

data/lib/after_migrate/collector.rb

@@ -12,9 +12,9 @@ module AfterMigrate
       return unless sql
       return unless sql.match?(/\A\s*(CREATE|ALTER|DROP|INSERT|UPDATE|DELETE|RENAME\s+TABLE|TRUNCATE)/i)
 
-      table_names = sql.scan(/(?:from|join|update|into|table)\s+((?:"\w+"|\w+)(?:\.(?:"\w+"|\w+))*)/i).flatten
+      table_names = parse_tables(sql)
       schema ||= fetch_schema
-      # AfterMigrate.log("[#{schema}] Detected change from '#{sql}' to table: #{table_names}") if table_names.present?
+      # AfterMigrate.log("[#{schema}] Detected change from '#{sql}' to tables: #{table_names}") if table_names.present?
       collect_tables(schema:, table_names:)
     end
 
@@ -36,5 +36,20 @@ module AfterMigrate
         quoted&.delete('"')
       end
     end
+
+    def parse_tables(sql)
+      connection = ActiveRecord::Base.connection
+      adapter = connection.adapter_name
+      case adapter
+      when 'PostgreSQL'
+        AfterMigrate::Postgresql.parse_tables(sql)
+      when 'SQLite'
+        AfterMigrate::Sqlite.parse_tables(sql)
+      when 'Mysql2', 'Trilogy'
+        AfterMigrate::Mysql.parse_tables(sql)
+      else
+        AfterMigrate.log("No maintenance implemented for #{adapter}")
+      end
+    end
   end
 end

data/lib/after_migrate/executor.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'after_migrate/adapters/sql'
 require 'after_migrate/adapters/mysql'
 require 'after_migrate/adapters/postgresql'
 require 'after_migrate/adapters/sqlite'

data/lib/after_migrate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AfterMigrate
-  VERSION = '0.1.0'
+  VERSION = '0.1.1'
 end

metadata

@@ -1,14 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: after_migrate
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Nikolay Moskvin
+autorequire:
 bindir: exe
 cert_chain: []
-date: 1980-01-02 00:00:00.000000000 Z
+date: 2025-12-11 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg_query
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.1'
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
@@ -48,6 +63,7 @@ files:
 - lib/after_migrate.rb
 - lib/after_migrate/adapters/mysql.rb
 - lib/after_migrate/adapters/postgresql.rb
+- lib/after_migrate/adapters/sql.rb
 - lib/after_migrate/adapters/sqlite.rb
 - lib/after_migrate/collector.rb
 - lib/after_migrate/current.rb
@@ -63,6 +79,7 @@ metadata:
   source_code_uri: https://github.com/moskvin/after_migrate
   changelog_uri: https://github.com/moskvin/after_migrate/blob/master/CHANGELOG.md
   rubygems_mfa_required: 'true'
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -77,7 +94,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 3.4.19
+signing_key:
 specification_version: 4
 summary: Automatically ANALYZE and VACUUM tables touched during Rails migrations.
 test_files: []