拦截Kernel的puts方法

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

irb(main):001:0> puts '一条日志'
一条日志
=> nil
irb(main):002:0> def log(arg)
irb(main):003:1> puts "#{Time.now} #{arg}"
irb(main):004:1> end
=> nil
irb(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.upcase
irb(main):019:2> end
irb(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
=> nil
NoMethodError: 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" ^C
irb(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> end
irb(main):040:1> end
=> #<Proc:0x000000030e4e78@(irb):6 (lambda)>
irb(main):041:0> puts 1
2014-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