长时间闲置后如何让irb退出?

时间:2014-03-28 10:58:19

标签: ruby irb

我是一名DBA,我偶然发现了这样的情况:开发人员运行irb会话(来自Ruby on Rails应用程序)。此irb保持数据库连接打开。有时候 - 他们忘记了它,它继续“运行” - 不做任何事情,但仍然使用一个数据库连接。

我想在他们的irb配置中添加某种“空闲超时”。可能吗?怎么做?

1 个答案:

答案 0 :(得分:1)

在这里,您可以快速了解如何实现这一点。

请注意,这并未考虑用户可能正在irb会话中执行一些长时间运行的任务。它只是查看最后一个输入的时间戳;如果它没有改变那么它只是平坦的杀死过程:

更新:它现在检查irb当前是否正在运行命令并忽略任何超时(如果是这种情况)。

# Add some methods to IRB::Context and IRB::Irb
# for easier timeout implementation.
class IRB::Irb
  def status
    @signal_status
  end
end

class IRB::Context
  attr_reader :irb
  attr_reader :line_no

  def is_evaluating?
    self.irb.status == :IN_EVAL
  end
end

# Implement an auto exit timer thread. Timeout is given in seconds.
module IRB
  def self.auto_exit_after(timeout = 60)
    Thread.new {
      context    = IRB.conf[:MAIN_CONTEXT]
      last_input = Time.now
      last_line  = context.line_no

      loop {
        sleep 10

        # Check if irb is running a command
        if context.is_evaluating?
          # Reset the input time and ignore any timeouts
          last_input = Time.now
          next
        end

        # Check for new input
        if last_line != context.line_no
          # Got new input
          last_line  = context.line_no
          last_input = Time.now
          next
        end

        # No new input, check if idle time exceeded
        if Time.now - last_input > timeout    
          $stderr.puts "\n** IRB exiting due to idle timeout. Goodbye..."
          Process.kill("KILL", Process.pid)
        end
      }
    }
  end
end

要使用它,请将代码添加到.irbrc,或者在启动irb时自动加载的其他位置,然后启动计时器:

IRB.auto_exit_after(60)