sequel-schema-sharding 0.10.0 → 0.11.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e8b1bb945b2bfeb51b502e79ae7199b9a15ac979
-  data.tar.gz: c644252bbcd60dc5c054208f366ef04706317d6d
+  metadata.gz: e6650670d2e7f69baf78e74a8078fb54199874b0
+  data.tar.gz: 397407436c70d317b8c18cdbf17e297225885045
 SHA512:
-  metadata.gz: fb2a36a6a4bbad5a40edfb9922ff132c0019f633f0f6ee03c9eabef4def14c13c3bbc0df82141939efb708b4ae845ed7e4ca04b2f49eab840514185143daf61f
-  data.tar.gz: 7d3dc3295513f7fc896b3686b48e467e1a06ba7026c0c6efea08c70f1b38efe31b8c49dae86cdc24d2e40d8ca3e776bfce45102afba393889b10148202e5e6db
+  metadata.gz: f9b175af8ac8e3ae327b66510f9ed6022ca1f978284770dd223a4f51937486e08dae239acb6075416a6f627bb67e6d0ac8593f04ef38188b11c10c90c5bc698b
+  data.tar.gz: 7064d7b970634745107c39eceda413d84b4df64afad13e1c4419302b792deec5848140050ebca22bedf9c06de1f7749c7542b4991b48b0a75e11f9ecf67654ac

data/lib/sequel/schema-sharding/connection_manager.rb

@@ -76,11 +76,15 @@ module Sequel
 
       def replica_hash_for(config)
         return {} if config['replicas'].nil?
-        i = rand(config['replicas'].size)
+        size = config['replicas'].size
+        i = rand(size)
         {
           :servers => {
             :read_only => ->(db) do
-              sequel_connection_config_for(config['replicas'][i % config['replicas'].size]).tap do
+              choice = i % size
+              probe = Sequel::SchemaSharding::DTraceProvider.provider.replica_hash_for
+              probe.fire(choice, size) if probe.enabled?
+              sequel_connection_config_for(config['replicas'][choice]).tap do
                 i += 1
               end
             end

data/lib/sequel/schema-sharding/dtrace_provider.rb

@@ -13,6 +13,14 @@ module Sequel
         @read_only_probe ||= provider.probe(:model, :read_only_shard_for, :string)
       end
 
+      def replica_hash_for
+        @replica_hash_for ||= provider.probe(:connection_manager, :replica_hash_for, :integer, :integer)
+      end
+
+      def shard_for
+        @shard_for_probe ||= provider.probe(:model, :shard_for, :string)
+      end
+
       def self.provider
         @provider ||= new.tap do |p|
           p.read_only_shard_for

data/lib/sequel/schema-sharding/model.rb

@@ -48,6 +48,7 @@ module Sequel
         # Return a valid Sequel::Dataset that is tied to the shard table and connection for the id and will load values
         # run by the query into the model.
         def shard_for(id)
+          Sequel::SchemaSharding::DTraceProvider.provider.shard_for.fire(id.to_s) if Sequel::SchemaSharding::DTraceProvider.provider.shard_for.enabled?
           result = self.result_for(id)
           ds = result.connection[schema_and_table(result)]
           ds.row_proc = self

data/lib/sequel/schema-sharding/version.rb

@@ -1,5 +1,5 @@
 module Sequel
   module SchemaSharding
-    VERSION = "0.10.0"
+    VERSION = "0.11.0"
   end
 end

data/spec/schema-sharding/dtrace_provider_spec.rb

@@ -11,6 +11,24 @@ describe Sequel::SchemaSharding::DTraceProvider do
   describe 'probes' do
     let(:provider) { Sequel::SchemaSharding::DTraceProvider.new }
 
+    describe '#replica_hash_for' do
+      it 'is a probe' do
+        expect(provider.replica_hash_for).to be_a USDT::Probe
+      end
+
+      it 'has :model for its function' do
+        expect(provider.replica_hash_for.function).to eq(:connection_manager)
+      end
+
+      it 'has :read_only_shard_for for its name' do
+        expect(provider.replica_hash_for.name).to eq(:replica_hash_for)
+      end
+
+      it 'takes a string argument' do
+        expect(provider.replica_hash_for.arguments).to eq([:integer, :integer])
+      end
+    end
+
     describe '#read_only_shard_for' do
       it 'is a probe' do
         expect(provider.read_only_shard_for).to be_a USDT::Probe
@@ -28,6 +46,24 @@ describe Sequel::SchemaSharding::DTraceProvider do
         expect(provider.read_only_shard_for.arguments).to eq([:string])
       end
     end
+
+    describe '#shard_for' do
+      it 'is a probe' do
+        expect(provider.shard_for).to be_a USDT::Probe
+      end
+
+      it 'has :model for its function' do
+        expect(provider.shard_for.function).to eq(:model)
+      end
+
+      it 'has :read_only_shard_for for its name' do
+        expect(provider.shard_for.name).to eq(:shard_for)
+      end
+
+      it 'takes a string argument' do
+        expect(provider.read_only_shard_for.arguments).to eq([:string])
+      end
+    end
   end
 
   describe '::provider' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sequel-schema-sharding
 version: !ruby/object:Gem::Version
-  version: 0.10.0
+  version: 0.11.0
 platform: ruby
 authors:
 - Paul Henry