pomelo-citrus-rpc 0.0.2

data/lib/citrus-rpc/rpc-client/proxy.rb

@@ -0,0 +1,37 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 4 July 2014
+
+module CitrusRpc
+  # RpcClient
+  #
+  #
+  module RpcClient
+    # Proxy
+    #
+    #
+    module Proxy
+      private
+
+      # Create proxy
+      #
+      # @param [Hash] args Options
+      #
+      # @option args [Class] remote
+      # @option args [Hash]  attach
+      # @option args [#call] proxy_cb
+      #
+      # @private
+      def create_proxy args={}
+        res = Object.new
+        methods = args[:remote].instance_methods
+        methods.each { |method|
+          res.define_singleton_method method, proc{ |*inner_args, &block|
+            args[:proxy_cb].call args[:service], method, args[:attach], false, *inner_args, &block
+          }
+        }
+        res
+      end
+    end
+  end
+end

data/lib/citrus-rpc/rpc-client/router.rb

@@ -0,0 +1,63 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 5 July 2014
+
+module CitrusRpc
+  # RpcClient
+  #
+  #
+  module RpcClient
+    # Router
+    #
+    #
+    module Router
+      # Calculate route info and return an appropriate server id
+      #
+      # @param [Object] session
+      # @param [Hash]   msg
+      # @param [Object] context
+      def df_route session, msg, context, &block
+      end
+
+      # Random algorithm for calculating server id
+      #
+      # @param [Object] client
+      # @param [String] server_type
+      # @param [Hash]   msg
+      def rd_route client, server_type, msg, &block
+      end
+
+      # Round-Robin algorithm for calculating server id
+      #
+      # @param [Object] client
+      # @param [String] server_type
+      # @param [Hash]   msg
+      def rr_route client, server_type, msg, &block
+      end
+
+      # Weight-Round-Robin algorithm for calculating server id
+      #
+      # @param [Object] client
+      # @param [String] server_type
+      # @param [Hash]   msg
+      def wrr_route client, server_type, msg, &block
+      end
+
+      # Least-Active algorithm for calculating server id
+      #
+      # @param [Object] client
+      # @param [String] server_type
+      # @param [Hash]   msg
+      def la_route client, server_type, msg, &block
+      end
+
+      # Consistent-Hash algorithm for calculating server id
+      #
+      # @param [Object] client
+      # @param [String] server_type
+      # @param [Hash]   msg
+      def ch_route client, server_type, msg, &block
+      end
+    end
+  end
+end

data/lib/citrus-rpc/rpc-server/acceptors/ws_acceptor.rb

@@ -0,0 +1,143 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 5 July 2014
+
+require 'websocket-eventmachine-server'
+
+module CitrusRpc
+  # RpcServer
+  #
+  #
+  module RpcServer
+    # WsAcceptor
+    #
+    #
+    class WsAcceptor
+      include Utils::EventEmitter
+
+      # Create a new websocket acceptor
+      #
+      # @param [Hash] args Options
+      #
+      # @option args [Boolean] buffer_msg
+      # @option args [Integer] interval
+      def initialize args={}, &block
+        @buffer_msg = args[:buffer_msg]
+        @interval = args[:interval]
+
+        @server = nil
+        @wss = {}
+
+        @msg_queues = {}
+        @callback = block
+
+        @listening = false
+        @closed = false
+      end
+
+      # Listen on port
+      #
+      # @param [Integer] port
+      def listen port
+        raise RuntimeError 'acceptor double listen' if @listening
+
+        begin
+          @server = WebSocket::EventMachine::Server.start(:host => '0.0.0.0', :port => port) { |ws|
+            ws.onopen {
+              @wss[ws.signature] = ws
+              peer_port, peer_host = Socket.unpack_sockaddr_in ws.get_peername
+              emit :connection, { :id => ws.signature, :ip => peer_host }
+            }
+
+            ws.onmessage { |msg, type|
+              begin
+                pkg = JSON.parse msg
+                if pkg.instance_of? Array
+                  process_msgs ws, pkg
+                else
+                  process_msg ws, pkg
+                end
+              rescue => err
+              end
+            }
+
+            ws.onclose {
+              @wss.delete ws.signature
+              @msg_queues.delete ws.signature
+            }
+            ws.onerror { |err| emit :error, err }
+          }
+        rescue => err
+          emit :error, err
+        end
+
+        on(:connection) { |obj| ip_filter obj }
+        @listening = true
+      end
+
+      # Close the acceptor
+      def close
+        return unless @listening && !@closed
+        EM.stop_server @server
+        @closed = true
+        emit :closed
+      end
+
+      private
+
+      # Clone error
+      #
+      # @private
+      def clone_error origin
+        { 'msg' => origin.message, 'stack' => nil }
+      end
+
+      # Process message
+      #
+      # @param [Object] ws
+      # @param [Hash]   pkg
+      #
+      # @private
+      def process_msg ws, pkg
+        @callback.call pkg['msg'] { |*args|
+          args.each_with_index { |arg, index|
+            args[index] = clone_error arg if arg.is_a? Exception
+          }
+
+          resp = { 'id' => pkg['id'], 'resp' => args }
+          if @buffer_msg
+            enqueue ws, resp
+          else
+            ws.send resp.to_json
+          end
+        }
+      end
+
+      # Batch version for process_msg
+      #
+      # @param [Object] ws
+      # @param [Array]  pkgs
+      #
+      # @private
+      def process_msgs ws, pkgs
+        pkgs.each { |pkg| process_msg ws, pkg }
+      end
+
+      # Enqueue the response
+      #
+      # @param [Object] ws
+      #
+      # @private
+      def enqueue ws, resp
+      end
+
+      # ip filter
+      #
+      # @param [Object] obj
+      #
+      # @private
+      def ip_filter obj
+      end
+    end
+  end
+end

data/lib/citrus-rpc/rpc-server/dispatcher.rb

@@ -0,0 +1,36 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 5 July 2014
+
+module CitrusRpc
+  # RpcServer
+  #
+  #
+  module RpcServer
+    # Dispatcher
+    #
+    #
+    module Dispatcher
+      # Dispatch message to appropriate service object
+      #
+      # @param [Hash] msg
+      # @param [Hash] services
+      def dispatch msg, services, &block
+        unless namespace = services[msg['namespace']]
+          block.call Exception.new 'no such namespace: ' + msg['namespace']
+          return
+        end
+        unless service = namespace[msg['service']]
+          block.call Exception.new 'no such service: ' + msg['service']
+          return
+        end
+        unless service.respond_to? msg['method']
+          block.call Exception.new 'no such method: ' + msg['method']
+          return
+        end
+
+        service.send msg['method'], *msg['args'], &block
+      end
+    end
+  end
+end

data/lib/citrus-rpc/rpc-server/gateway.rb

@@ -0,0 +1,58 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 5 July 2014
+
+require 'citrus-rpc/rpc-server/acceptors/ws_acceptor'
+require 'citrus-rpc/rpc-server/dispatcher'
+
+module CitrusRpc
+  # RpcServer
+  #
+  #
+  module RpcServer
+    # Gateway
+    #
+    #
+    class Gateway
+      include Dispatcher
+      include Utils::EventEmitter
+
+      # Create a gateway
+      #
+      # @param [Hash] args Options
+      #
+      # @option args [Integer] port
+      # @option args [Class]   acceptor_class
+      # @option args [Hash]    services
+      def initialize args={}
+        @port = args[:port] || 3050
+        @started = false
+        @stoped = false
+
+        @acceptor_class = args[:acceptor_class] || WsAcceptor
+        @services = args[:services]
+
+        @acceptor = @acceptor_class.new(args) { |msg, &block|
+          dispatch msg, @services, &block
+        }
+      end
+
+      # Start the gateway
+      def start
+        raise RuntimeError 'gateway already started' if @started
+        @started = true
+
+        @acceptor.on(:error) { |*args| emit :error, *args }
+        @acceptor.on(:closed) { |*args| emit :closed, *args }
+        @acceptor.listen @port
+      end
+
+      # Stop the gateway
+      def stop
+        return unless @started && !@stoped
+        @stoped = true
+        @acceptor.close
+      end
+    end
+  end
+end

data/lib/citrus-rpc/rpc-server/server.rb

@@ -0,0 +1,92 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 4 July 2014
+
+require 'citrus-rpc/rpc-server/gateway'
+
+module CitrusRpc
+  # RpcServer
+  #
+  #
+  module RpcServer
+    # Server
+    #
+    # @example
+    #
+    # Specifiy remote service interface paths
+    #
+    #  dirname = File.expand_path File.dirname(__FILE__)
+    #  records = [
+    #    { :namespace => 'user', :path => dirname + '/remote/test' }
+    #  ]
+    #
+    # Create a new rpc server and start it
+    #
+    #  Server.new(:records => records, :port => 3333).start
+    #
+    class Server
+      include CitrusLoader
+      include Utils::EventEmitter
+
+      # Create a new rpc server
+      #
+      # @param [Hash] args Options
+      #
+      # @option args [Integer] port
+      # @option args [Array]   records
+      def initialize args={}
+        raise ArgumentError, 'server port empty' unless args[:port]
+        raise ArgumentError, 'server port must be bigger than zero' unless args[:port] > 0
+        raise ArgumentError, 'records empty' unless args[:records]
+        raise ArgumentError, 'records must be an array' unless args[:records].respond_to? :to_a
+
+        @services = {}
+
+        create_namespace args[:records]
+        load_remote_services args[:records], args[:context]
+
+        args[:services] = @services
+
+        @gateway = Gateway.new args
+        @gateway.on(:error) { |*args| emit :error, *args }
+        @gateway.on(:closed) { |*args| emit :closed, *args }
+      end
+
+      # Start the rpc server
+      def start
+        @gateway.start
+      end
+
+      # Stop the rpc server
+      def stop
+        @gateway.stop
+      end
+
+      private
+
+      # Create remote services namespace
+      #
+      # @param [Array] records
+      #
+      # @private
+      def create_namespace records
+        records.each { |record| @services[record[:namespace]] ||= {} }
+      end
+
+      # Load remote services
+      #
+      # @param [Array]  records
+      # @param [Object] context
+      #
+      # @private
+      def load_remote_services records, context
+        records.each { |record|
+          remotes = load_app_remote record[:path]
+          remotes.each { |service, remote|
+            @services[record[:namespace]][service] = remote.new context
+          }
+        }
+      end
+    end
+  end
+end

data/lib/citrus-rpc/util/constants.rb

@@ -0,0 +1,20 @@
+# Author:: MinixLi (gmail: MinixLi1986)
+# Homepage:: http://citrus.inspawn.com
+# Date:: 5 July 2014
+
+module CitrusRpc
+  # Constants
+  #
+  #
+  module Constants
+    # DefaultParams
+    #
+    #
+    module DefaultParams
+      GraceTimeout = 3
+      Interval = 5
+      PendingSize = 1000
+      Timeout = 30
+    end
+  end
+end