如何在套接字关闭时让Kernel#select提前退出？

我编写了一个简单的Socket应用：

# Server
server = TCPServer.new(2000)
loop do
  client = server.accept
  sleep 10
end

# Client
s = TCPSocket.new('localhost', 2000)
th1 = Thread.new do
  begin
    Kernel.select([s], nil, [s], nil)
    puts 'Select returned'
  rescue => e
    puts "Select raised #{e}"
  end
end
th2 = Thread.new do
  sleep 0.5
  s.close
end
[th1, th2].map(&:join)

期望行为：当客户端线程在等待服务器响应时，若另一线程关闭了客户端套接字，Kernel#select调用应立即退出并抛出错误。该行为在macOS上正常，但在Linux上select会无限挂起（或直到超时）。

请问：

1. 有没有解决该问题的方法？
2. 我认为需要使用poll(2)系统调用，但Ruby中似乎没有提供该接口？

问题1：解决Linux下select挂起的方法 #

Linux与macOS在套接字关闭时对select的处理存在内核层面的差异：macOS会触发select立即返回并抛出错误，而Linux中，套接字被其他线程关闭后，select不会主动感知状态变化，会持续阻塞。

可以通过以下两种可靠方式解决：

• 管道中断法：创建一对管道，将读端加入select的监听列表，当需要中断select时，向写端写入任意数据，触发select即时返回。这种方式无需轮询，响应效率高：

# 初始化管道
rd, wr = IO.pipe

s = TCPSocket.new('localhost', 2000)
th1 = Thread.new do
  begin
    ready = Kernel.select([s, rd], nil, [s, rd], nil)
    if ready[0].include?(rd)
      puts 'Interrupted by pipe'
    elsif ready[0].include?(s)
      puts 'Select returned'
    end
  rescue => e
    puts "Select raised #{e}"
  ensure
    rd.close
    wr.close
  end
end
th2 = Thread.new do
  sleep 0.5
  s.close
  wr.write(' ') # 写入数据触发select返回
end
[th1, th2].map(&:join)

• 超时轮询+状态检查：给select设置较短的超时时间，每次超时后主动检查套接字是否已关闭，若关闭则退出循环：

th1 = Thread.new do
  begin
    loop do
      ready = Kernel.select([s], nil, [s], 0.1) # 0.1秒超时
      if ready
        puts 'Select returned'
        break
      end
      # 主动检查套接字状态
      break if s.closed?
    end
  rescue => e
    puts "Select raised #{e}"
  end
end

问题2：Ruby中是否有poll(2)接口 #

Ruby标准库确实没有直接暴露poll(2)系统调用，但可以通过两种方式间接实现类似效果：

• 使用IO#wait方法：Ruby的IO#wait底层在Linux环境下会调用poll(2)，它能在套接字就绪、出错或关闭时返回，适合替代select：

th1 = Thread.new do
  begin
    # wait(:read)会监听读事件，套接字关闭时返回nil
    result = s.wait(:read)
    if result.nil?
      puts 'Socket closed'
    else
      puts 'Wait returned'
    end
  rescue => e
    puts "Wait raised #{e}"
  end
end

• 使用第三方高效IO库：比如nio4r，它封装了Linux的epoll、macOS的kqueue等多路复用接口，比select性能更优，且能正确处理多线程下的套接字关闭场景。需先安装：gem install nio4r，示例代码：

require 'nio'

s = TCPSocket.new('localhost', 2000)
selector = NIO::Selector.new
selector.register(s, :r)

th1 = Thread.new do
  begin
    selector.select do |monitor|
      puts 'NIO selector returned'
    end
  rescue => e
    puts "NIO raised #{e}"
  ensure
    selector.close
  end
end
th2 = Thread.new do
  sleep 0.5
  s.close
  selector.wakeup # 主动唤醒selector
end
[th1, th2].map(&:join)

内容的提问来源于stack exchange，提问作者Neil Shweky