Kỹ thuật chia sẻ dữ liệu có thể thay đổi giữa Rust và Python bằng PyO3

Kết luận (Conclusion)

Việc PyO3 không thể trực tiếp phơi bày cho Python các struct sử dụng lifetime của Rust ban đầu có thể trông như một hạn chế. Tuy nhiên, thư viện chuẩn Rust và PyO3 cung cấp những công cụ mạnh mẽ để vượt qua giới hạn này. std::mem::take và std::mem::replace cho phép xử lý khéo léo giữa tham chiếu có thể thay đổi (mutable reference) và giá trị sở hữu (owned value), còn Arc và Mutex rất hữu ích khi cần phơi bày dữ liệu chia sẻ có thể thay đổi cho Python. Đặc biệt, MutexExt của PyO3 là công cụ thiết yếu để tránh deadlock khi sử dụng mutex cùng với Python.

---

Tóm tắt nội dung chính

Tài liệu này giải thích theo từng bước các vấn đề kỹ thuật gặp phải và quá trình giải quyết khi chia sẻ dữ liệu có thể thay đổi giữa Rust và Python trong một dự án tái hiện thực ngôn ngữ template của Django bằng Rust.

• Bối cảnh: Ngôn ngữ template của Django sử dụng một đối tượng gọi là context để cung cấp dữ liệu động cho template. Trong bản hiện thực bằng Rust của dự án, context này được định nghĩa là một struct Rust và phải được truyền dưới dạng tham chiếu có thể thay đổi (&mut Context) khi render các thẻ template.

• Vấn đề ban đầu: Cần truyền tham chiếu có thể thay đổi (&mut Context) từ mã Rust sang một hàm Python để thực thi custom tag. Tuy nhiên, Python không hiểu lifetime của Rust, và PyO3, thư viện liên kết Rust-Python, yêu cầu giá trị có quyền sở hữu (owned value), nên nếu truyền trực tiếp tham chiếu sẽ phát sinh lỗi biên dịch.

• Quá trình giải quyết:

  1. Giải quyết vấn đề sở hữu: Sử dụng std::mem::take để tạm thời lấy quyền sở hữu từ &mut Context, từ đó tạo một đối tượng Context có quyền sở hữu có thể truyền sang Python. Sau khi mã Python chạy xong, thử dùng std::mem::replace để đặt Context đã được xử lý trở lại vị trí tham chiếu ban đầu.
  2. Giải quyết lỗi 'Moved Value': Tuy nhiên, trong quá trình này, khi đối tượng Context đã được move vào hàm Python rồi lại bị dùng lại, sẽ xuất hiện lỗi biên dịch "use of moved value". Để xử lý, tài liệu đưa vào Arc (Atomic Reference Count) để bọc Context. Nhờ đó có thể truyền một tham chiếu đã được sao chép (clone) sang Python mà không cần chuyển quyền sở hữu.
  3. Xử lý khi Python giữ lại tham chiếu: Nếu Python tiếp tục giữ tham chiếu tới Context, việc thu hồi quyền sở hữu bằng Arc::try_unwrap có thể thất bại. Trong trường hợp này, có thể hiện thực một phương thức fallback như clone_ref để deep clone dữ liệu bên trong Context.
  4. Cho phép thay đổi dữ liệu từ Python: Cuối cùng, để mã Python không chỉ đọc mà còn có thể thay đổi Context, tài liệu đưa vào Mutex. Cấu trúc Arc<Mutex<Context>> đảm bảo dữ liệu có thể được truy cập và sửa đổi an toàn từ nhiều thread. Lúc này, để tránh deadlock với Python interpreter, cần dùng phương thức lock_py_attached của MutexExt do PyO3 cung cấp.