clickhouse-rails 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 53465e63150b708d9a01b7508e09a91e6ed599e6a9436d56b65576499984d483
+  data.tar.gz: fb517f02559a4f1d8d7140a4e0842f8c634941dadc73bcb67dff9d8de773a5de
+SHA512:
+  metadata.gz: 45503b480904734621b91fd5a3177f045998aedcd8712b3144351cfdad0f56c7dba68797c7cc2139760d8bd3ac0247d0e2337568fbc1a46ab2181ffd1fd5c9e7
+  data.tar.gz: ddfa5b82e2e061de0ace2786f140bf1c5d00637e10fff34e1230304ddba52a74fa1e2e8a0a39ca2ba84f62844b969d1c57c3adbc3b74c25010a2d188bd0f39b4

data/.gitignore

@@ -0,0 +1,52 @@
+*.gem
+*.rbc
+/.config
+/coverage/
+/InstalledFiles
+/pkg/
+/spec/reports/
+/spec/examples.txt
+/spec/tmp
+/test/tmp/
+/test/version_tmp/
+/tmp/
+/tmp/db/clickhouse
+
+# Used by dotenv library to load environment variables.
+# .env
+
+## Specific to RubyMotion:
+.dat*
+.repl_history
+build/
+*.bridgesupport
+build-iPhoneOS/
+build-iPhoneSimulator/
+
+## Specific to RubyMotion (use of CocoaPods):
+#
+# We recommend against adding the Pods directory to your .gitignore. However
+# you should judge for yourself, the pros and cons are mentioned at:
+# https://guides.cocoapods.org/using/using-cocoapods.html#should-i-check-the-pods-directory-into-source-control
+#
+# vendor/Pods/
+
+## Documentation cache and generated files:
+/.yardoc/
+/_yardoc/
+/doc/
+/rdoc/
+
+## Environment normalization:
+/.bundle/
+/vendor/bundle
+/lib/bundler/man/
+
+# for a library or gem, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+# Gemfile.lock
+# .ruby-version
+# .ruby-gemset
+
+# unless supporting rvm < 1.11.0 or doing something fancy, ignore this:
+.rvmrc

data/.rspec

@@ -0,0 +1 @@
+--require spec_helper

data/.rubocop.yml

@@ -0,0 +1,14 @@
+Metrics/BlockLength:
+  Max: 25
+  Exclude:
+    - './spec/**/*_spec.rb'
+
+Naming/FileName:
+  Exclude:
+    - 'lib/clickhouse-rails.rb'
+
+Style/Documentation:
+  Enabled: false
+
+Metrics/LineLength:
+  Max: 97

data/.travis.yml

@@ -0,0 +1,12 @@
+language: ruby
+cache: bundler
+rvm:
+  - 2.5
+services:
+  - docker
+before_install:
+  - bundle install
+  - docker run -e "TZ=Europe/Moscow" -d -p 127.0.0.1:8123:8123 --name test-clickhouse-server --ulimit nofile=262144:262144 yandex/clickhouse-server
+script:
+  - rubocop
+  - rspec

data/Dockerfile.test

@@ -0,0 +1,11 @@
+FROM ruby:2.5
+
+RUN mkdir /app
+WORKDIR /app
+ADD . /app
+COPY ./Gemfile* /app/
+
+RUN gem install bundler &&\
+    bundle install
+
+CMD ["bundle exec rspec"]

data/Gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+gemspec
+
+gem 'codecov', require: false, group: :test
+gem 'rubocop', require: false

data/Gemfile.lock

@@ -0,0 +1,158 @@
+PATH
+  remote: .
+  specs:
+    clickhouse-rails (0.1.0)
+      clickhouse
+      railties
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actionpack (5.2.3)
+      actionview (= 5.2.3)
+      activesupport (= 5.2.3)
+      rack (~> 2.0)
+      rack-test (>= 0.6.3)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.0, >= 1.0.2)
+    actionview (5.2.3)
+      activesupport (= 5.2.3)
+      builder (~> 3.1)
+      erubi (~> 1.4)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.0, >= 1.0.3)
+    activesupport (5.2.3)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+    addressable (2.6.0)
+      public_suffix (>= 2.0.2, < 4.0)
+    ammeter (1.1.4)
+      activesupport (>= 3.0)
+      railties (>= 3.0)
+      rspec-rails (>= 2.2)
+    ast (2.4.0)
+    builder (3.2.3)
+    clickhouse (0.1.10)
+      activesupport (>= 4.1.8)
+      bundler (>= 1.13.4)
+      erubis
+      faraday
+      launchy
+      pond
+      sinatra
+      thor
+    codecov (0.1.14)
+      json
+      simplecov
+      url
+    coderay (1.1.2)
+    concurrent-ruby (1.1.5)
+    crass (1.0.4)
+    diff-lcs (1.3)
+    docile (1.3.1)
+    erubi (1.8.0)
+    erubis (2.7.0)
+    faraday (0.15.4)
+      multipart-post (>= 1.2, < 3)
+    i18n (1.6.0)
+      concurrent-ruby (~> 1.0)
+    jaro_winkler (1.5.2)
+    json (2.2.0)
+    launchy (2.4.3)
+      addressable (~> 2.3)
+    loofah (2.2.3)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.5.9)
+    method_source (0.9.2)
+    mini_portile2 (2.4.0)
+    minitest (5.11.3)
+    multipart-post (2.0.0)
+    mustermann (1.0.3)
+    nokogiri (1.10.2)
+      mini_portile2 (~> 2.4.0)
+    parallel (1.17.0)
+    parser (2.6.2.1)
+      ast (~> 2.4.0)
+    pond (0.3.0)
+    pry (0.12.2)
+      coderay (~> 1.1.0)
+      method_source (~> 0.9.0)
+    psych (3.1.0)
+    public_suffix (3.0.3)
+    rack (2.0.7)
+    rack-protection (2.0.5)
+      rack
+    rack-test (1.1.0)
+      rack (>= 1.0, < 3)
+    rails-dom-testing (2.0.3)
+      activesupport (>= 4.2.0)
+      nokogiri (>= 1.6)
+    rails-html-sanitizer (1.0.4)
+      loofah (~> 2.2, >= 2.2.2)
+    railties (5.2.3)
+      actionpack (= 5.2.3)
+      activesupport (= 5.2.3)
+      method_source
+      rake (>= 0.8.7)
+      thor (>= 0.19.0, < 2.0)
+    rainbow (3.0.0)
+    rake (12.3.0)
+    rspec-core (3.8.0)
+      rspec-support (~> 3.8.0)
+    rspec-expectations (3.8.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-mocks (3.8.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-rails (3.8.2)
+      actionpack (>= 3.0)
+      activesupport (>= 3.0)
+      railties (>= 3.0)
+      rspec-core (~> 3.8.0)
+      rspec-expectations (~> 3.8.0)
+      rspec-mocks (~> 3.8.0)
+      rspec-support (~> 3.8.0)
+    rspec-support (3.8.0)
+    rubocop (0.67.2)
+      jaro_winkler (~> 1.5.1)
+      parallel (~> 1.10)
+      parser (>= 2.5, != 2.5.1.1)
+      psych (>= 3.1.0)
+      rainbow (>= 2.2.2, < 4.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 1.4.0, < 1.6)
+    ruby-progressbar (1.10.0)
+    simplecov (0.16.1)
+      docile (~> 1.1)
+      json (>= 1.8, < 3)
+      simplecov-html (~> 0.10.0)
+    simplecov-html (0.10.2)
+    sinatra (2.0.5)
+      mustermann (~> 1.0)
+      rack (~> 2.0)
+      rack-protection (= 2.0.5)
+      tilt (~> 2.0)
+    thor (0.20.3)
+    thread_safe (0.3.6)
+    tilt (2.0.9)
+    tzinfo (1.2.5)
+      thread_safe (~> 0.1)
+    unicode-display_width (1.5.0)
+    url (0.3.2)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  ammeter (~> 1.1.2)
+  clickhouse-rails!
+  codecov
+  pry
+  rspec-rails
+  rubocop
+
+BUNDLED WITH
+   2.0.1

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2019 Avramov Vsevolod
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.md

@@ -0,0 +1,55 @@
+# Clickhouse rails
+
+[![Build Status](https://travis-ci.com/vsevolod/clickhouse-rails.svg?branch=master)](https://travis-ci.com/vsevolod/clickhouse-rails)
+[![codecov](https://codecov.io/gh/vsevolod/clickhouse-rails/branch/master/graph/badge.svg)](https://codecov.io/gh/vsevolod/clickhouse-rails)
+
+## Install
+
+1. Add to Gemfile
+```ruby
+gem 'clickhouse-rails'
+```
+
+2. Run bundle
+```bash
+$ bundle install
+```
+
+3. Init config files and folders
+```bash
+$ rails g clickhouse:install
+```
+
+4. Change clickhouse.yml at `config/clickhouse.yml` path
+
+5. Create migrations
+```bash
+$ rails g clickhouse:migration add_tmp_table
+      create  db/clickhouse/migrate/002_add_tmp_table.rb
+```
+
+6. Edit file like this:
+```ruby
+# db/clickhouse/migrate/002_add_tmp_table.rb
+class AddTmpTable < Clickhouse::Rails::Migrations::Base
+  def self.up
+    create_table :tmp do |t|
+      t.date   :date
+      t.uint16 :id
+
+      t.engine "MergeTree(date, (date), 8192)"
+    end
+  end
+end
+```
+
+## TODO:
+
+1. Rollback migrations
+2. Alter table
+
+## Additional clickhouse links
+
+- [Base gem](https://github.com/archan937/clickhouse)
+- [List of data types](https://clickhouse.yandex/docs/en/data_types/)
+- [Table engines](https://clickhouse.yandex/docs/en/operations/table_engines/)

data/clickhouse-rails.gemspec

@@ -0,0 +1,23 @@
+require File.expand_path('lib/clickhouse/rails/version', __dir__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ['Vsevolod Avramov']
+  gem.email         = ['gsevka@gmail.com']
+  gem.summary       = 'A Rails database driver for ClickHouse'
+  gem.description   = 'A Rails database driver for ClickHouse'
+  gem.homepage      = 'https://github.com/vsevolod/clickhouse-rails'
+
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {test,spec,features}/*`.split('\n')
+  gem.name          = 'clickhouse-rails'
+  gem.require_paths = ['lib']
+  gem.version       = Clickhouse::Rails::VERSION
+  gem.licenses      = ['MIT']
+
+  gem.add_dependency 'clickhouse'
+  gem.add_dependency 'railties'
+
+  gem.add_development_dependency 'ammeter', '~> 1.1.2'
+  gem.add_development_dependency 'pry'
+  gem.add_development_dependency 'rspec-rails'
+end

data/docker-compose.yml

@@ -0,0 +1,17 @@
+version: '3'
+services:
+  ruby:
+    build:
+      dockerfile: ./Dockerfile.test
+      context: .
+    environment:
+      - "CLICKHOUSE_HOSTS=clickhouse:8123"
+      - "CLICKHOUSE_USER=default"
+      - "CLICKHOUSE_ENV=test"
+    command: bash -c "bundle install && rspec"
+    volumes:
+      - ./:/app
+    links:
+      - "clickhouse:clickhouse"
+  clickhouse:
+    image: yandex/clickhouse-server

data/lib/clickhouse/rails/config.rb

@@ -0,0 +1,33 @@
+require 'yaml'
+
+module Clickhouse
+  module Rails
+    module Config
+      class ConfigurationNotFound < NameError; end
+
+      CLICKHOUSE_ROOT = File.expand_path(::Rails.root.present? ? ::Rails.root : '.')
+      DEFAULT_CONFIG_PATH = File.join(CLICKHOUSE_ROOT, 'config', 'clickhouse.yml')
+
+      def self.init(config_path = nil)
+        config_path ||= DEFAULT_CONFIG_PATH
+        exists = config_path && File.exist?(config_path)
+        unless exists
+          raise ConfigurationNotFound, "could not find the \"#{config_path}\" configuration file"
+        end
+
+        content = File.read(config_path)
+        data = defined?(ERB) ? ERB.new(content).result : content
+        source = YAML.safe_load(data)[defined?(::Rails) ? ::Rails.env : ENV['CLICKHOUSE_ENV']]
+        config_mapper(source)
+      end
+
+      def self.config_mapper(source)
+        {
+          urls: source['hosts'].split(','),
+          username: source['username'],
+          password: source['password']
+        }
+      end
+    end
+  end
+end

data/lib/clickhouse/rails/migrations/base.rb

@@ -0,0 +1,89 @@
+module Clickhouse
+  module Rails
+    module Migrations
+      class Base
+        MIGRATION_TABLE = 'schema_migrations'.freeze
+
+        class << self
+          attr_accessor :migrations_list
+
+          def inherited(child)
+            @migrations_list ||= []
+            @migrations_list.push(child)
+          end
+
+          def run_up
+            return unless @migrations_list
+
+            @migrations_list.each do |migration|
+              next if migration.passed?
+
+              migration.up
+              migration.add_version
+            end
+          end
+
+          def passed?
+            return false unless table_exists?(MIGRATION_TABLE)
+
+            @rows ||= connection.select_row(select: 'version', from: MIGRATION_TABLE)
+            @rows.include?(name)
+          rescue Clickhouse::QueryError
+            false
+          end
+
+          def up; end
+
+          def add_version
+            connection.insert_rows(MIGRATION_TABLE) do |row|
+              row << { version: name }
+            end
+          end
+
+          def table_exists?(table_name)
+            connection.execute("EXISTS TABLE #{table_name}").strip == '1'
+          end
+
+          def soft_create_table(table_name, &block)
+            return if table_exists?(table_name)
+
+            create_table(table_name, &block)
+          end
+
+          def soft_drop_table(table_name)
+            return unless table_exists?(table_name)
+
+            drop_table(table_name)
+          end
+
+          delegate :create_table, to: :connection
+          delegate :drop_table, to: :connection
+
+          def alter_table(table_name)
+            @table_info = connection.describe_table(table_name)
+            @table_name = table_name
+
+            yield(table_name)
+          end
+
+          def fetch_column(column, type)
+            return if @table_info.find { |c_info| c_info.first == column.to_s }
+
+            type = type.to_s
+                       .gsub(/(^.|_\w)/) do
+              Regexp.last_match(1).upcase
+            end
+                       .gsub('Uint', 'UInt')
+                       .delete('_')
+
+            connection.execute("ALTER TABLE #{@table_name} ADD COLUMN #{column} #{type}")
+          end
+
+          def connection
+            Clickhouse.connection
+          end
+        end
+      end
+    end
+  end
+end

data/lib/clickhouse/rails/version.rb

@@ -0,0 +1,5 @@
+module Clickhouse
+  module Rails
+    VERSION = '0.1.0'.freeze
+  end
+end

data/lib/clickhouse/rails.rb

@@ -0,0 +1,37 @@
+require 'rails'
+require 'clickhouse/rails/config'
+
+module Clickhouse
+  module Rails
+    def self.init!
+      Clickhouse.establish_connection(config)
+    end
+
+    def self.config(config_path = nil)
+      @config ||= Clickhouse::Rails::Config.init(config_path)
+    end
+
+    class Railtie < ::Rails::Railtie
+      generators =
+        if config.respond_to?(:app_generators)
+          config.app_generators
+        else
+          config.generators
+        end
+      generators.integration_tool :clickhouse
+      generators.test_framework :clickhouse
+
+      generators do
+        require 'generators/clickhouse/install/install_generator'
+      end
+
+      rake_tasks do
+        load 'tasks/clickhouse/db/migrate.rake'
+      end
+
+      config.to_prepare do
+        Clickhouse::Rails.init!
+      end
+    end
+  end
+end

data/lib/clickhouse-rails.rb

@@ -0,0 +1,2 @@
+require 'clickhouse'
+require 'clickhouse/rails'

data/lib/generators/clickhouse/install/install_generator.rb

@@ -0,0 +1,23 @@
+module Clickhouse
+  module Generators
+    class InstallGenerator < ::Rails::Generators::Base
+      desc <<-DESC
+        Description:
+          Copy clickhouse file to your application
+      DESC
+
+      def self.source_root
+        @source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'templates'))
+      end
+
+      def copy_install_files
+        copy_file 'config/clickhouse.yml'
+        directory db_migrate_path
+      end
+
+      def db_migrate_path
+        'db/clickhouse/migrate'
+      end
+    end
+  end
+end

data/lib/generators/clickhouse/install/templates/config/clickhouse.yml

@@ -0,0 +1,3 @@
+development:
+  hosts: <%= ENV.fetch('CLICKHOUSE_HOSTS', 'localhost:8123') %>
+  username: <%= ENV.fetch('CLICKHOUSE_USER', 'default') %>

data/lib/generators/clickhouse/install/templates/db/clickhouse/migrate/001_init.rb

@@ -0,0 +1,9 @@
+class Init < Clickhouse::Rails::Migrations::Base
+  def self.up
+    create_table MIGRATION_TABLE do |t|
+      t.string :version
+
+      t.engine 'File(TabSeparated)'
+    end
+  end
+end

data/lib/generators/clickhouse/migration/migration_generator.rb

@@ -0,0 +1,36 @@
+module Clickhouse
+  module Generators
+    class MigrationGenerator < ::Rails::Generators::NamedBase
+      include ::Rails::Generators::Migration
+
+      desc <<-DESC
+        Description:
+          Add migration for clickhouse database
+      DESC
+
+      def self.source_root
+        @source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'templates'))
+      end
+
+      def self.next_migration_number(dirname)
+        number = current_migration_number(dirname) + 1
+        format('%03d', number)
+      end
+
+      def create_migration_file
+        set_local_assign!
+        migration_template @migration_template, File.join(db_migrate_path, "#{file_name}.rb")
+      end
+
+      private
+
+      def set_local_assign!
+        @migration_template = 'db/migrate/template.rb'
+      end
+
+      def db_migrate_path
+        'db/clickhouse/migrate'
+      end
+    end
+  end
+end

data/lib/generators/clickhouse/migration/templates/db/migrate/template.rb.tt

@@ -0,0 +1,4 @@
+class <%= migration_class_name %> < Clickhouse::Rails::Migrations::Base
+  def self.up
+  end
+end

data/lib/tasks/clickhouse/db/migrate.rake

@@ -0,0 +1,14 @@
+require_relative '../../../clickhouse/rails'
+require_relative '../../../clickhouse/rails/migrations/base'
+
+namespace :clickhouse do
+  namespace :db do
+    desc 'Migrate clickhouse migrations'
+    task :migrate do
+      path_to_migrations = './db/clickhouse/migrate/*.rb'
+      Dir[path_to_migrations].sort.each { |file| require file }
+
+      Clickhouse::Rails::Migrations::Base.run_up
+    end
+  end
+end

data/spec/config_spec.rb

@@ -0,0 +1,22 @@
+require_relative 'rails_helper'
+
+describe Clickhouse::Rails::Config do
+  describe '.init' do
+    subject(:init) { described_class.init }
+
+    context 'when config file does not exists' do
+      let(:error_class) { Clickhouse::Rails::Config::ConfigurationNotFound }
+      it { expect { init }.to raise_error(error_class) }
+    end
+
+    context 'when config file exists' do
+      let(:path) { File.dirname(__FILE__) + '/fixtures/files/clickhouse.yml' }
+
+      before do
+        stub_const('Clickhouse::Rails::Config::DEFAULT_CONFIG_PATH', path)
+      end
+
+      it { expect { init }.not_to raise_error }
+    end
+  end
+end

data/spec/fixtures/files/001_init.rb

@@ -0,0 +1,9 @@
+class Init < Clickhouse::Rails::Migrations::Base
+  def self.up
+    create_table MIGRATION_TABLE do |t|
+      t.string :version
+
+      t.engine 'File(TabSeparated)'
+    end
+  end
+end

data/spec/fixtures/files/clickhouse.yml

@@ -0,0 +1,3 @@
+test:
+  hosts: <%= ENV.fetch('CLICKHOUSE_HOSTS', 'localhost:8123') %>
+  username: <%= ENV.fetch('CLICKHOUSE_USER', 'default') %>

data/spec/generators/install/install_generator_spec.rb

@@ -0,0 +1,16 @@
+require_relative '../../rails_helper'
+require_relative '../../../lib/generators/clickhouse/install/install_generator'
+
+RSpec.describe Clickhouse::Generators::InstallGenerator, type: :generator do
+  setup_default_destination
+
+  it 'generates config/clickhouse.yml' do
+    run_generator
+    expect(file('config/clickhouse.yml')).to exist
+  end
+
+  it 'generates db/clickhouse/migrate/001_init.rb' do
+    run_generator
+    expect(file('db/clickhouse/migrate/001_init.rb')).to exist
+  end
+end

data/spec/generators/migration/migration_generator_spec.rb

@@ -0,0 +1,11 @@
+require_relative '../../rails_helper'
+require_relative '../../../lib/generators/clickhouse/migration/migration_generator'
+
+RSpec.describe Clickhouse::Generators::MigrationGenerator, type: :generator do
+  setup_default_destination
+
+  it 'generates db/clickhouse/migrate/temp.rb' do
+    run_generator(['temp'])
+    expect(file('db/clickhouse/migrate/001_temp.rb')).to exist
+  end
+end

data/spec/migrations/base_spec.rb

@@ -0,0 +1,65 @@
+require_relative '../rails_helper'
+require_relative '../../lib/clickhouse/rails/migrations/base'
+
+RSpec.describe Clickhouse::Rails::Migrations::Base do
+  describe '.run_up' do
+    subject(:run_up) { described_class.run_up }
+
+    include_context 'with init migration' do
+      it 'executes up and add version' do
+        expect(Init).to receive(:up)
+        expect(Init).to receive(:add_version)
+
+        subject
+      end
+
+      context 'when two Init in @migrations_list' do
+        before do
+          described_class.migrations_list << Init
+        end
+
+        it 'skips the second time' do
+          expect(Init).to receive(:up).once
+
+          subject
+        end
+      end
+    end
+  end
+
+  describe '.create_table' do
+    subject(:create_table) { described_class.create_table(table_name, &block) }
+
+    let(:config_file) { file_fixture('clickhouse.yml') }
+    let(:table_name) { 'example' }
+    let(:block) do
+      lambda do |t|
+        t.string 'field'
+
+        t.engine 'File(TabSeparated)'
+      end
+    end
+
+    before do
+      Clickhouse::Rails.config(config_file.realpath.to_s)
+      Clickhouse::Rails.init!
+      described_class.soft_drop_table(table_name)
+    end
+
+    it 'adds table to clickhouse' do
+      create_table
+
+      expect(described_class).to be_table_exists(table_name)
+    end
+  end
+
+  describe '.add_version' do
+    include_context 'with init migration'
+
+    it 'executes up and add version' do
+      described_class.add_version
+
+      expect(Clickhouse.connection.count(from: migration_table)).to eq(1)
+    end
+  end
+end

data/spec/rails_helper.rb

@@ -0,0 +1,18 @@
+ENV['RAILS_ENV'] ||= 'test'
+
+require 'simplecov'
+SimpleCov.start
+
+require 'codecov'
+SimpleCov.formatter = SimpleCov::Formatter::Codecov
+
+require 'rails'
+require 'action_view/railtie'
+require 'action_controller/railtie'
+require 'rails/generators'
+
+require 'ammeter/init'
+
+require_relative '../lib/clickhouse-rails'
+require_relative 'support/generators'
+require_relative 'support/init'

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+RSpec.configure do |config|
+  config.expect_with :rspec do |expectations|
+    expectations.include_chain_clauses_in_custom_matcher_descriptions = true
+  end
+
+  config.mock_with :rspec do |mocks|
+    mocks.verify_partial_doubles = true
+  end
+
+  config.shared_context_metadata_behavior = :apply_to_host_groups
+end

data/spec/support/generators.rb

@@ -0,0 +1,24 @@
+module Clickhouse
+  module Rails
+    module Generators
+      module Macros
+        def set_default_destination
+          destination File.expand_path('../../tmp', __dir__)
+        end
+
+        def setup_default_destination
+          set_default_destination
+          before { prepare_destination }
+        end
+      end
+
+      def self.included(klass)
+        klass.extend(Macros)
+      end
+    end
+  end
+end
+
+RSpec.configure do |config|
+  config.include Clickhouse::Rails::Generators, type: :generator
+end

data/spec/support/init.rb

@@ -0,0 +1,15 @@
+RSpec.shared_context 'with init migration' do
+  let(:config_file) { file_fixture('clickhouse.yml') }
+  let(:file) { file_fixture('001_init.rb') }
+  let(:migration_table) { described_class::MIGRATION_TABLE }
+
+  before do
+    require file.realpath
+
+    Clickhouse::Rails.config(config_file.realpath.to_s)
+    Clickhouse::Rails.init!
+
+    described_class.soft_drop_table(migration_table)
+    Init.up
+  end
+end

metadata

@@ -0,0 +1,146 @@
+--- !ruby/object:Gem::Specification
+name: clickhouse-rails
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Vsevolod Avramov
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-04-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: clickhouse
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ammeter
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.2
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec-rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A Rails database driver for ClickHouse
+email:
+- gsevka@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- ".travis.yml"
+- Dockerfile.test
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- clickhouse-rails.gemspec
+- docker-compose.yml
+- lib/clickhouse-rails.rb
+- lib/clickhouse/rails.rb
+- lib/clickhouse/rails/config.rb
+- lib/clickhouse/rails/migrations/base.rb
+- lib/clickhouse/rails/version.rb
+- lib/generators/clickhouse/install/install_generator.rb
+- lib/generators/clickhouse/install/templates/config/clickhouse.yml
+- lib/generators/clickhouse/install/templates/db/clickhouse/migrate/001_init.rb
+- lib/generators/clickhouse/migration/migration_generator.rb
+- lib/generators/clickhouse/migration/templates/db/migrate/template.rb.tt
+- lib/tasks/clickhouse/db/migrate.rake
+- spec/config_spec.rb
+- spec/fixtures/files/001_init.rb
+- spec/fixtures/files/clickhouse.yml
+- spec/generators/install/install_generator_spec.rb
+- spec/generators/migration/migration_generator_spec.rb
+- spec/migrations/base_spec.rb
+- spec/rails_helper.rb
+- spec/spec_helper.rb
+- spec/support/generators.rb
+- spec/support/init.rb
+homepage: https://github.com/vsevolod/clickhouse-rails
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.6
+signing_key: 
+specification_version: 4
+summary: A Rails database driver for ClickHouse
+test_files: []