防止在Python中

Question

关闭连接套接字我有简单的函数，像一个客户。它发送一个事件消息并等待一个确认（为简单起见，我删除了该部分）。之后，它会得到接收更多的消息：

def send_data(message): 
    """Connect to server and get requests""" 
    sock = socket.socket() 
    host = 'localhost' 
    port = 8000 
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    sock.connect((host,port)) 

    try: 
     if message: 
      sock.sendall(message) 

     # Look for the response 
     has_request = False 
     while not has_request: 
      data = sock.recv(1024) 

      #First I get ACK, code not listed here... 
      #Next iteration, I got my request message 
      has_request = True 

    finally: 
     print >> sys.stderr, 'Closing socket' 
     sock.close() 

    return data 

我称这种现象为__main__声明：

if __name__ == '__main__': 
    server_data = send_data(event) 

    if server_data: 
     parse_request(server_data) 

不过，我需要防止插座被自动关闭。因为我在分析请求后必须发送另一个消息。我删除了sock.close（），但套接字仍然关闭。这是因为我在回应数据，而且我不在函数范围内？

我只是想连接，然后接收和发送的消息没有任何问题，我在程序的任何时间。

Answer 1

当你走在send_data方法外，你的插座sock将不再被任何人引用，所以垃圾收集器会做他的工作，并删除对象，这将因此关闭套接字（即使没有close()调用）。

如果你回到你的数据Socket对象，并把它在你的主要部分，插座将仍然被引用，而不是关闭。

def create_socket(): 
    sock = socket.socket() 
    host = 'localhost' 
    port = 8000 
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    sock.connect((host,port)) 
    return sock 

def send_data(message, sock): 
    """Get requests""" 
    if message: 
     sock.sendall(message) 

    # Look for the response 
    has_request = False 
    while not has_request: 
     data = sock.recv(1024) 

     #First I get ACK, code not listed here... 
     #Next iteration, I got my request message 
     has_request = True 
    return data 

if __name__ == '__main__': 
    server_sock = create_socket() 
    server_data = send_data(event, server_sock) 

    if server_data: 
     parse_request(server_data, server_sock) 

    #no more use of server_socket ? 
    server_sock.close()