Context manager và câu lệnh with là gì?

Question

Accepted Answer

Một **context manager** là một đối tượng định nghĩa logic thiết lập và dọn dẹp, được dùng với câu lệnh **`with`**. Nó đảm bảo việc dọn dẹp (đóng file, giải phóng lock, rollback transaction) diễn ra tự động — ngay cả khi có lỗi xảy ra.

## Vấn đề nó giải quyết

```python
# ❌ dọn dẹp thủ công — dễ lỗi; nếu xử lý ném lỗi, file không bao giờ được đóng
f = open("file.txt")
data = f.read()
process(data)        # nếu cái này ném lỗi, f.close() bên dưới bị bỏ qua → rò rỉ file handle
f.close()

# ✅ câu lệnh with — đảm bảo file được đóng, bất kể điều gì
with open("file.txt") as f:
    data = f.read()
    process(data)
# f tự động được đóng ở đây — ngay cả khi process() ném ngoại lệ
```

Khối `with` đảm bảo `f.close()` chạy khi khối thoát ra, dù bình thường hay qua một ngoại lệ — loại bỏ tài nguyên rò rỉ do dọn dẹp bị quên hoặc bị bỏ qua.

## Các context manager phổ biến

```python
with open("f.txt") as f: ...           # file (tự đóng)
with lock:                              # threading lock (tự giải phóng)
    shared_resource += 1
with db.transaction():                  # transaction DB (tự commit/rollback)
    ...
with open("a") as a, open("b") as b:    # nhiều cái cùng lúc
    ...
```

## Viết của riêng bạn — cách dùng class

```python
class Timer:
    def __enter__(self):                 # THIẾT LẬP — chạy khi vào `with`
        self.start = time.perf_counter()
        return self                       # giá trị được gán cho `as`
    def __exit__(self, exc_type, exc_val, tb):  # DỌN DẸP — chạy khi thoát (kể cả khi lỗi)
        print(f"took {time.perf_counter() - self.start:.3f}s")
        # return False → lan truyền ngoại lệ; True → triệt tiêu nó

with Timer():
    do_work()
```

Một context manager hiện thực `__enter__` (thiết lập) và `__exit__` (dọn dẹp, nhận thông tin về bất kỳ ngoại lệ nào).

## Cách dễ hơn — contextlib

```python
from contextlib import contextmanager

@contextmanager
def timer():
    start = time.perf_counter()
    try:
        yield                            # mọi thứ trước yield = thiết lập; sau = dọn dẹp
    finally:
        print(f"took {time.perf_counter() - start:.3f}s")

with timer():
    do_work()
```

Decorator `@contextmanager` biến một generator thành một context manager — thiết lập trước `yield`, dọn dẹp sau (trong `finally` cho an toàn).

## Tại sao điều này quan trọng

Context manager và `with` là cách Pythonic để xử lý **quản lý tài nguyên an toàn** — đảm bảo file được đóng, lock được giải phóng, connection được trả về, và transaction được hoàn tất, ngay cả khi ngoại lệ xảy ra.

Điều này ngăn rò rỉ tài nguyên và trạng thái không nhất quán, một nguồn lỗi phổ biến với dọn dẹp thủ công.

Chúng được dùng liên tục (mọi thao tác file nên dùng `with`), và biết cách viết của riêng bạn (qua `__enter__`/`__exit__` hoặc `@contextmanager` đơn giản hơn) cho phép bạn đóng gói bất kỳ mẫu thiết lập/dọn dẹp nào một cách gọn gàng và đáng tin cậy.