octoball 0.1.4 → 0.1.6.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 260d7c95bac1ba724636272cfa7ecf444124b868b2c7a3cea4d6966e0ac947a9
-  data.tar.gz: 5036fc1b1f6f5f0d7b24c734951ad6cbff0fd48eabc529a3f5390e55f2ad50d2
+  metadata.gz: 973c85322876fbc5cbe08b6500c6baa749d62c24cfa3903b33844de8db585250
+  data.tar.gz: 0a14fdace9ea1b64c14a6473d1b1064333058cf21e9ce52ebc267ec6b7a9a943
 SHA512:
-  metadata.gz: 1d954e47f0a2603425d29e82669b9388d4f49abf7def04e268548554ee5ab52181a2b192a5ccfd34b288a67b34fd6cf8cac4d75b8304c918970f7523f3177285
-  data.tar.gz: 7b294deea783c80324d5694afd87984017411677f593d042fa9b78a35e5dba5d12584209bb36d45190c3e63d748e1a4f5096e6a6e992ef9ebfa71e7968a41c4d
+  metadata.gz: 8a7d59a53be55e87dc67295f4116bad650ca0c9583bc23cdad34369d39d3aa173fd0f86b328c7aa989ab498f06627c996e9235e0b9e1ec35e5d1039b0d595412
+  data.tar.gz: faa9f389bb5277d3e094c7f9146aea3688ddeb639b489655c3f4ffce726c958e2c4b47dcd984dd33ea327d17bcb0d6eec4316cd778c82ead1287fe49cc2f9d11

data/{.circleci → .github}/Gemfile

@@ -1,8 +1,8 @@
 source 'https://rubygems.org'
 
 if RUBY_VERSION < '3.0'
-  gem 'activerecord', '~> 6.1.0'
-  gem 'activesupport', '~> 6.1.0'
+  gem 'activerecord', '~> 7.1.1'
+  gem 'activesupport', '~> 7.1.1'
 end
 
 gemspec :path => '../'

data/.github/workflows/rspec.yml

@@ -0,0 +1,38 @@
+name: rspec
+on: push
+jobs:
+  rspec:
+    runs-on: ubuntu-latest
+    services:
+      mysql:
+        image: mysql:5
+        ports:
+          - 3306:3306
+        env:
+          MYSQL_ALLOW_EMPTY_PASSWORD: yes
+        options: >-
+          --health-cmd "mysqladmin ping"
+          --health-interval 5s
+          --health-timeout 3s
+    strategy:
+      fail-fast: true
+      matrix:
+        ruby: ["2.7","3.0","3.1","3.2"]
+    env:
+      BUNDLE_GEMFILE: .github/Gemfile
+      MYSQL_HOST: 127.0.0.1
+      RAILS_ENV: test
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v3
+      - name: Setup ruby
+        uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: ${{ matrix.ruby }}
+          bundler-cache: true
+      - name: Install rake
+        run: gem install rake
+      - name: Setup database
+        run: bundle exec rake db:prepare
+      - name: run rspec
+        run: bundle exec rspec

data/lib/octoball/connection_handling.rb

@@ -4,13 +4,15 @@ class Octoball
   module ConnectionHandlingAvoidAutoLoadProxy
     private
 
-    def swap_connection_handler(handler, &blk)
-      old_handler, ActiveRecord::Base.connection_handler = ActiveRecord::Base.connection_handler, handler
-      return_value = yield
-      return_value.load if !return_value.respond_to?(:ar_relation) && return_value.is_a?(ActiveRecord::Relation)
-      return_value
-    ensure
-      ActiveRecord::Base.connection_handler = old_handler
+    if ActiveRecord.gem_version < Gem::Version.new('7.1.0')
+      def swap_connection_handler(handler, &blk)
+        old_handler, ActiveRecord::Base.connection_handler = ActiveRecord::Base.connection_handler, handler
+        return_value = yield
+        return_value.load if !return_value.respond_to?(:ar_relation) && return_value.is_a?(ActiveRecord::Relation)
+        return_value
+      ensure
+        ActiveRecord::Base.connection_handler = old_handler
+      end
     end
   end
 

data/lib/octoball/log_subscriber.rb

@@ -12,7 +12,11 @@ class Octoball
     private
 
     def debug(progname = nil, &block)
-      conn = current_shard ? color("[Shard: #{current_shard}]", ActiveSupport::LogSubscriber::GREEN, true) : ''
+      if ActiveRecord.gem_version >= Gem::Version.new('7.1.0')
+        conn = current_shard ? color("[Shard: #{current_shard}]", ActiveSupport::LogSubscriber::GREEN, bold: true) : ''
+      else
+        conn = current_shard ? color("[Shard: #{current_shard}]", ActiveSupport::LogSubscriber::GREEN, true) : ''
+      end
       super(conn + progname.to_s, &block)
     end
   end

data/lib/octoball/using_shard.rb

@@ -0,0 +1,17 @@
+class Octoball
+  def self.using(shard, &block)
+    ActiveRecord::Base.connected_to(role: current_role, shard: shard&.to_sym, &block)
+  end
+
+  def self.current_role
+    ActiveRecord::Base.current_role || ActiveRecord::Base.writing_role
+  end
+
+  module UsingShard
+    def using(shard)
+      Octoball::RelationProxy.new(all, shard&.to_sym)
+    end
+  end
+
+  ::ActiveRecord::Base.singleton_class.prepend(UsingShard)
+end

data/lib/octoball/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class Octoball
-  VERSION = '0.1.4'
+  VERSION = '0.1.6.1'
 end

data/lib/octoball.rb

@@ -2,29 +2,15 @@
 
 require 'active_record'
 require 'octoball/version'
-require 'octoball/relation_proxy'
-require 'octoball/connection_adapters'
-require 'octoball/connection_handling'
-require 'octoball/current_shard_tracker'
-require 'octoball/association'
-require 'octoball/association_shard_check'
-require 'octoball/persistence'
-require 'octoball/log_subscriber'
 
-class Octoball
-  def self.using(shard, &block)
-    ActiveRecord::Base.connected_to(role: current_role, shard: shard&.to_sym, &block)
-  end
-
-  def self.current_role
-    ActiveRecord::Base.current_role || ActiveRecord::Base.writing_role
-  end
-
-  module UsingShard
-    def using(shard)
-      Octoball::RelationProxy.new(all, shard&.to_sym)
-    end
-  end
-
-  ::ActiveRecord::Base.singleton_class.prepend(UsingShard)
+ActiveSupport.on_load(:active_record) do
+  require 'octoball/relation_proxy'
+  require 'octoball/connection_adapters'
+  require 'octoball/connection_handling'
+  require 'octoball/current_shard_tracker'
+  require 'octoball/association_shard_check'
+  require 'octoball/persistence'
+  require 'octoball/association'
+  require 'octoball/log_subscriber'
+  require 'octoball/using_shard'
 end

data/spec/octoball/model_spec.rb

@@ -123,7 +123,7 @@ describe Octoball do
 
     it 'should clean #current_shard from proxy when using execute' do
       User.using(:canada).connection.execute('select * from users limit 1;')
-      expect(User.connection.current_shard).to eq(:default)
+      expect(User.connection.current_shard).to eq(:master)
     end
 
     it 'should allow scoping dynamically' do
@@ -264,7 +264,7 @@ describe Octoball do
 
   describe 'AR basic methods' do
     it 'connects_to' do
-      expect(CustomConnection.connection.current_database).to eq('octoball_shard_2')
+      expect(CustomConnection.connection.current_database).to eq('octoball_shard_3')
     end
 
     it 'reuses parent model connection' do
@@ -590,7 +590,7 @@ describe Octoball do
   describe 'custom connection' do
     context 'by default' do
       it 'with plain call should use custom connection' do
-        expect(CustomConnection.connection.current_database).to eq('octoball_shard_2')
+        expect(CustomConnection.connection.current_database).to eq('octoball_shard_3')
       end
 
       it 'should use model-specific shard' do
@@ -618,7 +618,7 @@ describe Octoball do
         CustomConnection.create(:value => 'custom value')
 
         # This is what Rails, Sidekiq etc call--this normally handles all connection pools in the app
-        expect { ActiveRecord::Base.clear_active_connections! }
+        expect { ActiveRecord::Base.connection_handler.clear_active_connections! }
           .to change { CustomConnection.connection_pool.active_connection? }
 
         expect(CustomConnection.connection_pool.active_connection?).to be_falsey

data/spec/octoball/scope_proxy_spec.rb

@@ -39,19 +39,19 @@ describe Octoball do
     end
 
     it 'allows multiple selection by string' do
-      expect(@evan.select('id, name').first.id).to be_a(Fixnum)
+      expect(@evan.select('id, name').first.id).to be_a(Integer)
     end
 
     it 'allows multiple selection by array' do
-      expect(@evan.select(%w(id name)).first.id).to be_a(Fixnum)
+      expect(@evan.select(%w(id name)).first.id).to be_a(Integer)
     end
 
     it 'allows multiple selection by symbol' do
-      expect(@evan.select(:id, :name).first.id).to be_a(Fixnum)
+      expect(@evan.select(:id, :name).first.id).to be_a(Integer)
     end
 
     it 'allows multiple selection by string and symbol' do
-      expect(@evan.select(:id, 'name').first.id).to be_a(Fixnum)
+      expect(@evan.select(:id, 'name').first.id).to be_a(Integer)
     end
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: octoball
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.6.1
 platform: ruby
 authors:
 - Tomoki Sekiyama
-autorequire:
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-12-24 00:00:00.000000000 Z
+date: 2024-05-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -116,8 +116,8 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".circleci/Gemfile"
-- ".circleci/config.yml"
+- ".github/Gemfile"
+- ".github/workflows/rspec.yml"
 - ".gitignore"
 - Gemfile
 - README.md
@@ -131,6 +131,7 @@ files:
 - lib/octoball/log_subscriber.rb
 - lib/octoball/persistence.rb
 - lib/octoball/relation_proxy.rb
+- lib/octoball/using_shard.rb
 - lib/octoball/version.rb
 - octoball.gemspec
 - spec/migration/1_test_tables.rb
@@ -153,7 +154,7 @@ homepage: https://github.com/aktsk/octoball
 licenses:
 - MIT
 metadata: {}
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -169,7 +170,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubygems_version: 3.1.6
-signing_key:
+signing_key: 
 specification_version: 4
 summary: Octopus-like Database Sharding Helper for ActiveRecord 6.1+
 test_files: []

data/.circleci/config.yml

@@ -1,48 +0,0 @@
-version: 2.1
-
-executors:
-  ruby:
-    parameters:
-      version:
-        type: string
-    docker:
-    - image: circleci/ruby:<< parameters.version >>
-    - image: circleci/mysql:5
-    environment:
-    - RAILS_ENV: test
-    - MYSQL_HOST: 127.0.0.1
-
-jobs:
-  rspec:
-    parameters:
-      version:
-        type: string
-    executor:
-      name: ruby
-      version: << parameters.version >>
-    steps:
-    - checkout
-    - run:
-        name: bundle install
-        command: |
-          gem update bundler
-          bundle config --local path ../vendor/bundle
-          bundle install --gemfile .circleci/Gemfile --jobs=4 --retry=3
-          bundle config --local path vendor/bundle
-    - run:
-        name: Setup databases
-        command: bundle exec rake db:prepare
-    - run:
-        name: Run tests
-        command: bundle exec rspec
-
-workflows:
-  version: 2
-  rspecs:
-    jobs:
-    - rspec:
-        matrix:
-          parameters:
-            version:
-            - "2.7"
-            - "3.0"