srvy 0.0.1

Sign up to get free protection for your applications and to get access to all the features.
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: 720038b1f82de00846abd452dcacb9e847c5467f
4
+ data.tar.gz: 4a6b6e85214b477add94f062b7a50c66272922fd
5
+ SHA512:
6
+ metadata.gz: 6e498289f089c7ae9fc73cc587db8cf6638d03c6dc1ad739a572fcf3901af34c82b3fe240b0922a6f9504e2139f52465a3d2a40c8a385bd2de14802a3de2f76a
7
+ data.tar.gz: 2f9579278a60f8b8e115238430516477a6d19ecc585a43a821943a325acbfa026a661d28fa975d9456af41a97971a8c8b9bbcd073f6fd4ba3f4f16b262e5c787
@@ -0,0 +1 @@
1
+ service_name: travis-ci
@@ -0,0 +1,18 @@
1
+ *.gem
2
+ *.rbc
3
+ .bundle
4
+ .config
5
+ .yardoc
6
+ Gemfile.lock
7
+ InstalledFiles
8
+ _yardoc
9
+ coverage
10
+ doc/
11
+ lib/bundler/man
12
+ pkg
13
+ rdoc
14
+ spec/reports
15
+ test/tmp
16
+ test/version_tmp
17
+ tmp
18
+ .DS_Store
@@ -0,0 +1,6 @@
1
+ language: ruby
2
+ rvm:
3
+ - 2.1.0
4
+ - 2.0.0
5
+ - 1.9.3
6
+ - jruby
data/Gemfile ADDED
@@ -0,0 +1,4 @@
1
+ source 'https://rubygems.org'
2
+
3
+ # Specify your gem's dependencies in srvy.gemspec
4
+ gemspec
@@ -0,0 +1,22 @@
1
+ Copyright (c) 2014 Scott Fleckenstein
2
+
3
+ MIT License
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining
6
+ a copy of this software and associated documentation files (the
7
+ "Software"), to deal in the Software without restriction, including
8
+ without limitation the rights to use, copy, modify, merge, publish,
9
+ distribute, sublicense, and/or sell copies of the Software, and to
10
+ permit persons to whom the Software is furnished to do so, subject to
11
+ the following conditions:
12
+
13
+ The above copyright notice and this permission notice shall be
14
+ included in all copies or substantial portions of the Software.
15
+
16
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
@@ -0,0 +1,50 @@
1
+ # Srvy
2
+ [![Build Status](https://travis-ci.org/nullstyle/srvy.png?branch=master)](https://travis-ci.org/nullstyle/srvy)
3
+ [![Coverage Status](https://coveralls.io/repos/nullstyle/srvy/badge.png)](https://coveralls.io/r/nullstyle/srvy)
4
+ [![Code Climate](https://codeclimate.com/github/nullstyle/srvy.png)](https://codeclimate.com/github/nullstyle/srvy)
5
+
6
+ A rubygem to integrate SRV-based service discovery into your application
7
+
8
+ ## Installation
9
+
10
+ Add this line to your application's Gemfile:
11
+
12
+ gem 'srvy'
13
+
14
+ And then execute:
15
+
16
+ $ bundle
17
+
18
+ Or install it yourself as:
19
+
20
+ $ gem install srvy
21
+
22
+ ## Usage
23
+
24
+ ```ruby
25
+ $srvy = Srvy::Resolver.new # default configuration, using the locally configured resolver
26
+
27
+ # picks a single server by randomized with weight in the highest priority group
28
+ $srvy.get_single("db-slaves.mydomain.com") # => "mysql01.mydomain.com:3306"
29
+
30
+ # to get all services in the highest priority group
31
+ $srvy.get_many("memcache.mydomain.com") # => ["memcache01.mydomain.com:11211", "memcache02.mydomain.com:11211"]
32
+
33
+ # get all records, including lower priority (e.g. backup) services
34
+ $srvy.get_all("db.mydomain.com") # => ["db01.mydomain.com:3306", "db01-failover.mydomain.com:3306"]
35
+
36
+ ```
37
+
38
+
39
+
40
+ #TODO
41
+
42
+ I havent currently thought of a good api to signal to Srvy that you want to use lower priority services... currently the only way is to use `get_all` and pick the services out yourself.
43
+
44
+ ## Contributing
45
+
46
+ 1. Fork it
47
+ 2. Create your feature branch (`git checkout -b my-new-feature`)
48
+ 3. Commit your changes (`git commit -am 'Add some feature'`)
49
+ 4. Push to the branch (`git push origin my-new-feature`)
50
+ 5. Create new Pull Request
@@ -0,0 +1,10 @@
1
+ require "bundler/gem_tasks"
2
+
3
+ require 'rspec/core/rake_task'
4
+
5
+ RSpec::Core::RakeTask.new do |t|
6
+ t.rspec_opts = ["-c", "-f progress", "-r ./spec/spec_helper.rb"]
7
+ end
8
+
9
+ task :default => [:spec]
10
+
@@ -0,0 +1,10 @@
1
+ require 'lru_redux'
2
+
3
+ module HasKey
4
+ def has_key?(key)
5
+ @data.has_key?(key)
6
+ end
7
+ end
8
+
9
+
10
+ LruRedux::Cache.send(:include, HasKey)
@@ -0,0 +1,36 @@
1
+ require 'net/dns'
2
+
3
+ Net::DNS::Resolver.class_eval do
4
+ def nameservers=(arg)
5
+ @config[:nameservers] = convert_nameservers_arg_to_ips(arg)
6
+ @logger.info "Nameservers list changed to value #{@config[:nameservers].inspect}"
7
+ end
8
+
9
+ private
10
+ def convert_nameservers_arg_to_ips(arg)
11
+ case arg
12
+ when IPAddr ; [arg]
13
+ when String ;
14
+ begin
15
+ [IPAddr.new(arg)]
16
+ rescue ArgumentError # arg is in the name form, not IP
17
+ nameservers_from_name(arg)
18
+ end
19
+ when Array ;
20
+ arg.map{|x| convert_nameservers_arg_to_ips(x) }.flatten
21
+ else
22
+ raise ArgumentError, "Wrong argument format, neither String, Array nor IPAddr"
23
+ end
24
+ end
25
+
26
+ def nameservers_from_name(arg)
27
+ arr = []
28
+ arg.split(" ").each do |name|
29
+ Net::DNS::Resolver.new.search(name).each_address do |ip|
30
+ arr << ip
31
+ end
32
+ end
33
+ arr
34
+ end
35
+
36
+ end
@@ -0,0 +1,10 @@
1
+ require "srvy/version"
2
+ require "net/dns/ext/resolver_nameserver_monkey_patch"
3
+
4
+ module Srvy
5
+ autoload :Resolver, "srvy/resolver"
6
+ autoload :Result, "srvy/result"
7
+ autoload :Host, "srvy/host"
8
+
9
+ autoload :Formatters, "srvy/formatters"
10
+ end
@@ -0,0 +1,23 @@
1
+ module Srvy
2
+ module Formatters
3
+ autoload :Base, "srvy/formatters/base"
4
+ autoload :HostPort, "srvy/formatters/host_port"
5
+ autoload :Dalli, "srvy/formatters/dalli"
6
+
7
+ def self.from_name(name)
8
+ case name.to_sym
9
+ when :host_port ; HostPort.new
10
+ when :dalli ; Dalli.new
11
+ else ; raise ArgumentError, "Unknown formatter name: #{name}"
12
+ end
13
+ end
14
+
15
+ def self.format_single(format_name, result)
16
+ from_name(format_name).format_single(result)
17
+ end
18
+
19
+ def self.format_many(format_name, result)
20
+ from_name(format_name).format_many(result)
21
+ end
22
+ end
23
+ end
@@ -0,0 +1,17 @@
1
+ module Srvy
2
+ module Formatters
3
+ class Base
4
+
5
+ def format_single(host)
6
+ raise NotImplementedError
7
+ end
8
+
9
+ def format_many(hosts)
10
+ # default to mapping over the collection with the format_single form
11
+ # by having a separate format_many call, we can build formatters
12
+ # that aggregate results in some way (such as producing a map of the hosts)
13
+ hosts.map{|h| format_single h}
14
+ end
15
+ end
16
+ end
17
+ end
@@ -0,0 +1,11 @@
1
+ module Srvy
2
+ module Formatters
3
+ class Dalli < Base
4
+
5
+ def format_single(host)
6
+ "#{host.host}:#{host.port}:#{host.weight}"
7
+ end
8
+
9
+ end
10
+ end
11
+ end
@@ -0,0 +1,11 @@
1
+ module Srvy
2
+ module Formatters
3
+ class HostPort < Base
4
+
5
+ def format_single(host)
6
+ "#{host.host}:#{host.port}"
7
+ end
8
+
9
+ end
10
+ end
11
+ end
@@ -0,0 +1,18 @@
1
+ module Srvy
2
+ class Host
3
+
4
+ attr_reader :host
5
+ attr_reader :port
6
+ attr_reader :weight
7
+ attr_reader :priority
8
+ attr_reader :ttl
9
+
10
+ def initialize(host, port, weight, priority, ttl)
11
+ @host = host
12
+ @port = port
13
+ @weight = weight
14
+ @priority = priority
15
+ @ttl = ttl
16
+ end
17
+ end
18
+ end
@@ -0,0 +1,61 @@
1
+ require "net/dns/ext/resolver_nameserver_monkey_patch"
2
+ require 'lru_redux/has_key_monkey_patch'
3
+
4
+ module Srvy
5
+ class Resolver
6
+ attr_reader :cache
7
+
8
+ def initialize(options={})
9
+ @dns = Net::DNS::Resolver.new
10
+
11
+ if options[:nameserver]
12
+ @dns.nameservers = options[:nameserver]
13
+ end
14
+
15
+ cache_size = options[:cache_size] || 100
16
+ @cache = LruRedux::Cache.new(cache_size) #cache of host -> result kv pairs
17
+ end
18
+
19
+ def inspect
20
+ "#<Srvy::Resolver:#{object_id} dns=#{@dns.nameservers.inspect}>"
21
+ end
22
+
23
+ def get_single(host, format=:host_port)
24
+ result = get(host).get_single
25
+
26
+ Srvy::Formatters.format_single(format, result)
27
+ end
28
+
29
+ def get_many(host, format=:host_port)
30
+ result = get(host).get_many
31
+ Srvy::Formatters.format_many(format, result)
32
+ end
33
+
34
+ def get_all(host, format=:host_port)
35
+ result = get(host).get_all
36
+ Srvy::Formatters.format_many(format, result)
37
+ end
38
+
39
+ def get_dns(host)
40
+ @dns.search(host, Net::DNS::SRV)
41
+ end
42
+
43
+ # Gets the Srvy::Result from the cache or from dns.
44
+ #
45
+ # host - The String hostname to query for SRV records.
46
+ #
47
+ #
48
+ # Returns the Srvy::Result for that hostname
49
+ def get(host)
50
+ result = @cache[host]
51
+
52
+ if !result || result.expired?
53
+ result = Srvy::Result.from_dns get_dns(host)
54
+ @cache[host] = result
55
+ end
56
+
57
+ result
58
+ end
59
+
60
+ end
61
+ end
@@ -0,0 +1,61 @@
1
+ module Srvy
2
+ class Result
3
+ EMPTY_RESULT_TTL = 10
4
+
5
+ def self.from_dns(dns_result)
6
+ # for each SRV record
7
+ srvs = dns_result.answer.select{|rr| rr.is_a?(Net::DNS::RR::SRV) }
8
+ hosts = srvs.map do |srv|
9
+ Srvy::Host.new(srv.host, srv.port, srv.weight, srv.priority, srv.ttl)
10
+ end
11
+
12
+ new(Time.now, hosts)
13
+ end
14
+
15
+ def initialize(created_at, hosts)
16
+ @created_at = created_at
17
+ @hosts = hosts
18
+ end
19
+
20
+ def get_single
21
+ return nil if @hosts.empty?
22
+
23
+ roll = rand(best_priority_cumulative_weight)
24
+ acc = 0
25
+
26
+ best_priority_hosts_by_weight.each do |host|
27
+ acc += host.weight
28
+ return host if roll < acc
29
+ end
30
+ end
31
+
32
+ def get_many
33
+ best_priority_hosts_by_weight
34
+ end
35
+
36
+ def get_all
37
+ @hosts
38
+ end
39
+
40
+ def expired?
41
+ now = Time.now
42
+ min_ttl = @hosts.map(&:ttl).min
43
+ min_ttl ||= EMPTY_RESULT_TTL # in case there are no hosts
44
+ @created_at + min_ttl < now
45
+ end
46
+
47
+ private
48
+ def best_priority_hosts_by_weight
49
+ # sorted weight descending
50
+ best_priority_hosts_by_weight ||= @hosts.select{|s| s.priority <= best_priority }.sort_by(&:weight).reverse
51
+ end
52
+
53
+ def best_priority_cumulative_weight
54
+ best_priority_hosts_by_weight.map(&:weight).inject(0, :+)
55
+ end
56
+
57
+ def best_priority
58
+ @best_priority ||= @hosts.map(&:priority).min
59
+ end
60
+ end
61
+ end
@@ -0,0 +1,3 @@
1
+ module Srvy
2
+ VERSION = "0.0.1"
3
+ end
@@ -0,0 +1,13 @@
1
+ require 'rspec'
2
+ require 'rspec/given'
3
+ require 'srvy'
4
+ require 'coveralls'
5
+ require 'timecop'
6
+ require 'active_support/all'
7
+
8
+ Coveralls.wear!
9
+
10
+ RSpec.configure do |config|
11
+
12
+
13
+ end
@@ -0,0 +1,10 @@
1
+ require 'spec_helper'
2
+
3
+ describe Srvy::Formatters::Base, "format_single" do
4
+ Given(:host) { Srvy::Host.new("test01.host.com", 11211, 0, 0, 100) }
5
+
6
+ When(:result){ subject.format_single(host) }
7
+
8
+ Then { expect(result).to have_failed(NotImplementedError) }
9
+ end
10
+
@@ -0,0 +1,21 @@
1
+ require 'spec_helper'
2
+
3
+ describe Srvy::Formatters::Dalli, "format_single" do
4
+ Given(:host) { Srvy::Host.new("test01.host.com", 11211, 10, 0, 100) }
5
+
6
+ When(:result){ subject.format_single(host) }
7
+
8
+ Then{ expect(result).to eq("test01.host.com:11211:10") }
9
+ end
10
+
11
+
12
+ describe Srvy::Formatters::Dalli, "format_many" do
13
+ Given(:hosts) { [
14
+ Srvy::Host.new("test01.host.com", 11211, 10, 0, 100),
15
+ Srvy::Host.new("test02.host.com", 11211, 5, 0, 100),
16
+ ] }
17
+
18
+ When(:result){ subject.format_many(hosts) }
19
+
20
+ Then{ expect(result).to eq(["test01.host.com:11211:10", "test02.host.com:11211:5"]) }
21
+ end
@@ -0,0 +1,18 @@
1
+ require 'spec_helper'
2
+
3
+ describe Srvy::Formatters::HostPort, "format_single" do
4
+ Given(:host) { Srvy::Host.new("test01.host.com", 11211, 0, 0, 100) }
5
+
6
+ When(:result){ subject.format_single(host) }
7
+
8
+ Then{ expect(result).to eq("test01.host.com:11211") }
9
+ end
10
+
11
+
12
+ describe Srvy::Formatters::HostPort, "format_many" do
13
+ Given(:hosts) { [Srvy::Host.new("test01.host.com", 11211, 0, 0, 100)] * 2 }
14
+
15
+ When(:result){ subject.format_many(hosts) }
16
+
17
+ Then{ expect(result).to eq(["test01.host.com:11211", "test01.host.com:11211"]) }
18
+ end
@@ -0,0 +1,40 @@
1
+ require 'spec_helper'
2
+
3
+ describe Srvy::Formatters, "#from_name" do
4
+
5
+ context "with an invalid name" do
6
+ Given(:name){ "some_formatter" }
7
+ When(:result) { Srvy::Formatters.from_name(name) }
8
+ Then { expect(result).to have_failed(ArgumentError, /Unknown/) }
9
+ end
10
+
11
+
12
+ context "with host_port" do
13
+ Given(:name){ "host_port" }
14
+ When(:result) { Srvy::Formatters.from_name(name) }
15
+ Then { expect(result).to be_kind_of(Srvy::Formatters::HostPort) }
16
+ end
17
+ end
18
+
19
+ describe Srvy::Formatters, "#format_single" do
20
+ Given(:formatter) { double("Formatter", :format_single => true)}
21
+ Given(:name) { "host_port" }
22
+ Given(:host) { Srvy::Host.new("test01.host.com", 11211, 0, 0, 100) }
23
+ Given { Srvy::Formatters.stub(:from_name){ formatter } }
24
+
25
+ When(:result) { Srvy::Formatters.format_single(name, host) }
26
+
27
+ Then{ expect(formatter).to have_received(:format_single).with(host) }
28
+ end
29
+
30
+
31
+ describe Srvy::Formatters, "#format_many" do
32
+ Given(:formatter) { double("Formatter", :format_many => true)}
33
+ Given(:name) { "host_port" }
34
+ Given(:hosts) { [Srvy::Host.new("test01.host.com", 11211, 0, 0, 100)] }
35
+ Given { Srvy::Formatters.stub(:from_name){ formatter } }
36
+
37
+ When(:result) { Srvy::Formatters.format_many(name, hosts) }
38
+
39
+ Then{ expect(formatter).to have_received(:format_many).with(hosts) }
40
+ end
@@ -0,0 +1,25 @@
1
+ require 'spec_helper'
2
+
3
+ describe Srvy::Resolver do
4
+ describe "cache setting" do
5
+ Given(:host) { "google.com"}
6
+
7
+ When{ subject.get host }
8
+
9
+ Then{ expect(subject.cache).to have_key(host) }
10
+
11
+ end
12
+
13
+ describe "cache usage" do
14
+
15
+ Given(:host) { "google.com"}
16
+ Given(:srvy_result) { Srvy::Result.new(Time.now, [Srvy::Host.new("test01.host.com", 11211, 10, 0, 100)]) }
17
+
18
+ Given { subject.cache[host] = srvy_result }
19
+
20
+ When(:result){ subject.get host }
21
+
22
+ Then{ expect(result).to eq(srvy_result) }
23
+
24
+ end
25
+ end
@@ -0,0 +1,178 @@
1
+ require 'spec_helper'
2
+
3
+ describe Srvy::Result do
4
+ Given(:now) { Time.at(Time.now.to_i) } # round to nearest second so jruby stops complaining on boundary cases
5
+ Given(:ttl) { 100 }
6
+ Given(:call_count) { 10000 }
7
+ Given(:allowance) { 0.02 }
8
+
9
+ # different hosts definitions
10
+ Given(:hosts_with_equal_weight) { [
11
+ Srvy::Host.new("test01.host.com", 11211, 10, 0, ttl),
12
+ Srvy::Host.new("test02.host.com", 11211, 10, 0, ttl),
13
+ ] }
14
+
15
+ Given(:hosts_with_unequal_weight) { [
16
+ Srvy::Host.new("test01.host.com", 11211, 30, 0, ttl),
17
+ Srvy::Host.new("test02.host.com", 11211, 10, 0, ttl),
18
+ ] }
19
+
20
+ Given(:hosts_with_unequal_priorities) { [
21
+ Srvy::Host.new("test01.host.com", 11211, 10, 0, ttl),
22
+ Srvy::Host.new("test02.host.com", 11211, 10, 1, ttl),
23
+ ] }
24
+ Given(:no_hosts) { [ ] }
25
+
26
+
27
+ describe "#expired?" do
28
+
29
+
30
+ context "with data" do
31
+ subject{ Srvy::Result.new(now, hosts_with_equal_weight) }
32
+
33
+ Then{ Timecop.freeze(now) { expect(subject).to_not be_expired } }
34
+ Then{ Timecop.freeze(now + ttl) { expect(subject).to_not be_expired } }
35
+ Then{ Timecop.freeze(now + ttl + 1) { expect(subject).to be_expired } }
36
+ Then{ Timecop.freeze(now - ttl) { expect(subject).to_not be_expired } }
37
+ Then{ Timecop.freeze(now - (ttl + 1)) { expect(subject).to_not be_expired } }
38
+ end
39
+
40
+ context "with no data" do
41
+ subject{ Srvy::Result.new(now, []) }
42
+
43
+ Then{ Timecop.freeze(now) { expect(subject).to_not be_expired } }
44
+ Then{ Timecop.freeze(now + 10) { expect(subject).to_not be_expired } }
45
+ Then{ Timecop.freeze(now + 11) { expect(subject).to be_expired } }
46
+ Then{ Timecop.freeze(now - 10) { expect(subject).to_not be_expired } }
47
+ Then{ Timecop.freeze(now - 11) { expect(subject).to_not be_expired } }
48
+ end
49
+ end
50
+
51
+ describe "#from_dns" do
52
+ subject{ Srvy::Result.from_dns(dns_result) }
53
+
54
+ Given(:dns_result) { Net::DNS::Resolver.new(:nameservers => "8.8.8.8").search("_xmpp-server._tcp.gmail.com", Net::DNS::SRV) }
55
+ Given(:srvs) { dns_result.answer.select{|rr| rr.is_a?(Net::DNS::RR::SRV) }}
56
+ Given(:dns_hosts) { srvs.map(&:host).sort }
57
+ Given(:srvys) { subject.get_all }
58
+ Given(:srvy_hosts) { srvys.map(&:host).sort }
59
+
60
+ Then{ expect(subject).to be_kind_of(Srvy::Result) }
61
+ Then{ expect(srvy_hosts).to eq(dns_hosts) }
62
+
63
+
64
+ Then do
65
+ srvys.each do |srvy|
66
+ dns = srvs.find{|d| d.host == srvy.host}
67
+
68
+ expect(dns).to_not be_nil
69
+
70
+ expect(srvy.host).to eq(dns.host)
71
+ expect(srvy.port).to eq(dns.port)
72
+ expect(srvy.weight).to eq(dns.weight)
73
+ expect(srvy.priority).to eq(dns.priority)
74
+ end
75
+ end
76
+ end
77
+
78
+
79
+ describe "#get_many" do
80
+ subject{ Srvy::Result.new(Time.now, hosts) }
81
+
82
+ When(:result){ subject.get_many }
83
+
84
+ context "hosts with equal priorities" do
85
+ Given(:hosts) { hosts_with_equal_weight.sort_by(&:host) }
86
+ Then{ expect(result.sort_by(&:host)).to eq(hosts) }
87
+ end
88
+
89
+ context "hosts with unequal priorities" do
90
+ Given(:hosts) { hosts_with_unequal_priorities }
91
+ # should only have the best (i.e. lowest) priority
92
+ Then{ expect(result).to eq([hosts[0]]) }
93
+ end
94
+
95
+ context "no hosts" do
96
+ Given(:hosts) { no_hosts }
97
+ Then{ expect(result).to eq([]) }
98
+ end
99
+ end
100
+
101
+ describe "#get_all" do
102
+ subject{ Srvy::Result.new(Time.now, hosts) }
103
+
104
+ When(:result){ subject.get_all }
105
+
106
+ context "with some hosts" do
107
+ Given(:hosts) { hosts_with_unequal_priorities }
108
+ Then{ expect(result).to eq(hosts) }
109
+ end
110
+
111
+ context "with no hosts" do
112
+ Given(:hosts) { [ ] }
113
+ Then{ expect(result).to eq(hosts) }
114
+ end
115
+ end
116
+
117
+ describe "#get_single" do
118
+ subject{ Srvy::Result.new(Time.now, hosts) }
119
+
120
+ When(:result_percentages) do
121
+ calls = call_count.times.map{ subject.get_single }
122
+
123
+ result_counts = calls.each_with_object({}) do |call_result, acc|
124
+ acc[call_result] ||= 0
125
+ acc[call_result] += 1
126
+ end
127
+ all_hosts_and_results = result_counts.keys | hosts
128
+ all_hosts_and_results.each_with_object({}) do |host, acc|
129
+ count = result_counts[host] || 0
130
+ acc[host] = count / call_count.to_f
131
+ end
132
+ end
133
+
134
+ context "When the result has hosts with equal weights" do
135
+ Given(:hosts){ hosts_with_equal_weight }
136
+
137
+ Then do
138
+ result_percentages.values.each do |percentage|
139
+ expect(percentage).to be_within(allowance).of(0.5)
140
+ end
141
+ end
142
+ end
143
+
144
+ context "hosts with unequal priority" do
145
+ Given(:hosts){ hosts_with_unequal_priorities }
146
+
147
+ Then do
148
+ expect(result_percentages[hosts[0]]).to be_within(allowance).of(1.0)
149
+ expect(result_percentages[hosts[1]]).to be_within(allowance).of(0.0)
150
+ end
151
+ end
152
+
153
+
154
+ context "hosts with unequal weights" do
155
+ Given(:hosts){ hosts_with_unequal_weight }
156
+
157
+ Then do
158
+ expect(result_percentages[hosts[0]]).to be_within(allowance).of(0.75)
159
+ expect(result_percentages[hosts[1]]).to be_within(allowance).of(0.25)
160
+ end
161
+ end
162
+
163
+ context "no hosts" do
164
+ Given(:hosts){ no_hosts }
165
+
166
+ Then do
167
+ expect(result_percentages[nil]).to be_within(allowance).of(1.0)
168
+ end
169
+ end
170
+ end
171
+ end
172
+
173
+
174
+
175
+
176
+
177
+
178
+
@@ -0,0 +1,33 @@
1
+ # coding: utf-8
2
+ lib = File.expand_path('../lib', __FILE__)
3
+ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
4
+ require 'srvy/version'
5
+
6
+ Gem::Specification.new do |spec|
7
+ spec.name = "srvy"
8
+ spec.version = Srvy::VERSION
9
+ spec.authors = ["Scott Fleckenstein"]
10
+ spec.email = ["nullstyle@gmail.com"]
11
+ spec.description = %q{SRV-based service discovery}
12
+ spec.summary = %q{SRV-based service discovery}
13
+ spec.homepage = "https://github.com/nullstyle/srvy"
14
+ spec.license = "MIT"
15
+
16
+ spec.files = `git ls-files`.split($/)
17
+ spec.executables = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
18
+ spec.test_files = spec.files.grep(%r{^(test|spec|features)/})
19
+ spec.require_paths = ["lib"]
20
+
21
+ spec.add_dependency "net-dns", "= 0.8.0" # pin this version of net-dns because our monkey patch only applies to it.
22
+ spec.add_dependency "lru_redux", "~> 0.8.1"
23
+
24
+ spec.add_development_dependency "bundler", "~> 1.3"
25
+ spec.add_development_dependency "rake"
26
+ spec.add_development_dependency "pry"
27
+
28
+ spec.add_development_dependency "rspec", ">= 2.14.1"
29
+ spec.add_development_dependency "rspec-given", ">= 3.5.0"
30
+ spec.add_development_dependency "coveralls", "~> 0.7.0"
31
+ spec.add_development_dependency "timecop", "~> 0.7.1"
32
+ spec.add_development_dependency "activesupport", ">= 4.0.0"
33
+ end
metadata ADDED
@@ -0,0 +1,218 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: srvy
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.0.1
5
+ platform: ruby
6
+ authors:
7
+ - Scott Fleckenstein
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2014-02-06 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: net-dns
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - '='
18
+ - !ruby/object:Gem::Version
19
+ version: 0.8.0
20
+ type: :runtime
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - '='
25
+ - !ruby/object:Gem::Version
26
+ version: 0.8.0
27
+ - !ruby/object:Gem::Dependency
28
+ name: lru_redux
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - ~>
32
+ - !ruby/object:Gem::Version
33
+ version: 0.8.1
34
+ type: :runtime
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - ~>
39
+ - !ruby/object:Gem::Version
40
+ version: 0.8.1
41
+ - !ruby/object:Gem::Dependency
42
+ name: bundler
43
+ requirement: !ruby/object:Gem::Requirement
44
+ requirements:
45
+ - - ~>
46
+ - !ruby/object:Gem::Version
47
+ version: '1.3'
48
+ type: :development
49
+ prerelease: false
50
+ version_requirements: !ruby/object:Gem::Requirement
51
+ requirements:
52
+ - - ~>
53
+ - !ruby/object:Gem::Version
54
+ version: '1.3'
55
+ - !ruby/object:Gem::Dependency
56
+ name: rake
57
+ requirement: !ruby/object:Gem::Requirement
58
+ requirements:
59
+ - - '>='
60
+ - !ruby/object:Gem::Version
61
+ version: '0'
62
+ type: :development
63
+ prerelease: false
64
+ version_requirements: !ruby/object:Gem::Requirement
65
+ requirements:
66
+ - - '>='
67
+ - !ruby/object:Gem::Version
68
+ version: '0'
69
+ - !ruby/object:Gem::Dependency
70
+ name: pry
71
+ requirement: !ruby/object:Gem::Requirement
72
+ requirements:
73
+ - - '>='
74
+ - !ruby/object:Gem::Version
75
+ version: '0'
76
+ type: :development
77
+ prerelease: false
78
+ version_requirements: !ruby/object:Gem::Requirement
79
+ requirements:
80
+ - - '>='
81
+ - !ruby/object:Gem::Version
82
+ version: '0'
83
+ - !ruby/object:Gem::Dependency
84
+ name: rspec
85
+ requirement: !ruby/object:Gem::Requirement
86
+ requirements:
87
+ - - '>='
88
+ - !ruby/object:Gem::Version
89
+ version: 2.14.1
90
+ type: :development
91
+ prerelease: false
92
+ version_requirements: !ruby/object:Gem::Requirement
93
+ requirements:
94
+ - - '>='
95
+ - !ruby/object:Gem::Version
96
+ version: 2.14.1
97
+ - !ruby/object:Gem::Dependency
98
+ name: rspec-given
99
+ requirement: !ruby/object:Gem::Requirement
100
+ requirements:
101
+ - - '>='
102
+ - !ruby/object:Gem::Version
103
+ version: 3.5.0
104
+ type: :development
105
+ prerelease: false
106
+ version_requirements: !ruby/object:Gem::Requirement
107
+ requirements:
108
+ - - '>='
109
+ - !ruby/object:Gem::Version
110
+ version: 3.5.0
111
+ - !ruby/object:Gem::Dependency
112
+ name: coveralls
113
+ requirement: !ruby/object:Gem::Requirement
114
+ requirements:
115
+ - - ~>
116
+ - !ruby/object:Gem::Version
117
+ version: 0.7.0
118
+ type: :development
119
+ prerelease: false
120
+ version_requirements: !ruby/object:Gem::Requirement
121
+ requirements:
122
+ - - ~>
123
+ - !ruby/object:Gem::Version
124
+ version: 0.7.0
125
+ - !ruby/object:Gem::Dependency
126
+ name: timecop
127
+ requirement: !ruby/object:Gem::Requirement
128
+ requirements:
129
+ - - ~>
130
+ - !ruby/object:Gem::Version
131
+ version: 0.7.1
132
+ type: :development
133
+ prerelease: false
134
+ version_requirements: !ruby/object:Gem::Requirement
135
+ requirements:
136
+ - - ~>
137
+ - !ruby/object:Gem::Version
138
+ version: 0.7.1
139
+ - !ruby/object:Gem::Dependency
140
+ name: activesupport
141
+ requirement: !ruby/object:Gem::Requirement
142
+ requirements:
143
+ - - '>='
144
+ - !ruby/object:Gem::Version
145
+ version: 4.0.0
146
+ type: :development
147
+ prerelease: false
148
+ version_requirements: !ruby/object:Gem::Requirement
149
+ requirements:
150
+ - - '>='
151
+ - !ruby/object:Gem::Version
152
+ version: 4.0.0
153
+ description: SRV-based service discovery
154
+ email:
155
+ - nullstyle@gmail.com
156
+ executables: []
157
+ extensions: []
158
+ extra_rdoc_files: []
159
+ files:
160
+ - .coveralls.yml
161
+ - .gitignore
162
+ - .travis.yml
163
+ - Gemfile
164
+ - LICENSE.txt
165
+ - README.md
166
+ - Rakefile
167
+ - lib/lru_redux/has_key_monkey_patch.rb
168
+ - lib/net/dns/ext/resolver_nameserver_monkey_patch.rb
169
+ - lib/srvy.rb
170
+ - lib/srvy/formatters.rb
171
+ - lib/srvy/formatters/base.rb
172
+ - lib/srvy/formatters/dalli.rb
173
+ - lib/srvy/formatters/host_port.rb
174
+ - lib/srvy/host.rb
175
+ - lib/srvy/resolver.rb
176
+ - lib/srvy/result.rb
177
+ - lib/srvy/version.rb
178
+ - spec/spec_helper.rb
179
+ - spec/srvy/formatters/base_spec.rb
180
+ - spec/srvy/formatters/dalli_spec.rb
181
+ - spec/srvy/formatters/host_port_spec.rb
182
+ - spec/srvy/formatters_spec.rb
183
+ - spec/srvy/resolver_spec.rb
184
+ - spec/srvy/result_spec.rb
185
+ - srvy.gemspec
186
+ homepage: https://github.com/nullstyle/srvy
187
+ licenses:
188
+ - MIT
189
+ metadata: {}
190
+ post_install_message:
191
+ rdoc_options: []
192
+ require_paths:
193
+ - lib
194
+ required_ruby_version: !ruby/object:Gem::Requirement
195
+ requirements:
196
+ - - '>='
197
+ - !ruby/object:Gem::Version
198
+ version: '0'
199
+ required_rubygems_version: !ruby/object:Gem::Requirement
200
+ requirements:
201
+ - - '>='
202
+ - !ruby/object:Gem::Version
203
+ version: '0'
204
+ requirements: []
205
+ rubyforge_project:
206
+ rubygems_version: 2.0.3
207
+ signing_key:
208
+ specification_version: 4
209
+ summary: SRV-based service discovery
210
+ test_files:
211
+ - spec/spec_helper.rb
212
+ - spec/srvy/formatters/base_spec.rb
213
+ - spec/srvy/formatters/dalli_spec.rb
214
+ - spec/srvy/formatters/host_port_spec.rb
215
+ - spec/srvy/formatters_spec.rb
216
+ - spec/srvy/resolver_spec.rb
217
+ - spec/srvy/result_spec.rb
218
+ has_rdoc: