changes for after create hooks

Author: Keith Gabryelski

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    partitioned (1.1.3)
+    partitioned (1.1.4)
       bulk_data_methods (= 1.0.0)
       pg
       rails (>= 3.2.8)
@@ -48,22 +48,22 @@ GEM
     hike (1.2.1)
     i18n (0.6.1)
     journey (1.0.4)
-    jquery-rails (2.1.4)
+    jquery-rails (2.2.0)
       railties (>= 3.0, < 5.0)
       thor (>= 0.14, < 2.0)
     json (1.7.6)
     mail (2.4.4)
       i18n (>= 0.4.0)
       mime-types (~> 1.16)
       treetop (~> 1.4.8)
-    mime-types (1.19)
+    mime-types (1.20.1)
     multi_json (1.5.0)
     pg (0.14.1)
     polyglot (0.3.3)
     rack (1.4.4)
     rack-cache (1.2)
       rack (>= 0.4)
-    rack-ssl (1.3.2)
+    rack-ssl (1.3.3)
       rack
     rack-test (0.6.2)
       rack (>= 1.0)
@@ -88,7 +88,7 @@ GEM
     rspec-core (2.12.2)
     rspec-expectations (2.12.1)
       diff-lcs (~> 1.1.3)
-    rspec-mocks (2.12.1)
+    rspec-mocks (2.12.2)
     rspec-rails (2.12.2)
       actionpack (>= 3.0)
       activesupport (>= 3.0)
@@ -101,7 +101,7 @@ GEM
       multi_json (~> 1.0)
       rack (~> 1.0)
       tilt (~> 1.1, != 1.3.0)
-    thor (0.16.0)
+    thor (0.17.0)
     tilt (1.3.3)
     treetop (1.4.12)
       polyglot

LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2010-2012, Fiksu, Inc.
+Copyright (c) 2010-2013, Fiksu, Inc.
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without

lib/partitioned/partitioned_base/configurator/data.rb

@@ -42,7 +42,8 @@ def self.foreign_key_to_foreign_table_name(foreign_key_field)
            :schema_name, :name_prefix, :base_name,
            :part_name, :table_name, :table_alias_name, :parent_table_schema_name,
            :parent_table_name, :check_constraint, :encoded_name,
-           :janitorial_creates_needed, :janitorial_archives_needed, :janitorial_drops_needed
+           :janitorial_creates_needed, :janitorial_archives_needed, :janitorial_drops_needed,
+           :after_partition_table_create_hooks
 
         def initialize
           @on_field = nil
@@ -70,6 +71,8 @@ def initialize
           @janitorial_creates_needed = nil
           @janitorial_archives_needed = nil
           @janitorial_drops_needed = nil
+
+          @after_partition_table_create_hooks = []
         end
       end
     end

lib/partitioned/partitioned_base/configurator/dsl.rb

@@ -263,6 +263,16 @@ def foreign_key(referencing_field, referenced_table = nil, referenced_field = :i
           end
         end
 
+        def after_partition_table_create_hook(method_name)
+          if method_name.is_a? Proc
+            data.after_partition_table_create_hooks << method_name
+          else
+            # XXX should be a symbol
+            data.after_partition_table_create_hooks << lambda {|model,*partition_key_values| model.send(method_name, *partition_key_values)}
+          end
+        end
+
+
         #
         # Define the check constraint for a given child table.
         #

lib/partitioned/partitioned_base/configurator/reader.rb

@@ -30,6 +30,7 @@ def initialize(most_derived_activerecord_class)
           @janitorial_creates_needed = nil
           @janitorial_archives_needed = nil
           @janitorial_drops_needed = nil
+          @after_partition_table_create_hooks = nil
         end
 
         #
@@ -167,6 +168,10 @@ def janitorial_drops_needed
           return @janitorial_drops_needed
         end
 
+        def run_after_partition_table_create_hooks(*partition_key_values)
+          collect_from_collection(*partition_key_values, &:after_partition_table_create_hooks)
+        end
+
         protected
 
         def configurators

lib/partitioned/partitioned_base/partition_manager.rb

@@ -127,6 +127,7 @@ def create_new_partition(*partition_key_values)
         create_partition_table(*partition_key_values)
         add_partition_table_index(*partition_key_values)
         add_references_to_partition_table(*partition_key_values)
+        configurator.run_after_partition_table_create_hooks(*partition_key_values)
       end
 
       ##

lib/partitioned/version.rb

@@ -1,4 +1,4 @@
 module Partitioned
   # the current version of this gem
-  VERSION = "1.1.3"
+  VERSION = "1.1.4"
 end

partitioned.gemspec

@@ -7,7 +7,7 @@ Gem::Specification.new do |s|
  s.name        = 'partitioned'
  s.version     = Partitioned::VERSION
  s.license     = 'New BSD License'
- s.date        = '2013-10-14'
+ s.date        = '2013-02-04'
  s.summary     = "Postgres table partitioning support for ActiveRecord."
  s.description = "A gem providing support for table partitioning in ActiveRecord. Support is only available for postgres databases. Other features include child table management (creation and deletion) and bulk data creating and updating."
  s.authors     = ["Keith Gabryelski", "Aleksandr Dembskiy"]