Module: KL::Agent

Defined in:
lib/kldockeragent/agent.rb

Constant Summary collapse

PIDFile = 
"#{KL.home}/kldockeragent.pid"
@@notifier_engine = 
KL::Notifier.new
@@web_server = 
KL::Server.new
@@global_app_uuid = 
KL.getApplicationId

Class Method Summary collapse

Class Method Details

.close ⇒ Object 



146
147
148
149
 
# File 'lib/kldockeragent/agent.rb', line 146

def self.close
  KL.logger.info "[main] Stopping agent..."
  @@main_enabled = false
end

.get_pid ⇒ Object 



121
122
123
124
125
126
127
 
# File 'lib/kldockeragent/agent.rb', line 121

def self.get_pid
  if pid.nil?
    puts "nil"
  else
    puts "#{pid}"
  end
end

.getUUID ⇒ Object 



151
152
153
 
# File 'lib/kldockeragent/agent.rb', line 151

def self.getUUID
  @@global_app_uuid
end

.getVersion ⇒ Object 



155
156
157
 
# File 'lib/kldockeragent/agent.rb', line 155

def self.getVersion
  File.read(File.dirname(__FILE__) + '/../../VERSION').strip
end

.is_linux ⇒ Object 



134
135
136
 
# File 'lib/kldockeragent/agent.rb', line 134

def self.is_linux
  (RbConfig::CONFIG['host_os'] =~ /linux/)
end

.is_windows ⇒ Object 



130
131
132
 
# File 'lib/kldockeragent/agent.rb', line 130

def self.is_windows
  (RbConfig::CONFIG['host_os'] =~ /mswin|mingw|cygwin/)
end

.notifier_engine ⇒ Object 



138
139
140
 
# File 'lib/kldockeragent/agent.rb', line 138

def self.notifier_engine
  @@notifier_engine
end

.pid ⇒ Object 



103
104
105
106
107
108
109
110
 
# File 'lib/kldockeragent/agent.rb', line 103

def self.pid
  begin
    pid = File.read(PIDFile).to_i
    return pid if Process.kill 0, pid
  rescue
  end
  nil
end

.start(opts = {}) ⇒ Object

Start the agent.

options: :daemon => true if running as a daemon, false if as a console application :port => port of web server will listen to :ssl => set true to enable HTTPS :certfile => certificate file path for HTTPS :keyfile => key file path for HTTPS



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

def self.start(opts={})
  KL.logger.info "[main] Starting agent: #{KL::Agent.getUUID}"
  puts "Starting agent..."

  @@config = opts
  Process.daemon() if opts[:daemon] and not opts[:mock]
  begin
     if not is_windows
      # trap signal
      ['INT', 'HUP'].each do |signal|
        trap(signal) {
          KL.logger.info "[main] Shutting down services..."
          web_server.stop
          notifier_engine.stop
          loop do
            break if notifier_engine.status == :stopped
            sleep 1
          end

          self.close
        }
      end
    end

     File.open(PIDFile, 'w', 0644) { |f| f.write($$.to_s) }

     notifier_engine.start
     web_server.start

     @@main_enabled = true
     while @@main_enabled 
       sleep(1)
     end

  rescue Exception => e
     KL.logger.error "Starting the agent [Failed] #{e}\n#{e.backtrace.join("\n")}"
     raise e
  end
end

.status ⇒ Object

Return agent’s PID if it is running, otherwise nil.



114
115
116
117
118
119
120
 
# File 'lib/kldockeragent/agent.rb', line 114

def self.status
  if pid.nil?
    puts "Agent is not running."
  else
    puts "Agent is running with PID #{pid}"
  end
end

.stop(opts = {}) ⇒ Object

Stop the agent’s daemon.



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

def self.stop(opts={})
   begin
     pid = File.read(PIDFile).to_i
     puts "Stopping agent with PID #{pid}..."
     Process.kill 'HUP', pid

    if not opts[:mock]
      begin
        sleep (KL::Notifier::SleepTime + 0.5)

        Process.kill 0, pid
        KL.logger.info "[main] Agent is still running."
        puts "Agent is still running."

        KL.logger.info "[main] Killing agent."
        puts "Killing agent."
        Process.kill 9, pid
      rescue
        KL.logger.info "[main] Agent has stopped."
        puts "Agent has stopped."
        File.delete(PIDFile) if File.exist?(PIDFile)
      end
    end

   rescue
     puts "Agent is not running."
     File.delete(PIDFile) if File.exist?(PIDFile)
   end
end

.web_server ⇒ Object 



142
143
144
 
# File 'lib/kldockeragent/agent.rb', line 142

def self.web_server
  @@web_server
end