lib/smallcage/application.rb in smallcage-0.0.9 vs lib/smallcage/application.rb in smallcage-0.1.0

@@ -3,24 +3,29 @@
 OPTIONS = {}
 OPTIONS[:original_argv] = ARGV.clone
 
 class SmallCage::Application
   require 'optparse'
-
-  @@signal_handlers = {
-    "INT" => [],
-    "TERM" => []
-  }
   
-  @@signal_handlers.keys.each do |signal|
-    Signal.trap(signal) do
-      @@signal_handlers[signal].each do |proc|
-        proc.call(signal)
+  @@signal_handlers = nil
+
+  def self.init_signal_handlers
+    @@signal_handlers = {
+      "INT" => [],
+      "TERM" => []
+    }
+
+    @@signal_handlers.keys.each do |signal|
+      Signal.trap(signal) do
+        @@signal_handlers[signal].each do |proc|
+          proc.call(signal)
+        end
       end
     end
   end
   
   def self.add_signal_handler(signal, handler)
+    init_signal_handlers if @@signal_handlers.nil?
     signal.to_a.each do |s|
       @@signal_handlers[s] << handler
     end
   end
 
\ No newline at end of file