pomelo-citrus 0.0.1

data/lib/citrus/connectors/commands/heartbeat.rb

@@ -0,0 +1,22 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 25 July 2014
+
+module Citrus
+  # Connectors
+  #
+  #
+  module Connectors
+    # Commands
+    #
+    #
+    module Commands
+      # HeartBeat
+      #
+      #
+      class HeartBeat
+
+      end
+    end
+  end
+end

data/lib/citrus/connectors/commands/kick.rb

@@ -0,0 +1,22 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 25 July 2014
+
+module Citrus
+  # Connectors
+  #
+  #
+  module Connectors
+    # Commands
+    #
+    #
+    module Commands
+      # Kick
+      #
+      #
+      class Kick
+
+      end
+    end
+  end
+end

data/lib/citrus/connectors/common/coder.rb

@@ -0,0 +1,21 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 25 July 2014
+
+module Citrus
+  # Connectors
+  #
+  #
+  module Connectors
+    # Common
+    #
+    #
+    module Common
+      # Coder
+      #
+      #
+      module Coder
+      end
+    end
+  end
+end

data/lib/citrus/connectors/common/handler.rb

@@ -0,0 +1,21 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 25 July 2014
+
+module Citrus
+  # Connectors
+  #
+  #
+  module Connectors
+    # Common
+    #
+    #
+    module Common
+      # Handler
+      #
+      #
+      module Handler
+      end
+    end
+  end
+end

data/lib/citrus/connectors/ws_connector.rb

@@ -0,0 +1,110 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 25 July 2014
+
+require 'json'
+require 'websocket-eventmachine-server'
+require 'citrus/connectors/ws_socket'
+
+module Citrus
+  # Connectors
+  #
+  #
+  module Connectors
+    # WsConnector
+    #
+    #
+    class WsConnector
+      include Utils::EventEmitter
+
+      # Create a new websocket connector
+      #
+      # @param [Integer] port
+      # @param [String]  host
+      # @param [Hash]    args
+      def initialize port, host, args={}
+        @port = port
+        @host = host
+        @args = args
+
+        @heartbeats = args[:heartbeats] || true
+        @heartbeat_timeout = args[:heartbeat_timeout] || 0.06
+        @heartbeat_interval = args[:heartbeat_interval] || 0.025
+
+        @cur_id = 0
+      end
+
+      # Start the connector to listen to the specified port
+      def start &block
+        begin
+          @server = WebSocket::EventMachine::Server.start(:host => @host, :port => @port) { |ws|
+            ws.onopen {
+              ws_socket = WsSocket.new @cur_id, ws
+              @cur_id += 1
+              emit :connection, ws_socket
+              ws_socket.on(:closing) { |reason|
+                ws_socket.send({ 'route' => 'on_kick', 'reason' => reason })
+              }
+            }
+          }
+        rescue => err
+        end
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Stop the connector
+      #
+      # @param [Boolean] force
+      def stop force=false, &block
+        @server.close
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Encode message
+      #
+      # @param [Integer, NilClass] req_id
+      # @param [String] route
+      # @param [Object] msg
+      def encode req_id, route, msg
+        if req_id
+          compose_response req_id, route, msg
+        else
+          componse_push route, msg
+        end
+      end
+
+      # Decode message
+      #
+      # @param [String] msg
+      def decode msg
+        begin
+          JSON.parse msg
+        rescue => err
+        end
+      end
+
+      private
+
+      # Compose response message
+      #
+      # @param [Integer] msg_id
+      # @param [String]  route
+      # @param [Hash]    msg_body
+      #
+      # @private
+      def compose_response msg_id, route, msg_body
+        { 'id' => msg_id, 'body' => msg_body }
+      end
+
+      # Compose push message
+      #
+      # @param [String] route
+      # @param [Hash]   msg_body
+      #
+      # @private
+      def compose_push route, msg_body
+        { 'route' => route, 'body' => msg_body }
+      end
+    end
+  end
+end

data/lib/citrus/connectors/ws_socket.rb

@@ -0,0 +1,75 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 25 July 2014
+
+require 'json'
+
+module Citrus
+  # Connectors
+  #
+  #
+  module Connectors
+    # WsSocket
+    #
+    #
+    class WsSocket
+      include Utils::EventEmitter
+
+      attr_reader :id, :remote_address
+
+      # Create a new ws socket
+      #
+      # @param [Integer] id
+      # @param [Object]  ws
+      def initialize id, ws
+        @id = id
+        @ws = ws
+
+        port, ip = Socket.unpack_sockaddr_in @ws.get_peername
+        @remote_address = {
+          :port => port,
+          :ip => ip
+        }
+
+        @ws.onclose { emit :disconnect }
+        @ws.onerror { |err| emit :error }
+        @ws.onmessage { |msg, type| emit :message, msg }
+
+        @state = :state_inited
+      end
+
+      # Send message to the client
+      #
+      # @param [Hash] msg
+      def send msg
+        return unless @state == :state_inited
+        @ws.send msg.to_json
+      end
+
+      # Disconnect the client
+      def disconnect
+        return if @state == :state_closed
+        @state = :state_closed
+        @ws.close
+      end
+
+      # Batch version for send
+      #
+      # @param [Array] msgs
+      def send_batch msgs
+        @ws.send encode_batch(msgs)
+      end
+
+      private
+
+      # Encode batch messages
+      #
+      # @param [Array] msgs
+      #
+      # @private
+      def encode_batch msgs
+        msgs.to_json
+      end
+    end
+  end
+end

data/lib/citrus/filters/handler/handler_filter.rb

@@ -0,0 +1,19 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 21 July 2014
+
+module Citrus
+  # HandlerFilter
+  #
+  #
+  class HandlerFilter
+    attr_accessor :name
+
+    #
+    #
+    #
+    def initialize
+      @name = ''
+    end
+  end
+end

data/lib/citrus/filters/handler/too_busy.rb

@@ -0,0 +1,16 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 17 July 2014
+
+require 'citrus/filters/handler/handler_filter'
+
+module Citrus
+  # TooBusy
+  #
+  #
+  class TooBusy < HandlerFilter
+    def initialize
+      @name = 'too_busy'
+    end
+  end
+end

data/lib/citrus/filters/rpc/rpc_filter.rb

@@ -0,0 +1,19 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 21 July 2014
+
+module Citrus
+  # RpcFilter
+  #
+  #
+  class RpcFilter
+    attr_accessor :name
+
+    #
+    #
+    #
+    def initialize
+      @name = ''
+    end
+  end
+end

data/lib/citrus/filters/rpc/too_busy.rb

@@ -0,0 +1,16 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 17 July 2014
+
+require 'citrus/filters/rpc/rpc_filter'
+
+module Citrus
+  # TooBusy
+  #
+  #
+  class TooBusy < RpcFilter
+    def initialize
+      @name = 'too_busy'
+    end
+  end
+end

data/lib/citrus/master/master.rb

@@ -0,0 +1,60 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 18 July 2014
+
+require 'citrus/master/starter'
+require 'citrus/util/module_util'
+
+module Citrus
+  # Master
+  #
+  #
+  module Master
+    # Master
+    #
+    #
+    class Master
+      include Starter
+      include Utils::ModuleUtil
+
+      # Create a new master
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args={}
+        @app = app
+        @master = true
+        @master_info = app.master
+        @modules = []
+        @close_watcher = args[:close_watcher]
+        @console_service = CitrusAdmin::ConsoleService.create_master_console(
+          args.merge({
+            :env => app.env,
+            :port => @master_info[:port]
+          })
+        )
+      end
+
+      # Start master
+      def start &block
+        register_default_modules
+        load_modules
+        @console_service.start { |err|
+          exit if err
+          start_modules { |err|
+            if err
+              block_given? and yield err
+              return
+            end
+            run_servers
+            block_given? and yield
+          }
+        }
+      end
+
+      # Stop master
+      def stop &block
+      end
+    end
+  end
+end

data/lib/citrus/master/starter.rb

@@ -0,0 +1,73 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 19 July 2014
+
+module Citrus
+  # Master
+  #
+  #
+  module Master
+    # Starter
+    #
+    #
+    module Starter
+      include Utils
+
+      # Run servers
+      def run_servers
+        condition = @app.start_id || @app.type
+        case condition
+        when :master
+        when :all
+          @app.servers_map.each { |server_id, server|
+            run_server server
+          }
+        else
+        end
+      end
+
+      # Run server
+      #
+      # @param [Hash] server
+      def run_server server, &block
+        if local? server[:host]
+          options = []
+          options << sprintf('%s', $0)
+          options << sprintf('env=%s', @app.env)
+          server.each { |key, value|
+            options << sprintf('%s=%s', key, value)
+          }
+          local_run 'ruby', nil, options
+        else
+        end
+      end
+
+      #
+      #
+      #
+      def ssh_run
+      end
+
+      #
+      #
+      #
+      def local_run cmd, host, options
+        spawn_process cmd, host, options
+      end
+
+      #
+      #
+      #
+      def spawn_process cmd, host, options
+        child = fork {
+          exec cmd + options.inject('') { |res, str| res += ' ' + str }
+        }
+        EM.watch_process child, Module.new {
+          define_method(:process_exited) {
+            Process.wait child
+          }
+        }
+      end
+    end
+  end
+end