ampedxx-yrpc 0.1.4

data/lib/yrpc/client.rb

@@ -0,0 +1,79 @@
+module Yrpc
+  class Client < SimpleDelegator
+
+    def initialize(service:, options: {}, client_options: {})
+      @base_klass = service
+      @service_klass = "#{base_klass}::Service".constantize
+      @opts = options || {}
+      @opts[:password] = options.fetch(:password, '').to_s
+      @opts[:hostname] = options.fetch(:hostname, Yrpc.default_client_host)
+      client = "#{service}::Stub".constantize.new(@opts[:hostname], build_ssl_credentials, interceptors: [*client_options[:interceptors]])
+      super(client)
+    end
+
+    attr_reader :service_klass
+    attr_reader :base_klass
+    attr_reader :opts
+
+    def invoke(request_method, params = {}, metadata={}, opts={}, &block)
+      request_method = request_method.to_sym
+      req = streaming_request?(request_method) ? params : request_object(request_method, params)
+      md = build_metadata(metadata)
+      call_sig = call_signature(request_method)
+      raise NotImplementedError, "The method #{request_method} has not been implemented in this service." unless call_sig
+      resp = execute(call_sig, req, md, opts, &block)
+      p resp
+      Response.new(resp)
+    end
+
+    def build_metadata(metadata = {})
+      unless opts[:password].empty?
+        username = opts.fetch(:username, 'grpc').to_s
+        username = username.empty? ? '' : "#{username}:"
+        auth_string = Base64.encode64("#{username}#{opts[:password]}")
+        metadata[:authorization] = "Basic #{auth_string}".tr("\n", '')
+      end
+      metadata
+    end
+
+    def execute(call_sig, req, metadata, opts = {}, &block)
+      # Timer.time do
+      opts[:return_op] = true
+      opts[:metadata] = metadata
+      # end
+      send(call_sig, req, opts, &block)
+    end
+
+    def rpc_desc(request_method)
+      service_klass.rpc_descs[request_method]
+    end
+
+    def request_object(request_method, params = {})
+      desc = rpc_desc(request_method)
+      desc && desc.input ? desc.input.new(params) : nil
+    end
+
+    def call_signature(request_method)
+      desc = rpc_desc(request_method)
+      desc && desc.name ? desc.name.to_s.underscore.to_sym : nil
+    end
+
+    def build_ssl_credentials
+      cert = nil
+      if opts[:ssl_certificate_file]
+        cert = File.read(opts[:ssl_certificate_file]).to_s.strip
+      elsif opts[:ssl_certificate]
+        cert = opts[:ssl_certificate].to_s.strip
+      end
+
+      cert ? GRPC::Core::ChannelCredentials.new(cert) : :this_channel_is_insecure
+    end
+
+    def streaming_request?(request_method)
+      desc = rpc_desc(request_method)
+      desc && (desc.client_streamer? || desc.bidi_streamer?)
+    end
+
+
+  end
+end

data/lib/yrpc/configuration.rb

@@ -0,0 +1,48 @@
+require_relative "interceptors/active_record"
+require 'logger'
+module Yrpc
+  module Configuration
+    VALID_CONFIG_KEYS = {
+        server_binding_url: '0.0.0.0:9001',
+        interceptors: nil,
+        use_default_interceptors: true,
+        default_client_host: '',
+        controllers_path: 'app/controllers',
+        server_options: {},
+        services: [],
+        logger:nil,
+        use_ssl: false,
+    }.freeze
+
+    attr_accessor *VALID_CONFIG_KEYS.keys
+
+    def self.extended(base)
+      base.reset
+    end
+
+    def configure
+      yield self
+    end
+
+    def options
+      opts = {}
+      VALID_CONFIG_KEYS.each_key do |k|
+        opts.merge!(k => send(k))
+      end
+      opts
+    end
+
+    #重制一些配置
+    def reset
+      VALID_CONFIG_KEYS.each do |k, v|
+        send((k.to_s + '='), v)
+      end
+      self.logger = ::Logger.new(STDOUT)
+      self.interceptors = Yrpc::Interceptors::Registry.new
+      if use_default_interceptors
+        interceptors.use(Yrpc::Interceptors::ActiveRecord::ResetConnection)
+      end
+      options
+    end
+  end
+end

data/lib/yrpc/controllers/base.rb

@@ -0,0 +1,47 @@
+module Yrpc
+  module Controllers
+    class Base
+      include ActiveSupport::Callbacks
+
+      define_callbacks :call
+
+      attr_reader :request
+
+
+      def initialize(method_key:, service:, rpc_desc:, active_call:, message:)
+        @request = Request.new(
+            method_key: method_key,
+            service: service,
+            rpc_desc: rpc_desc,
+            active_call: active_call,
+            message: message
+        )
+      end
+
+      def self.before_action(methods)
+        [*methods].each do |method|
+          set_callback :call, :before, method
+        end
+      end
+
+      def self.after_action(methods)
+        [*methods].each do |method|
+          set_callback :call, :after, method
+        end
+      end
+
+
+      def self.bind(service_class)
+        Yrpc.services << service_class.name.constantize
+        ServiceBinder.new(service_class).bind!(self)
+      end
+
+
+      def call(method_key, &block)
+        run_callbacks :call do
+          send(method_key, &block)
+        end
+      end
+    end
+  end
+end

data/lib/yrpc/controllers/request.rb

@@ -0,0 +1,91 @@
+module Yrpc
+  module Controllers
+
+    class Request
+
+      attr_reader :message
+      # @var [GRPC::ActiveCall] active_call
+      attr_reader :active_call
+      # @var [Symbol] method_key
+      attr_reader :method_key
+      # @var [Grpc::Controllers::Request::Type] type
+      attr_reader :type
+      # @var [Class] service
+      attr_reader :service
+
+      delegate :metadata, to: :active_call
+      delegate :messages, :client_streamer?, :server_streamer?, :bidi_streamer?, :request_response?, to: :type
+
+      ##
+      # Abstract representation of a gRPC request type
+      #
+      class Type
+        delegate :client_streamer?, :server_streamer?, :bidi_streamer?, :request_response?, to: :@rpc_desc
+
+        ##
+        # Initialize a new request type object
+        #
+        # @param [GRPC::RpcDesc] rpc_desc The RPC descriptor for the request type
+        #
+        def initialize(rpc_desc)
+          @rpc_desc = rpc_desc
+        end
+      end
+
+      ##
+
+      #struct GRPC::RpcDesc
+      def initialize(method_key:, service:, rpc_desc:, active_call:, message:)
+        @method_key = method_key
+        @service = service
+        @active_call = active_call
+        @message = message
+        @rpc_desc = rpc_desc
+        @type = Type.new(rpc_desc)
+      end
+
+
+      def service_key
+        @service.name.underscore.tr('/', '.').gsub('.service', '')
+      end
+
+      def method_missing(method_id, *arguments, &block)
+        method_message = *arguments.join
+        if (method_id.to_s =~ /^raise_[\w]+/) == 0
+          error_type = method_id.to_s.split('raise_')[1].upcase!
+          code = "Yrpc::Response::Code::#{error_type}".constantize
+          message = method_message
+          raise Yrpc::Error::ResponseError.new(code, message)
+        else
+          super
+        end
+      end
+
+
+      def response_class
+        @rpc_desc.output
+      end
+
+
+      def request_class
+        @rpc_desc.input
+      end
+
+      # grpc 中的 service.method
+      def method_name
+        "#{service_key}.#{method_key}"
+      end
+
+      # 返回这个request的所有mesage 要根据stremer类型来判断
+      def messages
+        if client_streamer?
+          message.call { |msg| yield msg }
+        elsif bidi_streamer?
+          message
+        else
+          [message]
+        end
+      end
+    end
+  end
+end

data/lib/yrpc/controllers/response.rb

@@ -0,0 +1,104 @@
+module Yrpc
+
+  class Response
+
+    module Code
+      # 添加模型常量国际化方法
+      # include Dictionary::Module::I18n
+
+      ################################################################################
+      #
+      # 20000 成功
+      #
+      ################################################################################
+
+      SUCCESS = '20000'
+
+      ################################################################################
+      #
+      # 3xxxx 数据相关
+      #
+      ################################################################################
+
+      # 用户绑定第三方账户
+
+      # 第三方账户已绑定其它用户
+      PROVIDER_BIND_ANOTHER_USER = '30010'
+
+      ################################################################################
+      #
+      # 4xxxx 业务相关
+      #
+      ################################################################################
+
+      # 非法请求
+      INVALID_REQUEST = '40300'
+
+      # 终端密钥错误
+      INVALID_TERMINAL_SESSION_KEY = '40301'
+
+      # 用户密钥错误
+      INVALID_USER_SESSION_KEY = '40302'
+
+      # 超出请求限制数
+      EXCEED_REQUEST_LIMIT = '40303'
+
+      # 版本号不适配
+      NOT_COMPATIBLE_REVISION = '40304'
+
+      # 访问令牌过期
+      ACCESS_TOKEN_EXPIRED = '40305'
+
+      ################################################################################
+      #
+      # 5xxxx 系统相关
+      #
+      ################################################################################
+
+      # 未知错误（通常在捕捉异常后使用）
+      ERROR = '50000'
+
+      # 请求参数缺失
+      # Todo: 移动到数据相关
+      MISS_REQUEST_PARAMS = '50001'
+
+      # 数据处理错误
+      # Todo: 移动到数据相关
+      DATA_PROCESS_ERROR = '51000'
+
+      # 数据缺失错误
+      # Todo: 移动到数据相关
+      DATA_MISS_ERROR = '51001'
+
+      ORDER_PAYMENT_COMPLETED = '55000'
+
+      NEED_LOGIN = '40100'
+
+      ERROR_VALIDATE_CODE = '40200'
+
+      # 全部
+      # ALL = get_all_constants.map { |constant| constant.to_s.downcase }
+    end
+
+
+    def initialize(operation, execution_time = nil)
+      @operation = operation
+      @message = operation.execute
+      @metadata = operation.metadata
+      @trailing_metadata = operation.trailing_metadata
+      @deadline = operation.deadline
+      @cancelled = operation.cancelled?
+      @execution_time = execution_time || 0.0
+    end
+
+    def message
+      @message ||= @operation.execute
+    end
+
+    def status
+      @message ||= @operation.execute
+      @message&.status
+    end
+
+  end
+end

data/lib/yrpc/controllers/service_binder.rb

@@ -0,0 +1,87 @@
+module Yrpc
+  module Controllers
+
+    class ServiceBinder
+      ##
+      #未来可能会有一些desc 的 拓展，先预留一个肉便器类。
+      #
+      class BoundDesc < SimpleDelegator;
+      end
+
+      #初始化。。没什么好解释的
+      def initialize(service)
+        @service = service
+      end
+
+
+      # 通过controller绑定所有的方法
+      def bind!(controller)
+        rpc_methods.each { |name, desc| bind_method(controller, name, desc) }
+      end
+
+      def rpc_desc(request_method)
+        @service.rpc_descs[request_method]
+      end
+
+      private
+
+
+      #打开类并且进行一个方法define，返回给这个方法打call。这里要注意的是request 会以参数形式构造进controller
+      #这里也涉及到stream 类型。
+      def bind_method(controller, method_name, desc)
+        method_desc = method_name.to_s.to_sym
+        method_key = method_name.to_s.underscore.to_sym
+        current_desc = rpc_desc(method_desc)
+        service_ref = @service
+        @service.class_eval do
+          if desc.request_response?
+            define_method(method_key) do |message, active_call|
+              c = controller.new(method_key: method_key, service: service_ref, message: message, active_call: active_call, rpc_desc: desc)
+              begin
+                c.call(method_key)
+              rescue => e
+                p method_key
+                p e.backtrace_locations
+                desc && desc.output ? desc.output.new(status: {code: e.respond_to?(:code) ? e.code :  Yrpc::Response::Code::ERROR , message: e.message}) : (raise e)
+              end
+            end
+          elsif desc.client_streamer?
+            define_method(method_key) do |active_call|
+              c = controller.new(method_key: method_key, service: service_ref, message: proc { |&block| active_call.each_remote_read(&block) }, active_call: active_call, rpc_desc: desc)
+              begin
+                c.call(method_key)
+              rescue => e
+                desc && desc.output ? desc.output.new(status: {code: e&.code || Yrpc::Response::Code::ERROR, message: e.message}) : (raise e)
+              end
+            end
+          elsif desc.server_streamer?
+            define_method(method_key) do |message, active_call, &block|
+              c = controller.new(method_key: method_key, service: service_ref, message: message, active_call: active_call, rpc_desc: desc)
+              begin
+                c.call(method_key)
+              rescue => e
+                desc && desc.output ? desc.output.new(status: {code: e&.code || Yrpc::Response::Code::ERROR , message: e.message}) : (raise e)
+              end
+            end
+          else # bidi
+            define_method(method_key) do |messages, active_call, &block|
+              c = controller.new(method_key: method_key, service: service_ref, message: messages, active_call: active_call, rpc_desc: desc)
+              begin
+                c.call(method_key)
+              rescue => e
+                desc && desc.output ? desc.output.new(status: {code: e&.code || Yrpc::Response::Code::ERROR, message: e.message}) : (raise e)
+              end
+            end
+          end
+        end
+      end
+
+      ##
+      # 讲方法都代理到boundDesc 上
+      #
+      def rpc_methods
+        @service.rpc_descs.map { |rd| BoundDesc.new(rd) }
+      end
+    end
+  end
+end