pomelo-citrus 0.0.1

data/lib/citrus/master/watchdog.rb

@@ -0,0 +1,83 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 22 July 2014
+
+module Citrus
+  # Master
+  #
+  #
+  module Master
+    # WatchDog
+    #
+    #
+    class WatchDog
+      # Create a new watchdog
+      #
+      # @param [Object] app
+      # @param [Object] console_service
+      def initialize app, console_service
+        @app = app
+        @service = console_service
+        @servers = {}
+        @listeners = []
+      end
+
+      # Add server
+      #
+      # @param [Hash] server
+      def add_server server
+        return unless server
+        @servers[server[:server_id]] = server
+        notify({ :action => 'add_server', :server => server })
+      end
+
+      # Remove server
+      def remove_server
+      end
+
+      # Reconnect server
+      def reconnect_server
+      end
+
+      # Subscribe
+      #
+      # @param [String] server_id
+      def subscribe server_id
+        @listeners << server_id
+      end
+
+      # Unsubscribe
+      #
+      # @param [String] server_id
+      def unsubscribe server_id
+        @listeners.delete server_id
+      end
+
+      # Query
+      def query
+        @servers.values
+      end
+
+      private
+
+      # Notify
+      #
+      # @param [Hash] msg
+      #
+      # @private
+      def notify msg
+        @listeners.each { |server_id|
+          @service.agent.notify server_id, ConsoleModules::MonitorWatcher.module_id, msg
+        }
+      end
+
+      # Notify
+      #
+      # @param [String] server_id
+      # @param [Hash] msg
+      def notify_by_id server_id, msg
+        @service.agent.notify server_id, ConsoleModules::MonitorWatcher.module_id, msg
+      end
+    end
+  end
+end

data/lib/citrus/modules/console.rb

@@ -0,0 +1,45 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+require 'citrus/modules/console_module'
+
+module Citrus
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # Console
+    #
+    #
+    class Console < ConsoleModule
+
+      @module_id = '__console__'
+
+      # Initialize the module
+      #
+      # @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/modules/console_module.rb

@@ -0,0 +1,35 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 21 July 2014
+
+module Citrus
+  # 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/modules/master_watcher.rb

@@ -0,0 +1,88 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+require 'citrus/master/watchdog'
+require 'citrus/modules/console_module'
+
+module Citrus
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # MaterWatcher
+    #
+    #
+    class MasterWatcher < ConsoleModule
+
+      @module_id = '__master_watcher__'
+
+      # Initialize the module
+      #
+      # @param [Hash]   args
+      # @param [Object] console_service
+      def initialize args={}, console_service
+        @app = args[:app]
+        @service = console_service
+        @server_id = @app.server_id
+        @watchdog = Master::WatchDog.new @app, @service
+        @service.on('register') { |*args| on_server_add *args }
+        @service.on('disconnect') { |*args| on_server_leave *args }
+        @service.on('reconnect') { |*args| on_server_reconnect *args }
+      end
+
+      # Start the module
+      def start &block
+        block_given? and yield
+      end
+
+      # Master handler
+      #
+      # @param [Object] agent
+      # @param [Hash]   msg
+      def master_handler agent, msg, &block
+        return if !msg
+        case msg[:action]
+        when 'subscribe'
+          handle_subscribe agent, msg, &block
+        else
+        end
+      end
+
+      private
+
+      # Server add listener
+      #
+      # @param [Hash] record
+      # @private
+      def on_server_add record
+        return if !record || record[:type] == 'client' || !record[:server_type]
+        @watchdog.add_server record
+      end
+
+      # Server leave listener
+      #
+      # @private
+      def on_server_leave
+      end
+
+      # Server reconnect listener
+      #
+      # @private
+      def on_server_reconnect
+      end
+
+      # Handle subscribe
+      #
+      # @param [Object] agent
+      # @param [Hash]   msg
+      #
+      # @private
+      def handle_subscribe agent, msg, &block
+        return if !msg
+        @watchdog.subscribe msg[:server_id]
+        block_given? and yield nil, @watchdog.query
+      end
+    end
+  end
+end

data/lib/citrus/modules/monitor_watcher.rb

@@ -0,0 +1,86 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+require 'citrus/modules/console_module'
+
+module Citrus
+  # ConsoleModules
+  #
+  #
+  module ConsoleModules
+    # MaterWatcher
+    #
+    #
+    class MonitorWatcher < ConsoleModule
+
+      @module_id = '__monitor_watcher__'
+
+      # Initialize the module
+      #
+      # @param [Hash]   args
+      # @param [Object] console_service
+      def initialize args={}, console_service
+        @app = args[:app]
+        @service = console_service
+        @server_id = @app.server_id
+      end
+
+      # Start the module
+      def start &block
+        subscribe_request &block
+      end
+
+      # Monitor handler
+      #
+      # @param [Object] agent
+      # @param [Hash]   msg
+      def monitor_handler agent, msg, &block
+        return if !msg || !msg[:action]
+        case msg[:action]
+        when 'add_server'
+          handle_add_server agent, msg
+        else
+        end
+      end
+
+      private
+
+      # Subscribe request
+      #
+      # @private
+      def subscribe_request &block
+        msg = { :action => 'subscribe', :server_id => @server_id }
+        @service.agent.request ConsoleModules::MasterWatcher.module_id, msg, proc{ |err, servers|
+          if err
+            block_given? and yield err
+            return
+          end
+          add_servers servers
+          block_given? and yield
+        }
+      end
+
+      # Handle add server
+      #
+      # @param [Object] agent
+      # @param [Hash]   msg
+      #
+      # @private
+      def handle_add_server agent, msg
+        return if !msg || !msg[:server]
+        add_servers [msg[:server]]
+      end
+
+      # Add servers
+      #
+      # @param [Array] sinfos
+      #
+      # @private
+      def add_servers sinfos
+        return unless sinfos && !sinfos.empty?
+        @app.add_servers sinfos
+      end
+    end
+  end
+end

data/lib/citrus/monitor/monitor.rb

@@ -0,0 +1,61 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 20 July 2014
+
+require 'citrus/util/module_util'
+
+module Citrus
+  # Monitor
+  #
+  #
+  module Monitor
+    # Monitor
+    #
+    #
+    class Monitor
+      include Utils::ModuleUtil
+
+      # Create a new monitor
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args={}
+        @app = app
+        @master = false
+        @master_info = @app.master
+        @server_info = @app.cur_server
+        @modules = []
+        @close_watcher = args[:close_watcher]
+        @console_service = CitrusAdmin::ConsoleService.create_monitor_console({
+          :env => @app.env,
+          :host => @master_info[:host],
+          :port => @master_info[:port],
+          :server_id => @server_info[:server_id],
+          :server_type => @app.server_type,
+          :server_info => @server_info,
+          :auth_server => nil
+        })
+      end
+
+      # Start master
+      def start &block
+        register_default_modules
+        load_modules
+        @console_service.start { |err|
+          if err
+            block_given? and yield err
+            return
+          end
+          start_modules { |err|
+            block_given? and yield err
+            return
+          }
+        }
+      end
+
+      # Stop master
+      def stop &block
+      end
+    end
+  end
+end

data/lib/citrus/push_schedulers/buffer.rb

@@ -0,0 +1,16 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 16 July 2014
+
+module Citrus
+  # PushSchedulers
+  #
+  #
+  module PushSchedulers
+    # Buffer
+    #
+    #
+    class Buffer
+    end
+  end
+end

data/lib/citrus/push_schedulers/direct.rb

@@ -0,0 +1,76 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 16 July 2014
+
+module Citrus
+  # PushSchedulers
+  #
+  #
+  module PushSchedulers
+    # Direct
+    #
+    #
+    class Direct
+      # Initialize the component
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args
+        @app = app
+      end
+
+      # Schedule
+      #
+      # @param [Integer] req_id
+      # @param [String]  route
+      # @param [Hash]    msg
+      # @param [Array]   recvs
+      # @param [Hash]    args
+      def schedule req_id, route, msg, recvs, args, &block
+        if args[:type] == 'broadcast'
+          do_broadcast msg, args[:user_args]
+        else
+          do_batch_push msg, recvs
+        end
+
+        EM.next_tick { block.call } if block_given?
+      end
+
+      private
+
+      # Do broadcast
+      #
+      # @param [Hash]  msg
+      # @param [Hash]  args
+      #
+      # @private
+      def do_broadcast msg, args={}
+        channel_service = @app.channel_service
+        session_service = @app.session_service
+
+        if args[:binded]
+          session_service.sessions.each { |session|
+            session_service.send_message_by_uid session.uid, msg
+          }
+        else
+          session_service.sessions.each { |session|
+            session_service.send_message session.id, msg
+          }
+        end
+      end
+
+      # Do batch push
+      #
+      # @param [Hash]  msg
+      # @param [Array] recvs
+      #
+      # @private
+      def do_batch_push msg, recvs
+        session_service = @app.session_service
+        recvs.each { |recv|
+          session_service.send_message recv, msg
+        }
+      end
+    end
+  end
+end