aboutsummaryrefslogtreecommitdiff
path: root/lib/mauve/heartbeat.rb
blob: 11f4d5b89deecf1354baa02706ad2414d413680d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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
47
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
require 'mauve/sender'
require 'mauve/proto'
require 'mauve/mauve_thread'
require 'log4r'

module Mauve

  #
  # This class is responsible for sending a heartbeat to another mauve instance elsewhere.
  #
  class Heartbeat < MauveThread

    include Singleton

    #
    # Allow access to some basics.
    #
    attr_reader   :raise_after, :destination, :summary, :detail, :send_every

    #
    # This sets up the Heartbeat singleton
    #
    def initialize
      super

      @destination    = nil
      @summary        = "Mauve alert server heartbeat failed"
      @detail         = "The Mauve server at #{Server.instance.hostname} has failed to send a heartbeat."
      @raise_after    = 310
      @send_every     = 60
    end

    #
    # This is the time period after which an alert is raised by the remote Mauve instance.
    # @param [Integer] i Seconds
    # @return [Integer] Seconds
    def raise_after=(i)
      raise ArgumentError "raise_after must be an integer" unless i.is_a?(Integer)      
      @raise_after = i
    end
    
    #
    # This is the time period after which an alert is raised by the remote Mauve instance.
    # @param [Integer] i Seconds
    # @return [Integer] Seconds
    def send_every=(i)
      raise ArgumentError "send_every must be an integer" unless i.is_a?(Integer)      
      @send_every = i
    end

    alias poll_every= send_every= 

    # Sets the summary of the heartbeat
    #
    # @param [String] s Summary
    def summary=(s)
      raise ArgumentError "summary must be a string" unless s.is_a?(String)
      @summary = s
    end

    # Sets the detail of the heartbeat
    #
    # @param [String] d Detail
    def detail=(d)
      raise ArgumentError "detail must be a string" unless d.is_a?(String)
      @detail = d
    end

    # Sets the destinantion Mauve instance
    #
    # @param [String] d Destination
    #
    def destination=(d)
      raise ArgumentError "destination must be a string" unless d.is_a?(String)
      @destination = d
    end

    # @return [Log4r::Logger]
    def logger
      @logger ||= Log4r::Logger.new(self.class.to_s)
    end

    private

    # @private This is the main heartbeat loop.
    def main_loop
      #
      # Don't send if no destination set.
      #
      return if @destination.nil?

      update = Mauve::Proto::AlertUpdate.new
      update.replace = false
      update.alert = []
      update.source = Server.instance.hostname
      update.transmission_id = rand(2**63)

      message = Mauve::Proto::Alert.new
      message.id = "mauve-heartbeat"
      message.summary = self.summary
      message.detail = self.detail
      message.raise_time = (Time.now.to_f+self.raise_after).to_i
      message.clear_time = Time.now.to_i

      update.alert << message

      Mauve::Sender.new(self.destination).send(update)
      logger.debug "Sent to #{self.destination}"

      sleep @send_every
    end

  end

end