pg_partitioner 0.4.3 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 45af2ccf7905618261ce0c46ad347356e2cb46144d0298c379777c55de3d4ec4
-  data.tar.gz: 21b9ee9fcd048f40c82ac119e77ca2e26e5f549f51ef9e1754017d0720402b64
+  metadata.gz: c2fde7de994fd9c1895b12d6347fce42e8fa84a622ad917ac73ed90d249daa27
+  data.tar.gz: '08bf1490d4a4462890291353eaadeec9a88cf101e54ccb982af8633c3eeab868'
 SHA512:
-  metadata.gz: fd11ac4386b85420f1ac10ea28fa9e84980a7cd691b2fe37b553a2c9c18ab1af310ef312006379df39af76ef28059bb727d1997c71e105c506347521b10d5ee3
-  data.tar.gz: 5d3ec6a976b909682f0c1ac46a3a8f9a1c6b5ac7ef972b6e1a07a77e2ccd6737a1957f657172d4b7ea6247849a5b5664e729ff3b11cc8230dc8edd4433f0093d
+  metadata.gz: 726173d02f3c6f173e784667f0543cf51d95170b9eb1446db7138707ebefa45774f727b25be8a5eb5cb3e6d3efa87f279817acc4ddd7e40772f08e735c2b30db
+  data.tar.gz: 571636fac008a954653af4c54b34eb43d3e1080c8dbd55a7c5aa4fe3357620849372ade89b97646c2b75daab6b5771276e9280c7c1e55ff4ff1c6e4b0b0154de

data/README.md

@@ -61,6 +61,16 @@ rails g migration make_partitioning_of_you_table
 
 add instructions to migration:
 
+```ruby
+class YouMigrationClassName
+  def change
+    YourModel.create_partitioning_by_week_trigger_sql
+    YourModel.create_current_week_table
+    YourModel.create_next_week_table
+  end
+end
+```
+
 ```ruby
 class YouMigrationClassName
   def change
@@ -71,9 +81,27 @@ class YouMigrationClassName
 end
 ```
 
-3) For correct work you need to create next partition every month.
+```ruby
+class YouMigrationClassName
+  def change
+    YourModel.create_partitioning_by_quater_trigger_sql
+    YourModel.create_current_quater_table
+    YourModel.create_next_quater_table
+  end
+end
+```
+
+3) For correct work you need to create next partition every month/quater.
   We recommend you to create a rake task and run it once a month by crontab. Code for a rake task:
 
 ```ruby
+YourModel.create_next_week_table
+```
+  or
+```ruby
 YourModel.create_next_month_table
 ```
+or
+```ruby
+YourModel.create_next_quater_table
+```

data/lib/pg_partitioner/separation_type/base.rb

@@ -0,0 +1,64 @@
+module PgPartitioner
+  module SeparationType
+    module Base
+      def drop_table(table_name)
+        sql = "DROP TABLE IF EXISTS #{table_name};"
+        execute_sql(sql)
+      end
+
+      def drop_partitioning_trigger_sql
+        sql = "DROP TRIGGER #{table_name}_insert ON #{table_name};
+         DROP FUNCTION #{table_name}_insert_trigger();
+         DROP TRIGGER #{table_name}_after_insert ON #{table_name};
+         DROP FUNCTION #{table_name}_delete_trigger();"
+
+        execute_sql(sql)
+      end
+
+      def create_partition_indexes(partition_table_name)
+        custom_indexes = partition_table_indexes.presence
+        return unless custom_indexes
+
+        custom_indexes.each { |custom_index| create_custom_index(partition_table_name, custom_index) }
+      end
+
+      def create_partition_named_indexes(partition_table_name)
+        custom_indexes = partition_table_named_indexes.presence
+        return unless custom_indexes
+
+        custom_indexes.map do |name, custom_index|
+          index_name = "index_#{partition_table_name}_#{name}"
+          create_custom_named_index(partition_table_name, custom_index, index_name)
+        end
+      end
+
+      def create_partition_unique_indexes(partition_table_name)
+        custom_unique_indexes = partition_table_unique_indexes.presence
+        return unless custom_unique_indexes
+
+        custom_unique_indexes.each { |custom_index| create_custom_index(partition_table_name, custom_index, true) }
+      end
+
+      def create_partition_named_unique_indexes(partition_table_name)
+        custom_indexes = partition_table_named_unique_indexes.presence
+        return unless custom_indexes
+
+        custom_indexes.map do |name, custom_index|
+          index_name = "index_#{partition_table_name}_#{name}"
+          create_custom_named_index(partition_table_name, custom_index, index_name, true)
+        end
+      end
+
+      def name_of_partition_table(date = Date.today, type:)
+        case type
+        when :month
+          date.strftime("#{table_name}_y%Ym%m")
+        when :quater
+          "#{table_name}_y#{date.year}q#{(((date.month - 1) / 3) + 1).to_i}"
+        when :week
+          "#{table_name}_y#{date.year}w#{date.cweek}"
+        end
+      end
+    end
+  end
+end

data/lib/pg_partitioner/separation_type/month.rb

@@ -10,13 +10,15 @@ module PgPartitioner
       end
 
       def drop_old_month_table
-        drop_month_table(Date.today.prev_month.prev_month)
+        table_name = name_of_partition_table(Date.today.prev_month.prev_month,
+                                             type: :quater)
+        drop_month_table(table_name)
       end
 
       def create_month_table(date = Date.today)
         date_start = date.at_beginning_of_month
         date_end = date.at_beginning_of_month.next_month
-        partition_table_name = name_of_partition_table(date)
+        partition_table_name = name_of_partition_table(date, type: :month)
         return 'Already exists' if connection.table_exists? partition_table_name
 
         sql = "CREATE TABLE IF NOT EXISTS #{partition_table_name} (
@@ -32,18 +34,13 @@ module PgPartitioner
         create_partition_named_unique_indexes(partition_table_name)
       end
 
-      def drop_month_table(date = Date.today)
-        sql = "DROP TABLE IF EXISTS #{name_of_partition_table(date)};"
-        execute_sql(sql)
-      end
-
       def create_partitioning_by_month_triggers
         sql = "CREATE OR REPLACE FUNCTION #{table_name}_insert_trigger() RETURNS trigger AS
     $$
            DECLARE
              curY varchar(4);
              curM varchar(2);
-             tbl varchar(63);
+             tbl varchar(121);
            BEGIN
               select cast(DATE_PART('year', new.#{parting_column}) as varchar) into curY;
               select lpad(cast(DATE_PART('month', new.#{parting_column}) as varchar), 2, '0') into curM;
@@ -78,53 +75,6 @@ module PgPartitioner
 
         execute_sql(sql)
       end
-
-      def drop_partitioning_by_month_trigger_sql
-        sql = "DROP TRIGGER #{table_name}_insert ON #{table_name};
-         DROP FUNCTION #{table_name}_insert_trigger();
-         DROP TRIGGER #{table_name}_after_insert ON #{table_name};
-         DROP FUNCTION #{table_name}_delete_trigger();"
-
-        execute_sql(sql)
-      end
-
-      def create_partition_indexes(partition_table_name)
-        custom_indexes = partition_table_indexes.presence
-        return unless custom_indexes
-
-        custom_indexes.each { |custom_index| create_custom_index(partition_table_name, custom_index) }
-      end
-
-      def create_partition_named_indexes(partition_table_name)
-        custom_indexes = partition_table_named_indexes.presence
-        return unless custom_indexes
-
-        custom_indexes.map do |name, custom_index|
-          index_name = "index_#{partition_table_name}_#{name}"
-          create_custom_named_index(partition_table_name, custom_index, index_name)
-        end
-      end
-
-      def create_partition_unique_indexes(partition_table_name)
-        custom_unique_indexes = partition_table_unique_indexes.presence
-        return unless custom_unique_indexes
-
-        custom_unique_indexes.each { |custom_index| create_custom_index(partition_table_name, custom_index, true) }
-      end
-
-      def create_partition_named_unique_indexes(partition_table_name)
-        custom_indexes = partition_table_named_unique_indexes.presence
-        return unless custom_indexes
-
-        custom_indexes.map do |name, custom_index|
-          index_name = "index_#{partition_table_name}_#{name}"
-          create_custom_named_index(partition_table_name, custom_index, index_name, true)
-        end
-      end
-
-      def name_of_partition_table(date = Date.today)
-        date.strftime("#{table_name}_y%Ym%m")
-      end
     end
   end
 end

data/lib/pg_partitioner/separation_type/quater.rb

@@ -0,0 +1,74 @@
+module PgPartitioner
+  module SeparationType
+    module Quater
+      def create_current_quater_table
+        create_quater_table(Date.today)
+      end
+
+      def create_next_quater_table
+        create_quater_table(Date.today.next_quarter)
+      end
+
+      def create_quater_table(date = Date.today)
+        date_start = date.at_beginning_of_quarter
+        date_end = date.end_of_quarter.next_month.at_beginning_of_month
+        partition_table_name = name_of_partition_table(date, type: :quater)
+        return 'Already exists' if connection.table_exists? partition_table_name
+
+        sql = "CREATE TABLE IF NOT EXISTS #{partition_table_name} (
+               CHECK ( #{parting_column} >= DATE('#{date_start}') AND #{parting_column} < DATE('#{date_end}') )
+               ) INHERITS (#{table_name});"
+        execute_sql(sql)
+        sql = "ALTER TABLE #{partition_table_name} ADD PRIMARY KEY (id);"
+        execute_sql(sql)
+
+        create_partition_indexes(partition_table_name)
+        create_partition_named_indexes(partition_table_name)
+        create_partition_unique_indexes(partition_table_name)
+        create_partition_named_unique_indexes(partition_table_name)
+      end
+
+      def create_partitioning_by_quater_triggers
+        sql = "CREATE OR REPLACE FUNCTION #{table_name}_insert_trigger() RETURNS trigger AS
+    $$
+           DECLARE
+             curY varchar(4);
+             curQ varchar(1);
+             tbl varchar(121);
+           BEGIN
+              select cast(DATE_PART('year', new.#{parting_column}) as varchar) into curY;
+              select lpad(cast(DATE_PART('quarter', new.#{parting_column}) as varchar), 2, '0') into curQ;
+              tbl := '#{table_name}_y' || curY || 'q' || curQ;
+              EXECUTE format('INSERT into %I values ($1.*);', tbl) USING NEW;
+              return NEW;
+           END;
+           $$
+         LANGUAGE plpgsql;
+
+         CREATE TRIGGER #{table_name}_insert
+         BEFORE INSERT ON #{table_name}
+         FOR EACH ROW
+         EXECUTE PROCEDURE #{table_name}_insert_trigger();
+
+         -- Trigger function to delete from the master table after the insert
+         CREATE OR REPLACE FUNCTION #{table_name}_delete_trigger() RETURNS trigger
+             AS $$
+         DECLARE
+             r #{table_name}%rowtype;
+         BEGIN
+             DELETE FROM ONLY #{table_name} where id = new.id returning * into r;
+             RETURN r;
+         end;
+         $$
+         LANGUAGE plpgsql;
+
+         CREATE TRIGGER #{table_name}_after_insert
+         AFTER INSERT ON #{table_name}
+         FOR EACH ROW
+         EXECUTE PROCEDURE #{table_name}_delete_trigger();"
+
+        execute_sql(sql)
+      end
+    end
+  end
+end

data/lib/pg_partitioner/separation_type/week.rb

@@ -0,0 +1,80 @@
+module PgPartitioner
+  module SeparationType
+    module Week
+      def create_current_week_table
+        create_week_table(Date.today)
+      end
+
+      def create_next_week_table
+        create_week_table(Date.today.next_week)
+      end
+
+      def drop_old_week_table
+        table_name = name_of_partition_table(Date.today.prev_month.prev_month,
+                                             type: :week)
+        drop_month_table(table_name)
+      end
+
+      def create_week_table(date = Date.today)
+        date_start = date.at_beginning_of_week
+        date_end = date.at_beginning_of_week.next_week
+        partition_table_name = name_of_partition_table(date, type: :week)
+        return 'Already exists' if connection.table_exists? partition_table_name
+
+        sql = "CREATE TABLE IF NOT EXISTS #{partition_table_name} (
+               CHECK ( #{parting_column} >= DATE('#{date_start}') AND #{parting_column} < DATE('#{date_end}') )
+               ) INHERITS (#{table_name});"
+        execute_sql(sql)
+        sql = "ALTER TABLE #{partition_table_name} ADD PRIMARY KEY (id);"
+        execute_sql(sql)
+
+        create_partition_indexes(partition_table_name)
+        create_partition_named_indexes(partition_table_name)
+        create_partition_unique_indexes(partition_table_name)
+        create_partition_named_unique_indexes(partition_table_name)
+      end
+
+      def create_partitioning_by_week_triggers
+        sql = "CREATE OR REPLACE FUNCTION #{table_name}_insert_trigger() RETURNS trigger AS
+    $$
+           DECLARE
+             curY varchar(4);
+             curW varchar(2);
+             tbl varchar(121);
+           BEGIN
+              select cast(DATE_PART('year', new.#{parting_column}) as varchar) into curY;
+              select lpad(cast(DATE_PART('week', new.#{parting_column}) as varchar), 2, '0') into curW;
+              tbl := '#{table_name}_y' || curY || 'w' || curW;
+              EXECUTE format('INSERT into %I values ($1.*);', tbl) USING NEW;
+              return NEW;
+           END;
+           $$
+         LANGUAGE plpgsql;
+
+         CREATE TRIGGER #{table_name}_insert
+         BEFORE INSERT ON #{table_name}
+         FOR EACH ROW
+         EXECUTE PROCEDURE #{table_name}_insert_trigger();
+
+         -- Trigger function to delete from the master table after the insert
+         CREATE OR REPLACE FUNCTION #{table_name}_delete_trigger() RETURNS trigger
+             AS $$
+         DECLARE
+             r #{table_name}%rowtype;
+         BEGIN
+             DELETE FROM ONLY #{table_name} where id = new.id returning * into r;
+             RETURN r;
+         end;
+         $$
+         LANGUAGE plpgsql;
+
+         CREATE TRIGGER #{table_name}_after_insert
+         AFTER INSERT ON #{table_name}
+         FOR EACH ROW
+         EXECUTE PROCEDURE #{table_name}_delete_trigger();"
+
+        execute_sql(sql)
+      end
+    end
+  end
+end

data/lib/pg_partitioner/version.rb

@@ -1,3 +1,3 @@
 module PartitionerPg
-  VERSION = '0.4.3'
+  VERSION = '0.5.0'
 end

data/lib/pg_partitioner.rb

@@ -1,9 +1,17 @@
 require 'pg_partitioner/version'
+require 'pg_partitioner/separation_type/base'
+require 'pg_partitioner/separation_type/week'
 require 'pg_partitioner/separation_type/month'
+require 'pg_partitioner/separation_type/quater'
 
 module PgPartitioner
   def self.extended(base)
-    base.extend(PgPartitioner::SeparationType::Month)
+    base.extend(
+      PgPartitioner::SeparationType::Base,
+      PgPartitioner::SeparationType::Week,
+      PgPartitioner::SeparationType::Month,
+      PgPartitioner::SeparationType::Quater
+    )
   end
 
   # Template method

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_partitioner
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.5.0
 platform: ruby
 authors:
 - Ovsapyan Mishel
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-20 00:00:00.000000000 Z
+date: 2023-10-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -52,7 +52,10 @@ files:
 - bin/console
 - bin/setup
 - lib/pg_partitioner.rb
+- lib/pg_partitioner/separation_type/base.rb
 - lib/pg_partitioner/separation_type/month.rb
+- lib/pg_partitioner/separation_type/quater.rb
+- lib/pg_partitioner/separation_type/week.rb
 - lib/pg_partitioner/version.rb
 - partitioner_pg.gemspec
 homepage: http://appodeal.com