dns-loadbalancer 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 00361dd09de0683a2c765fda0238e36a81cc019204fc7fd0950ae1c8fd1e548a
+  data.tar.gz: 53c2244c604b15fa6bb5d491a87418eaf119c8efcd6aa037febf44357df5ff9e
+SHA512:
+  metadata.gz: 6e6cdaf34cf4ef16ea43c85785c800ac4545fceec1088f253c02ee7500575e651deb5c8839a2ca6d479698253773399ac1299a7854e37aa013d9ae19e7f5a224
+  data.tar.gz: c999c74ff7f5975e7e12ed5f33976ba84782950593bbd60e5d80083874d8bff01f33f969738f510237dbc289fa1a001f92667e5b45cf20deb816d61f9e921ce0

data/lib/dns-loadbalancer.rb

@@ -0,0 +1,65 @@
+require 'net/ping/tcp'
+require 'resolv'
+require 'ipaddress'
+
+class DNSLoadBalancer
+  attr_reader :hosts, :port
+
+  def initialize(hosts, port, verbose: true)
+    @hosts = hosts.split(',')
+    @port = port
+    @verbose = verbose
+  end
+
+  def run
+    start = Time.now
+    ip_addresses = hosts.flat_map do |h|
+      result = resolve(h)
+      log "Resolved #{h} to IP addresses: #{result}"
+      result
+    end
+    result = find_closest(ip_addresses)
+    log "Found fastest IP #{result} in #{(Time.now - start).round(2)}s"
+    result
+  end
+
+  private
+
+  def resolve(host)
+    return host.to_s if IPAddress.valid?(host.to_s)
+
+    dns_obj = Resolv::DNS.new(nameserver: ['1.1.1.1', '8.8.8.8', '8.8.4.4'])
+    resources = dns_obj.getresources(host, Resolv::DNS::Resource::IN::A)
+    resources.map(&:address).map(&:to_s)
+  end
+
+  def find_closest(ip_addresses)
+    pings = ip_addresses.map do |address|
+      duration = ping(address.to_s, port)
+      log "Pinging #{address.ljust(16, ' ')} #{duration || 'not reachable'}"
+      [address, duration]
+    end.to_h
+
+    fastest = pings.select { |_, v| v }.min_by(&:last)
+    fastest&.first || raise('no server accessible')
+  end
+
+  def ping(ip_address, port)
+    durations = 3.times.map do
+      p = Net::Ping::TCP.new(ip_address, port, 0.5)
+      p.ping?
+      p.duration
+    end
+
+    durations.compact!
+    return nil if durations.empty?
+
+    durations.sum / durations.length
+  end
+
+  def log(msg)
+    return unless @verbose
+
+    puts "[DNSLoadBalancer] #{msg}"
+  end
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification
+name: dns-loadbalancer
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Ruben Stranders
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-09-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ipaddress
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+- !ruby/object:Gem::Dependency
+  name: net-ping
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: resolv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+description:
+email: ruben@fireservicerota.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/dns-loadbalancer.rb
+homepage: https://github.com/FireServiceRota/dns-loadbalancer
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.7
+signing_key:
+specification_version: 4
+summary: Gem for resolving a DNS host to the closest IP address
+test_files: []