Ruby套接字继承

Question

我正在写一个扩展TCPSocket的Ruby类。 假设它看起来像这样：

class FooSocket < TCPSocket
    def hello
        puts 'hello'
    end
end

我有一个TCPServer监听传入的连接

server = TCPServer.new 1234
socket = server.accept

当我的服务器最终接受连接时，它将返回TCPSocket 。 但是，我想要一个FooSocket以便可以调用socket.hello 。

如何将TCPSocket更改为FooSocket ？

我可以直接将想要的方法和属性直接插入TCPSocket类，但是我在其他地方使用了它，所以我不想这样做。

可能最简单的解决方案是编写一个封装TCPSocket的类，然后将accept返回的套接字作为参数传递。 但是，我很想知道如何通过继承来实现它-我一直在试图绕开它，但无法弄清楚。

提前致谢。

Answer 1

我认为最简单的方法是将FooSocket一个模块，然后extend其extend到需要特殊方法的TCPSocket 。 以下是一些如何使用它的示例：

module FooSocket
  def hello
    self.puts 'hello' # could also be written without the 'self',
  end                 # but it is clearer this way because it may
end                   # be mistaken for the Kernel method.

server = TCPServer.new 1234
socket = server.accept.extend FooSocket

您还可以轻松地重写TCPServer以返回以下其中一项：

class TCPServer
  alias old_accept accept
  def accept
    old_accept.extend FooSocket
  end
end

如果您不想覆盖标准类，则可以在服务器上定义一个单例类（这是我要做的），因为您很可能只有一台服务器。 为此，将上面示例中的行class TCPServer替换为class << server 。

Answer 2

您需要在TCPServer类中重写accept ，以通过重新打开类或子类并使用它来返回FooSocket而不是TCPSocket 。 我认为，在accept返回的TCPSocket实例后，对它进行装扮的想法比较干净。