对于系统资源如文件、数据库连接、socket而言,应用程序打开这些资源并执行完业务逻辑之后,必须做的一件事就是要关闭(断开)该资源。
在Python中打开过多的文件书写完成后没有及时关掉就会出现,Too many open files" 的错误,因为系统允许你打开的最大文件数量是有限的。
例如:在创建tcp和udp是用到的套接字,在程序运行结束后应该及时关掉
下面是利用上下文管理器通过面向对象来书写tcp,自动关闭套接字.
上下文管理器:任何实现了 __enter__() 和 __exit__() 方法的对象都可称之为上下文管理器,上下文管理器对象可以使用 with 关键字。
# 客户端
import socket
class Client(object):
def __init__(self,ip,dk):
self.ip = ip
self.dk = dk
def __enter__(self):
self.tcp_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
self.tcp_socket.connect((self.ip,self.dk))
send_mag = input('请输入信息:')
self.tcp_socket.send(send_mag.encode())
data = self.tcp_socket.recv(1024)
print(data.decode('gbk'))
def __exit__(self, *args):
self.tcp_socket.close()
if __name__ == '__main__':
with Client('192.168.90.22', 8080) astcp_socket:
print("就好哈")
# 服务器
import socket
class Server(object):
def __init__(self):
self.tcp_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
self.tcp_socket.bind(('',8888))
self.tcp_socket.listen(128)
def __enter__(self):
self.client_socket, address= self.tcp_socket.accept()
data = self.client_socket.recv(1024)
print(data.decode('gbk'))
mag = input('信息:')
self.client_socket.send(mag.encode())
def __exit__(self, *args):
self.client_socket.close()
self.tcp_socket.close()
with Server() as client_socket:
print('完成!!!....')