lhm-shopify 3.3.5

data/lib/lhm/timestamp.rb

@@ -0,0 +1,11 @@
+module Lhm
+  class Timestamp
+    def initialize(time)
+      @time = time
+    end
+
+    def to_s
+      @time.strftime "%Y_%m_%d_%H_%M_%S_#{ '%03d' % (@time.usec / 1000) }"
+    end
+  end
+end

data/lib/lhm/version.rb

@@ -0,0 +1,6 @@
+# Copyright (c) 2011 - 2013, SoundCloud Ltd., Rany Keddo, Tobias Bielohlawek, Tobias
+# Schmidt
+
+module Lhm
+  VERSION = '3.3.5'
+end

data/spec/.lhm.example

@@ -0,0 +1,4 @@
+mysqldir=/usr/local/mysql
+basedir=~/lhm-cluster
+master_port=3306
+slave_port=3307

data/spec/README.md

@@ -0,0 +1,58 @@
+# Preparing for master slave integration tests
+
+## Configuration
+
+create ~/.lhm:
+
+    mysqldir=/usr/local/mysql
+    basedir=~/lhm-cluster
+    master_port=3306
+    slave_port=3307
+
+mysqldir specifies the location of your mysql install. basedir is the
+directory master and slave databases will get installed into.
+
+## Automatic setup
+
+### Run
+
+    bin/lhm-spec-clobber.sh
+
+You can set the integration specs up to run against a master slave setup by
+running the included that. This deletes the configured lhm master slave setup and reinstalls and configures a master slave setup.
+
+Follow the manual instructions if you want more control over this process.
+
+## Manual setup
+
+### set up instances
+
+    bin/lhm-spec-setup-cluster.sh
+
+### start instances
+
+    basedir=/opt/lhm-luster
+    mysqld --defaults-file="$basedir/master/my.cnf"
+    mysqld --defaults-file="$basedir/slave/my.cnf"
+
+### run the grants
+
+    bin/lhm-spec-grants.sh
+
+## run specs
+
+Setup the dependency gems
+
+    export BUNDLE_GEMFILE=gemfiles/ar-4.2_mysql2.gemfile
+    bundle install
+
+To run specs in slave mode, set the MASTER_SLAVE=1 when running tests:
+
+    MASTER_SLAVE=1 bundle exec rake specs
+
+# connecting
+
+you can connect by running (with the respective ports):
+
+    mysql --protocol=TCP -p3307
+

data/spec/fixtures/bigint_table.ddl

@@ -0,0 +1,4 @@
+CREATE TABLE `bigint_table` (
+  `id` BIGINT(20) auto_increment,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/composite_primary_key.ddl

@@ -0,0 +1,7 @@
+
+CREATE TABLE `composite_primary_key` (
+  `id` bigint(20) NOT NULL AUTO_INCREMENT,
+  `shop_id` bigint(20) NOT NULL,
+  PRIMARY KEY (`shop_id`,`id`),
+  INDEX `index_key_id` (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/custom_primary_key.ddl

@@ -0,0 +1,6 @@
+CREATE TABLE `custom_primary_key` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `pk` varchar(255),
+  PRIMARY KEY (`pk`),
+  UNIQUE KEY `index_custom_primary_key_on_id` (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/destination.ddl

@@ -0,0 +1,6 @@
+CREATE TABLE `destination` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `destination` int(11) DEFAULT NULL,
+  `common` varchar(255) DEFAULT NULL,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/lines.ddl

@@ -0,0 +1,7 @@
+CREATE TABLE `lines` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `between` varchar(10),
+  `lines` int(11),
+  `key` varchar(10),
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/origin.ddl

@@ -0,0 +1,6 @@
+CREATE TABLE `origin` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `origin` int(11) DEFAULT NULL,
+  `common` varchar(255) DEFAULT NULL,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/permissions.ddl

@@ -0,0 +1,5 @@
+CREATE TABLE `permissions` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `track_id` int(11) DEFAULT NULL,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/small_table.ddl

@@ -0,0 +1,4 @@
+CREATE TABLE `small_table` (
+  `id` INT(11) auto_increment,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/tracks.ddl

@@ -0,0 +1,5 @@
+CREATE TABLE `tracks` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `public` int(4) DEFAULT 0,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/users.ddl

@@ -0,0 +1,14 @@
+CREATE TABLE `users` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `reference` int(11) DEFAULT NULL,
+  `username` varchar(255) DEFAULT NULL,
+  `group` varchar(255) DEFAULT 'Superfriends',
+  `created_at` datetime DEFAULT NULL,
+  `comment` varchar(20) DEFAULT NULL,
+  `description` text,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `index_users_on_reference` (`reference`),
+  KEY `index_users_on_username_and_created_at` (`username`,`created_at`),
+  KEY `index_with_a_custom_name` (`username`,`group`)
+
+) ENGINE=InnoDB DEFAULT CHARSET=utf8

data/spec/fixtures/wo_id_int_column.ddl

@@ -0,0 +1,6 @@
+-- Without id int column
+CREATE TABLE `wo_id_int_column` (
+  `id` varchar(15) NOT NULL,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8
+

data/spec/integration/atomic_switcher_spec.rb

@@ -0,0 +1,93 @@
+# Copyright (c) 2011 - 2013, SoundCloud Ltd., Rany Keddo, Tobias Bielohlawek, Tobias
+# Schmidt
+
+require File.expand_path(File.dirname(__FILE__)) + '/integration_helper'
+
+require 'lhm/table'
+require 'lhm/migration'
+require 'lhm/atomic_switcher'
+
+describe Lhm::AtomicSwitcher do
+  include IntegrationHelper
+
+  before(:each) { connect_master! }
+
+  describe 'switching' do
+    before(:each) do
+      Thread.abort_on_exception = true
+      @origin      = table_create('origin')
+      @destination = table_create('destination')
+      @migration   = Lhm::Migration.new(@origin, @destination)
+      @logs = StringIO.new
+      Lhm.logger = Logger.new(@logs)
+      @connection.execute('SET GLOBAL innodb_lock_wait_timeout=3')
+      @connection.execute('SET GLOBAL lock_wait_timeout=3')
+    end
+
+    after(:each) do
+      Thread.abort_on_exception = false
+    end
+
+    it 'should retry and log on lock wait timeouts' do
+      connection = mock()
+      connection.stubs(:data_source_exists?).returns(true)
+      connection.stubs(:execute).raises(ActiveRecord::StatementInvalid, 'Lock wait timeout exceeded; try restarting transaction.').then.returns(true)
+
+      switcher = Lhm::AtomicSwitcher.new(@migration, connection, retriable: {base_interval: 0})
+
+      assert switcher.run
+
+      log_messages = @logs.string.split("\n")
+      assert_equal(2, log_messages.length)
+      assert log_messages[0].include? "Starting run of class=Lhm::AtomicSwitcher"
+      assert log_messages[1].include? "[AtomicSwitcher] ActiveRecord::StatementInvalid: 'Lock wait timeout exceeded; try restarting transaction.' - 1 tries"
+    end
+
+    it 'should give up on lock wait timeouts after a configured number of tries' do
+      connection = mock()
+      connection.stubs(:data_source_exists?).returns(true)
+      connection.stubs(:execute).twice.raises(ActiveRecord::StatementInvalid, 'Lock wait timeout exceeded; try restarting transaction.')
+
+      switcher = Lhm::AtomicSwitcher.new(@migration, connection, retriable: {tries: 2, base_interval: 0})
+
+      assert_raises(ActiveRecord::StatementInvalid) { switcher.run }
+    end
+
+    it 'should raise on non lock wait timeout exceptions' do
+      switcher = Lhm::AtomicSwitcher.new(@migration, connection)
+      switcher.send :define_singleton_method, :atomic_switch do
+        'SELECT * FROM nonexistent'
+      end
+      -> { switcher.run }.must_raise(ActiveRecord::StatementInvalid)
+    end
+
+    it "should raise when destination doesn't exist" do
+      connection = mock()
+      connection.stubs(:data_source_exists?).returns(false)
+
+      switcher = Lhm::AtomicSwitcher.new(@migration, connection)
+
+      assert_raises(Lhm::Error) { switcher.run }
+    end
+
+    it 'rename origin to archive' do
+      switcher = Lhm::AtomicSwitcher.new(@migration, connection)
+      switcher.run
+
+      slave do
+        data_source_exists?(@origin).must_equal true
+        table_read(@migration.archive_name).columns.keys.must_include 'origin'
+      end
+    end
+
+    it 'rename destination to origin' do
+      switcher = Lhm::AtomicSwitcher.new(@migration, connection)
+      switcher.run
+
+      slave do
+        data_source_exists?(@destination).must_equal false
+        table_read(@origin.name).columns.keys.must_include 'destination'
+      end
+    end
+  end
+end

data/spec/integration/chunk_insert_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path(File.dirname(__FILE__)) + '/integration_helper'
+require 'lhm/migration'
+
+describe Lhm::ChunkInsert do
+  include IntegrationHelper
+
+  describe 'insert_and_return_count_of_rows_created' do
+    before(:each) do
+      connect_master!
+      @origin = table_create(:origin)
+      @destination = table_create(:destination)
+      @migration = Lhm::Migration.new(@origin, @destination)
+      execute("insert into origin set id = 1001")
+      @instance = Lhm::ChunkInsert.new(@migration, connection, 1001, 1001)
+    end
+
+    it "returns the count" do
+      assert_equal 1, @instance.insert_and_return_count_of_rows_created
+    end
+
+    it "inserts the record into the slave" do
+      @instance.insert_and_return_count_of_rows_created
+
+      slave do
+        count_all(@destination.name).must_equal(1)
+      end
+    end
+  end
+end

data/spec/integration/chunker_spec.rb

@@ -0,0 +1,185 @@
+# Copyright (c) 2011 - 2013, SoundCloud Ltd., Rany Keddo, Tobias Bielohlawek, Tobias
+# Schmidt
+
+require File.expand_path(File.dirname(__FILE__)) + '/integration_helper'
+require 'lhm/table'
+require 'lhm/migration'
+
+describe Lhm::Chunker do
+  include IntegrationHelper
+
+  before(:each) { connect_master! }
+
+  describe 'copying' do
+    before(:each) do
+      @origin = table_create(:origin)
+      @destination = table_create(:destination)
+      @migration = Lhm::Migration.new(@origin, @destination)
+    end
+
+    it 'should copy 1 row from origin to destination even if the id of the single row does not start at 1' do
+      execute("insert into origin set id = 1001 ")
+      printer = Lhm::Printer::Base.new
+
+      def printer.notify(*) ;end
+      def printer.end(*) [] ;end
+
+      Lhm::Chunker.new(@migration, connection, {:throttler => Lhm::Throttler::Time.new(:stride => 100), :printer => printer} ).run
+
+      slave do
+        count_all(@destination.name).must_equal(1)
+      end
+
+    end
+
+    it 'should create the modified destination, even if the source is empty' do
+      execute("truncate origin ")
+      printer = Lhm::Printer::Base.new
+
+      def printer.notify(*) ;end
+      def printer.end(*) [] ;end
+
+      Lhm::Chunker.new(@migration, connection, {:throttler => Lhm::Throttler::Time.new(:stride => 100), :printer => printer} ).run
+
+      slave do
+        count_all(@destination.name).must_equal(0)
+      end
+
+    end
+
+    it 'should copy 23 rows from origin to destination in one shot, regardless of the value of the id' do
+      23.times { |n| execute("insert into origin set id = '#{ n * n + 23 }'") }
+
+      printer = MiniTest::Mock.new
+      printer.expect(:notify, :return_value, [Integer, Integer])
+      printer.expect(:end, :return_value, [])
+
+      Lhm::Chunker.new(
+        @migration, connection, { :throttler => Lhm::Throttler::Time.new(:stride => 100), :printer => printer }
+      ).run
+
+      slave do
+        count_all(@destination.name).must_equal(23)
+      end
+
+      printer.verify
+
+    end
+
+    it 'should copy all the records of a table, even if the last chunk starts with the last record of it.' do
+      11.times { |n| execute("insert into origin set id = '#{ n + 1 }'") }
+
+      printer = Lhm::Printer::Base.new
+
+      def printer.notify(*) ;end
+      def printer.end(*) [] ;end
+
+      Lhm::Chunker.new(
+        @migration, connection, { :throttler => Lhm::Throttler::Time.new(:stride => 10), :printer => printer }
+      ).run
+
+      slave do
+        count_all(@destination.name).must_equal(11)
+      end
+
+    end
+
+    it 'should copy 23 rows from origin to destination in one shot with slave lag based throttler, regardless of the value of the id' do
+      23.times { |n| execute("insert into origin set id = '#{ 100000 + n * n + 23 }'") }
+
+      printer = MiniTest::Mock.new
+      printer.expect(:notify, :return_value, [Integer, Integer])
+      printer.expect(:end, :return_value, [])
+
+      Lhm::Chunker.new(
+        @migration, connection, { :throttler => Lhm::Throttler::SlaveLag.new(:stride => 100), :printer => printer }
+      ).run
+
+      slave do
+        count_all(@destination.name).must_equal(23)
+      end
+
+      printer.verify
+    end
+
+    it 'should throttle work stride based on slave lag' do
+      15.times { |n| execute("insert into origin set id = '#{ (n * n) + 1 }'") }
+
+      printer = mock()
+      printer.expects(:notify).with(instance_of(Integer), instance_of(Integer)).twice
+      printer.expects(:end)
+
+      throttler = Lhm::Throttler::SlaveLag.new(:stride => 10, :allowed_lag => 0)
+      def throttler.max_current_slave_lag
+        1
+      end
+
+      Lhm::Chunker.new(
+        @migration, connection, { :throttler => throttler, :printer => printer }
+      ).run
+
+      assert_equal(Lhm::Throttler::SlaveLag::INITIAL_TIMEOUT * 2 * 2, throttler.timeout_seconds)
+
+      slave do
+        count_all(@destination.name).must_equal(15)
+      end
+    end
+
+    it 'should detect a single slave with no lag in the default configuration' do
+      15.times { |n| execute("insert into origin set id = '#{ (n * n) + 1 }'") }
+
+      printer = mock()
+      printer.expects(:notify).with(instance_of(Integer), instance_of(Integer)).twice
+      printer.expects(:verify)
+      printer.expects(:end)
+
+      throttler = Lhm::Throttler::SlaveLag.new(:stride => 10, :allowed_lag => 0)
+
+      def throttler.slave_hosts
+        ['127.0.0.1']
+      end
+
+      if master_slave_mode?
+        def throttler.slave_connection(slave)
+          config = ActiveRecord::Base.connection_pool.spec.config.dup
+          config[:host] = slave
+          config[:port] = 3307
+          ActiveRecord::Base.send('mysql2_connection', config)
+        end
+      end
+
+      Lhm::Chunker.new(
+        @migration, connection, { :throttler => throttler, :printer => printer }
+      ).run
+
+      assert_equal(Lhm::Throttler::SlaveLag::INITIAL_TIMEOUT, throttler.timeout_seconds)
+      assert_equal(0, throttler.send(:max_current_slave_lag))
+
+      slave do
+        count_all(@destination.name).must_equal(15)
+      end
+
+      printer.verify
+    end
+
+    it 'should abort early if the triggers are removed' do
+      15.times { |n| execute("insert into origin set id = '#{ (n * n) + 1 }'") }
+
+      printer = mock()
+
+      failer = Proc.new { false }
+
+      exception = assert_raises do
+        Lhm::Chunker.new(
+          @migration, connection, { :verifier => failer, :printer => printer, :throttler => Lhm::Throttler::Time.new(:stride => 100) }
+        ).run
+      end
+
+      assert_match "Verification failed, aborting early", exception.message
+
+      slave do
+        count_all(@destination.name).must_equal(0)
+      end
+    end
+  end
+end