在 MCP SDK 源码中，with 语句随处可见，特别是异步上下文管理器更是遍地开花。通过异步上下文管理器提供了一种优雅的方式来管理外部资源，例如文件、锁和网络连接。通过 with 语句，我们可以利用上下文管理器自动处理资源的设置和释放，避免资源泄漏并提高代码的可读性。
什么是上下文管理器？
在 Python 中，上下文管理器 (Context Manager) 是一种用于管理资源的对象，定义了在进入和退出代码块时需要执行的操作。上下文管理器最常用的场景是使用 with 语句。
上下文管理器是一种实现了上下文管理协议的对象。该协议定义了两个特殊方法：__enter__() 和 __exit__()。
__enter__()方法：在进入with语句块时被调用，负责资源的获取和设置。__exit__()方法：在离开with语句块时被调用，负责资源的释放和清理。
Python 中的资源管理
在程序开发中，正确管理外部资源至关重要。这些资源包括：
- 文件读写
- 网络访问
- 数据库操作
- 线程锁
如果资源在使用完毕后没有及时释放，可能会导致资源泄漏，最终影响程序的性能甚至崩溃。
例如，当我们需要操作文件时，通常会使用 open() 函数打开文件，然后进行读写操作。但是，如果忘记在操作完成后调用 close() 函数关闭文件，文件句柄将一直被占用，直到程序结束。这不仅浪费了系统资源，还可能导致其他程序无法访问该文件。
上下文管理器的优势
使用上下文管理器可以有效地解决资源管理问题。with 语句可以确保在代码块执行完毕后，无论是否发生异常，__exit__() 方法都会被调用，从而保证资源得到及时释放。
此外，上下文管理器还可以简化代码，提高可读性。通过将资源的获取和释放逻辑封装在上下文管理器中，我们可以将关注点放在核心业务逻辑上，而无需过多关注资源管理细节。
自定义上下文管理器
Python 提供了两种方式来创建自定义上下文管理器：
基于类的上下文管理器
通过定义一个类，并实现 __enter__() 和 __exit__() 方法，可以创建基于类的上下文管理器。
Python
python
class FileManager:
def __init__(self, filename, mode):
self.filename = filename
self.mode = mode
self.file = None
def __enter__(self):
self.file = open(self.filename, self.mode)
return self.file
def __exit__(self, exc_type, exc_val, exc_tb):
if self.file:
self.file.close()
# 使用 with 语句管理文件资源
with FileManager("my_file.txt", "w") as f:
f.write("Hello, world!")2. 基于生成器的上下文管理器
通过使用 contextlib.contextmanager 装饰器，可以将一个生成器函数转换为上下文管理器。
Python
python
from contextlib import contextmanager
@contextmanager
def file_manager(filename, mode):
f = open(filename, mode)
try:
yield f
finally:
f.close()
# 使用 with 语句管理文件资源
with file_manager("my_file.txt", "w") as f:
f.write("Hello, world!")上下文管理器的应用场景
除了资源管理，上下文管理器还可以应用于其他场景，例如：
- 数据库事务管理
- 线程锁的获取和释放
- 程序运行时间的统计
异步上下文管理器
asynccontextmanager 是 contextlib 模块中的一个装饰器，用于将一个异步生成器函数转换为一个异步上下文管理器 (asynchronous context manager)。异步上下文管理器主要用于管理需要进行异步操作（例如，网络请求、文件 I/O 等）的资源。
应用场景
- 异步数据库连接: 在进入 async with 时建立连接，在退出时关闭连接。
- 异步文件操作: 在进入 async with 时打开文件，在退出时关闭文件。
- 异步网络连接: 在进入 async with 时建立网络连接，在退出时关闭连接。
- 任何需要在异步操作前后执行特定设置和清理逻辑的场景。
为什么使用上下文管理器
资源管理自动化： 确保资源在使用完毕后总是被正确地释放或清理，即使在发生异常的情况下也能保证。这可以避免资源泄漏（例如未关闭的文件句柄、未释放的锁）。 代码更清晰和简洁： 将资源的获取和释放逻辑与主要业务逻辑分离，使代码更易读和维护。避免了显式的 try...finally 块来处理资源清理。 提高代码可靠性： 通过确保清理操作总是被执行，提高了代码的健壮性。
总结
上下文管理器是 Python 中一个强大的工具，可以帮助我们优雅地管理外部资源，避免资源泄漏，提高代码可读性。通过学习上下文管理器的使用，我们可以编写出更加健壮和高效的 Python 程序。
希望这篇文章能够帮助您理解 Python 上下文管理器的概念和使用方法。如果您有任何疑问，欢迎在评论区留言。