diplomatic_bag 2.2.1

data/lib/diplomat.rb

@@ -0,0 +1,62 @@
+require 'json'
+require 'base64'
+require 'faraday'
+
+# Top level namespace ensures all required libraries are included and initializes the gem configration.
+module Diplomat
+  class << self
+    attr_accessor :root_path
+    attr_accessor :lib_path
+    attr_accessor :configuration
+
+    # Internal: Requires internal Faraday libraries.
+    # @param *libs One or more relative String names to Faraday classes.
+    # @return [nil]
+    def require_libs(*libs)
+      libs.each do |lib|
+        require "#{lib_path}/#{lib}"
+      end
+    end
+
+    alias require_lib require_libs
+  end
+
+  raise 'Diplomat only supports ruby >= 2.0.0' unless RUBY_VERSION.to_f >= 2.0
+
+  self.root_path = File.expand_path __dir__
+  self.lib_path = File.expand_path 'diplomat', __dir__
+
+  require_libs 'configuration', 'rest_client', 'kv', 'datacenter', 'service',
+               'members', 'node', 'nodes', 'check', 'health', 'session', 'lock',
+               'error', 'event', 'acl', 'maintenance', 'query', 'agent', 'status'
+  self.configuration ||= Diplomat::Configuration.new
+
+  class << self
+    # Build optional configuration by yielding a block to configure
+    # @yield [Diplomat::Configuration]
+    def configure
+      self.configuration ||= Diplomat::Configuration.new
+      yield(configuration)
+    end
+
+    private
+
+    # Send all other unknown commands to Diplomat::Kv
+    # @deprecated Please use Diplomat::Kv instead.
+    # @param name [Symbol] Method to send to Kv
+    # @param *args List of arguments to send to Kv
+    # @param &block block to send to Kv
+    # @return [Object]
+    def method_missing(name, *args, &block)
+      Diplomat::Kv.new.send(name, *args, &block) || super
+    end
+
+    # Make `respond_to_missing?` fall back to super
+    #
+    # @param meth_id [Symbol] the tested method
+    # @oaram with_private if private methods should be tested too
+    def respond_to_missing?(meth_id, with_private = false)
+      access_method?(meth_id) || super
+    end
+  end
+end

data/lib/diplomatic_bag/datacenters.rb

@@ -0,0 +1,11 @@
+# Usefull usage of Diplomat lib: Datacenter functions
+module DiplomaticBag
+  def self.get_datacenters_list(dc, options = {})
+    dcs = []
+    datacenters = Diplomat::Datacenter.get(nil, options)
+    dc.each do |c|
+      dcs.concat(datacenters.select { |d| d[/#{c}/] })
+    end
+    dcs.uniq
+  end
+end

data/lib/diplomatic_bag/info.rb

@@ -0,0 +1,32 @@
+# Usefull usage of Diplomat lib: Info functions
+module DiplomaticBag
+  # rubocop:disable AbcSize
+  def self.consul_info(options = {})
+    consul_self = Diplomat::Agent.self(options)
+    puts 'Server: ' + consul_self['Config']['NodeName']
+    puts 'Datacenter: ' + consul_self['Config']['Datacenter']
+    puts 'Consul Version: ' + consul_self['Config']['Version']
+    if consul_self['Stats']['consul']['leader_addr']
+      puts 'Leader Address: ' + consul_self['Stats']['consul']['leader_addr']
+      puts 'applied_index: ' + consul_self['Stats']['raft']['applied_index']
+      puts 'commit_index: ' + consul_self['Stats']['raft']['commit_index']
+    end
+    members = Diplomat::Members.get(options)
+    servers = members.select { |member| member['Tags']['role'] == 'consul' }.sort_by { |n| n['Name'] }
+    nodes = members.select { |member| member['Tags']['role'] == 'node' }
+    leader = Diplomat::Status.leader(options).split(':')[0]
+    puts 'Servers Count: ' + servers.count.to_s
+    puts 'Nodes Count: ' + nodes.count.to_s
+    puts 'Servers:'
+    servers.map do |s|
+      if s['Tags']['role'] == 'consul'
+        if s['Addr'] == leader
+          puts '  ' + s['Name'] + ' ' + s['Addr'] + ' *Leader*'
+        else
+          puts '  ' + s['Name'] + ' ' + s['Addr']
+        end
+      end
+    end
+  end
+  # rubocop:enable AbcSize
+end

data/lib/diplomatic_bag/nodes.rb

@@ -0,0 +1,28 @@
+# Usefull usage of Diplomat lib: Nodes functions
+module DiplomaticBag
+  def self.get_duplicate_node_id(options = {})
+    status = {
+      1 => 'Alive',
+      2 => '?',
+      3 => 'Left',
+      4 => 'Failed'
+    }
+    result = []
+    members = Diplomat::Members.get(options)
+    grouped = members.group_by do |row|
+      [row['Tags']['id']]
+    end
+    filtered = grouped.values.select { |a| a.size > 1 }
+    filtered.each do |dup|
+      instance = {}
+      instance[:node_id] = dup[0]['Tags']['id']
+      nodes = []
+      dup.each do |inst|
+        nodes << { name: inst['Name'], status: status[inst['Status']] }
+      end
+      instance[:nodes] = nodes
+      result << instance
+    end
+    result
+  end
+end

data/lib/diplomatic_bag/service.rb

@@ -0,0 +1,20 @@
+# Usefull usage of Diplomat lib: Service functions
+module DiplomaticBag
+  def self.get_service_info(service, options = {})
+    result = {}
+    health = Diplomat::Health.service(service, options)
+    result[service] = {}
+    health.each do |h|
+      result[service][h['Node']['Node']] = {
+        'Address': h['Node']['Address'],
+        'Port': h['Service']['Port']
+      }
+      checks = {}
+      h['Checks'].each do |c|
+        checks[c['Name']] = { 'status': c['Status'], 'output': c['Output'] }
+      end
+      result[service][h['Node']['Node']]['Checks'] = checks
+    end
+    result
+  end
+end

data/lib/diplomatic_bag/services.rb

@@ -0,0 +1,38 @@
+# Usefull usage of Diplomat lib: Services functions
+module DiplomaticBag
+  def self.compute_service_name(s)
+    return '/node::health/' if s.nil? || s == ''
+
+    s
+  end
+
+  def self.get_all_services_status(options = {})
+    result = {}
+    services = Diplomat::Health.state('any', options)
+    grouped_by_service = services.group_by { |h| compute_service_name(h['ServiceName']) }.values
+    grouped_by_service.each do |s|
+      grouped_by_status = s.group_by { |h| h['Status'] }.values
+      status = {}
+      grouped_by_status.each do |state|
+        status[state[0]['Status']] = state.count
+      end
+      service_name = compute_service_name(s[0]['ServiceName'])
+      result[service_name] = status
+    end
+    result
+  end
+
+  def self.get_services_list(service, options = {})
+    services = Diplomat::Service.get_all(options)
+    services.to_h.keys.grep(/#{service}/)
+  end
+
+  def self.get_services_info(service, options = {})
+    result = []
+    services = get_services_list(service, options)
+    services.each do |s|
+      result << get_service_info(s.to_s, options)
+    end
+    result
+  end
+end

data/lib/diplomatic_bag.rb

@@ -0,0 +1,7 @@
+require 'diplomat'
+
+# Usefull usage of Diplomat lib.
+module DiplomaticBag
+  # Load all module files
+  Dir[File.join(__dir__, 'diplomatic_bag', '*.rb')].each { |file| require file }
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: diplomatic_bag
+version: !ruby/object:Gem::Version
+  version: 2.2.1
+platform: ruby
+authors:
+- Nicolas Benoit
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-04-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: diplomat
+  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: Toolbox for Consul
+email:
+- n.benoit@criteo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- features/configuration.feature
+- features/step_definitions/setup_diplomat.rb
+- features/step_definitions/test_key_value.rb
+- lib/diplomat.rb
+- lib/diplomat/acl.rb
+- lib/diplomat/agent.rb
+- lib/diplomat/check.rb
+- lib/diplomat/configuration.rb
+- lib/diplomat/datacenter.rb
+- lib/diplomat/error.rb
+- lib/diplomat/event.rb
+- lib/diplomat/health.rb
+- lib/diplomat/kv.rb
+- lib/diplomat/lock.rb
+- lib/diplomat/maintenance.rb
+- lib/diplomat/members.rb
+- lib/diplomat/node.rb
+- lib/diplomat/nodes.rb
+- lib/diplomat/query.rb
+- lib/diplomat/rest_client.rb
+- lib/diplomat/service.rb
+- lib/diplomat/session.rb
+- lib/diplomat/status.rb
+- lib/diplomat/version.rb
+- lib/diplomatic_bag.rb
+- lib/diplomatic_bag/datacenters.rb
+- lib/diplomatic_bag/info.rb
+- lib/diplomatic_bag/nodes.rb
+- lib/diplomatic_bag/service.rb
+- lib/diplomatic_bag/services.rb
+homepage: https://github.com/WeAreFarmGeek/diplomat
+licenses:
+- BSD-3-Clause
+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.0.3
+signing_key: 
+specification_version: 4
+summary: Toolbox for Consul
+test_files: []