clickhouse-activerecord 1.0.2 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e5a31d2ddff53bb618e1624e325c0be76f2f602655565a74a96106674f3a375e
-  data.tar.gz: 762b41c929d36e7ff0f0ceff557da5b85d0b231e8df3e6c6bd41b6085eb940b0
+  metadata.gz: 194430a00c6f085ec0bf6b1cf18911c06b69ff2e87ef18fc50183b1d7d1f20c7
+  data.tar.gz: 54e418ac9c841e6852e7e87e1ea169335f0c0940537eda3f95d645f3e63ed81e
 SHA512:
-  metadata.gz: a6f8143c312efbc6f276b0fdc2b9e5ea8ab811f06a865d54ff82dcaa0614c30b26c44f65f60098f0895dcf935aa97c86e8b168e7e1bc5b7e54899662b71ad86a
-  data.tar.gz: d9f4ba22a3f13bd157a225870e4d5eb025ca6a35ac21dc73731c19a8bb0d7464e479479109007d117d29f1aa788bb2f3cd5ae0350f1fa13596c36da3ed99a766
+  metadata.gz: 9d5a90636c9eaa390ce1b75f65581061312dd598fbaf64e0402a40d3e2a1ab6b8079a0ae438ac86e1b03d4df3a1d1ab6308db961cf988320a39b13cf65014cc5
+  data.tar.gz: d961eaf6d2d5798cbf220dc7249330c91b43fe86a68d64ff0371bb0a0f7cba754dd5e95fb1fb666fadcbd74fe71fd89f6ea5b2d85a6d478e7c624e541f00b768

data/README.md

@@ -124,11 +124,11 @@ If you using multiple databases, for example: PostgreSQL, Clickhouse.
 
 Schema dump to `db/clickhouse_schema.rb` file:
 
-    $ rake clickhouse:schema:dump
+    $ rake db:schema:dump:clickhouse
     
 Schema load from `db/clickhouse_schema.rb` file:
 
-    $ rake clickhouse:schema:load
+    $ rake db:schema:load:clickhouse
 
 For export schema to PostgreSQL, you need use:
 
@@ -238,6 +238,7 @@ Donations to this project are going directly to [PNixx](https://github.com/PNixx
 * BTC address: `1H3rhpf7WEF5JmMZ3PVFMQc7Hm29THgUfN`
 * ETH address: `0x6F094365A70fe7836A633d2eE80A1FA9758234d5`
 * XMR address: `42gP71qLB5M43RuDnrQ3vSJFFxis9Kw9VMURhpx9NLQRRwNvaZRjm2TFojAMC8Fk1BQhZNKyWhoyJSn5Ak9kppgZPjE17Zh`
+* TON address: `UQBt0-s1igIpJoEup0B1yAUkZ56rzbpruuAjNhQ26MVCaNlC`
 
 ## Development
 

data/lib/active_record/connection_adapters/clickhouse/oid/uuid.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module ConnectionAdapters
+    module Clickhouse
+      module OID # :nodoc:
+        class Uuid < Type::Value # :nodoc:
+          ACCEPTABLE_UUID = %r{\A(\{)?([a-fA-F0-9]{4}-?){8}(?(1)\}|)\z}
+
+          alias :serialize :deserialize
+
+          def type
+            :uuid
+          end
+
+          def changed?(old_value, new_value, _)
+            old_value.class != new_value.class ||
+              new_value && old_value.casecmp(new_value) != 0
+          end
+
+          def changed_in_place?(raw_old_value, new_value)
+            raw_old_value.class != new_value.class ||
+              new_value && raw_old_value.casecmp(new_value) != 0
+          end
+
+          private
+
+          def cast_value(value)
+            casted = value.to_s
+            casted if casted.match?(ACCEPTABLE_UUID)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/clickhouse/schema_statements.rb

@@ -64,7 +64,7 @@ module ActiveRecord
 
         def do_system_execute(sql, name = nil)
           log_with_debug(sql, "#{adapter_name} #{name}") do
-            res = @connection.post("/?#{@config.to_param}", "#{sql} FORMAT JSONCompact", 'User-Agent' => "Clickhouse ActiveRecord #{ClickhouseActiverecord::VERSION}")
+            res = @connection.post("/?#{@connection_config.to_param}", "#{sql} FORMAT JSONCompact", 'User-Agent' => "Clickhouse ActiveRecord #{ClickhouseActiverecord::VERSION}")
 
             process_response(res)
           end
@@ -73,7 +73,7 @@ module ActiveRecord
         def do_execute(sql, name = nil, format: 'JSONCompact', settings: {})
           log(sql, "#{adapter_name} #{name}") do
             formatted_sql = apply_format(sql, format)
-            request_params = @config || {}
+            request_params = @connection_config || {}
             res = @connection.post("/?#{request_params.merge(settings).to_param}", formatted_sql, 'User-Agent' => "Clickhouse ActiveRecord #{ClickhouseActiverecord::VERSION}")
 
             process_response(res)
@@ -100,6 +100,16 @@ module ActiveRecord
           end
         end
 
+        # Fix insert_all method
+        # https://github.com/PNixx/clickhouse-activerecord/issues/71#issuecomment-1923244983
+        def with_yaml_fallback(value) # :nodoc:
+          if value.is_a?(Array)
+            value
+          else
+            super
+          end
+        end
+
         private
 
         def apply_format(sql, format)

data/lib/active_record/connection_adapters/clickhouse_adapter.rb

@@ -1,13 +1,14 @@
 # frozen_string_literal: true
 
 require 'arel/visitors/clickhouse'
+require 'arel/nodes/final'
 require 'arel/nodes/settings'
 require 'arel/nodes/using'
-require 'clickhouse-activerecord/migration'
 require 'active_record/connection_adapters/clickhouse/oid/array'
 require 'active_record/connection_adapters/clickhouse/oid/date'
 require 'active_record/connection_adapters/clickhouse/oid/date_time'
 require 'active_record/connection_adapters/clickhouse/oid/big_integer'
+require 'active_record/connection_adapters/clickhouse/oid/uuid'
 require 'active_record/connection_adapters/clickhouse/schema_definitions'
 require 'active_record/connection_adapters/clickhouse/schema_creation'
 require 'active_record/connection_adapters/clickhouse/schema_statements'
@@ -44,7 +45,7 @@ module ActiveRecord
           raise ArgumentError, 'No database specified. Missing argument: database.'
         end
 
-        ConnectionAdapters::ClickhouseAdapter.new(logger, connection, { user: config[:username], password: config[:password], database: database }.compact, config)
+        ConnectionAdapters::ClickhouseAdapter.new(logger, connection, config)
       end
     end
   end
@@ -63,7 +64,7 @@ module ActiveRecord
 
   module ModelSchema
     module ClassMethods
-      delegate :final, :settings, to: :all
+      delegate :final, :final!, :settings, :settings!, to: :all
 
       def is_view
         @is_view || false
@@ -119,33 +120,20 @@ module ActiveRecord
       include Clickhouse::SchemaStatements
 
       # Initializes and connects a Clickhouse adapter.
-      def initialize(logger, connection_parameters, config, full_config)
+      def initialize(logger, connection_parameters, config)
         super(nil, logger)
         @connection_parameters = connection_parameters
+        @connection_config = { user: config[:username], password: config[:password], database: config[:database] }.compact
+        @debug = config[:debug] || false
         @config = config
-        @debug = full_config[:debug] || false
-        @full_config = full_config
 
         @prepared_statements = false
 
         connect
       end
 
-      # Support SchemaMigration from v5.2.2 to v6+
-      def schema_migration # :nodoc:
-        ClickhouseActiverecord::SchemaMigration.new(self)
-      end
-
-      def internal_metadata # :nodoc:
-        ClickhouseActiverecord::InternalMetadata.new(self)
-      end
-
       def migrations_paths
-        @full_config[:migrations_paths] || 'db/migrate_clickhouse'
-      end
-
-      def migration_context # :nodoc:
-        ClickhouseActiverecord::MigrationContext.new(migrations_paths, schema_migration, internal_metadata)
+        @config[:migrations_paths] || 'db/migrate_clickhouse'
       end
 
       def arel_visitor # :nodoc:
@@ -211,6 +199,9 @@ module ActiveRecord
           register_class_with_limit m, %r(UInt64), Type::UnsignedInteger
           #register_class_with_limit m, %r(UInt128), Type::UnsignedInteger #not implemnted in clickhouse
           register_class_with_limit m, %r(UInt256), Type::UnsignedInteger
+
+          m.register_type %r(bool)i, ActiveModel::Type::Boolean.new
+          m.register_type %r{uuid}i, Clickhouse::OID::Uuid.new
           # register_class_with_limit m, %r(Array), Clickhouse::OID::Array
           m.register_type(%r(Array)) do |sql_type|
             Clickhouse::OID::Array.new(sql_type)
@@ -275,7 +266,7 @@ module ActiveRecord
       def create_database(name)
         sql = apply_cluster "CREATE DATABASE #{quote_table_name(name)}"
         log_with_debug(sql, adapter_name) do
-          res = @connection.post("/?#{@config.except(:database).to_param}", sql)
+          res = @connection.post("/?#{@connection_config.except(:database).to_param}", sql)
           process_response(res)
         end
       end
@@ -311,7 +302,7 @@ module ActiveRecord
           raise 'Set a cluster' unless cluster
 
           distributed_options =
-            "Distributed(#{cluster}, #{@config[:database]}, #{table_name}, #{sharding_key})"
+            "Distributed(#{cluster}, #{@connection_config[:database]}, #{table_name}, #{sharding_key})"
           create_table(distributed_table_name, **options.merge(options: distributed_options), &block)
         end
       end
@@ -320,7 +311,7 @@ module ActiveRecord
       def drop_database(name) #:nodoc:
         sql = apply_cluster "DROP DATABASE IF EXISTS #{quote_table_name(name)}"
         log_with_debug(sql, adapter_name) do
-          res = @connection.post("/?#{@config.except(:database).to_param}", sql)
+          res = @connection.post("/?#{@connection_config.except(:database).to_param}", sql)
           process_response(res)
         end
       end
@@ -374,15 +365,15 @@ module ActiveRecord
       end
 
       def cluster
-        @full_config[:cluster_name]
+        @config[:cluster_name]
       end
 
       def replica
-        @full_config[:replica_name]
+        @config[:replica_name]
       end
 
       def use_default_replicated_merge_tree_params?
-        database_engine_atomic? && @full_config[:use_default_replicated_merge_tree_params]
+        database_engine_atomic? && @config[:use_default_replicated_merge_tree_params]
       end
 
       def use_replica?
@@ -390,11 +381,11 @@ module ActiveRecord
       end
 
       def replica_path(table)
-        "/clickhouse/tables/#{cluster}/#{@config[:database]}.#{table}"
+        "/clickhouse/tables/#{cluster}/#{@connection_config[:database]}.#{table}"
       end
 
       def database_engine_atomic?
-        current_database_engine = "select engine from system.databases where name = '#{@config[:database]}'"
+        current_database_engine = "select engine from system.databases where name = '#{@connection_config[:database]}'"
         res = select_one(current_database_engine)
         res['engine'] == 'Atomic' if res
       end

data/lib/arel/nodes/final.rb

@@ -0,0 +1,7 @@
+module Arel # :nodoc: all
+  module Nodes
+    class Final < Arel::Nodes::Unary
+      delegate :empty?, to: :expr
+    end
+  end
+end

data/lib/arel/visitors/clickhouse.rb

@@ -13,12 +13,6 @@ module Arel
         end
       end
 
-      def visit_Arel_Table o, collector
-        collector = super
-        collector << ' FINAL' if o.final
-        collector
-      end
-
       def visit_Arel_Nodes_SelectOptions(o, collector)
         maybe_visit o.settings, super
       end
@@ -34,6 +28,12 @@ module Arel
         maybe_visit o.limit, collector
       end
 
+      def visit_Arel_Nodes_Final(o, collector)
+        visit o.expr, collector
+        collector << ' FINAL'
+        collector
+      end
+
       def visit_Arel_Nodes_Settings(o, collector)
         return collector if o.expr.empty?
 
@@ -53,6 +53,16 @@ module Arel
         collector
       end
 
+      def visit_Arel_Nodes_Matches(o, collector)
+        op = o.case_sensitive ? " LIKE " : " ILIKE "
+        infix_value o, collector, op
+      end
+
+      def visit_Arel_Nodes_DoesNotMatch(o, collector)
+        op = o.case_sensitive ? " NOT LIKE " : " NOT ILIKE "
+        infix_value o, collector, op
+      end
+
       def sanitize_as_setting_value(value)
         if value == :default
           'DEFAULT'

data/lib/clickhouse-activerecord/tasks.rb

@@ -2,7 +2,7 @@
 
 module ClickhouseActiverecord
   class Tasks
-    delegate :connection, :establish_connection, :clear_active_connections!, to: ActiveRecord::Base
+    delegate :connection, :establish_connection, to: ActiveRecord::Base
 
     def initialize(configuration)
       @configuration = configuration.with_indifferent_access
@@ -25,7 +25,7 @@ module ClickhouseActiverecord
     end
 
     def purge
-      clear_active_connections!
+      ActiveRecord::Base.connection_handler.clear_active_connections!(:all)
       drop
       create
     end

data/lib/clickhouse-activerecord/version.rb

@@ -1,3 +1,3 @@
 module ClickhouseActiverecord
-  VERSION = '1.0.2'
+  VERSION = '1.0.4'
 end

data/lib/clickhouse-activerecord.rb

@@ -2,8 +2,11 @@
 
 require 'active_record/connection_adapters/clickhouse_adapter'
 
+require 'core_extensions/active_record/internal_metadata'
 require 'core_extensions/active_record/relation'
+require 'core_extensions/active_record/schema_migration'
 
+require 'core_extensions/arel/nodes/select_core'
 require 'core_extensions/arel/nodes/select_statement'
 require 'core_extensions/arel/select_manager'
 require 'core_extensions/arel/table'
@@ -21,8 +24,11 @@ end
 
 module ClickhouseActiverecord
   def self.load
+    ActiveRecord::InternalMetadata.singleton_class.prepend(CoreExtensions::ActiveRecord::InternalMetadata::ClassMethods)
     ActiveRecord::Relation.prepend(CoreExtensions::ActiveRecord::Relation)
+    ActiveRecord::SchemaMigration.singleton_class.prepend(CoreExtensions::ActiveRecord::SchemaMigration::ClassMethods)
 
+    Arel::Nodes::SelectCore.prepend(CoreExtensions::Arel::Nodes::SelectCore)
     Arel::Nodes::SelectStatement.prepend(CoreExtensions::Arel::Nodes::SelectStatement)
     Arel::SelectManager.prepend(CoreExtensions::Arel::SelectManager)
     Arel::Table.prepend(CoreExtensions::Arel::Table)

data/lib/core_extensions/active_record/internal_metadata.rb

@@ -0,0 +1,46 @@
+module CoreExtensions
+  module ActiveRecord
+    module InternalMetadata
+      module ClassMethods
+
+        def []=(key, value)
+          row = final.find_by(key: key)
+          if row.nil? || row.value != value
+            create!(key: key, value: value)
+          end
+        end
+
+        def [](key)
+          final.where(key: key).pluck(:value).first
+        end
+
+        def create_table
+          return super unless connection.is_a?(::ActiveRecord::ConnectionAdapters::ClickhouseAdapter)
+          return if table_exists? || !enabled?
+
+          key_options = connection.internal_string_options_for_primary_key
+          table_options = {
+            id: false,
+            options: connection.adapter_name.downcase == 'clickhouse' ? 'ReplacingMergeTree(created_at) PARTITION BY key ORDER BY key' : '',
+            if_not_exists: true
+          }
+          full_config = connection.instance_variable_get(:@config) || {}
+
+          if full_config[:distributed_service_tables]
+            table_options.merge!(with_distributed: table_name, sharding_key: 'cityHash64(created_at)')
+
+            distributed_suffix = "_#{full_config[:distributed_service_tables_suffix] || 'distributed'}"
+          else
+            distributed_suffix = ''
+          end
+
+          connection.create_table(table_name + distributed_suffix.to_s, **table_options) do |t|
+            t.string :key, **key_options
+            t.string :value
+            t.timestamps
+          end
+        end
+      end
+    end
+  end
+end

data/lib/core_extensions/active_record/relation.rb

@@ -38,17 +38,14 @@ module CoreExtensions
       #   # SELECT users.* FROM users FINAL
       #
       # An <tt>ActiveRecord::ActiveRecordError</tt> will be raised if database not ClickHouse.
-      # @param [Boolean] final
-      def final(final = true)
-        spawn.final!(final)
+      def final
+        spawn.final!
       end
 
-      # @param [Boolean] final
-      def final!(final = true)
+      def final!
         assert_mutability!
         check_command('FINAL')
-        @table = @table.dup
-        @table.final = final
+        @values[:final] = true
         self
       end
 
@@ -79,6 +76,7 @@ module CoreExtensions
       def build_arel(aliases = nil)
         arel = super
 
+        arel.final! if @values[:final].present?
         arel.settings(@values[:settings]) if @values[:settings].present?
         arel.using(@values[:using]) if @values[:using].present?
 

data/lib/core_extensions/active_record/schema_migration.rb

@@ -0,0 +1,48 @@
+module CoreExtensions
+  module ActiveRecord
+    module SchemaMigration
+      module ClassMethods
+
+        def create_table
+          return super unless connection.is_a?(::ActiveRecord::ConnectionAdapters::ClickhouseAdapter)
+
+          return if table_exists?
+
+          version_options = connection.internal_string_options_for_primary_key
+          table_options = {
+            id: false, options: 'ReplacingMergeTree(ver) ORDER BY (version)', if_not_exists: true
+          }
+          full_config = connection.instance_variable_get(:@config) || {}
+
+          if full_config[:distributed_service_tables]
+            table_options.merge!(with_distributed: table_name, sharding_key: 'cityHash64(version)')
+
+            distributed_suffix = "_#{full_config[:distributed_service_tables_suffix] || 'distributed'}"
+          else
+            distributed_suffix = ''
+          end
+
+          connection.create_table(table_name + distributed_suffix.to_s, **table_options) do |t|
+            t.string :version, **version_options
+            t.column :active, 'Int8', null: false, default: '1'
+            t.datetime :ver, null: false, default: -> { 'now()' }
+          end
+        end
+
+        def delete_version(version)
+          return super unless connection.is_a?(::ActiveRecord::ConnectionAdapters::ClickhouseAdapter)
+
+          im = Arel::InsertManager.new(arel_table)
+          im.insert(arel_table[primary_key] => version.to_s, arel_table['active'] => 0)
+          connection.insert(im, "#{self.class} Create Rollback Version", primary_key, version)
+        end
+
+        def all_versions
+          return super unless connection.is_a?(::ActiveRecord::ConnectionAdapters::ClickhouseAdapter)
+
+          final.where(active: 1).order(:version).pluck(:version)
+        end
+      end
+    end
+  end
+end

data/lib/core_extensions/arel/nodes/select_core.rb

@@ -0,0 +1,19 @@
+module CoreExtensions
+  module Arel # :nodoc: all
+    module Nodes
+      module SelectCore
+        attr_accessor :final
+
+        def source
+          return super unless final
+
+          ::Arel::Nodes::Final.new(super)
+        end
+
+        def eql?(other)
+          super && final == other.final
+        end
+      end
+    end
+  end
+end

data/lib/core_extensions/arel/select_manager.rb

@@ -2,6 +2,11 @@ module CoreExtensions
   module Arel
     module SelectManager
 
+      def final!
+        @ctx.final = true
+        self
+      end
+
       # @param [Hash] values
       def settings(values)
         @ast.settings = ::Arel::Nodes::Settings.new(values)

data/lib/core_extensions/arel/table.rb

@@ -1,8 +1,6 @@
 module CoreExtensions
   module Arel
     module Table
-      attr_accessor :final
-
       def is_view
         type_caster.is_view
       end

data/lib/tasks/clickhouse.rake

@@ -12,15 +12,16 @@ namespace :clickhouse do
   end
 
   namespace :schema do
-    # TODO: not testing
+    # TODO: deprecated
     desc 'Load database schema'
     task load: %i[prepare_internal_metadata_table] do
       simple = ENV['simple'] || ARGV.any? { |a| a.include?('--simple') } ? '_simple' : nil
-      config = ActiveRecord::Base.configurations.configs_for(env_name: Rails.env, name: 'clickhouse')
-      ClickhouseActiverecord::SchemaMigration.new(ActiveRecord::Base.establish_connection(config).connection).drop_table
+      ActiveRecord::Base.establish_connection(:clickhouse)
+      ActiveRecord::SchemaMigration.drop_table
       load(Rails.root.join("db/clickhouse_schema#{simple}.rb"))
     end
 
+    # TODO: deprecated
     desc 'Dump database schema'
     task dump: :environment do |_, args|
       simple = ENV['simple'] || args[:simple] || ARGV.any? { |a| a.include?('--simple') } ? '_simple' : nil

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: clickhouse-activerecord
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.4
 platform: ruby
 authors:
 - Sergey Odintsov
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-12 00:00:00.000000000 Z
+date: 2024-02-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -103,21 +103,25 @@ files:
 - lib/active_record/connection_adapters/clickhouse/oid/big_integer.rb
 - lib/active_record/connection_adapters/clickhouse/oid/date.rb
 - lib/active_record/connection_adapters/clickhouse/oid/date_time.rb
+- lib/active_record/connection_adapters/clickhouse/oid/uuid.rb
 - lib/active_record/connection_adapters/clickhouse/schema_creation.rb
 - lib/active_record/connection_adapters/clickhouse/schema_definitions.rb
 - lib/active_record/connection_adapters/clickhouse/schema_statements.rb
 - lib/active_record/connection_adapters/clickhouse_adapter.rb
+- lib/arel/nodes/final.rb
 - lib/arel/nodes/settings.rb
 - lib/arel/nodes/using.rb
 - lib/arel/visitors/clickhouse.rb
 - lib/clickhouse-activerecord.rb
-- lib/clickhouse-activerecord/migration.rb
 - lib/clickhouse-activerecord/railtie.rb
 - lib/clickhouse-activerecord/schema.rb
 - lib/clickhouse-activerecord/schema_dumper.rb
 - lib/clickhouse-activerecord/tasks.rb
 - lib/clickhouse-activerecord/version.rb
+- lib/core_extensions/active_record/internal_metadata.rb
 - lib/core_extensions/active_record/relation.rb
+- lib/core_extensions/active_record/schema_migration.rb
+- lib/core_extensions/arel/nodes/select_core.rb
 - lib/core_extensions/arel/nodes/select_statement.rb
 - lib/core_extensions/arel/select_manager.rb
 - lib/core_extensions/arel/table.rb

data/lib/clickhouse-activerecord/migration.rb

@@ -1,103 +0,0 @@
-require 'active_record/migration'
-
-module ClickhouseActiverecord
-
-  class SchemaMigration < ::ActiveRecord::SchemaMigration
-    def create_table
-      return if table_exists?
-
-      version_options = connection.internal_string_options_for_primary_key
-      table_options = {
-        id: false, options: 'ReplacingMergeTree(ver) ORDER BY (version)', if_not_exists: true
-      }
-      full_config = connection.instance_variable_get(:@full_config) || {}
-
-      if full_config[:distributed_service_tables]
-        table_options.merge!(with_distributed: table_name, sharding_key: 'cityHash64(version)')
-
-        distributed_suffix = "_#{full_config[:distributed_service_tables_suffix] || 'distributed'}"
-      end
-
-      connection.create_table(table_name + distributed_suffix.to_s, **table_options) do |t|
-        t.string :version, **version_options
-        t.column :active, 'Int8', null: false, default: '1'
-        t.datetime :ver, null: false, default: -> { 'now()' }
-      end
-    end
-
-    def versions
-      table = arel_table.dup
-      table.final = true
-      sm = Arel::SelectManager.new(table)
-      sm.project(arel_table[primary_key])
-      sm.order(arel_table[primary_key].asc)
-      sm.where([arel_table['active'].eq(1)])
-
-      connection.select_values(sm, "#{self.class} Load")
-    end
-
-    def delete_version(version)
-      im = Arel::InsertManager.new(arel_table)
-      im.insert(arel_table[primary_key] => version.to_s, arel_table['active'] => 0)
-      connection.insert(im, "#{self.class} Create Rollback Version", primary_key, version)
-    end
-  end
-
-  class InternalMetadata < ::ActiveRecord::InternalMetadata
-
-    def create_table
-      return if table_exists? || !enabled?
-
-      key_options = connection.internal_string_options_for_primary_key
-      table_options = {
-        id: false,
-        options: connection.adapter_name.downcase == 'clickhouse' ? 'ReplacingMergeTree(created_at) PARTITION BY key ORDER BY key' : '',
-        if_not_exists: true
-      }
-      full_config = connection.instance_variable_get(:@full_config) || {}
-
-      if full_config[:distributed_service_tables]
-        table_options.merge!(with_distributed: table_name, sharding_key: 'cityHash64(created_at)')
-
-        distributed_suffix = "_#{full_config[:distributed_service_tables_suffix] || 'distributed'}"
-      end
-
-      connection.create_table(table_name + distributed_suffix.to_s, **table_options) do |t|
-        t.string :key, **key_options
-        t.string :value
-        t.timestamps
-      end
-    end
-
-    private
-
-    def update_entry(key, new_value)
-      create_entry(key, new_value)
-    end
-
-    def select_entry(key)
-      table = arel_table.dup
-      table.final = true
-      sm = Arel::SelectManager.new(table)
-      sm.project(Arel::Nodes::SqlLiteral.new("*"))
-      sm.where(table[primary_key].eq(Arel::Nodes::BindParam.new(key)))
-      sm.order(table[primary_key].asc)
-      sm.limit = 1
-
-      connection.select_all(sm, "#{self.class} Load").first
-    end
-  end
-
-  class MigrationContext < ::ActiveRecord::MigrationContext #:nodoc:
-
-    def get_all_versions
-      if schema_migration.table_exists?
-        schema_migration.versions.map(&:to_i)
-      else
-        []
-      end
-    end
-
-  end
-
-end