discourse_sd 0.0.7

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3fc3cfdab012516746db93ef82db8e640eeab501
+  data.tar.gz: 3d66a320135732d196b38dd2d698909cd5f89466
+SHA512:
+  metadata.gz: 8f42790d9530178a2c1d6cc5ecf529b8b59fedc9640f9353dac03a611552050baec082d4ccf952841a58ea01667f878d621e74403955b43660aeb57de261d702
+  data.tar.gz: c4f2e4c4074c3430e8223033427886050ea91e9ff2ca9c505ede895b6bc82f3583d5f4a0a47606c212fd728e99ab046e0110f88295b914b17289ace0452656f2

data/bin/discourse_sd

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+
+require 'discourse_sd_cli'

data/lib/discourse_sd.rb

@@ -0,0 +1,86 @@
+require 'resolv'
+require 'dns-sd'
+
+require_relative 'discourse_sd/version'
+
+class DiscourseSD
+  attr_reader :zone
+
+  # Create an instance of DiscourseSD to lookup services in the zone
+  # determined from the environment.
+  #
+  # @return [DiscourseSD] the DiscourseSD instance
+  def self.create_from_env
+    if sd_dns = Resolv::DNS.new.getresources("powerdns._domain._tcp", Resolv::DNS::Resource::IN::SRV).first
+      parts = sd_dns.target.to_s.split('.')
+      new(parts[parts.rindex('sd')..-1].join('.'))
+    else
+      raise RuntimeError, "No PowerDNS servers were found!"
+    end
+  end
+
+  # Create an instance of DiscourseSD to lookup services in the given zone.
+  #
+  # @param zone [String] The zone in which all services should be found.
+  def initialize(zone)
+    @zone = zone
+  end
+
+  # Looks up all the services relevant to a Discourse app, and returns
+  # a hash with their IPv6 addresses and ports using the same key names
+  # that are used in the Discourse config.
+  #
+  # @param prefix [String] The name of the Discourse app cluster.
+  #
+  # @return [Hash] IP addresses and ports for services.
+  def service_instances(prefix)
+    config = {}
+    sd = DNSSD.new(@zone)
+
+    redises = redis_service_instances(sd, prefix)
+    pgbouncers = pgbouncer_service_instances(sd, prefix)
+
+    Resolv::DNS.open do |dns|
+      redis_master = redises.first
+      redis_slave = redises.size > 1 ? redises.last : nil
+
+      if redis_master
+        config[:redis_host] = aaaa_record(dns, redis_master.hostname).to_s
+        config[:redis_port] = redis_master.port
+      end
+      if redis_slave
+        config[:redis_slave_host] = aaaa_record(dns, redis_slave.hostname).to_s
+        config[:redis_slave_port] = redis_slave.port
+      end
+
+      db_master = pgbouncers.first
+      db_replica = pgbouncers.size > 1 ? pgbouncers.last : nil
+
+      if db_master
+        config[:db_host] = aaaa_record(dns, db_master.hostname).to_s
+        config[:db_port] = db_master.port
+      end
+      if db_replica
+        config[:db_replica_host] = aaaa_record(dns, db_replica.hostname).to_s
+        config[:db_replica_port] = db_replica.port
+      end
+    end
+
+    config
+  end
+
+  private
+
+  def redis_service_instances(sd, prefix)
+    sd.service_instance(prefix, 'redis', :TCP).targets
+  end
+
+  def pgbouncer_service_instances(sd, prefix)
+    sd.service_instance(prefix, 'pgbouncer', :TCP).targets
+  end
+
+  def aaaa_record(dns, name)
+    dns.getresources(name, Resolv::DNS::Resource::IN::AAAA).first&.address
+  end
+
+end

data/lib/discourse_sd/version.rb

@@ -0,0 +1,3 @@
+class DiscourseSD
+  VERSION = '0.0.7'
+end

data/lib/discourse_sd_cli.rb

@@ -0,0 +1,25 @@
+require 'discourse_sd'
+require 'json'
+
+def usage
+  puts "discourse_sd v#{DiscourseSD::VERSION}"
+  puts ''
+  puts 'usage:', ''
+  puts ' discourse_sd rails-config <name>'
+  puts '   Outputs a JSON hash of a rails configuration for the app with the given name'
+  puts ''
+end
+
+if ARGV.size == 0
+  usage
+else
+  sd = DiscourseSD.create_from_env
+
+  case ARGV[0]
+  when 'rails-config'
+    usage && exit unless ARGV.size == 2
+    puts sd.service_instances(ARGV[1]).to_json
+  else
+    usage
+  end
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: discourse_sd
+version: !ruby/object:Gem::Version
+  version: 0.0.7
+platform: ruby
+authors:
+- Neil Lalonde
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-03-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dns-sd
+  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: yard
+  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'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+description: For internal Discourse use
+email: neil.lalonde@discourse.org
+executables:
+- discourse_sd
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/discourse_sd
+- lib/discourse_sd.rb
+- lib/discourse_sd/version.rb
+- lib/discourse_sd_cli.rb
+homepage: 
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2.1
+signing_key: 
+specification_version: 4
+summary: Utilities for using DNS-SD in our hosting
+test_files: []