code0-zero_track 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d6150c745af141848ee0848ec9beda4db5c1809ece022f90b9006cf69df88f10
-  data.tar.gz: 73128d1d5ab33dec7887b195831c8a1b591eb46999c5cb4e20eb79387d5d138e
+  metadata.gz: 5af11d8b9714d7d557f46cfe5ea9dbd6702d3bc85f5aa67e9ce09e44c6963238
+  data.tar.gz: 5fc99a08521c502aec93ebe725d9c82c18d9b49e6a7120fba5127db9bb620f6b
 SHA512:
-  metadata.gz: 129416f538b904432c213a42a22d95244220fb9c35548b089d9a9ef4836b11b1b3fd0a93c910ab748dfd51d024d56d00ac525e7fd22855330916462b848a3824
-  data.tar.gz: 78bba0f4d37e6a08b2d5b02a46478060e7ea944ee8a40c407a4f63c819c450ae41e86b33df9b47a28472ebb202dad28a49c2c95fa221de041d672de1164e8a6d
+  metadata.gz: fded2cb69f93aff4704e89b6c3094a0c9dd47a9891748ddd40af68e72eb2d432c150ecd62bf104cd864ceee65a5e782b3eef656fd7ac69ea338898f2cae713a7
+  data.tar.gz: db2a0b2c23fa93a22442e7f8e0c956ddae670a05bf4105c0a77ea98968c5827541b442b119331bdfb6f8b564b151ccedb9b1418cf45a8dd2fe243c7402748bb8

data/README.md

@@ -87,9 +87,10 @@ config.zero_track.db_partitioning.base_ar_class = 'ActiveRecord::Base' # default
 Include the migration helpers by inheriting from `Code0::ZeroTrack::Database::Migration[1.0]` (or the
 appropriate version). The following methods become available:
 
-`create_partition_by_date_table(table_name, partition_column:, **options, &block)` creates a table
-partitioned by range on the given column. It automatically sets up a composite primary key
-of `(id, partition_column)`.
+`create_partition_by_date_table(table_name, partition_column:, primary_key: nil, **options, &block)` creates a table
+partitioned by range on the given column. By default, it adds a `bigserial` id column and sets up a composite
+primary key of `(id, partition_column)`. If `primary_key` is provided (e.g. `primary_key: %i[date other_column]`),
+the id column is omitted and the given columns are used as the primary key instead.
 
 `create_dynamic_partition_schema` / `drop_dynamic_partition_schema` creates or drops the schema
 used for storing dynamic partitions.

data/lib/code0/zero_track/database/migration_helpers/table_partitioning.rb

@@ -5,21 +5,23 @@ module Code0
     module Database
       module MigrationHelpers
         module TablePartitioning
-          def create_partition_by_date_table(table_name, partition_column:, **options, &block)
+          def create_partition_by_date_table(table_name, partition_column:, primary_key: nil, **options, &block)
             options[:options] = "PARTITION BY RANGE (#{quote_column_name(partition_column)})"
             options[:id] = false
 
             create_table(table_name, **options) do |t|
-              t.bigserial :id, null: false
+              t.bigserial :id, null: false unless primary_key
 
               block.call(t)
             end
 
+            pk_columns = primary_key || [:id, partition_column]
+
             reversible do |dir|
               dir.up do
                 execute <<~SQL.squish
                   ALTER TABLE #{quote_table_name(table_name)}
-                  ADD PRIMARY KEY (#{quote_column_name(:id)}, #{quote_column_name(partition_column)})
+                  ADD PRIMARY KEY (#{pk_columns.map { |c| quote_column_name(c) }.join(', ')})
                 SQL
               end
             end

data/lib/code0/zero_track/version.rb

@@ -2,6 +2,6 @@
 
 module Code0
   module ZeroTrack
-    VERSION = '0.0.8'
+    VERSION = '0.0.9'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: code0-zero_track
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors:
 - Niklas van Schrick