pomelo-citrus-admin 0.0.1

data/lib/citrus-admin/modules/console_module.rb

@@ -0,0 +1,35 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 21 July 2014
+
+module CitrusAdmin
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # ConsoleModule
+    #
+    #
+    class ConsoleModule
+      #
+      #
+      #
+      def self.inherited subclass
+        class << subclass
+          attr_reader :module_id
+        end
+      end
+
+      attr_reader :type, :delay, :interval
+
+      #
+      #
+      #
+      def initialize
+        @type = ''
+        @delay = 0
+        @interval = 0
+      end
+    end
+  end
+end

data/lib/citrus-admin/modules/monitor_log.rb

@@ -0,0 +1,45 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+require 'citrus-admin/modules/console_module'
+
+module CitrusAdmin
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # MonitorLog
+    #
+    #
+    class MonitorLog < ConsoleModule
+
+      @module_id = 'monitor_log'
+
+      #
+      #
+      # @param [Hash]   args
+      # @param [Object] console_service
+      def initialize args={}, console_service
+      end
+
+      # Monitor handler
+      #
+      #
+      def monitor_handler
+      end
+
+      # Master handler
+      #
+      #
+      def master_handler
+      end
+
+      # Client handler
+      #
+      #
+      def client_handler
+      end
+    end
+  end
+end

data/lib/citrus-admin/modules/process_info.rb

@@ -0,0 +1,45 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+require 'citrus-admin/modules/console_module'
+
+module CitrusAdmin
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # ProcessInfo
+    #
+    #
+    class ProcessInfo < ConsoleModule
+
+      @module_id = 'process_info'
+
+      #
+      #
+      # @param [Hash]   args
+      # @param [Object] console_service
+      def initialize args={}, console_service
+      end
+
+      # Monitor handler
+      #
+      #
+      def monitor_handler
+      end
+
+      # Master handler
+      #
+      #
+      def master_handler
+      end
+
+      # Client handler
+      #
+      #
+      def client_handler
+      end
+    end
+  end
+end

data/lib/citrus-admin/modules/system_info.rb

@@ -0,0 +1,45 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+require 'citrus-admin/modules/console_module'
+
+module CitrusAdmin
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # SystemInfo
+    #
+    #
+    class SystemInfo < ConsoleModule
+
+      @module_id = 'system_info'
+
+      #
+      #
+      # @param [Hash]   args
+      # @param [Object] console_service
+      def initialize args={}, console_service
+      end
+
+      # Monitor handler
+      #
+      #
+      def monitor_handler
+      end
+
+      # Master handler
+      #
+      #
+      def master_handler
+      end
+
+      # Client handler
+      #
+      #
+      def client_handler
+      end
+    end
+  end
+end

data/lib/citrus-admin/modules/watch_server.rb

@@ -0,0 +1,45 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 8 July 2014
+
+require 'citrus-admin/modules/console_module'
+
+module CitrusAdmin
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # WatchServer
+    #
+    #
+    class WatchServer < ConsoleModule
+
+      @module_id = 'watch_server'
+
+      #
+      #
+      # @param [Hash]   args
+      # @param [Object] console_service
+      def initialize args={}, console_service
+      end
+
+      # Monitor handler
+      #
+      #
+      def monitor_handler
+      end
+
+      # Master handler
+      #
+      #
+      def master_handler
+      end
+
+      # Client handler
+      #
+      #
+      def client_handler
+      end
+    end
+  end
+end

data/lib/citrus-admin/monitor_agent.rb

@@ -0,0 +1,166 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 9 July 2014
+
+module CitrusAdmin
+  # MonitorAgent
+  #
+  #
+  class MonitorAgent
+    include Protocol
+    include Utils::EventEmitter
+
+    # Create a new monitor agent
+    #
+    # @param [Hash] args Options
+    #
+    # @option args [Object]  :console_service
+    # @option args [String]  :server_id
+    # @option args [String]  :server_type
+    # @option args [Object]  :server_info
+    def initialize args={}
+      @console_service = args[:console_service]
+      @server_id = args[:server_id]
+      @server_type = args[:server_type]
+      @server_info = args[:server_info]
+      @req_id = 1
+      @callbacks = {}
+      @state = :state_inited
+    end
+
+    # Register and connect to master server
+    #
+    # @param [Integer] port
+    # @param [String]  host
+    def connect port, host, &block
+      if @state != :state_inited
+        return
+      end
+      begin
+        @ws = WebSocket::EventMachine::Client.connect :uri => 'ws://' + host + ':' + port.to_s
+        @ws.onopen {
+          @state = :state_connected
+          event = 'register'
+          msg = {
+            :type => 'monitor',
+            :server_id => @server_id,
+            :server_type => @server_type,
+            :server_info => @server_info,
+            :pid => Process.pid
+          }
+          @console_service.auth_server.call(msg, @console_service.env) { |token|
+            msg[:token] = token
+            @ws.send [event, msg].to_json
+          }
+        }
+        @ws.onmessage { |msg, type|
+          begin
+            event, msg = parse msg
+            case event
+            when 'register'
+              process_register_msg msg, &block
+            when 'monitor'
+              process_monitor_msg msg
+            else
+            end
+          rescue => err
+          end
+        }
+        @ws.onclose { |code, reason|
+          @state = :state_closed
+          emit 'close'
+        }
+        @ws.onerror { |err|
+          if @state == :state_inited
+            block_given? and yield err
+          else
+            emit 'error', err
+          end
+        }
+      rescue => err
+        block_given? and yield err
+      end
+    end
+
+    # Close the monitor agent
+    def close
+      return if @state == :state_closed
+      @state = :state_closed
+      @ws.close
+    end
+
+    # Request master server with callback
+    #
+    # @param [String] module_id
+    # @param [Object] msg
+    def request module_id, msg, block
+      if @state != :state_registered
+        return
+      end
+      req_id = @req_id
+      @req_id += 1
+      @callbacks[req_id] = block
+      @ws.send ['monitor', compose_request(req_id, module_id, msg)].to_json
+    end
+
+    # Notify master server without callback
+    #
+    # @param [String] module_id
+    # @param [Object] msg
+    def notify module_id, msg
+      if @state != :state_registered
+        return
+      end
+      @ws.send ['monitor', compose_request(nil, module_id, msg)].to_json
+    end
+
+    private
+
+    # Process register message
+    #
+    # @param [Object] msg
+    #
+    # @private
+    def process_register_msg msg, &block
+      if msg && msg[:code] == PRO_OK
+        @state = :state_registered
+        block_given? and yield
+      else
+        emit 'close'
+      end
+    end
+
+    # Process monitor message
+    #
+    # @param [Object] msg
+    #
+    # @private
+    def process_monitor_msg msg, &block
+      return unless @state == :state_registered
+      if msg[:command]
+        # command from master server
+        @console_service.command(msg[:command], msg[:module_id], msg[:body]) { |err, res|
+          # notify should not have a callback
+        }
+      else
+        if resp_id = msg[:resp_id]
+          # response from master server
+          if !callback = @callbacks[resp_id]
+            return
+          end
+          @callbacks.delete resp_id
+          callback.call msg[:err], msg[:body]
+          return
+        end
+        # request from master server
+        @console_service.execute(msg[:module_id], 'monitor_handler', msg[:body]) { |err, res|
+          if is_request? msg
+            @ws.send ['monitor', compose_response(msg, err, res)].to_json
+          else
+             # notify should not have a callback
+          end
+        }
+      end
+    end
+  end
+end

data/lib/citrus-admin/util/protocol.rb

@@ -0,0 +1,83 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 9 July 2014
+
+module CitrusAdmin
+  # Protocol
+  #
+  #
+  module Protocol
+    #
+    #
+    #
+    PRO_OK = 1
+    PRO_FAIL = -1
+
+    # Componse request
+    #
+    # @param [Integer] req_id
+    # @param [String]  module_id
+    # @param [Object]  body
+    def compose_request req_id, module_id, body
+      if req_id
+        # request message
+        { :req_id => req_id, :module_id => module_id, :body => body }
+      else
+        # notify message
+        { :module_id => module_id, :body => body }
+      end
+    end
+
+    # Compose response
+    #
+    # @param [Object] msg
+    # @param [Object] err
+    # @param [Object] res
+    def compose_response msg, err, res
+      return nil unless msg[:req_id]
+      { :resp_id => msg[:req_id], :err => clone_error(err), :body => res }
+    end
+
+    # Compose command
+    #
+    # @param [Integer] req_id
+    # @param [String]  command
+    # @param [String]  module_id
+    # @param [Object]  body
+    def compose_command req_id, command, module_id, body
+      if req_id
+        # command message
+        { :req_id => req_id, :command => command, :module_id => module_id, :body => body }
+      else
+        { :command => command, :module_id => module_id, :body => body }
+      end
+    end
+
+    # Parse message
+    #
+    # @param [String] msg
+    def parse msg
+      begin
+        JSON.parse msg, { :symbolize_names => true }
+      rescue => err
+      end
+    end
+
+    # Determine if a message is a request
+    #
+    # @param [Object] msg
+    def is_request? msg
+      msg && msg[:req_id]
+    end
+
+    # Clone error
+    #
+    # @private
+    def clone_error origin
+      if origin.is_a? Exception
+        return { :msg => origin.message, :stack => nil }
+      end
+      return origin
+    end
+  end
+end