sequel_bitemporal 0.9.0 → 0.9.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4f44bf401ad71c80c9815ff7d2e3508edf8e5bdbef4112b36bf86d06a6d02523
-  data.tar.gz: 19b320461ebbdac3b2e7ace0117ffa2921180dd602b59e5145d2c1771987d993
+  metadata.gz: 7f6daa2d7f025cd1281e8cbfc7235d0a6ddd85d2251f8d9195d85cc5340aafb3
+  data.tar.gz: 5d5de257ee011436a1c3959206c31c985d6b11597d63fcab468876865394f9ee
 SHA512:
-  metadata.gz: 1e9928807145999a4ba2215b10e3169bb26e63bc42137c28b9fc05ad6ac9a2e66f10c69ccdda40ae1ae8d7bbfabad0f1f921122fc3e0bb155da559b335ffa97f
-  data.tar.gz: a7fac94af384e4478fd9d6f98a6381bf03278e1310d8d97dff5bea449f2e3c5baeb962e3552774535a14b0a021a5b0ce22d76b659b08f7fcf9fe9a17a36d31dc
+  metadata.gz: '0309ac260a37ec550e417efb01eaa72120583a5c26c6306f52c15910d36c8dc24f9e74d39319f763e3a566192ddd5887b0b022ce747f820beee388a444726dc8'
+  data.tar.gz: f4f61ba42c6510bcc744f96a27ca9a91fe965762ce4ec6206720441346ec1e5402475e27e87a7b838d342898c47df99a1e8025abf5ebce59691250417bc0015f

data/.ruby-version

@@ -1 +1 @@
-2.3.1
+2.5.3

data/README.md

@@ -1,9 +1,9 @@
 sequel_bitemporal
 =================
 
-[![Build Status](https://travis-ci.org/TalentBox/sequel_bitemporal.png?branch=master)](https://travis-ci.org/TalentBox/sequel_bitemporal)
+[![Build Status](https://travis-ci.org/TalentBox/sequel_bitemporal.svg?branch=master)](https://travis-ci.org/TalentBox/sequel_bitemporal)
 
-Bitemporal versioning for sequel.
+Bitemporal versioning for [Sequel].
 
 Dependencies
 ------------
@@ -47,7 +47,12 @@ Thanks to Evgeniy L (@fiscal-cliff) for his contributions:
 Thanks to Ksenia Zalesnaya (@ksenia-zalesnaya) for her contributions:
 - define setter methods for versioned columns
 
+Thanks to Denis Kalesnikov (@DenisKem) for his contributions:
+- allow composite primary key
+
 License
 -------
 
 sequel_bitemporal is Copyright © 2011 TalentBox SA. It is free software, and may be redistributed under the terms specified in the LICENSE file.
+
+[Sequel]: http://sequel.jeremyevans.net/

data/Rakefile

@@ -3,4 +3,4 @@ require "rspec/core/rake_task"
 RSpec::Core::RakeTask.new("spec").tap do |config|
   config.rspec_opts = "--color"
 end
-task :default => :spec
+task :default => :spec

data/lib/sequel/plugins/bitemporal.rb

@@ -31,19 +31,25 @@ module Sequel
         Thread.current[THREAD_NOW_KEY] || DateTime.now
       end
 
-      def self.bitemporal_version_columns
-        @bitemporal_version_columns ||= [:master_id, :valid_from, :valid_to, :created_at, :expired_at]
+      def self.version_foreign_keys(master = nil)
+        return :master_id unless master
+        primary_key = [*master.primary_key]
+        primary_key.size > 1 ? primary_key : :master_id
       end
 
-      def self.bitemporal_excluded_columns
-        @bitemporal_excluded_columns ||= [:id, *bitemporal_version_columns]
+      def self.bitemporal_version_columns(master = nil)
+        [*version_foreign_keys(master), :valid_from, :valid_to, :created_at, :expired_at]
+      end
+
+      def self.bitemporal_excluded_columns(master = nil)
+        [:id, *bitemporal_version_columns(master)]
       end
 
       def self.configure(master, opts = {})
         version = opts[:version_class]
         raise Error, "please specify version class to use for bitemporal plugin" unless version
         return version.db.log_info("Version table does not exist for #{version.name}") unless version.db.table_exists?(version.table_name)
-        missing = bitemporal_version_columns - version.columns
+        missing = bitemporal_version_columns(master) - version.columns
         raise Error, "bitemporal plugin requires the following missing column#{"s" if missing.size>1} on version class: #{missing.join(", ")}" unless missing.empty?
 
         if Sequel::Plugins::Bitemporal.jdbc?(master.db)
@@ -76,7 +82,7 @@ module Sequel
           @audit_updated_by_method = opts.fetch(:audit_updated_by_method){ :updated_by }
           @propagate_per_column = opts.fetch(:propagate_per_column, false)
           @version_uses_string_nilifier = version.plugins.map(&:to_s).include? "Sequel::Plugins::StringNilifier"
-          @excluded_columns = Sequel::Plugins::Bitemporal.bitemporal_excluded_columns
+          @excluded_columns = Sequel::Plugins::Bitemporal.bitemporal_excluded_columns(master)
           @excluded_columns += Array opts[:excluded_columns] if opts[:excluded_columns]
           @use_ranges = if opts[:ranges]
             db = self.db
@@ -100,8 +106,8 @@ module Sequel
             end
           end
         end
-        master.one_to_many :versions, class: version, key: :master_id, graph_alias_base: master.versions_alias
-        master.one_to_one :current_version, class: version, key: :master_id, graph_alias_base: master.current_version_alias, :graph_block=>(proc do |j, lj, js|
+        master.one_to_many :versions, class: version, key: version_foreign_keys(master), graph_alias_base: master.versions_alias
+        master.one_to_one :current_version, class: version, key: version_foreign_keys(master), graph_alias_base: master.current_version_alias, :graph_block=>(proc do |j, lj, js|
           t = Sequel.delay{ ::Sequel::Plugins::Bitemporal.point_in_time }
           n = Sequel.delay{ ::Sequel::Plugins::Bitemporal.now }
           if master.use_ranges
@@ -135,7 +141,7 @@ module Sequel
             )
           )
         end
-        master.one_to_many :current_or_future_versions, class: version, key: :master_id, :graph_block=>(proc do |j, lj, js|
+        master.one_to_many :current_or_future_versions, class: version, key: version_foreign_keys(master), :graph_block=>(proc do |j, lj, js|
           t = Sequel.delay{ ::Sequel::Plugins::Bitemporal.point_in_time }
           n = Sequel.delay{ ::Sequel::Plugins::Bitemporal.now }
           if master.use_ranges
@@ -169,7 +175,7 @@ module Sequel
             Sequel.negate(Sequel.qualify(:current_or_future_versions, :id) => nil)
           )
         end
-        version.many_to_one :master, class: master, key: :master_id
+        version.many_to_one :master, class: master, key: version_foreign_keys(master)
         version.class_eval do
           if Sequel::Plugins::Bitemporal.jdbc?(master.db)
             plugin :typecast_on_load, *columns
@@ -319,7 +325,13 @@ module Sequel
 
         def attributes=(attributes)
           @pending_version ||= begin
-            current_attributes = {master_id: id}
+            current_attributes =
+              if composite_primary_key?
+                version_values
+              else
+                { master_id: id }
+              end
+
             current_version.keys.each do |key|
               next if excluded_columns.include? key
               current_attributes[key] = current_version.send key
@@ -329,6 +341,10 @@ module Sequel
           pending_version.set_all attributes
         end
 
+        def version_values
+          version_foreign_keys.map { |k| [k, public_send(k)] }.to_h
+        end
+
         def update_attributes(attributes={})
           self.attributes = attributes
           if save raise_on_failure: false
@@ -364,6 +380,10 @@ module Sequel
           _refresh_set_values @values
         end
 
+        def composite_primary_key?
+          [*primary_key].size > 1
+        end
+
         def destroy
           point_in_time = ::Sequel::Plugins::Bitemporal.point_in_time
           versions_dataset.where(
@@ -471,6 +491,10 @@ module Sequel
           @propagated_during_last_save ||= []
         end
 
+        def version_foreign_keys
+          composite_primary_key? ? primary_key : :master_id
+        end
+
       private
 
         def prepare_pending_version
@@ -608,7 +632,7 @@ module Sequel
           columns.detect do |column|
             new_value = pending_version.send column
             case column
-            when :id, :master_id, :created_at, :expired_at
+            when :id, :created_at, :expired_at, *version_foreign_keys
               false
             when :valid_from
               pending_version.values.has_key?(:valid_from) && (

data/sequel_bitemporal.gemspec

@@ -3,7 +3,7 @@ $:.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = "sequel_bitemporal"
-  s.version     = "0.9.0"
+  s.version     = "0.9.1"
   s.authors     = ["Joseph HALTER", "Jonathan TRON"]
   s.email       = ["joseph.halter@thetalentbox.com", "jonathan.tron@thetalentbox.com"]
   s.homepage    = "https://github.com/TalentBox/sequel_bitemporal"

data/spec/bitemporal_date_spec.rb

@@ -1,6 +1,6 @@
 require "spec_helper"
 
-describe "Sequel::Plugins::Bitemporal" do
+RSpec.describe "Sequel::Plugins::Bitemporal" do
   before :all do
     db_setup
   end
@@ -791,7 +791,7 @@ describe "Sequel::Plugins::Bitemporal" do
   end
 end
 
-describe "Sequel::Plugins::Bitemporal", "with audit" do
+RSpec.describe "Sequel::Plugins::Bitemporal", "with audit" do
   before :all do
     @audit_class = Class.new do
       def self.audit(*args); end
@@ -889,7 +889,7 @@ describe "Sequel::Plugins::Bitemporal", "with audit" do
     end
   end
 end
-describe "Sequel::Plugins::Bitemporal", "with audit, specifying how to get the author" do
+RSpec.describe "Sequel::Plugins::Bitemporal", "with audit, specifying how to get the author" do
   before :all do
     @audit_class = Class.new do
       def self.audit(*args); end

data/spec/bitemporal_date_with_range_spec.rb

@@ -1,6 +1,6 @@
 require "spec_helper"
 if DbHelpers.pg?
-  describe "Sequel::Plugins::Bitemporal", "with ranges" do
+  RSpec.describe "Sequel::Plugins::Bitemporal", "with ranges" do
     before :all do
       db_setup ranges: true
     end
@@ -637,7 +637,7 @@ if DbHelpers.pg?
     end
   end
 
-  describe "Sequel::Plugins::Bitemporal", "with audit and ranges" do
+  RSpec.describe "Sequel::Plugins::Bitemporal", "with audit and ranges" do
     before :all do
       @audit_class = Class.new do
         def self.audit(*args); end
@@ -735,7 +735,7 @@ if DbHelpers.pg?
       end
     end
   end
-  describe "Sequel::Plugins::Bitemporal", "with audit, ranges, specifying how to get the author" do
+  RSpec.describe "Sequel::Plugins::Bitemporal", "with audit, ranges, specifying how to get the author" do
     before :all do
       @audit_class = Class.new do
         def self.audit(*args); end

data/spec/bitemporal_serialization_spec.rb

@@ -1,7 +1,7 @@
 require "spec_helper"
 require "json"
 
-describe "Sequel::Plugins::Bitemporal", :skip_jdbc_sqlite do
+RSpec.describe "Sequel::Plugins::Bitemporal", :skip_jdbc_sqlite do
   before :all do
     db_setup
     @version_class.instance_eval do

data/spec/bitemporal_time_spec.rb

@@ -1,6 +1,6 @@
 require "spec_helper"
 
-describe "Sequel::Plugins::Bitemporal" do
+RSpec.describe "Sequel::Plugins::Bitemporal" do
   let(:hour){ 3600 }
   before :all do
     db_setup use_time: true
@@ -381,7 +381,7 @@ describe "Sequel::Plugins::Bitemporal" do
     expect(@master_class.with_current_or_future_versions.all.size).to eq(2)
   end
 end
-describe "Sequel::Plugins::Bitemporal", "with audit" do
+RSpec.describe "Sequel::Plugins::Bitemporal", "with audit" do
   before :all do
     @audit_class = Class.new
     db_setup use_time: true, audit_class: @audit_class
@@ -435,7 +435,7 @@ describe "Sequel::Plugins::Bitemporal", "with audit" do
   end
 end
 
-describe "Sequel::Plugins::Bitemporal", "with audit, specifying how to get the author" do
+RSpec.describe "Sequel::Plugins::Bitemporal", "with audit, specifying how to get the author" do
   before :all do
     @audit_class = Class.new
     db_setup use_time: true, audit_class: @audit_class, audit_updated_by_method: :author

data/spec/bitemporal_time_with_range_spec.rb

@@ -1,6 +1,6 @@
 require "spec_helper"
 if DbHelpers.pg?
-  describe "Sequel::Plugins::Bitemporal" do
+  RSpec.describe "Sequel::Plugins::Bitemporal" do
     let(:hour){ 3600 }
     before :all do
       db_setup use_time: true, ranges: true
@@ -381,7 +381,7 @@ if DbHelpers.pg?
       expect(@master_class.with_current_or_future_versions.all.size).to eq(2)
     end
   end
-  describe "Sequel::Plugins::Bitemporal", "with audit" do
+  RSpec.describe "Sequel::Plugins::Bitemporal", "with audit" do
     before :all do
       @audit_class = Class.new
       db_setup use_time: true, audit_class: @audit_class, ranges: true
@@ -435,7 +435,7 @@ if DbHelpers.pg?
     end
   end
 
-  describe "Sequel::Plugins::Bitemporal", "with audit, specifying how to get the author" do
+  RSpec.describe "Sequel::Plugins::Bitemporal", "with audit, specifying how to get the author" do
     before :all do
       @audit_class = Class.new
       db_setup(

data/spec/composite_primary_key_spec.rb

@@ -0,0 +1,59 @@
+require "spec_helper"
+
+RSpec.describe "Sequel::Plugins::Bitemporal", "composite_primary_key" do
+  before :all do
+    setup_composite_primary_key
+  end
+
+  describe "::version_foreign_keys" do
+    context "when is invoked without args" do
+      it "returns master_id" do
+        expect(Sequel::Plugins::Bitemporal.bitemporal_excluded_columns).to(
+          eq([:id, :master_id, :valid_from, :valid_to, :created_at, :expired_at])
+        )
+      end
+    end
+  end
+
+  describe "missing required foreign keys in version table" do
+    ERROR_TEXT = "bitemporal plugin requires the following missing columns on version class: department_id, team_id"
+
+    it "raises error" do
+      expect do
+        setup_composite_primary_key(with_foreign_key: false)
+      end.to raise_error Sequel::Error, ERROR_TEXT
+    end
+  end
+
+  describe "versions work correctly" do
+    let(:master) { @master_class.new(name: "john Smith") }
+    let(:junior) { "Junior Ruby-developer" }
+    let(:senior) { "Senior Ruby-developer" }
+    let(:two_years_in_seconds) { 2 * 365 * 24 * 60 * 60 }
+
+    before do
+      setup_composite_primary_key
+      master.department_id = 1
+      master.team_id = 1
+      master.save
+
+      Sequel::Plugins::Bitemporal.at(Time.now) do
+        master.update_attributes(position: junior)
+      end
+
+      Sequel::Plugins::Bitemporal.at(Time.now + two_years_in_seconds) do
+        master.update_attributes(position: senior)
+      end
+    end
+
+    specify do
+      Sequel::Plugins::Bitemporal.at(Time.now) do
+        expect(master.reload.position).to eq(junior)
+      end
+
+      Sequel::Plugins::Bitemporal.at(Time.now + two_years_in_seconds) do
+        expect(master.reload.position).to eq(senior)
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -35,6 +35,7 @@ end
 RSpec.configure do |config|
   config.include DbHelpers
   config.filter_run_excluding rspec_exclusions
+  config.disable_monkey_patching!
   config.before :each do
     db_truncate
   end

data/spec/support/db.rb

@@ -63,6 +63,48 @@ module DbHelpers
     end
   end
 
+  def setup_composite_primary_key(with_foreign_key: true)
+    DB.drop_table(:employee_versions) if DB.table_exists?(:employee_versions)
+    DB.drop_table(:employees) if DB.table_exists?(:employees)
+
+    DB.create_table! :employees do
+      Integer :department_id
+      Integer :team_id
+      primary_key [:department_id, :team_id]
+      String :name, null: false
+    end
+
+    DB.create_table! :employee_versions do
+      if with_foreign_key
+        Integer :department_id
+        Integer :team_id
+
+        foreign_key [:department_id, :team_id], :employees
+      end
+
+      primary_key :id
+      String :position
+      Date :created_at
+      Date :expired_at
+      Date :valid_from
+      Date :valid_to
+    end
+
+    @version_class = Class.new Sequel::Model do
+      set_dataset :employee_versions
+    end
+
+    bitemporal_options = {
+      version_class: @version_class,
+      key: [:department_id, :team_id]
+    }
+
+    @master_class = Class.new Sequel::Model do
+      set_dataset :employees
+      plugin :bitemporal, bitemporal_options
+    end
+  end
+
   def db_truncate
     if DbHelpers.pg?
       @version_class.truncate cascade: true

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sequel_bitemporal
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.9.1
 platform: ruby
 authors:
 - Joseph HALTER
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-11-29 00:00:00.000000000 Z
+date: 2019-01-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel
@@ -98,6 +98,7 @@ files:
 - spec/bitemporal_serialization_spec.rb
 - spec/bitemporal_time_spec.rb
 - spec/bitemporal_time_with_range_spec.rb
+- spec/composite_primary_key_spec.rb
 - spec/spec_helper.rb
 - spec/support/bitemporal_matchers.rb
 - spec/support/db.rb
@@ -131,6 +132,7 @@ test_files:
 - spec/bitemporal_serialization_spec.rb
 - spec/bitemporal_time_spec.rb
 - spec/bitemporal_time_with_range_spec.rb
+- spec/composite_primary_key_spec.rb
 - spec/spec_helper.rb
 - spec/support/bitemporal_matchers.rb
 - spec/support/db.rb