Dưới đây là cơ chế bọc (Wrapper) kết nối để khi lập trình viên gọi `conn.close()`, nó thực chất là trả kết nối về Pool thay vì đóng hẳn.

```python
class PooledConnectionWrapper:
    def __init__(self, pool, conn: psycopg.Connection):
        self._pool = pool
        self._conn = conn
        self._closed = False

    def __getattr__(self, name: str) -> Any:
        # Pass-through mọi lệnh (cursor, commit...) tới connection thật
        return getattr(self._conn, name)

    def close(self):
        # ĐÁNH CHẶN: Khi gọi close, trả về pool
        if not self._closed:
            self._pool.putconn(self._conn)
            self._closed = True

    def __enter__(self):
        return self._conn.__enter__()
        
    def __exit__(self, exc_type, exc_val, exc_tb):
        try:
            self._conn.__exit__(exc_type, exc_val, exc_tb)
        finally:
            self.close()
```