wye 0.0.3 → 0.1.0

data/lib/wye/active_record/base.rb

@@ -5,7 +5,7 @@ module Wye
     module Base
       module ClassMethods
         def on(alternate, &blk)
-          connection_handler.switch.on(alternate, &blk)
+          connection_handler.switch.on(self, alternate, &blk)
         end
       end
     end

data/lib/wye/active_record/connection_handler.rb

@@ -19,6 +19,10 @@ module Wye
         @class_to_alternate_pools.map { |(klass,atp)| atp.map(&:first) }.flatten.uniq
       end
 
+      def distributor(type)
+        Distributor.new(type, alternates << nil)
+      end
+
       def establish_connection(name, spec)
         super.tap do
           @class_to_alternate_pools[name] ||= {}

data/lib/wye/distributor/round_robin.rb

@@ -0,0 +1,15 @@
+module Wye
+  module Distributor
+    class RoundRobin
+      def initialize(values)
+        @cycle = values.cycle
+      end
+
+      def next(*arguments)
+        @cycle.next
+      rescue StopIteration
+        nil
+      end
+    end
+  end
+end

data/lib/wye/distributor/sticky.rb

@@ -0,0 +1,15 @@
+require 'zlib'
+
+module Wye
+  module Distributor
+    class Sticky
+      def initialize(values)
+        @values = values
+      end
+
+      def next(id)
+        (mod = @values.length) > 0 ? @values[Zlib.crc32(id.to_s) % mod] : nil
+      end
+    end
+  end
+end

data/lib/wye/distributor.rb

@@ -0,0 +1,12 @@
+module Wye
+  module Distributor
+    autoload :RoundRobin, 'wye/distributor/round_robin'
+    autoload :Sticky, 'wye/distributor/sticky'
+
+    def self.new(type, values)
+      klass = type.to_s.classify
+      raise "unknown distributor type" unless const_defined?(klass, false)
+      const_get(klass).new(values)
+    end
+  end
+end

data/lib/wye/version.rb

@@ -1,3 +1,3 @@
 module Wye
-  VERSION = '0.0.3'
+  VERSION = '0.1.0'
 end

data/lib/wye.rb

@@ -1,5 +1,6 @@
 module Wye
   autoload :ActiveRecord, 'wye/active_record'
+  autoload :Distributor, 'wye/distributor'
   autoload :Switch, 'wye/switch'
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: wye
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -56,6 +56,9 @@ files:
 - lib/wye/active_record/base.rb
 - lib/wye/active_record/connection_handler.rb
 - lib/wye/active_record.rb
+- lib/wye/distributor/round_robin.rb
+- lib/wye/distributor/sticky.rb
+- lib/wye/distributor.rb
 - lib/wye/railtie.rb
 - lib/wye/switch.rb
 - lib/wye/version.rb