module-puppetdb/lib/puppet/util/puppetdb_validator.rb

require 'puppet/network/http_pool'

module Puppet
  module Util
    class PuppetdbValidator
      attr_reader :puppetdb_server
      attr_reader :puppetdb_port

      def initialize(puppetdb_server, puppetdb_port, use_ssl=true)
        @puppetdb_server = puppetdb_server
        @puppetdb_port   = puppetdb_port
        @use_ssl         = use_ssl
      end

      # Utility method; attempts to make an https connection to the puppetdb server.
      # This is abstracted out into a method so that it can be called multiple times
      # for retry attempts.
      #
      # @return true if the connection is successful, false otherwise.
      def attempt_connection
        # All that we care about is that we are able to connect successfully via
        # https, so here we're simpling hitting a somewhat arbitrary low-impact URL
        # on the puppetdb server.
        path = "/v2/metrics/mbean/java.lang:type=Memory"
        headers = {"Accept" => "application/json"}
        if @use_ssl
          conn = Puppet::Network::HttpPool.http_instance(@puppetdb_server, @puppetdb_port, @use_ssl)
        else
          # the Puppet httppool only supports disabling ssl in Puppet > 3.x
          # this code allows ssl to be disabled for the connection on both 2.7 and 3.x
          conn = Net::HTTP.new(@puppetdb_server, @puppetdb_port)
          conn.read_timeout = Puppet[:configtimeout]
          conn.open_timeout = Puppet[:configtimeout]
        end

        response = conn.get(path, headers)
        unless response.kind_of?(Net::HTTPSuccess)
          Puppet.notice "Unable to connect to puppetdb server (#{@puppetdb_server}:#{@puppetdb_port}): [#{response.code}] #{response.msg}"
          return false
        end
        return true
      rescue Exception => e
        Puppet.notice "Unable to connect to puppetdb server (#{@puppetdb_server}:#{@puppetdb_port}): #{e.message}"
        return false
      end
    end
  end
end