欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Ruby中区分运行来源的方法

程序员文章站 2022-05-26 07:56:26
当我们在写模块的时候,或多或少需要直接运行这个文件也可以执行一些方法,但是这样对于当这个模块被require或者include时,显得不好,在ruby里,有没有区分运行来自...

当我们在写模块的时候,或多或少需要直接运行这个文件也可以执行一些方法,但是这样对于当这个模块被require或者include时,显得不好,在ruby里,有没有区分运行来自当前文件,还是被require的目标文件调用呢?

python可以

比如像python这样

复制代码 代码如下:

if __name__ == '__main__':
    print "from direct running"

ruby当然也可以

对于处处为程序员着想,拥有快乐编程理念的ruby来说当然是可以区别的。其原理就是判断启动文件是否为模块的代码文件。

复制代码 代码如下:

if __file__ == $0
    puts 'called from direct running'
end

举个例子

工具类模块utils.rb

复制代码 代码如下:

module utils
    class stringutils
        def self.test
            puts "test method myfile=" + __file__ + ';load from ' +  $0
        end
    end
end

if __file__ == $0
    puts 'called from direct running'
    utils::stringutils.test()
end

直接运行,结果,if条件成立,执行了输出

复制代码 代码如下:

20:04:37-androidyue~/rubydir/test$ ruby utils.rb
called from direct running
test method myfile=utils.rb;load from utils.rb

引用utils的类test.rb

复制代码 代码如下:

require './utils'
utils::stringutils.test()

运行结果,引入模块的条件不成立,没有输出called from direct running

复制代码 代码如下:

20:08:07-androidyue~/rubydir/test$ ruby test.rb
test method myfile=/home/androidyue/rubydir/test/utils.rb;load from test.rb