module RPC
require 'rack/rpc'
#require rpc/*.rb文件
Dir.glob(File.join(File.dirname(__FILE__), 'rpc', "*.rb")) do |file|
require file
end
class Runner < Rack::RPC::Server
#include rpc/*.rb and regsiter rpc call
#eg. rpc/god.rb god.hello
@@rpc_list = []
Dir.glob(File.join(File.dirname(__FILE__), 'rpc', "*.rb")) do |file|
rpc_class = File.basename(file).split('.rb')[0].capitalize
rpc_list = []
#加载module下的方法到Runner这个类下面
eval "include Frigga::RPC::#{rpc_class}"
#获取声明的RPC接口
eval "rpc_list = Frigga::RPC::#{rpc_class}::RPC_LIST"
rpc_list.each do |rpc_name|
#alias一个新的rpc方法,叫old_xxxx_xxxx
eval "alias :old_#{rpc_class.downcase}_#{rpc_name} :#{rpc_name}"
#重新定义rpc方法,添加一行日志打印功能,然后再调用old_xxxx_xxxx rpc方法
define_method "#{rpc_class.downcase}_#{rpc_name}".to_sym do |*arg|
Logger.info "[#{request.ip}] called #{rpc_class.downcase}.#{rpc_name} #{arg.join(', ')}"
eval "old_#{rpc_class.downcase}_#{rpc_name} *arg"
end
#注册RPC调用
rpc "#{rpc_class.downcase}.#{rpc_name}" => "#{rpc_class.downcase}_#{rpc_name}".to_sym
#添加到全局变量,汇总所有的rpc方法
@@rpc_list << "#{rpc_class.downcase}.#{rpc_name}"
end
end
def help
rpc_methods = (['help'] + @@rpc_list.sort).join("\n")
end
rpc "help" => :help
end
end #RPC