Bạn có thể quản lý các thao tác đang chạy bằng context.Context của Go. Context là một giá trị dữ liệu chuẩn của Go, có khả năng cho biết thao tác tổng thể mà nó đại diện đã bị huỷ và không còn cần thiết nữa hay chưa. Bằng cách truyền một context.Context qua các lệnh gọi hàm và các dịch vụ trong ứng dụng, những thành phần đó có thể dừng làm việc sớm và trả về lỗi khi phần xử lý của chúng không còn cần đến nữa. Tìm hiểu thêm về Context tại Go Concurrency Patterns: Context.
Ví dụ, bạn có thể muốn:
- Kết thúc những thao tác chạy lâu, bao gồm các thao tác cơ sở dữ liệu mất quá nhiều thời gian để hoàn thành.
- Lan truyền các yêu cầu huỷ đến từ nơi khác, chẳng hạn khi một client đóng kết nối.
Nhiều API dành cho lập trình viên Go có các hàm nhận đối số Context, giúp bạn dùng Context xuyên suốt ứng dụng của mình dễ dàng hơn.
Huỷ thao tác cơ sở dữ liệu sau một timeout
Bạn có thể dùng Context để đặt một timeout hoặc deadline, sau mốc đó thao tác sẽ bị huỷ. Để tạo ra một Context có timeout hoặc deadline, hãy gọi context.WithTimeout hoặc context.WithDeadline.
Mã trong ví dụ về timeout dưới đây tạo ra một Context rồi truyền nó vào method QueryContext của sql.DB.
func QueryWithTimeout(ctx context.Context) {
// Create a Context with a timeout.
queryCtx, cancel := context.WithTimeout(ctx, 5*time.Second)
defer cancel()
// Pass the timeout Context with a query.
rows, err := db.QueryContext(queryCtx, "SELECT * FROM album")
if err != nil {
log.Fatal(err)
}
defer rows.Close()
// Handle returned rows.
}Khi một context được dẫn xuất từ một context bên ngoài, như queryCtx được dẫn xuất từ ctx trong ví dụ này, thì nếu context bên ngoài bị huỷ, context dẫn xuất cũng tự động bị huỷ theo. Ví dụ, trong các máy chủ HTTP, method http.Request.Context trả về một context gắn với request đó. Context ấy sẽ bị huỷ nếu client HTTP ngắt kết nối hoặc huỷ request HTTP (điều này khả thi trong HTTP/2). Truyền context của một request HTTP vào QueryWithTimeout ở trên sẽ khiến truy vấn cơ sở dữ liệu dừng sớm khi toàn bộ request HTTP bị huỷ, hoặc khi truy vấn chạy quá năm giây.
Lưu ý: Luôn defer lệnh gọi hàm cancel được trả về khi bạn tạo một Context mới có timeout hoặc deadline. Việc này giải phóng các tài nguyên mà Context mới đang giữ khi hàm chứa nó kết thúc. Nó cũng huỷ luôn queryCtx, nhưng tới lúc hàm trả về thì lẽ ra không còn gì đang dùng queryCtx nữa.