Config variable to enable the trace agent (#302)

Author: DDRBoxman

manifests/init.pp

@@ -151,7 +151,9 @@
 #   $pup_log_file
 #       Specifies the log file location for the pup system
 #       String. Default: empty
-#
+#   $apm_enabled
+#       Boolean to enable or disable the trace agent
+#       Boolean. Default: false
 #
 # Actions:
 #
@@ -238,6 +240,7 @@
   $package_name = $datadog_agent::params::package_name,
   $dd_user = $datadog_agent::params::dd_user,
   $dd_group = $datadog_agent::params::dd_group,
+  $apm_enabled = false,
 ) inherits datadog_agent::params {
 
   validate_string($dd_url)
@@ -298,6 +301,7 @@
   validate_string($sd_template_dir)
   validate_bool($sd_jmx_enable)
   validate_string($consul_token)
+  validate_bool($apm_enabled)
 
   if $hiera_tags {
     $local_tags = hiera_array('datadog_agent::tags')

spec/classes/datadog_agent_spec.rb

@@ -203,6 +203,12 @@
                     'content' => /^# sd_jmx_enable: no\n/,
                     )}
                 end
+
+                context 'for APM' do
+                    it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                    'content' => /^apm_enabled: false\n/,
+                    )}
+                end
             end
 
             context 'with user provided paramaters' do
@@ -500,6 +506,12 @@
                     'content' => /^syslog_port: 8080\n/,
                 )}
             end
+            context 'with apm_enabled set to true' do
+                let(:params) {{:apm_enabled  => true }}
+                it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                    'content' => /^apm_enabled: true\n/,
+                )}
+            end
             context 'with service_discovery enabled' do
                 let(:params) {
                     {:service_discovery_backend  => 'docker',

templates/datadog.conf.erb

@@ -401,6 +401,11 @@ sd_jmx_enable: <%= @sd_jmx_enable %>
 consul_token: <%= @consul_token %>
 <% end -%>
 
+# ========================================================================== #
+# Trace                                                                      #
+# ========================================================================== #
+
+apm_enabled: <%= @apm_enabled %>
 
 <% if not @extra_template.empty? -%>