require 'socket' require 'openssl' require 'thread' require 'uri' module Le module Host class HTTP API_SERVER = 'api.logentries.com' API_PORT = 10000 include Le::Host::InstanceMethods attr_accessor :token, :queue, :started, :thread, :conn, :local, :debug def initialize(token, local, debug) if defined?(Rails) @logger_console = Logger.new("log/#{Rails.env}.log") else @logger_console = Logger.new(STDOUT) end @token = token @local = local @debug = debug @queue = Queue.new @started = false @thread = nil if @debug then self.init_debug end end def init_debug filePath = "logentriesGem.log" if File.exist?('log/') filePath = "log/logentriesGem.log" end @debug_logger = Logger.new(filePath) end def dbg(message) if @debug then @debug_logger.add(Logger::Severity::DEBUG,message) end end def write(message) if @local then @logger_console.add(Logger::Severity::UNKNOWN,message) end @queue << "#{@token}#{message}\n" if @started then check_async_thread else start_async_thread end end def start_async_thread @thread = Thread.new{run()} dbg "LE: Asynchronous socket writer started" @started = true end def check_async_thread if not @thread.alive? @thread = Thread.new{run()} dbg "LE: Asyncrhonous socket writer restarted" end end def close dbg "LE: Closing asynchronous socket writer" @started = false end def openConnection dbg "LE: Reopening connection to Logentries API server" @conn = TCPSocket.new(API_SERVER, API_PORT) dbg "LE: Connection established" end def reopenConnection closeConnection root_delay = 0.1 while true begin openConnection break rescue TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEDOUT, EOFError => e dbg "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 dbg "LE: Waiting for " + wait_for.to_s + "ms" sleep(wait_for) end end def closeConnection if @conn != nil @conn.sysclose @conn = nil end end def run reopenConnection while true data = @queue.pop while true begin @conn.write(data) rescue TimeoutError, Errno::EHOSTUNREACH, Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::ETIMEOUT, EOFError => e reopenConnection next end break end end dbg "LE: Closing Asyncrhonous socket writer" closeConnection end end end end