ampedxx-yrpc 0.1.4

data/lib/yrpc/error/response_error.rb

@@ -0,0 +1,12 @@
+module Yrpc
+  module Error
+    class ResponseError < StandardError
+      attr_reader :code
+
+      def initialize(code, message)
+        super(message)
+        @code = code
+      end
+    end
+  end
+end

data/lib/yrpc/executor/executor.rb

@@ -0,0 +1,29 @@
+module Yrpc
+
+  class Executor
+
+
+    def run
+      server = Yrpc::Server.new(Yrpc.server_options)
+      Yrpc.services.each { |service| server.add_service(service) }
+      server.grpc_server.run
+    end
+
+    def auto_load_all
+
+    end
+
+
+  end
+
+  module Ruler
+    def self.to_underscore(string)
+      string.gsub(/::/, '/').
+          gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2').
+          gsub(/([a-z\d])([A-Z])/, '\1_\2').
+          tr("-", "_").
+          downcase
+    end
+  end
+
+end

data/lib/yrpc/helpers/autoload.rb

@@ -0,0 +1,13 @@
+module Yrpc
+  module AutoLoading
+    def Object.const_missing name
+      $LOAD_PATH << File.join("./","app","controllers")  unless  $LOAD_PATH.include?("./app/controllers")
+      $LOAD_PATH << File.join("./","app","models") unless  $LOAD_PATH.include?("./app/models")
+      require Yrpc::Ruler.to_underscore(name.to_s)
+      klass = Object.const_get(name)
+      klass
+    end
+
+  end
+
+end

data/lib/yrpc/interceptors/active_record.rb

@@ -0,0 +1,26 @@
+module Yrpc
+  module Interceptors
+    module ActiveRecord
+      class ResetConnection
+
+        def invoke
+          if enabled? && !::ActiveRecord::Base.connection.active?
+            ::ActiveRecord::Base.establish_connection
+          end
+          yield
+        ensure
+          ::ActiveRecord::Base.clear_active_connections! if enabled?
+        end
+
+        private
+
+        ##
+        # @return [Boolean] If AR is loaded, we can enable this hook safely
+        #
+        def enabled?
+          defined?(::ActiveRecord::Base)
+        end
+      end
+    end
+  end
+end

data/lib/yrpc/interceptors/base.rb

@@ -0,0 +1,19 @@
+module Yrpc
+  module Interceptors
+    class Base
+
+      attr_reader :request
+
+      attr_reader :error
+
+      attr_reader :options
+
+      def initialize(request, error, options = {})
+        @request = request
+        @error = error
+        @options = options || {}
+      end
+
+    end
+  end
+end

data/lib/yrpc/interceptors/client_interceptor.rb

@@ -0,0 +1,97 @@
+# Copyright (c) 2017-present, BigCommerce Pty. Ltd. All rights reserved
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+# documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit
+# persons to whom the Software is furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+# Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+# OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+module Yrpc
+  module Interceptors
+    ##
+    # Intercepts outbound client requests to provide a unified interface and request context
+    #
+    class ClientInterceptor < GRPC::ClientInterceptor
+
+
+      ##
+      # Handles interception of outbound calls. Implement this in your derivative interceptor implementation.
+      #
+      # @param [Yrpc::Outbound::RequestContext] request_context The context of the outbound request
+      # @return [Object] This method must return the response from the yielded block
+      #
+      def call(request_context:)
+        Yrpc.logger "Logging client interceptor for request: #{request_context.method}"
+        yield
+      end
+
+      ##
+      # Call the interceptor from the request_response call
+      #
+      # @param [Object] request The request being sent
+      # @param [GRPC::ActiveCall] call The GRPC ActiveCall object
+      # @param [Method] method The method being called
+      # @param [Hash] metadata A hash of outgoing metadata
+      # @return [Object] The response message
+      #
+      def request_response(request: nil, call: nil, method: nil, metadata: nil)
+        rc = Yrpc::Outbound::RequestContext.new(type: :request_response, requests: [request], call: call, method: method, metadata: metadata)
+        call(request_context: rc)
+        matadata= metadata.merge!(rc.metadata)
+        yield
+      end
+
+      ##
+      # Call the interceptor from the client_streamer call
+      #
+      # @param [Enumerable] requests An enumerable of requests being sent
+      # @param [GRPC::ActiveCall] call The GRPC ActiveCall object
+      # @param [Method] method The method being called
+      # @param [Hash] metadata A hash of outgoing metadata
+      # @return [Object] The response message
+      #
+      def client_streamer(requests: nil, call: nil, method: nil, metadata: nil)
+        rc = Yrpc::Outbound::RequestContext.new(type: :client_streamer, requests: requests, call: call, method: method, metadata: metadata)
+        call(request_context: rc)
+        matadata= metadata.merge!(rc.metadata)
+        yield
+      end
+
+      ##
+      # Call the interceptor from the server_streamer call
+      #
+      # @param [Object] request The request being sent
+      # @param [GRPC::ActiveCall] call The GRPC ActiveCall object
+      # @param [Method] method The method being called
+      # @param [Hash] metadata A hash of outgoing metadata
+      # @return [Object] The response message
+      #
+      def server_streamer(request: nil, call: nil, method: nil, metadata: nil)
+        rc = Yrpc::Outbound::RequestContext.new(type: :server_streamer, requests: [request], call: call, method: method, metadata: metadata)
+        call(request_context: rc)
+        yield
+      end
+
+      ##
+      # Call the interceptor from the bidi_streamer call
+      #
+      # @param [Enumerable] requests An enumerable of requests being sent
+      # @param [GRPC::ActiveCall] call The GRPC ActiveCall object
+      # @param [Method] method The method being called
+      # @param [Hash] metadata A hash of outgoing metadata
+      #
+      def bidi_streamer(requests: nil, call: nil, method: nil, metadata: nil)
+        rc = Yrpc::Outbound::RequestContext.new(type: :bidi_streamer, requests: requests, call: call, method: method, metadata: metadata)
+        call(request_context: rc)
+        yield
+      end
+    end
+  end
+end

data/lib/yrpc/interceptors/register.rb

@@ -0,0 +1,28 @@
+module Yrpc
+  module Interceptors
+    class Registry
+
+      def initialize
+        @interceptors ||= []
+      end
+
+      def use(interceptor_class, options = {})
+        interceptors_mutex do
+          @interceptors << {
+              klass: interceptor_class,
+              options: options
+          }
+        end
+      end
+
+      #ruby 互斥锁，防止线程对资源进行竞争，导致资源不同步
+      def interceptors_mutex(&block)
+        @interceptors_mutex ||= begin
+          require 'monitor'
+          Monitor.new
+        end
+        @interceptors_mutex.synchronize(&block)
+      end
+    end
+  end
+end

data/lib/yrpc/outbound/request_context.rb

@@ -0,0 +1,69 @@
+# Copyright (c) 2017-present, BigCommerce Pty. Ltd. All rights reserved
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+# documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
+# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit
+# persons to whom the Software is furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
+# Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
+# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+# OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+module Yrpc
+  module Outbound
+    ##
+    # Encapsulates the context of an outbound client request
+    #
+    class RequestContext
+      # @var [Symbol]
+      attr_reader :type
+      # @var [Enumerable] requests
+      attr_reader :requests
+      # @var [GRPC::ActiveCall]
+      attr_reader :call
+      # @var [Method] method
+      attr_reader :method
+      # @var [Hash] metadata
+      attr_accessor :metadata
+
+      ##
+      # Initialize the new request context
+      #
+      # @param [Symbol] type The type of request
+      # @param [Enumerable] requests An enumerable of requests being sent
+      # @param [GRPC::ActiveCall] call The GRPC ActiveCall object
+      # @param [Method] method The method being called
+      # @param [Hash] metadata A hash of outgoing metadata
+      #
+      def initialize(type:, requests:, call:, method:, metadata:)
+        @type = type
+        @requests = requests
+        @call = call
+        @method = method
+        @metadata = metadata
+      end
+
+      ##
+      # Return the name of the method being called, e.g. GetThing
+      #
+      # @return [String]
+      #
+      def method_name
+        @method.to_s.split('/').last
+      end
+
+      ##
+      # Return the proper routing key for the request
+      #
+      # @return [String]
+      #
+      def route_key
+        @method[1..-1].underscore.tr('/', '.')
+      end
+    end
+  end
+end

data/lib/yrpc/server.rb

@@ -0,0 +1,82 @@
+module Yrpc
+  class Server
+
+    def initialize(options = {})
+      @options = options || {}
+      @interceptors = options.fetch(:interceptor_registry, Yrpc.interceptors)
+      @interceptors = Yrpc::Interceptors::Registry.new unless @interceptors.is_a?(Yrpc::Interceptors::Registry)
+      @services = []
+      @started = false
+      @stop_server = false
+      @stop_server_cv = ConditionVariable.new
+      @stop_server_mu = Monitor.new
+      @server_mu = Monitor.new
+      @hostname = options.fetch(:hostname, Yrpc.server_binding_url)
+      load_models
+      load_controllers
+    end
+
+    attr_reader :options
+
+    def controllers_path
+      options.fetch(:controllers_path, Yrpc.controllers_path)
+    end
+
+    def grpc_server
+      # 互斥锁 操作相同的资源
+      @server_mu.synchronize do
+        @server ||= begin
+          server = GRPC::RpcServer.new(@options)
+          @port = server.add_http2_port(@hostname, ssl_credentials)
+          @services.each do |service|
+            server.handle(service)
+          end
+          server
+        end
+      end
+    end
+
+    def add_service(service)
+      @services << service unless @services.include?(service)
+    end
+
+    def ssl_credentials
+      if options.fetch(:use_ssl, Yrpc.use_ssl)
+        private_key = File.read(options.fetch(:ssl_key_file, Yrpc.ssl_key_file))
+        cert_chain = File.read(options.fetch(:ssl_crt_file, Yrpc.ssl_crt_file))
+        certs = [nil, [{private_key: private_key, cert_chain: cert_chain}], false]
+        GRPC::Core::ServerCredentials.new(*certs)
+      else
+        :this_port_is_insecure
+      end
+    end
+
+    def load_controllers
+      return unless File.directory?(controllers_path)
+      path = File.realpath(controllers_path)
+      $LOAD_PATH.unshift(path)
+      p path
+      Dir["#{path}/**/*.rb"].each do |f|
+        # 把pb文件给exclue
+        next if f.include?('_pb')
+        Yrpc.logger.info "- Loading gRPC service file: #{f}"
+        load File.realpath(f)
+      end
+    end
+
+    def load_models
+      return unless File.directory?("app/models")
+      path = File.realpath("app/models")
+      $LOAD_PATH.unshift(path)
+      Dir["#{path}/**/*.rb"].each do |f|
+        # 把pb文件给exclue
+        next if f.include?('_pb')
+        Yrpc.logger.info "- Loading gRPC Model file: #{f}"
+        load File.realpath(f)
+      end
+    end
+
+  end
+
+
+end

data/lib/yrpc/version.rb

@@ -0,0 +1,3 @@
+module Yrpc
+  VERSION = "0.1.4"
+end

data/yrpc.gemspec

@@ -0,0 +1,35 @@
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "yrpc/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "ampedxx-yrpc"
+  spec.version = Yrpc::VERSION
+  spec.authors = ["icepoint0"]
+  spec.email = ["351711778@qq.com"]
+
+  spec.summary = "yprc"
+  spec.description = "yrpc"
+  spec.homepage = "http://ypr.com"
+  spec.license = "MIT"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+
+  spec.files = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "bin"
+  spec.executables   = ["yrpc-cli"]
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency 'bundler', '~> 1.11'
+  spec.add_development_dependency 'rake', '~> 10.0'
+
+  spec.add_runtime_dependency 'grpc', '~> 1.10'
+  spec.add_runtime_dependency 'grpc-tools', '~> 1.10'
+  spec.add_runtime_dependency 'activesupport', '~> 5.0'
+  spec.add_runtime_dependency 'slop', '~> 4.6'
+end

metadata

@@ -0,0 +1,166 @@
+--- !ruby/object:Gem::Specification
+name: ampedxx-yrpc
+version: !ruby/object:Gem::Version
+  version: 0.1.4
+platform: ruby
+authors:
+- icepoint0
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-06-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: grpc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: grpc-tools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: slop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.6'
+description: yrpc
+email:
+- 351711778@qq.com
+executables:
+- yrpc-cli
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".idea/.rakeTasks"
+- ".idea/inspectionProfiles/Project_Default.xml"
+- ".idea/misc.xml"
+- ".idea/modules.xml"
+- ".idea/vcs.xml"
+- ".idea/workspace.xml"
+- ".idea/yrpc.iml"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- bin/yrpc-cli
+- lib/yrpc.rb
+- lib/yrpc/client.rb
+- lib/yrpc/configuration.rb
+- lib/yrpc/controllers/base.rb
+- lib/yrpc/controllers/request.rb
+- lib/yrpc/controllers/response.rb
+- lib/yrpc/controllers/service_binder.rb
+- lib/yrpc/error/response_error.rb
+- lib/yrpc/executor/executor.rb
+- lib/yrpc/helpers/autoload.rb
+- lib/yrpc/interceptors/active_record.rb
+- lib/yrpc/interceptors/base.rb
+- lib/yrpc/interceptors/client_interceptor.rb
+- lib/yrpc/interceptors/register.rb
+- lib/yrpc/outbound/request_context.rb
+- lib/yrpc/server.rb
+- lib/yrpc/version.rb
+- yrpc.gemspec
+homepage: http://ypr.com
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.12
+signing_key: 
+specification_version: 4
+summary: yprc
+test_files: []