foreign_key_checker 0.2.2 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cefd15f97041e89257c6487ba55dfeed549c4b97abcd577af614e38a2a23c54b
-  data.tar.gz: 322939db9133f937b95bb7f7324c418eae65758a0f4c50f422abc0387e7706ea
+  metadata.gz: 04ba348e8860c85305d58ec740c7d49dd318a11f2d80fff5905e434281b0f75e
+  data.tar.gz: 0e40f9cff0e3af05525c529217c28eb360e008f6e20cf1b6268d0146c71955e7
 SHA512:
-  metadata.gz: d8a8b3c7e1e308935781008935029c850d57db6cfb17127eced235580f3b2eaf75dcd9d2b8d1692130e4f83f6dabead19410fe7d15c886b1105b3524a66db60e
-  data.tar.gz: aab87280fdcba657ac5737ac4bf793713528d689f95d7a73ffaa558f3ab75bc107af6b61dd635b6faaa7a6df143973cf77e618b2687013d10f930ff0184a2ca2
+  metadata.gz: f66ab3eacf8fd3238740483184683f871bbf0cd482ae3fa644d7120a8f01bde616ef571ab19638c6bae9e2a677650838370858a9f2e8ab776829a17d48a75f8a
+  data.tar.gz: 53fd8d518399d7b61ee42cb7e47ac01f5604852e724e78971c7fd63a2009ac8b95afbc4ddd17a433da453dc7b4b0427d3c813a241015c0525b233473af9bb319

data/README.md

@@ -23,6 +23,12 @@ ForeignKeyChecker.check.each do |key, result|
 end
 ```
 
+Get general information about foreign keys (only postgresql, mysql2)
+```ruby
+ForeignKeyChecker::Utils.get_foreign_keys_hash
+# =>  {"users"=>[#<ForeignKeyChecker::Utils::Result:0x00005645e51756e8 @from_table="user_rating_changes", @from_column="user_id", @to_table="users", @to_column="id">]}
+```
+
 ## Installation
 Add this line to your application's Gemfile:
 

data/lib/foreign_key_checker.rb

@@ -1,4 +1,5 @@
 require "foreign_key_checker/railtie"
+require 'foreign_key_checker/utils'
 
 module ForeignKeyChecker
   class TypeMismatch < StandardError; end

data/lib/foreign_key_checker/utils.rb

@@ -0,0 +1,61 @@
+module ForeignKeyChecker
+  module Utils
+    class Result
+      attr_reader :from_table, :to_table, :from_column, :to_column
+      def initialize(args)
+        args.each { |k, v| instance_variable_set("@#{k}", v) }
+      end
+    end
+    class UnsupportedConnectionAdapter < StandardError; end
+    def self.get_foreign_keys(model = ActiveRecord::Base)
+      adapter = model.connection_config[:adapter]
+      raise(UnsupportedConnectionAdapter, adapter) unless %w[postgresql mysql2].include?(adapter)
+
+      connection = model.connection
+      send("get_#{adapter}_foreign_keys", connection)
+    end
+
+    def self.get_foreign_keys_hash(model = ActiveRecord::Base)
+      get_foreign_keys(model).to_a.each_with_object({}) do |datum, obj|
+        obj[datum.to_table] ||= []
+        obj[datum.to_table].push(datum)
+      end
+    end
+
+    def self.get_mysql2_foreign_keys(connection)
+      res = connection.select_all <<-SQL
+        SELECT
+          fks.TABLE_NAME AS from_table,
+          fks.COLUMN_NAME AS from_column,
+          fks.REFERENCED_TABLE_NAME AS to_table,
+          fks.REFERENCED_COLUMN_NAME AS to_column
+        FROM information_schema.KEY_COLUMN_USAGE AS fks
+        INNER JOIN information_schema.REFERENTIAL_CONSTRAINTS rules ON rules.CONSTRAINT_NAME = fks.CONSTRAINT_NAME
+        WHERE
+          fks.CONSTRAINT_SCHEMA = DATABASE()
+          AND rules.CONSTRAINT_SCHEMA = DATABASE();
+      SQL
+      res.to_a.map{|i| Result.new(i) }
+    end
+
+    def self.get_postgresql_foreign_keys(connection)
+      res = connection.select_all <<-SQL
+        SELECT
+          tc.table_name AS from_table,
+          kcu.column_name AS from_column,
+          ccu.table_name AS to_table,
+          ccu.column_name AS to_column
+        FROM
+          information_schema.table_constraints AS tc
+          JOIN information_schema.key_column_usage AS kcu
+            ON tc.constraint_name = kcu.constraint_name
+            AND tc.table_schema = kcu.table_schema
+          JOIN information_schema.constraint_column_usage AS ccu
+            ON ccu.constraint_name = tc.constraint_name
+            AND ccu.table_schema = tc.table_schema
+        WHERE tc.constraint_type = 'FOREIGN KEY';
+      SQL
+      res.to_a.map{ |i| Result.new(i) }
+    end
+  end
+end

data/lib/foreign_key_checker/version.rb

@@ -1,3 +1,3 @@
 module ForeignKeyChecker
-  VERSION = '0.2.2'
+  VERSION = '0.3.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: foreign_key_checker
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.0
 platform: ruby
 authors:
 - AnatolyShirykalov
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-10-24 00:00:00.000000000 Z
+date: 2019-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -50,6 +50,7 @@ files:
 - Rakefile
 - lib/foreign_key_checker.rb
 - lib/foreign_key_checker/railtie.rb
+- lib/foreign_key_checker/utils.rb
 - lib/foreign_key_checker/version.rb
 - lib/generators/foreign_key_checker/migration/migration_generator.rb
 - lib/generators/foreign_key_checker/migration/templates/migrations/fix_foreign_keys.rb.erb