cognizant 0.0.1 → 0.0.2

data/lib/cognizant/server/daemon.rb

@@ -4,6 +4,7 @@ require "cognizant"
 require "cognizant/logging"
 require "cognizant/process"
 require "cognizant/server/interface"
+require "cognizant/system"
 
 module Cognizant
   module Server
@@ -85,9 +86,9 @@ module Cognizant
       # @return [String] Defaults to nil
       attr_accessor :group
 
-      # Array of processes being managed.
+      # Hash of processes being managed.
       # @private
-      # @return [Array]
+      # @return [Hash]
       attr_accessor :processes
 
       # Initializes and starts the cognizant daemon with the given options
@@ -121,10 +122,10 @@ module Cognizant
         @pids_dir = File.expand_path(@pids_dir)
         @logs_dir = File.expand_path(@logs_dir)
 
-        self.processes = {}
+        self.processes = Hash.new
 
-        # # Only available through a config file/stdin.
-        # load_processes(options[:monitor]) if options.has_key?(:monitor)
+        # Only available through a config file/stdin.
+        load_processes(options[:monitor]) if options.has_key?(:monitor)
       end
 
       def bootup
@@ -134,7 +135,8 @@ module Cognizant
         EventMachine.run do
           start_interface_server
           start_periodic_ticks
-          daemonize
+          daemonize_process
+          write_pid
         end
       end
 
@@ -149,14 +151,13 @@ module Cognizant
 
       def monitor(process_name = nil, attributes = {}, &block)
         process = Cognizant::Process.new(process_name, attributes, &block)
-        self.processes[process_name] = process
+        self.processes[process.name] = process
         process.monitor
       end
 
       # Stops the TCP server and the tick loop, and performs cleanup.
       def shutdown
         log.info "Shutting down cognizantd..."
-
         EventMachine.next_tick do
           EventMachine.stop
           logger.close
@@ -175,13 +176,14 @@ module Cognizant
 
       # Starts the TCP server with the set socket lock file or port.
       def start_interface_server
+        stop_previous_server
         if port = @port
           log.info "Starting the TCP server at #{@port}..."
           host = "127.0.0.1"
           splitted = port.to_s.split(":")
           host, port = splitted if splitted.size > 1
           EventMachine.start_server(host, port, Server::Interface)
-        else  
+        else
           log.info "Starting the UNIX domain server with socket #{@socket}..."
           EventMachine.start_unix_domain_server(@socket, Server::Interface)
         end
@@ -191,9 +193,8 @@ module Cognizant
       def start_periodic_ticks
         log.info "Starting the periodic tick..."
         EventMachine.add_periodic_timer(1) do
-          self.processes.each do |group, process|
-            process.tick
-          end
+          System.reset_data
+          self.processes.values.map(&:tick)
         end
       end
 
@@ -220,18 +221,55 @@ module Cognizant
         Signal.trap('QUIT', &terminator)
       end
 
+      def stop_previous_server
+        if @pidfile and File.exists?(@pidfile)
+          if previous_daemon_pid = File.read(@pidfile).to_i
+            # Only attempt to stop automatically if the daemon will run in background.
+            if @daemonize and System.pid_running?(previous_daemon_pid)
+              # Ensure that the process stops within 5 seconds or force kill.
+              signals = ["TERM", "KILL"]
+              timeout = 2
+              catch :stopped do
+                signals.each do |stop_signal|
+                  # Send the stop signal and wait for it to stop.
+                  System.signal(stop_signal, previous_daemon_pid)
+
+                  # Poll to see if it's stopped yet. Minimum 2 so that we check at least once again.
+                  ([timeout / signals.size, 2].max).times do
+                    throw :stopped unless System.pid_running?(previous_daemon_pid)
+                    sleep 1
+                  end
+                end
+              end
+            end
+          end
+
+          # Alert the user to manually stop the previous daemon, if it is [still] alive.
+          if System.pid_running?(previous_daemon_pid)
+            raise "There is already a daemon running with pid #{previous_daemon_pid}."
+          else
+            begin
+              File.unlink(@pidfile) if @pidfile
+            rescue Errno::ENOENT
+              nil
+            end
+          end
+        end
+      end
+
       # Daemonize the current process and save it pid in a file.
-      def daemonize
+      def daemonize_process
         if @daemonize
           log.info "Daemonizing into the background..."
           ::Process.daemon
+        end
+      end
 
-          pid = ::Process.pid
-
-          if @pidfile
-            log.info "Writing the daemon pid (#{pid}) to the pidfile..."
-            File.open(@pidfile, "w") { |f| f.write(pid) }
-          end
+      def write_pid
+        pid = ::Process.pid
+        if @pidfile
+          log.info "Writing the daemon pid (#{pid}) to the pidfile..."
+          File.open(@pidfile, "w") { |f| f.write(pid) }
         end
       end
     end

data/lib/cognizant/server/interface.rb

@@ -1,4 +1,3 @@
-require "json"
 require "eventmachine"
 
 require "cognizant/server/commands"
@@ -6,55 +5,82 @@ require "cognizant/server/commands"
 module Cognizant
   module Server
     class Interface < EventMachine::Connection
+
+      attr_accessor :authenticated, :username, :password
+
       def post_init
-        # puts "-- someone connected to the server!"
+        @authenticated = false
+        @username      = nil
+        @password      = nil
+        if Cognizant::Server.daemon.username and Cognizant::Server.daemon.password
+          post_authentication_challenge
+        end
       end
 
       def receive_data(args)
+        if not @authenticated and (Cognizant::Server.daemon.username and Cognizant::Server.daemon.password)
+          return post_authentication_challenge(args.to_s)
+        end
+
         args = [*args.split]
         command = args.shift.to_s.strip.downcase
         if command.size > 0
           begin
             Commands.send(command, *args) do |response|
-              send_data "#{response.to_json}\r\n\r\n"
+              send_data "#{response}"
             end
           rescue => e
-            send_data "#{e.inspect.to_json}\r\n\r\n"
+            send_data "#{e.message}"
           end
         end
-        # close_connection_after_writing
+        close_connection_after_writing
       end
 
       def unbind
+        @authenticated = false
         # puts "-- someone disconnected from the server!"
       end
+
+      def post_authentication_challenge(data = nil)
+        data = data.to_s.strip
+        if data.to_s.size > 0
+          if @username
+            @password = data
+          else
+            @username = data
+          end
+        end
+
+        unless @username.to_s.size > 0
+          if Cognizant::Server.daemon.username and Cognizant::Server.daemon.username.size != nil
+            send_data "Username: "
+            return
+          end
+        end
+
+        unless @password.to_s.size > 0
+          if Cognizant::Server.daemon.password and Cognizant::Server.daemon.password.size != nil
+            send_data "Password: "
+            return
+          end
+        end
+
+        if @username and @password
+          validate
+        end
+      end
+
+      def validate
+        @authenticated = @username == Cognizant::Server.daemon.username and @password == Cognizant::Server.daemon.password
+        @username = nil
+        @password = nil
+        if @authenticated
+          send_data "Authenticated. Welcome!\r\n\r\n"
+        else
+          send_data "Authentication failed.\r\n\r\n"
+          post_authentication_challenge
+        end
+      end
     end
   end
 end
-
-# module Cognizant
-#   module Server
-#     class Interface < EventMachine::Connection
-#       include EventMachine::Protocols::ObjectProtocol
-#       # include EventMachine::Protocols::SASLauth
-# 
-#       def post_init
-#         @obj = Hash.new
-#         puts "-- someone connected to the echo server!"
-#       end
-# 
-#       def receive_object(method)
-#         send_object @obj.send(*method)
-#       end
-# 
-#       def unbind
-#         @obj = nil
-#         puts "-- someone disconnected from the echo server!"
-#       end
-# 
-#       # def validate(usr, psw, sys, realm)
-#       #   usr == TestUser and psw == TestPsw
-#       # end
-#     end
-#   end
-# end

data/lib/cognizant/system/ps.rb

@@ -0,0 +1,61 @@
+module Cognizant
+  module System
+    module PS
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        # Fields to fetch from ps.
+        IDX_MAP = {
+          :pid  => 0,
+          :ppid => 1,
+          :pcpu => 2,
+          :rss  => 3
+        }
+
+        def cpu_usage(pid)
+          ps_axu[pid] && ps_axu[pid][IDX_MAP[:pcpu]].to_f
+        end
+
+        def memory_usage(pid)
+          ps_axu[pid] && ps_axu[pid][IDX_MAP[:rss]].to_f
+        end
+
+        def get_children(parent_pid)
+          child_pids = Array.new
+          ps_axu.each_pair do |pid, chunks|
+            child_pids << chunks[IDX_MAP[:pid]].to_i if chunks[IDX_MAP[:ppid]].to_i == parent_pid.to_i
+          end
+          child_pids
+        end
+
+        def reset_data
+          store.clear unless store.empty?
+        end
+
+        def ps_axu
+          # TODO: Need a mutex here?
+          store[:ps_axu] ||= begin
+            # BSD style ps invocation.
+            lines = `ps axo #{IDX_MAP.keys.join(",")}`.split("\n")
+
+            lines.inject(Hash.new) do |mem, line|
+              chunks = line.split(/\s+/)
+              chunks.delete_if {|c| c.strip.empty? }
+              pid = chunks[IDX_MAP[:pid]].strip.to_i
+              mem[pid] = chunks
+              mem
+            end
+          end
+        end
+
+        private
+
+        def store
+          @store ||= Hash.new
+        end
+      end
+    end
+  end
+end

data/lib/cognizant/system/signal.rb

@@ -0,0 +1,37 @@
+module Cognizant
+  module System
+    module Signal
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        def signal(signal, pid)
+          ::Process.kill(signal, pid) if pid and pid != 0
+        end
+
+        def send_signals(pid, options = {})
+          # Return if the process is not running.
+          return true unless pid_running?(pid)
+
+          signals = options[:signals] || ["TERM", "INT", "KILL"]
+          timeout = options[:timeout] || 10
+
+          catch :stopped do
+            signals.each do |stop_signal|
+              # Send the stop signal and wait for it to stop.
+              signal(stop_signal, pid)
+
+              # Poll to see if it's stopped yet. Minimum 2 so that we check at least once again.
+              ([timeout / signals.size, 2].max).times do
+                throw :stopped unless pid_running?(pid)
+                sleep 1
+              end
+            end
+          end
+          not pid_running?(pid)
+        end
+      end
+    end
+  end
+end

data/lib/cognizant/system.rb

@@ -0,0 +1,28 @@
+require "cognizant/system/signal"
+require "cognizant/system/ps"
+
+module Cognizant
+  module System
+    extend self
+
+    include Cognizant::System::Signal
+    include Cognizant::System::PS
+
+    def pid_running?(pid)
+      return false unless pid and pid != 0
+      signal(0, pid)
+      # It's running since no exception was raised.
+      true
+    rescue Errno::ESRCH
+      # No such process.
+      false
+    rescue Errno::EPERM
+      # Probably running, but we're not allowed to pass signals.
+      # TODO: Is this a sign of problems ahead?
+      true
+    else
+      # Possibly running.
+      true
+    end
+  end
+end

data/lib/cognizant/util/rotational_array.rb

@@ -0,0 +1,17 @@
+module Cognizant
+  module Util
+    class RotationalArray < Array
+      def initialize(size)
+        @capacity = size
+        super # no size - intentionally
+      end
+
+      def push(value)
+        super(value)
+        self.shift if self.length > @capacity
+        self
+      end
+      alias_method :<<, :push
+    end
+  end
+end

data/lib/cognizant/util/symbolize_hash_keys.rb

@@ -0,0 +1,19 @@
+class Hash
+  # Destructively convert all keys by using the block operation.
+  # This includes the keys from the root hash and from all
+  # nested hashes.
+  def deep_transform_keys!(&block)
+    keys.each do |key|
+      value = delete(key)
+      self[yield(key)] = value.is_a?(Hash) ? value.deep_transform_keys!(&block) : value
+    end
+    self
+  end
+
+  # Destructively convert all keys to symbols, as long as they respond
+  # to +to_sym+. This includes the keys from the root hash and from all
+  # nested hashes.
+  def deep_symbolize_keys!
+    deep_transform_keys!{ |key| key.to_sym rescue key }
+  end
+end

data/lib/cognizant/version.rb

@@ -1,3 +1,3 @@
 module Cognizant
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/cognizant.rb

@@ -1,7 +1,10 @@
+require "active_support/core_ext/numeric"
+require "active_support/duration"
+
 require "cognizant/server"
 
 module Cognizant
-  def self.monitor(process_name, &block)
-    Cognizant::Server.daemon.monitor(process_name, &block)
+  def self.monitor(process_name = nil, attributes = {}, &block)
+    Cognizant::Server.daemon.monitor(process_name, attributes, &block)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cognizant
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-11-01 00:00:00.000000000 Z
+date: 2012-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -91,7 +91,56 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
-description: Process monitoring and management framework
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: gli
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: formatador
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Cognizant is a process management system utility that supervises your
+  processes, ensuring their state based on a flexible criteria.
 email:
 - contact@gurpartap.com
 executables:
@@ -116,8 +165,6 @@ files:
 - images/logo.png
 - images/logo.pxm
 - lib/cognizant.rb
-- lib/cognizant/client/cli.rb
-- lib/cognizant/client/interface.rb
 - lib/cognizant/logging.rb
 - lib/cognizant/process.rb
 - lib/cognizant/process/actions.rb
@@ -125,6 +172,14 @@ files:
 - lib/cognizant/process/actions/start.rb
 - lib/cognizant/process/actions/stop.rb
 - lib/cognizant/process/attributes.rb
+- lib/cognizant/process/condition_check.rb
+- lib/cognizant/process/conditions.rb
+- lib/cognizant/process/conditions/always_true.rb
+- lib/cognizant/process/conditions/cpu_usage.rb
+- lib/cognizant/process/conditions/flapping.rb
+- lib/cognizant/process/conditions/memory_usage.rb
+- lib/cognizant/process/conditions/poll_condition.rb
+- lib/cognizant/process/conditions/trigger_condition.rb
 - lib/cognizant/process/execution.rb
 - lib/cognizant/process/pid.rb
 - lib/cognizant/process/status.rb
@@ -132,9 +187,14 @@ files:
 - lib/cognizant/server/commands.rb
 - lib/cognizant/server/daemon.rb
 - lib/cognizant/server/interface.rb
+- lib/cognizant/system.rb
+- lib/cognizant/system/ps.rb
+- lib/cognizant/system/signal.rb
+- lib/cognizant/util/rotational_array.rb
+- lib/cognizant/util/symbolize_hash_keys.rb
 - lib/cognizant/validations.rb
 - lib/cognizant/version.rb
-homepage: http://gurpartap.github.com/cognizant/
+homepage: http://github.com/Gurpartap/cognizant
 licenses: []
 post_install_message: 
 rdoc_options: []
@@ -148,7 +208,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 2933141880858945843
+      hash: -687515942623630851
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -157,13 +217,13 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 2933141880858945843
+      hash: -687515942623630851
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
-summary: Cognizant is an advanced and efficient process monitoring and management
-  framework.
+summary: Cognizant is a process management system utility that supervises your processes,
+  ensuring their state based on a flexible criteria.
 test_files: []
 has_rdoc: 

data/lib/cognizant/client/cli.rb

@@ -1,9 +0,0 @@
-module Cognizant
-  module Client
-    class CLI
-      def initialize(command)
-        puts "got #{command}"
-      end
-    end
-  end
-end

data/lib/cognizant/client/interface.rb

@@ -1,33 +0,0 @@
-require "eventmachine"
-
-module Cognizant
-  module Client
-    class Interface
-      def initialize(server = nil)
-        @sock = EventMachine.connect(server, Client::Interface::Handler)
-      end
-
-      def method_missing(*meth, &blk)
-        @sock.queue << blk
-        @sock.send_object(meth)
-      end
-
-      class Handler < EventMachine::Connection
-        # include EventMachine::Protocols::SASLauthclient
-        include EventMachine::Protocols::ObjectProtocol
-
-        attr_reader :queue
-
-        def post_init
-          @queue = []
-        end
-
-        def receive_object(obj)
-          if callback = @queue.shift
-            callback.call(obj)
-          end
-        end
-      end
-    end
-  end
-end