Quản lý Ngữ cảnh Giao dịch với __enter__ và __exit__ Rollback

Mục tiêu kiến thức

• Hiểu sâu giao thức Context Manager: __enter__ và __exit__.
• Thực hiện cơ chế Rollback trạng thái khi xảy ra ngoại lệ bên trong khối with.
• Lan truyền ngoại lệ ra ngoài (không nuốt lỗi vô cớ) bằng cách trả về False trong __exit__.

Mô tả bài toán

Cài đặt lớp DatabaseTransactionContext(db: dict):

• __init__(self, db: dict): Nhận tham chiếu đến từ điển db đại diện cho cơ sở dữ liệu in-memory.
• __enter__(self):
• Tạo một checkpoint bản sao độc lập của db (dict(self.db) hoặc sao chép nông/sâu phù hợp).
• Trả về đối tượng giao dịch (hoặc self.db).
• __exit__(self, exc_type, exc_val, exc_tb) -> bool:
• Nếu xảy ra ngoại lệ (exc_type is not None):
• Hoàn nguyên (Rollback) toàn bộ nội dung của self.db về trạng thái checkpoint ban đầu (xóa các key mới thêm, phục hồi các key đã sửa/xóa).
• Trả về False để Python lan truyền ngoại lệ lên trên.
• Nếu không có ngoại lệ: Commit thay đổi và trả về True hoặc None.

Ràng buộc

• Thời gian chạy tối đa: 1000ms.
• Giới hạn bộ nhớ: 256MB.
• Dữ liệu đầu vào tuân thủ đúng kiểu dữ liệu và miền giá trị được mô tả.