Question

我有一个可以在两台计算机上运行的应用程序X，但不能同时运行一台。我有另一个应用程序Y，用Python编写，给出两个可能的IP地址需要找出哪个计算机正在运行应用程序X（如果有的话）。我通过拥有一个侦听端口的UDP服务并在收到一些数据时以“Hello”响应来部分地解决了这个问题。客户端可以尝试将数据发送到每个地址的应用程序X端口，如果得到响应，我知道应用程序正在该计算机上运行。

到目前为止我的代码看起来像这样：

 def ipaddress(self):
    """Test which side responds on the status port."""
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    try:
        s.settimeout(5)
        s.sendto("status", (ADDR_A, PORT))
        s.recvfrom(1024)
    except socket.timeout:
        try:
            s.sendto("status", (ADDR_B, PORT))
            s.recvfrom(1024)
        except:
            pass
        else:
            return ADDR_B
    else:
        return ADDR_A
    finally:
        s.close()
    return None

这个函数的问题在于，每当我想要与运行应用程序X的计算机对话时，它都会定期调用。它将始终首先测试ADDR_A，如果它没有运行应用程序X，那么我必须等待套接字超时尝试ADDR_B。虽然经常不会发生，但每当我再次尝试时，应用程序X可能会切换计算机。

有更好的方法吗？我想知道是否有可能并行连接到两台计算机并在响应时立即返回？或者我应该在上次调用该函数时缓存哪个IP地址首先响应？我该如何编写这些或其他想法？

感谢。

编辑：这是我使用select修改后的代码：

def ipaddress(addr_a, addr_b, timeout=5):
    """Test which side responds on the status port."""

    # Create UDP sockets for each address 
    socks = [ socket.socket(socket.AF_INET, socket.SOCK_DGRAM),
              socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
              ]

    # Send some data to each socket 
    for sock, addr in zip(socks, (addr_a, addr_b)):
        sock.connect(addr)  # do explicit connect so getpeername works
        sock.send("status")

    # Wait for the first to respond if any
    while socks:
        waiting = select.select(socks, [], socks, timeout)[0]
        if waiting:
            for sock in waiting:
                try:
                    data = sock.recv(1024)
                    if data:
                        return sock.getpeername()[0]
                except Exception, e:
                    # Occasionally get [Errno 10054] which means socket isn't really
                    # available, so see if other responds instead...
                    socks.remove(sock)
        else:
            break   # timeout occurred
    return None

Answer 1

您应该查看select.select()，它提供了您正在寻找的并行查看两台计算机的功能。

同时测试一对网络套接字

1 个答案: