ruby - 仅当从命令行调用脚本时才运行代码

Question

所以当我从命令行调用我的脚本时，我希望它接受一个 int 并使用该值做一些事情:

ruby script.rb


puts ARGV[0], etc...

但是，每当加载或需要脚本而不是从命令行调用脚本时，我想完全跳过这部分代码。如何检测脚本是通过命令行调用还是刚刚加载？谢谢!

Answer 1

通常将它放在脚本的底部:

if __FILE__==$0
  # this will only run if the script was the main, not load'd or require'd
end

因为我喜欢在文件顶部看到主要操作，所以我通常将 def run! 作为文件中的第一个方法，然后以以下方式结束文件:

run! if __FILE__==$0