pomelo-citrus 0.0.1

data/lib/citrus/components/proxy.rb

@@ -0,0 +1,195 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 20 July 2014
+
+require 'digest/crc32'
+require 'citrus/components/component'
+require 'citrus/util/path_util'
+
+module Citrus
+  # Components
+  #
+  #
+  module Components
+    # Proxy
+    #
+    #
+    class Proxy < Component
+      @name = 'proxy'
+
+      include Utils::PathUtil
+
+      # Initialize the component
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args={}
+        args[:buffer_msg] = args[:buffer_msg] || false
+        args[:interval]  = args[:interval] || 0.03
+
+        args[:router] = gen_router
+        args[:context] = app
+        args[:route_context] = app
+
+        @args = args
+        @client = CitrusRpc::RpcClient::Client.new @args
+
+        @app = app
+        @app.on(:add_servers) { |servers| add_servers servers }
+        @app.on(:remove_servers) { |ids| remove_servers ids }
+        @app.on(:replace_servers) { |servers| replace_servers servers }
+      end
+
+      # Start the component
+      def start &block
+        unless @app.rpc_befores.empty?
+          @client.before @app.rpc_befores
+        end
+
+        unless @app.rpc_afters.empty?
+          @client.after @app.rpc_afters
+        end
+
+        if @app.rpc_error_handler
+          @client.set_error_handler @app.rpc_error_handler
+        end
+
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Component lifecycle callback
+      def after_start &block
+        @app.rpc = @client.proxies.user
+        @app.sysrpc = @client.proxies.sys
+
+        @app.define_singleton_method :rpc_invoke, proc{ |*args, &block|
+          @client.rpc_invoke *args, &block
+        }
+
+        @client.start &block
+      end
+
+      # Add remote servers
+      #
+      # @param [Array] servers
+      def add_servers servers
+        return unless servers
+        return if servers.empty
+
+        gen_proxies servers
+
+        @client.add_servers servers
+      end
+
+      # Remove remote servers
+      #
+      # @param [Array] ids
+      def remove_servers ids
+        @client.remove_servers ids
+      end
+
+      # Replace remote servers
+      #
+      # @param [Array] servers
+      def replace_servers servers
+        return unless servers
+        return if servers.empty
+
+        @client.proxies = {}
+        gen_proxies servers
+
+        @client.replace_servers servers
+      end
+
+      # Proxy for rpc client's rpc_invoke
+      #
+      # @param [String] server_id
+      # @param [Hash]   msg
+      def rpc_invoke server_id, msg, &block
+        @client.rpc_invoke server_id, msg, &block
+      end
+
+      private
+
+      # Generate proxies for the server infos
+      #
+      # @param [Array] sinfos
+      #
+      # @private
+      def gen_proxies sinfos
+        sinfos.each { |sinfo|
+          @client.add_proxies get_proxy_records(sinfo) unless has_proxy? sinfo
+        }
+      end
+
+      # Check whether a proxy has been generated
+      #
+      # @param [Hash] sinfo
+      #
+      # @private
+      def has_proxy? sinfo
+        @client.proxies.sys && @client.proxies.sys.respond_to? sinfo[:server_type]
+      end
+
+      # Get proxy path for rpc client
+      #
+      # @param [Hash] sinfo
+      #
+      # @private
+      def get_proxy_records sinfo
+        records = []
+
+        role = @app.frontend? sinfo ? :frontend : :backend
+        server_type = sinfo[:server_type]
+
+        record = get_sys_remote_path role
+        records << remote_path_record('sys', server_type, record) if record
+
+        record = get_user_remote_path @app.base, server_type
+        records << remote_path_record('user', server_type, record) if record
+
+        records
+      end
+
+      # Generate router
+      #
+      # @private
+      def gen_router
+        Proc.new { |session, msg, &block|
+          routers = @app.routers
+          unless routers
+            default_router session, msg, &block
+            return
+          end
+
+          type = msg['server_type']
+          router = routers[type] || routers['default']
+
+          if router
+            router.call session, msg, &block
+          else
+            default_router session, msg, &block
+          end
+        }
+      end
+
+      # Default router
+      #
+      # @param [Object] session
+      # @param [Hash]   msg
+      #
+      # @private
+      def default_router session, msg, &block
+        list = @app.get_servers_by_type msg['server_type']
+        unless list && !list.empty?
+          block_given? and yield Exception.new 'can not find server info for type: ' + msg['server_type']
+          return
+        end
+
+        uid = session ? (session.uid || '') : ''
+        idx = (Digest::CRC32.hexdigest uid).to_i % list.length
+        block_given? and yield nil, list[idx][:server_id]
+      end
+    end
+  end
+end

data/lib/citrus/components/push_scheduler.rb

@@ -0,0 +1,74 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 17 July 2014
+
+require 'citrus/components/component'
+require 'citrus/push_schedulers/direct'
+
+module Citrus
+  # Components
+  #
+  #
+  module Components
+    # PushScheduler
+    #
+    #
+    class PushScheduler < Component
+      @name = 'push_scheduler'
+
+      # Initialize the component
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args={}
+        @app = app
+        @scheduler = get_scheduler app, args
+      end
+
+      # Component lifecycle callback
+      def after_start &block
+        if @scheduler.respond_to? :start
+          @scheduler.start &block
+        else
+          EM.next_tick { block.call } if block_given?
+        end
+      end
+
+      # Component lifecycle callback
+      def stop &block
+        if @scheduler.respond_to? :stop
+          @scheduler.stop &block
+        else
+          EM.next_tick { block.call } if block_given?
+        end
+      end
+
+      # Schedule how the message to send
+      #
+      # @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 @scheculer.respond_to? :schedule
+          @scheduler.schedule req_id, route, msg, recvs, args, &block
+        else
+        end
+      end
+
+      private
+
+      # Get scheduler
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      #
+      # @private
+      def get_scheduler app, args={}
+        scheduler = args[:scheduler] || Citrus::PushSchedulers::Direct
+        scheduler.new app, args
+      end
+    end
+  end
+end

data/lib/citrus/components/remote.rb

@@ -0,0 +1,71 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 17 July 2014
+
+require 'citrus/components/component'
+require 'citrus/util/path_util'
+
+module Citrus
+  # Components
+  #
+  #
+  module Components
+    # Remote
+    #
+    #
+    class Remote < Component
+      @name = 'remote'
+
+      include Utils::PathUtil
+
+      # Initialize the component
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args={}
+        args[:buffer_msg] = args[:buffer_msg] || false
+        args[:interval]  = args[:interval] || 0.03
+        @app = app
+        @args = args
+      end
+
+      # Start the component
+      def start &block
+        @args[:port] = @app.cur_server[:port]
+        @args[:paths] = get_remote_paths
+        @args[:context] = @app
+
+        @remote = CitrusRpc::RpcServer::Server.new @args
+        @remote.start
+
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Stop the component
+      #
+      # @param [Boolean] force
+      def stop force=false, &block
+        @remote.stop force
+        EM.next_tick { block_given? and yield }
+      end
+
+      private
+
+      # Get remote paths
+      def get_remote_paths
+        paths = []
+
+        role = @app.frontend? sinfo ? :frontend : :backend
+        server_type = sinfo[:server_type]
+
+        sys_path = get_sys_remote_path role
+        paths << remote_path_record('sys', server_type, sys_path) if sys_path
+
+        user_path = get_user_remote_path @app.base, server_type
+        paths << remote_path_record('user', server_type, user_path) if user_path
+
+        paths
+      end
+    end
+  end
+end

data/lib/citrus/components/server.rb

@@ -0,0 +1,61 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 17 July 2014
+
+require 'citrus/components/component'
+require 'citrus/server/server'
+
+module Citrus
+  # Components
+  #
+  #
+  module Components
+    # Server
+    #
+    #
+    class Server < Component
+      @name = 'server'
+
+      # Initialize the component
+      #
+      # @param [Object] app
+      def initialize app
+        @server = Citrus::Server::Server.new app
+      end
+
+      # Start the component
+      def start &block
+        @server.start
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Component lifecycle callback
+      def after_start &block
+        @server.after_start
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Stop the component
+      def stop force=false, &block
+        @server.stop
+        EM.next_tick { block_given? and yield }
+      end
+
+      # Proxy server handle
+      #
+      # @param [Hash]   msg
+      # @param [Object] session
+      def handle msg, session, &block
+        @server.handle msg, session, &block
+      end
+
+      # Proxy server global handle
+      #
+      # @param [Hash]   msg
+      # @param [Object] session
+      def global_handle msg, session, &block
+        @server.global_handle msg, session, &block
+      end
+    end
+  end
+end

data/lib/citrus/components/session.rb

@@ -0,0 +1,41 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 17 July 2014
+
+require 'citrus/components/component'
+require 'citrus/common/service/session_service'
+
+module Citrus
+  # Components
+  #
+  #
+  module Components
+    # Session
+    #
+    #
+    class Session < Component
+      @name = 'session'
+
+      attr_reader :service
+
+      # Initialize the component
+      #
+      # @param [Object] app
+      # @param [Hash]   args
+      def initialize app, args={}
+        @app = app
+        @service = Common::Service::SessionService.new args
+
+        this = self
+        @app.define_singleton_method :session_service, proc{ this }
+      end
+
+      # Proxy for connection service
+      #
+      # @param [String] name
+      def method_missing name, *args, &block
+        @service.send name, *args, &block
+      end
+    end
+  end
+end

data/lib/citrus/connectors/commands/handshake.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
+      # HandShake
+      #
+      #
+      class HandShake
+
+      end
+    end
+  end
+end