Merge pull request #215 from DataDog/jaime/kong-integration

Add integration for Kong

Author: truthbk

manifests/integrations/kong.pp

@@ -0,0 +1,46 @@
+# Class: datadog_agent::integrations::kong
+#
+# This class will install the necessary configuration for the Kong integration
+#
+# Note: if you're Cassandra data-store is large in size the `/status` page may
+# take a long time to return.
+# <https://github.com/Mashape/kong/issues/1323>
+#
+# Parameters:
+#   $instances:
+#       Array of hashes for all Kong instances and associated tags. See example
+#
+# Sample Usage:
+#
+#   class { 'datadog_agent::integrations::kong':
+#     instances => [
+#         {
+#             'status_url'  => http://localhost:8001/status/',
+#         },
+#         {
+#             'status_url'  => 'http://localhost:8001/status/',
+#             'tags' => ['instance:foo'],
+#         },
+#     ],
+#   }
+#
+class datadog_agent::integrations::kong (
+  $instances = [
+    {
+      'status_url' => 'http://localhost:8001/status/',
+      'tags' => []
+    }
+  ]
+) inherits datadog_agent::params {
+  include datadog_agent
+
+  file { "${datadog_agent::params::conf_dir}/kong.yaml":
+    ensure  => file,
+    owner   => $datadog_agent::params::dd_user,
+    group   => $datadog_agent::params::dd_group,
+    mode    => '0644',
+    content => template('datadog_agent/agent-conf.d/kong.yaml.erb'),
+    require => Package[$datadog_agent::params::package_name],
+    notify  => Service[$datadog_agent::params::service_name]
+  }
+}

spec/classes/datadog_agent_integrations_kong_spec.rb

@@ -0,0 +1,40 @@
+require 'spec_helper'
+
+describe 'datadog_agent::integrations::kong' do
+  let(:facts) {{
+    operatingsystem: 'Ubuntu',
+  }}
+  let(:conf_dir) { '/etc/dd-agent/conf.d' }
+  let(:dd_user) { 'dd-agent' }
+  let(:dd_group) { 'root' }
+  let(:dd_package) { 'datadog-agent' }
+  let(:dd_service) { 'datadog-agent' }
+  let(:conf_file) { "#{conf_dir}/kong.yaml" }
+
+  it { should compile.with_all_deps }
+  it { should contain_file(conf_file).with(
+    owner: dd_user,
+    group: dd_group,
+    mode: '0644',
+  )}
+  it { should contain_file(conf_file).that_requires("Package[#{dd_package}]") }
+  it { should contain_file(conf_file).that_notifies("Service[#{dd_service}]") }
+
+  context 'with default parameters' do
+    it { should contain_file(conf_file).with_content(%r{kong_status_url: http://localhost:8001/status/}) }
+  end
+
+  context 'with params set' do
+    let(:params) {{
+      instances: [
+        {
+          'status_url' => 'http://foo.bar:8080/status/',
+          'tags' => ['baz']
+        }
+      ]
+    }}
+
+    it { should contain_file(conf_file).with_content(%r{tags:\n.*- baz}) }
+    it { should contain_file(conf_file).with_content(%r{kong_status_url: http://foo.bar:8080/status/}) }
+  end
+end

templates/agent-conf.d/kong.yaml.erb

@@ -0,0 +1,16 @@
+#
+# MANAGED BY PUPPET
+#
+
+init_config:
+
+instances:
+<% @instances.each do |instance| -%>
+  - kong_status_url: <%= instance["status_url"] %>
+  <%- if instance.key?("tags") and not instance["tags"].empty? -%>
+    tags:
+    <%- instance["tags"].each do |tag| -%>
+      - <%= tag %>
+    <%- end -%>
+  <%- end -%>
+<% end -%>