Class: KL::Notifier

Inherits:
Object
  • Object
show all
Defined in:
lib/kldockeragent/notifier.rb

Constant Summary collapse

SleepTime = 
10
NotifierLockFile = 
"#{KL.home}/notifier.lock.#{Time.now.to_i}"

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(p = {}) ⇒ Notifier

Returns a new instance of Notifier.



11
12
13
14
15
16
17
18
 
# File 'lib/kldockeragent/notifier.rb', line 11

def initialize(p={})
	@lock = Mutex.new
	@enabled = false
	@status = :stopped
	@index = 0
	config = KL.config
	@rate = config['notifier']['rate']
end

Instance Attribute Details

#enabled ⇒ Object (readonly)

Returns the value of attribute enabled.



9
10
11
 
# File 'lib/kldockeragent/notifier.rb', line 9

def enabled
  @enabled
end

#mode ⇒ Object (readonly)

Returns the value of attribute mode.



9
10
11
 
# File 'lib/kldockeragent/notifier.rb', line 9

def mode
  @mode
end

#status ⇒ Object (readonly)

Returns the value of attribute status.



9
10
11
 
# File 'lib/kldockeragent/notifier.rb', line 9

def status
  @status
end

Instance Method Details

#notify_data ⇒ Object 



88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/kldockeragent/notifier.rb', line 88

def notify_data
	KL.logger.info "[heartbeat]"
	if KL.server_registered
		if (@index <= (@rate/SleepTime))
			@index = @index + 1
		else
			@index = 0
			KL.logger.info "[notifier] Executing notifier..."
			KL.Command_notifyStatus
			KL.logger.info "[notifier] Notifier executed."
		end
	else
		KL.logger.info "[notifier] Executing notifier..."
		KL.Command_registerNewAgent
		KL.logger.info "[notifier] Notifier executed."
	end
end

#register_notifier_thread(mode = nil) ⇒ Object 



68
69
70
71
72
73
74
75
76
77
 
# File 'lib/kldockeragent/notifier.rb', line 68

def register_notifier_thread(mode=nil)
	File.open(NotifierLockFile, File::RDWR|File::CREAT, 0644) { |f|
		f.flock(File::LOCK_EX)
		value = (mode == :reset ? 0 : (f.read.to_i + 1))
		f.rewind
		f.write(value.to_s)
		f.flush
		f.truncate(f.pos)
	}
end

#run_notifier ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/kldockeragent/notifier.rb', line 51

def run_notifier
	KL.logger.info "[notifier] Executing notifier engine every #{@rate} seconds"
	while @enabled
		begin
			wait_for_notifier?

			notify_data

			sleep SleepTime
			
		rescue Exception => e
			KL.logger.error "Error on executing notifier \n#{e}\n#{e.backtrace.join("\n")}"
			sleep SleepTime
		end
	end
end

#start ⇒ Object 



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
 
# File 'lib/kldockeragent/notifier.rb', line 24

def start
	@enabled = true
	@lock.synchronize {
		return if @status == :running
		@status = :running
	}
	Thread.new {
		begin
			register_notifier_thread(:reset)

			system("rm -f #{KL.home}/operator.*.lock")
			system("rm -f #{KL.home}/notifier.lock.*")

			KL.logger.info "[notifier] Notifier engine is running"

			self.run_notifier

			File.delete(NotifierLockFile) if File.exist?(NotifierLockFile)
			KL.logger.info "[notifier] Notifier engine has stopped."
		
		rescue Exception => exp
			KL.logger.error "Cannot start notifier engine => #{exp}\n#{exp.backtrace.join("\n")}"
		end
		@status = :stopped
	}
end

#stop ⇒ Object 



20
21
22
 
# File 'lib/kldockeragent/notifier.rb', line 20

def stop
	@enabled = false
end

#wait_for_notifier? ⇒ Boolean

Returns:

  • (Boolean) 


79
80
81
82
83
84
85
86
 
# File 'lib/kldockeragent/notifier.rb', line 79

def wait_for_notifier?
	total_notifier = 1
	loop do
		total_notifier = (File.exist?(NotifierLockFile) ? File.read(NotifierLockFile).to_i : 0)
		return if total_notifier <= 0 or not @enabled
		sleep 1
	end
end