pg_party 0.4.2 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 85fa2f5792258d5519df5df148eab1fce17d2f3a
-  data.tar.gz: f1881d32c2ba397dc1fc4ae99304480ee2234eed
+  metadata.gz: 2fdbe60dbdafaf57fb4fff60f9cc484640226e39
+  data.tar.gz: fdfa7113d06c6605e5c9ce6936ab688b14364b9d
 SHA512:
-  metadata.gz: 9f15450a444d4611195a21463b26e2fdcf840598e4893593df832f53ada93d9641e5935c8591c45f536dfb41a682e944a254b3a6e4b5e08f600ba46fdfff85b5
-  data.tar.gz: bf3c288225c95debf9c1e10060ed1c9b41eb96ec55f2d1d86931fadb0290acda0c758a8815c9b6927f0f7d144483595710f62e30736ad37bcf5354bf22d4b02d
+  metadata.gz: f50e755596d95d0063436bd0a65a045103d1e3fc89f3657b69c1204a6cb73f6172dcd90e27152c072b47f8a631bf9c526a56a1096444323bc4b93398852ba03c
+  data.tar.gz: eeddf72d17854248814a845470238695b36d2852555b7ba98a2cec79d88ce32786f525722fc5271518acb38d5987abefebc8c55763f42d64592b14f41ac6072f

data/lib/pg_party.rb

@@ -11,11 +11,15 @@ ActiveSupport.on_load(:active_record) do
   ActiveRecord::ConnectionAdapters::AbstractAdapter.class_eval do
     include PgParty::Adapter::AbstractMethods
   end
+  
+  begin
+    require "active_record/connection_adapters/postgresql_adapter"
+    require "pg_party/adapter/postgresql_methods"
 
-  require "pg_party/adapter/postgresql_methods"
-  require "active_record/connection_adapters/postgresql_adapter"
-
-  ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.class_eval do
-    include PgParty::Adapter::PostgreSQLMethods
+    ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.class_eval do
+      include PgParty::Adapter::PostgreSQLMethods
+    end
+  rescue LoadError
+    # migration methods will not be available
   end
 end

data/lib/pg_party/adapter_decorator.rb

@@ -75,15 +75,20 @@ module PgParty
       modified_options[:id]      = false
       modified_options[:options] = "PARTITION BY #{type.to_s.upcase} ((#{quote_partition_key(partition_key)}))"
 
-      create_table(table_name, modified_options) do |td|
+      result = create_table(table_name, modified_options) do |td|
         if id == :uuid
-          td.send(id, primary_key, null: false, default: uuid_function)
+          td.column(primary_key, id, null: false, default: uuid_function)
         elsif id
-          td.send(id, primary_key, null: false)
+          td.column(primary_key, id, null: false)
         end
 
         yield(td) if block_given?
       end
+
+      # Rails 4 has a bug where uuid columns are always nullable
+      change_column_null(table_name, primary_key, false) if id == :uuid
+
+      result
     end
 
     def create_partition_of(table_name, child_table_name, constraint_clause, **options)
@@ -93,13 +98,12 @@ module PgParty
 
       raise ArgumentError, "composite primary key not supported" if primary_key.is_a?(Array)
 
-      partition_clause = <<-SQL
+      execute(<<-SQL)
+        CREATE TABLE #{quote_table_name(child_table_name)}
         PARTITION OF #{quote_table_name(table_name)}
         FOR VALUES #{constraint_clause}
       SQL
 
-      create_table(child_table_name, id: false, options: partition_clause)
-
       if primary_key
         execute(<<-SQL)
           ALTER TABLE #{quote_table_name(child_table_name)}
@@ -108,7 +112,13 @@ module PgParty
       end
 
       if index && partition_key && primary_key != partition_key
-        add_index(child_table_name, "((#{quote_partition_key(partition_key)}))")
+        index_name = index_name(child_table_name, partition_key)
+
+        execute(<<-SQL)
+          CREATE INDEX #{quote_table_name(index_name)}
+          ON #{quote_table_name(child_table_name)}
+          USING btree ((#{quote_partition_key(partition_key)}))
+        SQL
       end
 
       child_table_name
@@ -127,7 +137,11 @@ module PgParty
     end
 
     def supports_partitions?
-      postgresql_version >= 100000
+      __getobj__.send(:postgresql_version) >= 100000
+    end
+
+    def index_name(table_name, partition_key)
+      "index_#{table_name}_on_#{partition_key.to_s.split("::").join("_")}"
     end
   end
 end

data/lib/pg_party/model/shared_methods.rb

@@ -3,6 +3,16 @@ require "pg_party/model_decorator"
 module PgParty
   module Model
     module SharedMethods
+      def reset_primary_key
+        PgParty::ModelDecorator.new(self).partition_primary_key
+      end
+
+      def table_exists?
+        return @table_exists if defined?(@table_exists)
+
+        @table_exists = PgParty::ModelDecorator.new(self).partition_table_exists?
+      end
+
       def partitions
         PgParty::ModelDecorator.new(self).partitions
       end

data/lib/pg_party/model_decorator.rb

@@ -1,5 +1,21 @@
 module PgParty
   class ModelDecorator < SimpleDelegator
+    def partition_primary_key
+      if self != base_class
+        base_class.primary_key
+      elsif partition = partitions.first
+        child_class(partition).get_primary_key(base_class.name)
+      else
+        get_primary_key(base_class.name)
+      end
+    end
+
+    def partition_table_exists?
+      target_table = partitions.first || table_name
+
+      connection.schema_cache.send(table_exists_method, target_table)
+    end
+
     def in_partition(table_name)
       child_class(table_name).all
     end
@@ -51,6 +67,12 @@ module PgParty
 
     private
 
+    def table_exists_method
+      [:data_source_exists?, :table_exists?].detect do |meth|
+        connection.schema_cache.respond_to?(meth)
+      end
+    end
+
     def child_class(table_name)
       Class.new(__getobj__) do
         self.table_name = table_name

data/lib/pg_party/version.rb

@@ -1,3 +1,3 @@
 module PgParty
-  VERSION = "0.4.2"
+  VERSION = "0.5.0"
 end

metadata

@@ -1,29 +1,35 @@
 --- !ruby/object:Gem::Specification
 name: pg_party
 version: !ruby/object:Gem::Version
-  version: 0.4.2
+  version: 0.5.0
 platform: ruby
 authors:
 - Ryan Krage
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-10-07 00:00:00.000000000 Z
+date: 2017-10-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.2'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '5.0'
+        version: '6'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.2'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '5.0'
+        version: '6'
 - !ruby/object:Gem::Dependency
   name: pg
   requirement: !ruby/object:Gem::Requirement