sequel-schema-sharding 0.9.1 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ea4a2b527ce17f8307ddd24be978f68bcd39b6bc
-  data.tar.gz: d4bb27788548838a33fd44f0c22b805911c275b0
+  metadata.gz: e8b1bb945b2bfeb51b502e79ae7199b9a15ac979
+  data.tar.gz: c644252bbcd60dc5c054208f366ef04706317d6d
 SHA512:
-  metadata.gz: 25c8ec3fe83a20bc5e5a4709f92f6a2627efbd2ae18d8e7ed806a6fa02cc73e9b7ab4f429d906f972264c8d6fd82796997c23e67a87d876c24ba812484ecb9d0
-  data.tar.gz: 408b96aae129a274e553e7abab512da3c19cb8a9de1f43c93736e1bf91bdebeacfda95aeb06aa89b5a4ee3019bcb114295cedaf59a7633f28d8be2bce78450d3
+  metadata.gz: fb2a36a6a4bbad5a40edfb9922ff132c0019f633f0f6ee03c9eabef4def14c13c3bbc0df82141939efb708b4ae845ed7e4ca04b2f49eab840514185143daf61f
+  data.tar.gz: 7d3dc3295513f7fc896b3686b48e467e1a06ba7026c0c6efea08c70f1b38efe31b8c49dae86cdc24d2e40d8ca3e776bfce45102afba393889b10148202e5e6db

data/Gemfile

@@ -8,4 +8,7 @@ group :test do
   gem 'mocha', require: false
   gem 'pry-nav'
   gem 'guard-rspec'
+
+  # for testing purposes until pull request is accepted
+  gem 'ruby-usdt', github: 'sax/ruby-usdt', ref: 'probe-introspection', submodules: true
 end

data/lib/sequel/schema-sharding.rb

@@ -1,6 +1,7 @@
 require 'logger'
 require 'sequel'
 require 'sequel/schema-sharding/version'
+require 'sequel/schema-sharding/dtrace_provider'
 require 'sequel/schema-sharding/configuration'
 require 'sequel/schema-sharding/connection_manager'
 require 'sequel/schema-sharding/database_manager'

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

@@ -0,0 +1,25 @@
+require 'usdt'
+
+module Sequel
+  module SchemaSharding
+    class DTraceProvider
+      attr_reader :provider
+
+      def initialize
+        @provider = USDT::Provider.create(:ruby, :sequel_schema_sharding)
+      end
+
+      def read_only_shard_for
+        @read_only_probe ||= provider.probe(:model, :read_only_shard_for, :string)
+      end
+
+      def self.provider
+        @provider ||= new.tap do |p|
+          p.read_only_shard_for
+          p.provider.enable
+        end
+      end
+    end
+  end
+end
+

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

@@ -57,6 +57,7 @@ module Sequel
         end
 
         def read_only_shard_for(id)
+          Sequel::SchemaSharding::DTraceProvider.provider.read_only_shard_for.fire(id.to_s) if Sequel::SchemaSharding::DTraceProvider.provider.read_only_shard_for.enabled?
           shard_for(id).server(:read_only)
         end
 

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

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

data/sequel-schema-sharding.gemspec

@@ -21,6 +21,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'sequel'
   spec.add_dependency 'pg'
   spec.add_dependency 'sequel-replica-failover'
+  spec.add_dependency 'ruby-usdt'
 
   spec.add_development_dependency 'bundler', '~> 1.3'
   spec.add_development_dependency 'rake'

data/spec/schema-sharding/dtrace_provider_spec.rb

@@ -0,0 +1,40 @@
+require 'spec_helper'
+
+describe Sequel::SchemaSharding::DTraceProvider do
+  describe 'initialize' do
+    it 'creates a new provider' do
+      USDT::Provider.expects(:create).with(:ruby, :sequel_schema_sharding)
+      Sequel::SchemaSharding::DTraceProvider.new
+    end
+  end
+
+  describe 'probes' do
+    let(:provider) { Sequel::SchemaSharding::DTraceProvider.new }
+
+    describe '#read_only_shard_for' do
+      it 'is a probe' do
+        expect(provider.read_only_shard_for).to be_a USDT::Probe
+      end
+
+      it 'has :model for its function' do
+        expect(provider.read_only_shard_for.function).to eq(:model)
+      end
+
+      it 'has :read_only_shard_for for its name' do
+        expect(provider.read_only_shard_for.name).to eq(:read_only_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
+    it 'returns a DTraceProvider' do
+      provider = Sequel::SchemaSharding::DTraceProvider.provider
+      expect(provider).to be_a(Sequel::SchemaSharding::DTraceProvider)
+    end
+  end
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sequel-schema-sharding
 version: !ruby/object:Gem::Version
-  version: 0.9.1
+  version: 0.10.0
 platform: ruby
 authors:
 - Paul Henry
@@ -54,6 +54,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: ruby-usdt
+  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: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -109,6 +123,7 @@ files:
 - lib/sequel/schema-sharding/connection_manager.rb
 - lib/sequel/schema-sharding/database_manager.rb
 - lib/sequel/schema-sharding/database_manager/schema_iterator.rb
+- lib/sequel/schema-sharding/dtrace_provider.rb
 - lib/sequel/schema-sharding/extensions/migrations_ext.rb
 - lib/sequel/schema-sharding/extensions/sequel_ext.rb
 - lib/sequel/schema-sharding/finder.rb
@@ -127,6 +142,7 @@ files:
 - spec/schema-sharding/connection_manager_spec.rb
 - spec/schema-sharding/database_manager/schema_iterator_spec.rb
 - spec/schema-sharding/database_manager_spec.rb
+- spec/schema-sharding/dtrace_provider_spec.rb
 - spec/schema-sharding/finder_spec.rb
 - spec/schema-sharding/model_spec.rb
 - spec/schema-sharding/ring_spec.rb
@@ -165,6 +181,7 @@ test_files:
 - spec/schema-sharding/connection_manager_spec.rb
 - spec/schema-sharding/database_manager/schema_iterator_spec.rb
 - spec/schema-sharding/database_manager_spec.rb
+- spec/schema-sharding/dtrace_provider_spec.rb
 - spec/schema-sharding/finder_spec.rb
 - spec/schema-sharding/model_spec.rb
 - spec/schema-sharding/ring_spec.rb