sequel-replica-failover 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 184f9367a55260f7b22c13d31ee8bb3ee39d5ece
-  data.tar.gz: 2d5023b1cf3dff63cc77b08ead3b0dfc25277c33
+  metadata.gz: 77c8e355ea71e9f4eccfee4cd410733251ebc73f
+  data.tar.gz: a593201985a1b9f9c85921102a8d40b8498da2d6
 SHA512:
-  metadata.gz: c8a3edbdf975c3a2e6a99a19766dfb3cb1bd008207f8eb5f31eb71998de5ac01170a1fd941dab0b2e1499bb737396120f4db8f248ccb17c1c437550d5d80a167
-  data.tar.gz: 3d378d43a33d6babab0d587c334b8e003b1ceb59041168a927a3e9b4daf77082ad81a65a2ebea0de7483571e615ee88234ed556a40a8fecd7d85857510e36f71
+  metadata.gz: 32af56509abc4a23fdb48e83b3fa5401160e52a7a8e60f2c8b52faccde878cdfbaaf4395be121bc58baee5a74b2db06aa06175fe25261ca05ed29b261903d499
+  data.tar.gz: fe29a8ca07a06a7985d592af5131a43e748e848703c253d957b0eef2508c31839ff0d2af01d964e74d688d01040590caf2cb6ef8d603614fa6664b650c37096f

data/Gemfile

@@ -2,3 +2,6 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in sequel-replica-failover.gemspec
 gemspec
+
+# for testing purposes until pull request is accepted
+gem 'ruby-usdt', github: 'sax/ruby-usdt', ref: 'probe-introspection', submodules: true

data/Guardfile

@@ -0,0 +1,9 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+guard 'rspec', :version => 2 do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})     { |m| "spec/lib/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec" }
+end
+

data/lib/sequel-replica-failover.rb

@@ -1,5 +1,6 @@
 require 'sequel-replica-failover/version'
 require 'sequel/connection_pool/sharded_single_failover'
+require 'sequel-replica-failover/dtrace_provider'
 
 module Sequel
   module ReplicaFailover

data/lib/sequel-replica-failover/dtrace_provider.rb

@@ -0,0 +1,29 @@
+require 'usdt'
+
+module Sequel
+  module ReplicaFailover
+    class DTraceProvider
+      attr_reader :provider
+
+      def initialize
+        @provider = USDT::Provider.create(:ruby, :sequel_replica_failover)
+      end
+
+      def stick
+        @stick_probe ||= provider.probe(:connection, :stick)
+      end
+
+      def unstick
+        @unstick_probe ||= provider.probe(:connection, :unstick, :string)
+      end
+
+      def self.provider
+        @provider ||= new.tap do |p|
+          p.stick
+          p.unstick
+          p.provider.enable
+        end
+      end
+    end
+  end
+end

data/lib/sequel-replica-failover/version.rb

@@ -1,5 +1,5 @@
 module Sequel
   module ReplicaFailover
-    VERSION = '0.0.3'
+    VERSION = '0.1.0'
   end
 end

data/lib/sequel/connection_pool/sharded_single_failover.rb

@@ -42,6 +42,7 @@ class Sequel::ShardedSingleFailoverConnectionPool < Sequel::ShardedSingleConnect
   private
 
   def unstick(server)
+    probe(server.to_s) { |p| p.unstick }
     disconnect_server(server)
     @conns[server] = nil
     @stuck_at = nil
@@ -49,8 +50,19 @@ class Sequel::ShardedSingleFailoverConnectionPool < Sequel::ShardedSingleConnect
   end
 
   def stick
+    probe { |p| p.stick }
     @stuck_at ||= Time.now
     @stuck_times ||= 0
     @stuck_times += 1
   end
+
+  def probe(*args)
+    p = yield(Sequel::ReplicaFailover::DTraceProvider.provider)
+    return unless p.enabled?
+    if args.any?
+      p.fire(*args)
+    else
+      p.fire
+    end
+  end
 end

data/sequel-replica-failover.gemspec

@@ -19,9 +19,11 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency "sequel", ">= 4.3.0"
+  spec.add_dependency "ruby-usdt"
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "rake"
   spec.add_development_dependency "rspec"
   spec.add_development_dependency "timecop"
+  spec.add_development_dependency "guard-rspec"
 end

data/spec/sequel-replica-failover/dtrace_provider_spec.rb

@@ -0,0 +1,53 @@
+require 'spec_helper'
+
+describe Sequel::ReplicaFailover::DTraceProvider do
+  describe 'initialize' do
+    it 'creates a new provider' do
+      USDT::Provider.should_receive(:create).with(:ruby, :sequel_replica_failover)
+      Sequel::ReplicaFailover::DTraceProvider.new
+    end
+  end
+
+  describe 'probes' do
+    let(:provider) { Sequel::ReplicaFailover::DTraceProvider.new }
+
+    describe '#stick' do
+      it 'is a probe' do
+        expect(provider.stick).to be_a USDT::Probe
+      end
+
+      it 'has :replica_failover for its function' do
+        expect(provider.stick.function).to eq(:connection)
+      end
+
+      it 'has :stick for its name' do
+        expect(provider.stick.name).to eq(:stick)
+      end
+    end
+
+    describe '#unstick' do
+      it 'is a probe' do
+        expect(provider.unstick).to be_a USDT::Probe
+      end
+
+      it 'has :replica_failover for its function' do
+        expect(provider.unstick.function).to eq(:connection)
+      end
+
+      it 'has :unstick for its name' do
+        expect(provider.unstick.name).to eq(:unstick)
+      end
+
+      it 'takes a string argument' do
+        expect(provider.unstick.arguments).to eq([:string])
+      end
+    end
+  end
+
+  describe '::provider' do
+    it 'returns a DTraceProvider' do
+      provider = Sequel::ReplicaFailover::DTraceProvider.provider
+      expect(provider).to be_a(Sequel::ReplicaFailover::DTraceProvider)
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sequel-replica-failover
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Paul Henry
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-11-04 00:00:00.000000000 Z
+date: 2013-11-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel
@@ -24,6 +24,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: 4.3.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
@@ -80,6 +94,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard-rspec
+  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: Automatically failover between replicas when they go down.
 email:
 - paul@wanelo.com
@@ -90,13 +118,16 @@ files:
 - .gitignore
 - .rspec
 - Gemfile
+- Guardfile
 - LICENSE.txt
 - README.md
 - Rakefile
 - lib/sequel-replica-failover.rb
+- lib/sequel-replica-failover/dtrace_provider.rb
 - lib/sequel-replica-failover/version.rb
 - lib/sequel/connection_pool/sharded_single_failover.rb
 - sequel-replica-failover.gemspec
+- spec/sequel-replica-failover/dtrace_provider_spec.rb
 - spec/sequel/connection_pool/sharded_single_failover_spec.rb
 - spec/spec_helper.rb
 homepage: ''
@@ -124,5 +155,6 @@ signing_key:
 specification_version: 4
 summary: Automatically failover when replicas go down.
 test_files:
+- spec/sequel-replica-failover/dtrace_provider_spec.rb
 - spec/sequel/connection_pool/sharded_single_failover_spec.rb
 - spec/spec_helper.rb