le 2.0 → 2.1

data/LE.gemspec

@@ -3,7 +3,7 @@ require File.expand_path(File.join(dir, 'lib', 'le'))
 
 Gem::Specification.new do |s|
   s.name	= "le"
-  s.version	= "2.0"
+  s.version	= "2.1"
   s.date	= Time.now
   s.summary	= "Logentries plugin"
   s.description	=<<EOD

data/lib/le.rb

@@ -1,12 +1,3 @@
-#!/usr/bin/env ruby
-# coding: utf-8
-
-#
-# Logentries Ruby monitoring agent
-# Copyright 2010,2011 Logentries, Jlizard
-# Mark Lacomber <marklacomber@gmail.com>
-#
-
 require File.join(File.dirname(__FILE__), 'le', 'host')
 
 require 'logger'
@@ -20,19 +11,17 @@ module Le
    host = Le::Host.new(token, local)      
    logger = Logger.new(host)
    
-   logger.formatter = host.formatter
+   if host.respond_to?(:formatter)
+	logger.formatter = host.formatter
+   end
 
    logger  
  end
 
  def self.checkParams(token)
-	if token == nil
-		puts "\nLE: Incorrect token parameter for Logentries Plugin!\n"
-	end
-
-	# Check if the key is valid UUID format
-	if (token =~ /\A(urn:uuid:)?[\da-f]{8}-([\da-f]{4}-){3}[\da-f]{12}\z/i) == nil
-		puts "\nLE: It appears the LOGENTRIES_TOKEN you entered is invalid!\n"
-	end
+    # Check if the key is valid UUID format
+    if (token =~ /\A(urn:uuid:)?[\da-f]{8}-([\da-f]{4}-){3}[\da-f]{12}\z/i) == nil
+       puts "\nLE: It appears the LOGENTRIES_TOKEN you entered is invalid!\n"
+    end
  end
 end

data/lib/le/host.rb

@@ -1,48 +1,35 @@
-#!/usr/bin/env ruby
-# coding: utf-8
-
-#
-# Logentries Ruby monitoring agent
-# Copyright 2010,2011 Logentries, Jlizard
-# Mark Lacomber <marklacomber@gmail.com>
-#
-
 module Le
   module Host
-    
-    # Creates a new Logentries host, based on a user-key and location of destination file on logentries,
-    # both must be provided correctly for a connection to be made.
 
     def self.new(token, local)
 
-      Le::Host::HTTPS.new(token, local)
+      Le::Host::HTTP.new(token, local)
       
     end
   
-    module HelperMethods
- 
+    module InstanceMethods
       def formatter
 	proc do |severity, datetime, progname, msg|
-	  message = "#{datetime} "
-          message << format_message(msg, severity)
-        end
-      end
+	  	message = "#{datetime} "
+      	message << format_message(msg, severity)
+    end
+	end
   
-      def format_message(msg_in, severity)
-	msg_in = msg_in.lstrip
-	msg_out = ""
-	msg_out << "severity=#{severity}, "
+      def format_message(message_in, severity)
+		message_in = message_in.lstrip
 
-	case msg_in
-	when String
-		msg_out << msg_in
-	else
-		msg_out << msg_in.inspect
-        end
-	msg_out
+		message_out = ""
+		message_out = "severity=#{severity}, "
+		case message_in
+		when String
+		  message_out << message_in
+		else
+		  message_out << message_in.inspect
+		end
+		message_out
       end
     end
   end
 end
 
-require File.join(File.dirname(__FILE__), 'host', 'https')
+require File.join(File.dirname(__FILE__), 'host', 'http')

data/lib/le/host/http.rb

@@ -0,0 +1,103 @@
+require 'socket'
+require 'openssl'
+require 'thread'
+require 'uri'
+
+module Le
+  module Host
+    class HTTP
+      include Le::Host::InstanceMethods
+      attr_accessor :token, :queue, :started, :thread, :conn, :local
+
+      def initialize(token, local)
+		@token = token
+		@local = local
+        @queue = Queue.new
+		@started = false
+      end
+
+      def write(message)
+		if @local then
+		  puts message
+		  return
+		end
+
+		@queue << "#{@token}#{message}\n"
+
+	    if not @started then
+			puts "LE: Starting asynchronous socket writer"
+			@thread = Thread.new{run()}
+			@started = true
+		end
+      end
+
+      def close
+		puts "LE: Closing asynchronous socket writer"
+		@thread.raise Interrupt
+      end
+
+	  def openConnection
+		puts "LE: Reopening connection to Logentries API server"
+		@conn = TCPSocket.new('api.logentries.com', 10000)
+
+		#@conn = OpenSSL::SSL::SSLSocket.new(@sock, OpenSSL::SSL::SSLContext.new())
+		#@conn.connect
+
+		puts "LE: Connection established"
+	  end
+
+	  def reopenConnection
+		closeConnection
+		root_delay = 0.1
+		while true
+			begin
+				openConnection
+				break
+			rescue OpenSSL::SSL::SSLError, TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEDOUT, EOFError => e
+				puts "LE: Unable to connect to Logentries"
+			end
+			root_delay *= 2
+			if root_delay >= 10 then
+				root_delay = 10
+			end
+			wait_for = (root_delay + rand(root_delay)).to_i
+			puts "LE: Waiting for " + wait_for.to_s + "ms"
+			sleep(wait_for)
+		end
+	  end
+
+	  def closeConnection
+		if @conn != nil
+			@conn.sysclose
+			@conn = nil
+		end
+		#if @sock != nil
+		#	@sock.close
+		#	@sock = nil
+		#end
+	  end
+
+	  def run
+		begin
+		reopenConnection
+
+			while true
+				data = @queue.pop
+				while true
+					begin
+						@conn.write(data)
+					rescue OpenSSL::SSL::SSLError, TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEOUT, EOFError => e
+						reopenConnection
+						next
+					end
+					break
+				end
+			end
+		rescue Interrupt 
+		  puts "LE: Asynchronous socket writer interrupted"
+		end
+		closeConnection
+	  end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: le
 version: !ruby/object:Gem::Version
-  version: '2.0'
+  version: '2.1'
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-09-28 00:00:00.000000000 Z
+date: 2012-12-15 00:00:00.000000000 Z
 dependencies: []
 description: ! '
 
@@ -22,8 +22,7 @@ files:
 - LE.gemspec
 - ./lib/le.rb
 - ./lib/le/host.rb
-- ./lib/le/host/https.rb
-- ./lib/le/host/https/tcp.rb
+- ./lib/le/host/http.rb
 homepage: 
 licenses: []
 post_install_message: 

data/lib/le/host/https.rb

@@ -1,42 +0,0 @@
-#!/usr/bin/env ruby
-# coding: utf-8
-
-#
-# Logentries Ruby monitoring agent
-# Copyright 2010,2011 Logentries, Jlizard
-# Mark Lacomber <marklacomber@gmail.com>
-#
-
-require File.join(File.dirname(__FILE__), 'https', 'tcp')
-
-module Le
-  module Host
-    class HTTPS
-      include Le::Host::HelperMethods	
-      
-      attr_reader :deliverer, :local_bool
-
-      def initialize(token, local)
-	@local_bool = local
-	if not local
-        	@deliverer = Le::Host::HTTPS::TCPSOCKET.new(token)
-        end   
-      end
-
-      def write(message)
-	
-	if @local_bool
-        	puts message
-        else
-        	# Deliver the message to logentries via TCP
-		@deliverer.deliver(message)
-	end
-      end
-
-      def close
-	nil
-      end
-    
-    end
-  end
-end

data/lib/le/host/https/tcp.rb

@@ -1,64 +0,0 @@
-#!/usr/bin/env ruby
-# coding: utf-8
-
-#
-# Logentries Ruby monitoring agent
-# Copyright 2010,2011 Logentries, Jlizard
-# Mark Lacomber <marklacomber@gmail.com>
-#
-
-require 'socket'
-require 'openssl'
-
-module Le
-  module Host
-    class HTTPS
-
-      class TCPSOCKET
-
-	attr_accessor :conn, :token
-	def initialize(token)
-
-          @token = token
-	  begin
-          	createSocket()
-	  rescue TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEDOUT, EOFError => e
-		$stderr.puts "WARNING: #{e.class} Could not create the connection to Logentries. #{e.message}"
-          end
-	end
-
-        def createSocket()
-
-          # Open the TCP connection to the Logentries Server
-          @conn = TCPSocket.new('api.logentries.com', 10000)
-
-        end
-	
-	def deliver(message)
-
-          if @conn == nil
-             begin
-                createSocket()
-             rescue TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEDOUT, EOFError => e
-                $stderr.puts "WARNING: #{e.class} Could not write log. No connection to Logentries #{e.message}"
-                return
-             end
-          end
-          # Sends the log to the Logentries Server
-          begin
-	    @conn.puts(@token + message)
-	  rescue TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ENOTCONN, Errno::ECONNRESET, Errno::ETIMEDOUT, EOFError => e
-	    $stderr.puts "WARNING: #{e.class} Could not send log to Logentries #{e.message}"
-              begin
-                createSocket()
-              rescue TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEDOUT, EOFError => e
-                $stderr.puts "WARNING: #{e.class} Could not create the connection to Logentries. #{e.message}"
-              end
-	  end
-        end
-	
-      end
-
-    end
-  end
-end