From 0260e315364466bf1ba5e5f84a39f54b6217e126 Mon Sep 17 00:00:00 2001
From: "John E. Vincent" <lusis.org+github.com@gmail.com>
Date: Mon, 2 Jul 2012 00:21:27 -0400
Subject: [PATCH] experimental datadog event output

---
 lib/logstash/outputs/datadog.rb | 92 +++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 lib/logstash/outputs/datadog.rb

diff --git a/lib/logstash/outputs/datadog.rb b/lib/logstash/outputs/datadog.rb
new file mode 100644
index 000000000..015ed031b
--- /dev/null
+++ b/lib/logstash/outputs/datadog.rb
@@ -0,0 +1,92 @@
+require "logstash/outputs/base"
+require "logstash/namespace"
+
+class LogStash::Outputs::Datadog < LogStash::Outputs::Base
+  # This output lets you send events (for now. soon metrics) to
+  # DataDogHQ based on Logstash events
+  #
+  # Note that since Logstash maintains no state
+  # these will be one-shot events
+  #
+
+  config_name "datadog"
+  plugin_status "experimental"
+
+  # Your DatadogHQ API key
+  config :api_key, :validate => :string, :required => true
+
+  # Title
+  config :title, :validate => :string, :default => "Logstash event for %{@source_host}"
+
+  # Text
+  config :text, :validate => :string, :default => "%{@message}"
+
+  # Date Happened
+  config :date_happened, :validate => :string
+
+  # Source type name
+  config :source_type_name, :validate => ["nagios", "hudson", "jenkins", "user", "my apps", "feed", "chef", "puppet", "git", "bitbucket"], :default => "my apps"
+ 
+  # Alert type
+  config :alert_type, :validate => ["info", "error", "warning", "success"]
+
+  # Priority
+  config :priority, :validate => ["normal", "low"]
+
+  # Tags
+  # Set any custom tags for this event
+  # Default are the Logstash tags if any
+  config :dd_tags, :validate => :array
+
+  public
+  def register
+    require "net/https"
+    require "uri"
+    @url = "https://app.datadoghq.com/api/v1/events"
+    @uri = URI.parse(@url)
+    @client = Net::HTTP.new(@uri.host, @uri.port)
+    @client.use_ssl = true
+    @client.verify_mode = OpenSSL::SSL::VERIFY_NONE
+    @logger.debug("Client", :client => @client.inspect)
+  end # def register
+
+  public
+  def receive(event)
+    return unless output?(event)
+
+
+    dd_event = Hash.new
+    dd_event['title'] = event.sprintf(@title)
+    dd_event['text'] = event.sprintf(@text)
+    dd_event['source_type_name'] = @source_type_name
+    dd_event['alert_type'] = @alert_type if @alert_type
+    dd_event['priority'] = @priority if @priority
+
+    if @date_happened
+      dd_event['date_happened'] = event.sprintf(@date_happened)
+    else
+      dd_event['date_happened'] = event.unix_timestamp.to_i
+    end
+
+    if @dd_tags
+      tagz = @dd_tags.collect {|x| event.sprintf(x) }
+    else
+      tagz = event.tags
+    end
+    dd_event['tags'] = tagz if tagz
+
+    @logger.debug("DataDog event", :dd_event => dd_event)
+
+    request = Net::HTTP::Post.new("#{@uri.path}?api_key=#{@api_key}")
+    
+    begin
+      request.body = dd_event.to_json
+      request.add_field("Content-Type", 'application/json')
+      response = @client.request(request)
+      @logger.info("DD convo", :request => request.inspect, :response => response.inspect)
+      raise unless response.code == '200'
+    rescue Exception => e
+      @logger.warn("Unhandled exception", :request => request.inspect, :response => response.inspect, :exception => e.inspect)
+    end
+  end # def receive
+end