ci-helper 0.8.0 → 0.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6752c279e742404a6b104c4d4e2bccfaf4c18e17cfb0c6693ce1ddb0ce2b58e1
-  data.tar.gz: 4e9960ccc6a0f0bd7b1ecf65a4c6df063cf579cdc71338dddb90248065368332
+  metadata.gz: 2c5d7ac1dcb58a235887b801308afcfa0e6f4cee4a52c829c769d12424866107
+  data.tar.gz: 446bfe1be5110ece9d61eec181a4718999bd20e0c7f70e2f3fcf04b3c9b7f72f
 SHA512:
-  metadata.gz: 3d8da6805297e53be7955d7219289dbece6f2decacdf760b81061e5745c4066a458b8b23ffa64cf06691ac270b00d58259941523e39ed919d056acc28aff0a85
-  data.tar.gz: 97b48c63728b3ef1c3a3457a1b5c90e4f2954d250783db38eed9ae6414aeb177cacfca9038d9d46226f6c307f0439a608c38e9bee83ec5999f93d39ba8a2a141
+  metadata.gz: 1b35b1ea59c8fb0eb4bd2c293f807c4a421b6326852f3fb8e7a879f8c6a5a05dc5010e4f494846236c5733f2a59fec89b7994c4295b9254c49bc237a8acb92aa
+  data.tar.gz: 9cb25a481ea5c3d03afd5fbc45f276454a88c880d1296b24b898941305b50ef1144fd8df84447e897cf8d9edb0757aef4f82a6591a9ff0b3a5f6e40c6a0be9bc

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ci-helper (0.8.0)
+    ci-helper (0.9.0)
       colorize (~> 1.1)
       dry-inflector (~> 1.0)
       umbrellio-sequel-plugins (~> 0.14)

data/lib/ci_helper/commands/check_db_development.rb

@@ -6,6 +6,8 @@ module CIHelper
       def call
         create_and_migrate_database!
         execute("bundle exec rake db:seed")
+        create_and_migrate_clickhouse_database! if with_clickhouse?
+        0
       end
 
       private
@@ -13,6 +15,10 @@ module CIHelper
       def env
         :development
       end
+
+      def with_clickhouse?
+        boolean_option(:with_clickhouse)
+      end
     end
   end
 end

data/lib/ci_helper/commands/check_db_rollback.rb

@@ -7,6 +7,13 @@ module CIHelper
         create_and_migrate_database!
         execute_with_env("bundle exec rake db:rollback_new_migrations")
         execute_with_env("bundle exec rake db:migrate")
+
+        if with_clickhouse?
+          create_and_migrate_clickhouse_database!
+          execute_with_env("bundle exec rake ch:rollback_new_migrations")
+          execute_with_env("bundle exec rake ch:migrate")
+        end
+        0
       end
 
       private
@@ -14,6 +21,10 @@ module CIHelper
       def env
         :test
       end
+
+      def with_clickhouse?
+        boolean_option(:with_clickhouse)
+      end
     end
   end
 end

data/lib/ci_helper/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CIHelper
-  VERSION = "0.8.0"
+  VERSION = "0.9.0"
 end

data/lib/tasks/sequel_management.rake

@@ -13,6 +13,7 @@ class SequelManagement
     self.logger = Logger.new($stdout)
 
     define_db_task!
+    define_ch_task!
   end
 
   private
@@ -23,6 +24,10 @@ class SequelManagement
     self.class::MIGRATIONS_PATH
   end
 
+  def ch_migrations_path
+    "db/migrate/clickhouse"
+  end
+
   def define_db_task!
     namespace :db do
       desc "Rollback all migrations, which doesn't exist in master branch"
@@ -47,6 +52,31 @@ class SequelManagement
     end
   end
 
+  def define_ch_task!
+    namespace :ch do
+      desc "Rollback all migrations, which doesn't exist in master branch"
+      task rollback_new_migrations: :environment do
+        abort "Shouldn't run in production mode!" if Rails.env.production?
+
+        logger.info "Begin rolling back new migrations"
+
+        git_command = "git ls-tree --name-only origin/master #{ch_migrations_path}/"
+        migration_files, error, status = Open3.capture3(git_command)
+        abort "Can't get list of migration files:\n#{error}" unless status.success?
+
+        original_migrations = migration_files.split.map { |path| File.basename(path) }
+        migrations_to_rollback = (ch_migrator.applied_migrations - original_migrations).sort.reverse
+
+        next if migrations_to_rollback.empty?
+
+        logger.info "Rolling back migrations:"
+        logger.info migrations_to_rollback.join("\n")
+
+        ch_rollback!(migrations_to_rollback)
+      end
+    end
+  end
+
   def git_command
     "git ls-tree --name-only origin/master #{migrations_path}/"
   end
@@ -58,6 +88,14 @@ class SequelManagement
     end
   end
 
+  def ch_migrator
+    @ch_migrator ||= begin
+      full_path = Rails.root.join(ch_migrations_path)
+      Sequel::TimestampMigrator.new(DB, full_path, allow_missing_migration_files: true,
+                                                   table: "clickhouse_migrations")
+    end
+  end
+
   def rollback!(migrations)
     migrations.each do |migration|
       migrator.undo(migration.to_i)
@@ -69,6 +107,18 @@ class SequelManagement
       end
     end
   end
+
+  def ch_rollback!(migrations)
+    migrations.each do |migration|
+      ch_migrator.undo(migration.to_i)
+    rescue Sequel::Migrator::Error => error
+      if error.message.include?("does not exist in the filesystem")
+        logger.info error.message
+      else
+        raise error
+      end
+    end
+  end
 end
 
 SequelManagement.new

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ci-helper
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.9.0
 platform: ruby
 authors:
 - JustAnotherDude