mirror of
https://github.com/elastic/logstash.git
synced 2025-04-24 22:57:16 -04:00
434a633013
hand-code. This method is as fast as the hand-coded method and with less
writing.
- Make message registration easier with a simple 'register' function:
class Foo < Message
register
end
We dynamically generate the message 'type' by the class name.
73 lines
1.6 KiB
Ruby
Executable file
73 lines
1.6 KiB
Ruby
Executable file
#!/usr/bin/env ruby
|
|
|
|
require 'rubygems'
|
|
require 'lib/net/client'
|
|
require 'lib/net/messages/indexevent'
|
|
require 'lib/net/messages/quit'
|
|
require 'lib/file/tail/since'
|
|
require 'stomp'
|
|
require 'socket'
|
|
|
|
|
|
class Agent < LogStash::Net::MessageClient
|
|
def initialize(host, port)
|
|
super(username="", password="", host=host, port=port)
|
|
@hostname = Socket.gethostname
|
|
@msgs = []
|
|
end # def initialize
|
|
|
|
def start_log_watcher
|
|
#@t1 = Thread.new do
|
|
#File::Tail::Since.new("/b/logs/auth.log.scorn").tail do |line|
|
|
#line.chomp!
|
|
#index("linux-syslog", line)
|
|
#end
|
|
#end
|
|
|
|
@t2 = Thread.new do
|
|
count = 0
|
|
#File::Tail::Since.new("/b/access.100").tail do |line|
|
|
File.open("/b/access.10k").each do |line|
|
|
line.chomp!
|
|
count += 1
|
|
index("httpd-access", line)
|
|
puts count
|
|
break if count >= 10
|
|
end
|
|
sendmsg("/queue/logstash", LogStash::Net::Messages::QuitRequest.new)
|
|
end
|
|
end # def start_log_watcher
|
|
|
|
def index(type, string)
|
|
ier = LogStash::Net::Messages::IndexEventRequest.new
|
|
ier.log_type = type
|
|
ier.log_data = string
|
|
ier.metadata["source_host"] = @hostname
|
|
|
|
#puts "Sending: #{ier}"
|
|
sendmsg("/queue/logstash", ier)
|
|
end # def index
|
|
|
|
def IndexEventResponseHandler(msg)
|
|
if msg.code != 0
|
|
puts msg.inspect
|
|
end
|
|
end # def IndexEventResponseHandler
|
|
|
|
def run
|
|
start_log_watcher
|
|
super
|
|
end
|
|
end
|
|
|
|
|
|
if $0 == __FILE__
|
|
if ARGV.length == 0
|
|
puts "Usage: #{$0} host:port"
|
|
exit 1
|
|
end
|
|
Thread::abort_on_exception = true
|
|
host, port = ARGV[0].split(":")
|
|
agent = Agent.new(host, port)
|
|
agent.run
|
|
end
|