Merge pull request #1517 from tbetbetbe/grpc-ruby-dont-extend-string

make underscore a class method of GenericService
pull/1485/head
Michael Lumish 10 years ago
commit b1f3104815
  1. 5
      src/ruby/lib/grpc/generic/rpc_server.rb
  2. 38
      src/ruby/lib/grpc/generic/service.rb
  3. 18
      src/ruby/spec/generic/service_spec.rb

@ -468,10 +468,11 @@ module GRPC
route = "/#{cls.service_name}/#{name}".to_sym route = "/#{cls.service_name}/#{name}".to_sym
fail "already registered: rpc #{route} from #{spec}" if specs.key? route fail "already registered: rpc #{route} from #{spec}" if specs.key? route
specs[route] = spec specs[route] = spec
rpc_name = GenericService.underscore(name.to_s).to_sym
if service.is_a?(Class) if service.is_a?(Class)
handlers[route] = cls.new.method(name.to_s.underscore.to_sym) handlers[route] = cls.new.method(rpc_name)
else else
handlers[route] = service.method(name.to_s.underscore.to_sym) handlers[route] = service.method(rpc_name)
end end
logger.info("handling #{route} with #{handlers[route]}") logger.info("handling #{route} with #{handlers[route]}")
end end

@ -30,24 +30,6 @@
require 'grpc/generic/client_stub' require 'grpc/generic/client_stub'
require 'grpc/generic/rpc_desc' require 'grpc/generic/rpc_desc'
# Extend String to add a method underscore
class String
# creates a new string that is the underscore separate version of this one.
#
# E.g,
# PrintHTML -> print_html
# AMethod -> a_method
# AnRpc -> an_rpc
def underscore
word = dup
word.gsub!(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
word.gsub!(/([a-z\d])([A-Z])/, '\1_\2')
word.tr!('-', '_')
word.downcase!
word
end
end
# GRPC contains the General RPC module. # GRPC contains the General RPC module.
module GRPC module GRPC
# Provides behaviour used to implement schema-derived service classes. # Provides behaviour used to implement schema-derived service classes.
@ -55,6 +37,22 @@ module GRPC
# Is intended to be used to support both client and server # Is intended to be used to support both client and server
# IDL-schema-derived servers. # IDL-schema-derived servers.
module GenericService module GenericService
# creates a new string that is the underscore separate version of s.
#
# E.g,
# PrintHTML -> print_html
# AMethod -> a_method
# AnRpc -> an_rpc
#
# @param s [String] the string to be converted.
def self.underscore(s)
s.gsub!(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
s.gsub!(/([a-z\d])([A-Z])/, '\1_\2')
s.tr!('-', '_')
s.downcase!
s
end
# Used to indicate that a name has already been specified # Used to indicate that a name has already been specified
class DuplicateRpcName < StandardError class DuplicateRpcName < StandardError
def initialize(name) def initialize(name)
@ -171,7 +169,7 @@ module GRPC
# Used define_method to add a method for each rpc_desc. Each method # Used define_method to add a method for each rpc_desc. Each method
# calls the base class method for the given descriptor. # calls the base class method for the given descriptor.
descs.each_pair do |name, desc| descs.each_pair do |name, desc|
mth_name = name.to_s.underscore.to_sym mth_name = GenericService.underscore(name.to_s).to_sym
marshal = desc.marshal_proc marshal = desc.marshal_proc
unmarshal = desc.unmarshal_proc(:output) unmarshal = desc.unmarshal_proc(:output)
route = "/#{route_prefix}/#{name}" route = "/#{route_prefix}/#{name}"
@ -207,7 +205,7 @@ module GRPC
# implemented. # implemented.
def assert_rpc_descs_have_methods def assert_rpc_descs_have_methods
rpc_descs.each_pair do |m, spec| rpc_descs.each_pair do |m, spec|
mth_name = m.to_s.underscore.to_sym mth_name = GenericService.underscore(m.to_s).to_sym
unless instance_methods.include?(mth_name) unless instance_methods.include?(mth_name)
fail "#{self} does not provide instance method '#{mth_name}'" fail "#{self} does not provide instance method '#{mth_name}'"
end end

@ -56,15 +56,6 @@ end
GenericService = GRPC::GenericService GenericService = GRPC::GenericService
Dsl = GenericService::Dsl Dsl = GenericService::Dsl
describe 'String#underscore' do
it 'should convert CamelCase to underscore separated' do
expect('AnRPC'.underscore).to eq('an_rpc')
expect('AMethod'.underscore).to eq('a_method')
expect('PrintHTML'.underscore).to eq('print_html')
expect('PrintHTMLBooks'.underscore).to eq('print_html_books')
end
end
describe Dsl do describe Dsl do
it 'can be included in new classes' do it 'can be included in new classes' do
blk = proc { Class.new { include Dsl } } blk = proc { Class.new { include Dsl } }
@ -73,6 +64,15 @@ describe Dsl do
end end
describe GenericService do describe GenericService do
context '#underscore' do
it 'should convert CamelCase to underscore separated' do
expect(GenericService.underscore('AnRPC')).to eq('an_rpc')
expect(GenericService.underscore('AMethod')).to eq('a_method')
expect(GenericService.underscore('PrintHTML')).to eq('print_html')
expect(GenericService.underscore('SeeHTMLBooks')).to eq('see_html_books')
end
end
describe 'including it' do describe 'including it' do
it 'adds a class method, rpc' do it 'adds a class method, rpc' do
c = Class.new do c = Class.new do

Loading…
Cancel
Save