在Python中关闭套接字

Question

我正在修改具有以下形式的Python代码：

def foo(self):
    try:
        connect socket
    except Exception, e:
        some error reporting stuff
        return an error

    use the socket
    do some other stuff

    if some condition:
        return

    do some more stuff
    socket.close()
    return normally

作为一名从Java转来的开发者，我希望在整个过程中都使用try-finally语句块以确保套接字被关闭。是否应该在这段代码中也加入try-finally语句块呢？还是说Python自己有某种魔力可以自动处理这个问题呢？

我在Python文档中读到过套接字会在垃圾回收时关闭。但是仅仅依靠垃圾回收器来关闭套接字并不是一个好的做法。

- Jörgen Lundberg

Answer 1

你可以使用Python 2.5中添加的try-finally块：

try:
    open socket
    do stuff with socket
finally:
    close socket

你也可以使用with语句，它在Python 2.6中被添加（并且可以在2.5中使用from __future__ import with_statement声明）：

with open_the_socket() as s:
    use s

当内部代码块退出时，无论是正常退出还是通过异常退出，只要socket类在其__exit__()方法中关闭，这将自动关闭套接字。

在Python 2.7.2中，socket类没有实现__exit__()方法。

- Adam Rosenfield

---

Python一直都有try-finally语句。自2.5版本起，它将try-except-finally语句进行了统一。详见http://docs.python.org/reference/compound_stmts.html#the-try-statement。 - JimB

Answer 2

你似乎想在2.5中添加finally块到try/except中

http://docs.python.org/whatsnew/2.5.html#pep-341-unified-try-except-finally

你说得对，依靠垃圾回收时自动关闭是一种不好的做法，最好在使用完之后关闭它们。使用try/finally是一个不错的方式。

别忘了在关闭之前调用shutdown()。

- Vinko Vrsalovic

Answer 3

除了使用try/finally块外，还可以考虑设置套接字的默认超时时间，在指定的时间后自动关闭套接字。

http://docs.python.org/library/socket.html?highlight=setdefaulttimeout#socket.setdefaulttimeout

- twils

---

套接字超时不会关闭套接字。它会超时读取（可能还有连接/写入），但保持套接字打开，以便用户可以重试操作。 - Dima Tisnek