拦截Kernel的puts方法

来源：互联网发布：nba个人数据统计编辑：程序博客网时间：2024/06/06 04:24

最近在对selenium做一个基于ruby的封装, 遇到到这样的需求:

irb(main):001:0> puts '一条日志'一条日志=> nilirb(main):002:0> def log(arg)irb(main):003:1> puts "#{Time.now} #{arg}"irb(main):004:1> end=> nilirb(main):005:0> log('一条日志')2014-12-12 15:53:11 +0800 一条日志=> nil

我们经常用log('xxxx')的方式为信息添加统一的装饰, 但是这样仍需要编写脚本的工作人员了解统一的输出格式. 所以为了减少脚本开发人员的工作量，我尝试改写常用的输出方法 'puts'.

irb(main):007:0> method 'puts'=> #<Method: Object(Kernel)#puts>

用method方法查看一下puts方法在Kernel下. 但是很遗憾, 并不能通过class Kernel 这样的方式来重写puts.

可是既然我们获得到了 puts 的Method对象, 那么这个对象一定可以做点什么.

通常来说我们可以像这样动态建立一个方法

irb(main):016:0> define_method(:shout) do |*args|irb(main):017:1* args.each do |arg|irb(main):018:2* puts arg.upcaseirb(main):019:2> endirb(main):020:1> end=> #<Proc:0x000000034c6b40@(irb):16 (lambda)>irb(main):021:0> shout 'hi, alex'HI, ALEX=> ["hi, alex"]

接着我们干掉puts

irb(main):028:0> undef puts=> nilNoMethodError: undefined method `puts' for main:Object        from (irb):29        from C:/software/Ruby200-x64/bin/irb:12:in `<main>'

然后重新建立puts方法

irb(main):033:0> define_method(:puts) do |*args|irb(main):034:1* args.each do |arg|irb(main):035:2* Kernel.puts "irb(main):036:2" ^Cirb(main):036:0> define_method(:puts) do |*args|irb(main):037:1* args.each do |arg|irb(main):038:2* Kernel.puts "#{Time.now} #{arg}"irb(main):039:2> endirb(main):040:1> end=> #<Proc:0x000000030e4e78@(irb):6 (lambda)>irb(main):041:0> puts 12014-12-12 16:59:31 +0800 1=> [1]

便成功的在pust时添加了时间戳.

PS : 也可以直接给Method和某个方法名建立个关联, 如下:

irb(main):025:0> define_method(:kernel_puts, method('puts'))=> #<Method: Object(Kernel)#puts>irb(main):026:0> kernel_puts 'yo'yo=> nil

0 0