Question

如何检查Ruby中的当前脚本是否正在关闭？特别是，如果程序正在关闭，我想将@reconnect设置为false，而不是允许web-socket重新连接。我试过了Signal.trap("TERM")，但它似乎没有用。

@reconnect是WebsocketClient类中的一个实例变量，我无法在课外的脚本中直接更改它。

class WebsocketClient
    def ws_closed(event)
      $logger.warn "WS CLOSED"
      Signal.trap("TERM") { 
        @stop = true
        @reconnect = false
      }

      unless $reauth
        if @stop
          EM.stop
        elsif @reconnect
          $logger.warn "Reconnecting..."
          EM.add_timer(@reconnect_after){ connect! }
        end
      end
    end
end

at_exit {
  $logger.fatal "Application terminated. Shutting down gracefully..."
  # ...
  # Do some exit work...
  # ...
  exit!
}

CTRL-C上的输出

01-02-2018 12:00:54.59  WARN    > WS CLOSED
01-02-2018 12:00:54.595 WARN    > Reconnecting...
01-02-2018 12:00:54.596 FATAL   > Application terminated. Shutting down gracefully..

Answer 1

请参阅以下摘录Here，但与您目前所附的问题相比，您的问题似乎更为恰当：

你最好的选择可能比信号陷阱更容易。 Kernel Module实际上为您提供了#at_exit方法，该方法将在程序实际退出之前执行。

用法:(来自Kernel#at_exit Docs）

def do_at_exit(str1)
  at_exit { print str1 }
end
at_exit { puts "cruel world" }
do_at_exit("goodbye ")
exit

“生产：”

goodbye cruel world

如您所见，您可以定义多个处理程序，这些处理程序将在程序退出时以相反的顺序执行。

由于Kernel中包含Object，您可以处理Object具体信息，例如

class People
  at_exit {puts "The #{self.name} have left"}
end
exit
# The People have left

甚至是实例

p = People.new
p.send(:at_exit, &->{puts "We are leaving"})
# We are leaving
# The People have left

此外，对于基于Object的更具体的实施，您可以查看ObjectSpace.define_finalizer。

用法示例：

class Person
  def self.finalize(name)
    proc {puts "Goodbye Cruel World -#{name}"}
  end 
  def initialize(name)
    @name = name
    ObjectSpace.define_finalizer(self, self.class.finalize(@name))
  end
end

用法：

p = Person.new("engineersmnky")
exit
# Goodbye Cruel World -engineersmnky

这可能不是你想要的特别之处，因为当Object被垃圾收集时也会触发（对于短暂的对象来说不是很好）但是如果你有整个应用程序应该存在的对象，那么这可能仍然是与at_exit类似。实施例

# requiring WeakRef to allow garbage collection 
# See: https://ruby-doc.org/stdlib-2.3.3/libdoc/weakref/rdoc/WeakRef.html
require 'weakref' # 
p1 = Person.new("Engineer")
p2 = Person.new("Engineer's Monkey")
p2 = WeakRef.new(p2)
GC.start # just for this example
# Goodbye Cruel World -Engineer's Monkey
#=> nil
p2
#=> WeakRef::RefError: Invalid Reference - probably recycled
exit
# Goodbye Cruel World -Engineer

正如您所看到的p2已定义的终结器已被触发，因为Person已被gc'd但该程序尚未退出。 p1的终结器等待直到退出，因为它在整个应用程序中保留了它的引用。

Ruby检查程序当前是否正在关闭

1 个答案: