tako 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a1b4936efb2bb9c024c85a7e41d835a1c0c23041
-  data.tar.gz: d8392ebb554a5186db289f7ecfcdc7f81ebf9188
+  metadata.gz: dff98c4a83f4ce6d0ed8154a4307b94c38589e1d
+  data.tar.gz: d671907d64c06501c442293125f728fa2d13c517
 SHA512:
-  metadata.gz: 7e6af9b9c02d28b361469c626ec181bddfc448804aa5444a4f16b19c813ab0f5e6a8a327712f1c3d6098853e573dc50cb333f9d87aae425afa05e0c44ca977d7
-  data.tar.gz: afc1f92c71a81c670b3ad139a0a654f629d51c96e53853511b60742c258342cca5b6e0d690f2efe8bfa93ad239c080c990b4399057bd0226f00d9872a5aa9dbc
+  metadata.gz: 188ddfbd46646a1103b29f2fd4391db979d89d5a23a1fbe36bb6181e36fc8db3afda0aea71dafdc520e4b79f392733895ddd05dd7e27ca21b55995e1417ddfb8
+  data.tar.gz: 9f4f1f4f50d3e929a70ade86bc7cd22bd036c72d2c83ad4af4e0f6af05c3fba76aeed92c5f5cf38169bad925530c3168761844f61036edc721b45f83139b35bd

data/.gitignore

@@ -8,3 +8,4 @@
 /spec/reports/
 /tmp/
 *.swp
+*.gem

data/README.md

@@ -1,6 +1,7 @@
 # Tako
 
 Provides Horizontal-Sharding for ActiveRecord.
+Rails 5 Ready.
 
 ## Installation
 
@@ -50,6 +51,11 @@ After checking out the repo, run `bin/setup` to install dependencies. Then, run
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 
+### Run test
+
+Run `bundle exec rake` to run rspec
+Run `bundle exec rake` in `spec/dummy5` will run rspec with rails 5
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/the40san/tako. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.

data/Rakefile

@@ -1,6 +1,8 @@
 require "bundler/gem_tasks"
 require "rspec/core/rake_task"
 
-RSpec::Core::RakeTask.new(:spec)
+RSpec::Core::RakeTask.new(:spec) do |t|
+  t.exclude_pattern = "spec/dummy*/**/*"
+end
 
 task :default => :spec

data/lib/tako/multi_shard_execution.rb

@@ -0,0 +1,15 @@
+module Tako
+  module MultiShardExecution
+    def with_all_shards
+      shard_names.each do |shard_name|
+        shard(shard_name) do
+          yield
+        end
+      end
+    end
+
+    def shard_names
+      Tako::Repository.shard_names
+    end
+  end
+end

data/lib/tako/query_chain.rb

@@ -10,7 +10,13 @@ module Tako
 
     def method_missing(method, *args)
       @proxy.in_proxy do
-        base_object.send(method, *args)
+        if block_given?
+          base_object.send(method, *args) do
+            yield
+          end
+        else
+          base_object.send(method, *args)
+        end
       end
     end
 

data/lib/tako/railtie.rb

@@ -0,0 +1,15 @@
+if defined?(::Rails)
+  module Tako
+    class Railtie < Rails::Railtie
+      rake_tasks do
+        load "tako/railties/databases.rake"
+      end
+
+      initializer "tako.initialize_database" do
+        ActiveSupport.on_load(:active_record) do
+          Tako.load_connections_from_yaml
+        end
+      end
+    end
+  end
+end

data/lib/tako/railties/databases.rake

@@ -0,0 +1,57 @@
+require 'tako'
+
+namespace :db do
+  namespace :tako do
+    task :load_config do
+      Tako.load_connections_from_yaml
+    end
+
+    task :check_protected_environments => [:environment, :load_config] do
+      ActiveRecord::Tasks::DatabaseTasks.check_protected_environments!
+    end
+
+    task :create do
+      (Tako.config[Tako.env] || []).values.each do |conf|
+        ActiveRecord::Tasks::DatabaseTasks.create(conf)
+      end
+    end
+
+    task :migrate => [:environment] do
+      paths = ActiveRecord::Tasks::DatabaseTasks.migrations_paths
+      # load all migration files
+      paths.each do |path|
+        Dir[File.join(path, "*")].each {|f| require f }
+      end
+
+      Tako.with_all_shards do
+        version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+        revert = !!ENV["DOWN_MIGRATION"]
+
+        migrations = if version
+          [
+            ActiveRecord::Migrator.migrations(paths).find {|proxy| proxy.version == version }.name.constantize
+          ]
+        else
+          ActiveRecord::Migrator.migrations(paths).map(&:name).map(&:constantize)
+        end
+
+        ActiveRecord::Migration.run(*migrations, revert: revert)
+      end
+    end
+
+    namespace :migrate do
+      task :up => ['db:tako:migrate']
+
+      task :down  => [:environment] do
+        ENV["DOWN_MIGRATION"] = "y"
+        Rake::Task['db:tako:migrate'].invoke
+      end
+    end
+
+    task :drop => [:load_config, :check_protected_environments] do
+      (Tako.config[Tako.env] || []).values.each do |conf|
+        ActiveRecord::Tasks::DatabaseTasks.drop(conf)
+      end
+    end
+  end
+end

data/lib/tako/repository.rb

@@ -2,30 +2,31 @@ module Tako
   class Repository
     class << self
       def proxy_configs
-        @proxy_configs
+        @proxy_configs ||= {}
       end
 
       def proxy_connections
-        @proxy_connections
+        @proxy_connections ||= {}
       end
 
       def add(shard_name, conf)
-        @proxy_configs ||= {}
-        @proxy_connections ||= {}
-
         shard_name = shard_name.to_sym
-        return if @proxy_configs[shard_name]
+        return if proxy_configs[shard_name]
 
         temporary_class = Class.new(ActiveRecord::Base)
         const_set("TAKO_AR_CLASS_#{shard_name.upcase}", temporary_class)
         temporary_class.establish_connection(conf)
 
-        @proxy_connections[shard_name] = temporary_class.connection
-        @proxy_configs[shard_name] = conf
+        proxy_connections[shard_name] = temporary_class.connection
+        proxy_configs[shard_name] = conf
       end
 
       def shard(shard_name, base = nil)
-        Proxy.new(shard_name, @proxy_connections[shard_name.to_sym], base)
+        Proxy.new(shard_name, proxy_connections[shard_name.to_sym], base)
+      end
+
+      def shard_names
+        proxy_configs.keys
       end
     end
   end

data/lib/tako/version.rb

@@ -1,3 +1,3 @@
 module Tako
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/tako.rb

@@ -6,8 +6,11 @@ require "tako/repository"
 require "tako/proxy_stack"
 require "tako/proxy"
 require "tako/query_chain"
+require "tako/multi_shard_execution"
 
 module Tako
+  extend MultiShardExecution
+
   class << self
     def shard(shard_name)
       if block_given?
@@ -19,11 +22,19 @@ module Tako
       end
     end
 
-    def load_connections_from_yaml(config = Tako::Config.shards_yml)
-      (config[:tako][Tako::Config.env] || []).each do |shard_name, conf|
+    def load_connections_from_yaml
+      (config[env] || []).each do |shard_name, conf|
         Tako::Repository.add(shard_name, conf)
       end
     end
+
+    def config
+      Tako::Config.shards_yml[:tako]
+    end
+
+    def env
+      Tako::Config.env
+    end
   end
 end
 
@@ -36,3 +47,5 @@ end
 ActiveRecord::Base.class_eval do
   extend Tako::ActiveRecordExt::Base::Extend
 end
+
+require 'tako/railtie' if defined?(::Rails)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tako
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Masashi AKISUE
@@ -100,9 +100,12 @@ files:
 - lib/tako.rb
 - lib/tako/active_record_ext.rb
 - lib/tako/config.rb
+- lib/tako/multi_shard_execution.rb
 - lib/tako/proxy.rb
 - lib/tako/proxy_stack.rb
 - lib/tako/query_chain.rb
+- lib/tako/railtie.rb
+- lib/tako/railties/databases.rake
 - lib/tako/repository.rb
 - lib/tako/version.rb
 - tako.gemspec