Class: PromMultiProc::Writer

Inherits:
Object
  • Object
show all
Defined in:
lib/prom_multi_proc/writer.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(socket:, batch_size: 1, batch_timeout: 3, validate: false) ⇒ Writer

Returns a new instance of Writer.



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/prom_multi_proc/writer.rb', line 5

def initialize(socket:, batch_size: 1, batch_timeout: 3, validate: false)
  if !batch_size.is_a?(Integer) || batch_size <= 0
    raise PromMultiProcError.new("Invalid batch size: #{batch_size}")
  end
  if !batch_timeout.is_a?(Integer) || batch_timeout <= 0
    raise PromMultiProcError.new("Invalid batch timeout: #{batch_timeout}")
  end
  @batch_size = batch_size
  @batch_timeout = batch_timeout
  @validate = !!validate

  @lock = Mutex.new
  @thread = Thread.new {
    loop do
      sleep(batch_timeout)
      flush(force: true)
    end
  }
  @messages = []

  @socket = socket
end

Instance Attribute Details

#batch_size ⇒ Object (readonly)

Returns the value of attribute batch_size.



3
4
5
 
# File 'lib/prom_multi_proc/writer.rb', line 3

def batch_size
  @batch_size
end

#batch_timeout ⇒ Object (readonly)

Returns the value of attribute batch_timeout.



3
4
5
 
# File 'lib/prom_multi_proc/writer.rb', line 3

def batch_timeout
  @batch_timeout
end

#socket ⇒ Object (readonly)

Returns the value of attribute socket.



3
4
5
 
# File 'lib/prom_multi_proc/writer.rb', line 3

def socket
  @socket
end

Instance Method Details

#flush(force: false) ⇒ Object 



59
60
61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/prom_multi_proc/writer.rb', line 59

def flush(force: false)
  @lock.synchronize do
    if force && @messages.length > 0 || @messages.length >= batch_size
      begin
        write_socket(JSON.generate(@messages))
      ensure
        @messages.clear
      end
    else
      true
    end
  end
end

#socket? ⇒ Boolean

Returns:

  • (Boolean) 


73
74
75
 
# File 'lib/prom_multi_proc/writer.rb', line 73

def socket?
  !!write_socket("\n")
end

#validate? ⇒ Boolean

Returns:

  • (Boolean) 


28
29
30
 
# File 'lib/prom_multi_proc/writer.rb', line 28

def validate?
  @validate
end

#write(metric, method, value, labels) ⇒ Object 



32
33
34
35
36
37
38
39
 
# File 'lib/prom_multi_proc/writer.rb', line 32

def write(metric, method, value, labels)
  @lock.synchronize do
    metric.validate!(method, value, labels) if validate?
    @messages << metric.to_msg(method, value, labels)
  end

  flush
end

#write_multi(metrics) ⇒ Object

array of arrays where inner array is length 4 matching arguments for signature of #write



43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/prom_multi_proc/writer.rb', line 43

def write_multi(metrics)
  @lock.synchronize do
    if validate?
      metrics.each do |m, method, value, labels|
        m.validate!(method, value, labels)
      end
    end

    metrics.each do |m, method, value, labels|
      @messages << m.to_msg(method, value, labels)
    end
  end

  flush
end