Class: PuavoRest::FluentRelay

Inherits:
PuavoSinatra show all
Defined in:
resources/fluent_relay.rb

Constant Summary

Constant Summary

Constants inherited from PuavoSinatra

PuavoSinatra::ROOT

Class Method Summary (collapse)

Instance Method Summary (collapse)

Methods inherited from PuavoSinatra

#auth, #basic_auth, #flog, #flog=, #from_post, #json_params, #kerberos, #legacy_server_auth, #limit, #pw_mgmt_server_auth, #server_auth, #txt

Class Method Details

+ (Object) fluent_logger 



8
9
10
 
# File 'resources/fluent_relay.rb', line 8

def self.fluent_logger
  @@logger ||= Fluent::Logger::FluentLogger.new(nil, :host=>'localhost', :port=>24224)
end

+ (Object) fluent_logger=(logger) 



16
17
18
 
# File 'resources/fluent_relay.rb', line 16

def self.fluent_logger=(logger)
  @@logger = logger
end

Instance Method Details

- (Object) fluent_logger 



12
13
14
 
# File 'resources/fluent_relay.rb', line 12

def fluent_logger
  self.class.fluent_logger
end

- (Object) handle_json 



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'resources/fluent_relay.rb', line 20

def handle_json
  json_parser = Yajl::Parser.new
  records = json_parser.parse(request.body)

  klass = records.class
  if klass == Hash
    records = [records]
  elsif klass == Array
    records = records
  else
    raise BadInput, :user => "Invalid json type #{ klass }"
  end

  puts "Relaying #{ records.size } fluentd records from http to local fluentd"

  records.each do |r|
    tag = r["_tag"]
    time = r["_time"]
    r.delete("_tag")
    r.delete("_time")

    # https://github.com/fluent/fluent-logger-ruby/blob/master/lib/fluent/logger/fluent_logger.rb#L115
    if not fluent_logger.post_with_time(tag, r, time)
      raise InternalError, :user => "Failed to relay fluent packages"
    end
  end
end

- (Object) handle_msgpack 



48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'resources/fluent_relay.rb', line 48

def handle_msgpack
  body_size = request.body.size
  self.flog.info("Relaying fluentd chunk", {
    :format => "msgpack",
    :bytes => body_size
  })

  parsing_start = Time.now
  u = MessagePack::Unpacker.new(request.body)
  parsing_time = Time.now - parsing_start

  puts "Relaying #{ request.body.size } bytes of msgpack data to fluentd"


  relaying_start = Time.now
  i = 0
  u.each do |(tag, time, r)|
    i += 1
    if not fluent_logger.post_with_time(tag, r, time)
      raise InternalError, :user => "Failed to relay fluent packages"
    end
  end
  relaying_time = Time.now - relaying_start

  self.flog.info("Relayed fluentd chunk", {
    :format => "msgpack",
    :parse_ms => parsing_time * 1000,
    :relay_ms => relaying_time * 1000,
    :total_ms => (relaying_time+parsing_time) * 1000,
    :bytes => body_size,
    :count => i
  })

  puts "Relayed #{ i } records to fluentd"
end