Golang'da WithTimeout ve WithCancel context ile veritabanı sorgularını iptal etme

Aşağıda listelenen iki senaryoyu kullanarak tüm aktif sorguları context ile durduracağız.

• Sorgu 5 saniyeden uzun sürerse.
  • Bu, WithTimeout özelliğini kullanan alt context'in sorumluluğu olacaktır. Sorgu duracak ve context deadline exceeded hatası verecek.


• Uygulama durdurulursa/iptal edilirse (sorgunun süresi ne olursa olsun).
  • Bu, WithCancel özelliğini kullanan ana context'in sorumluluğu olacaktır. Sorgu duracak ve context canceled hatası verecek.

Yukarıda açıklandığı gibi, tüm uygulama için bir ana context ve ana içerikten türetilen bir alt içeriğe sahibiz. Sonuç olarak:

• Alt context iptal edilirse, yalnızca ona bağlı işlemler etkilenir. Ana context bozulmadan kalır.


• Ana içerik iptal edilirse, ona bağlı olan tüm işlemler etkilenir. Bu, tüm türetilmiş alt contextleri de içerir. Teknik olarak tüm uygulama etkilenir.

Örnek

Ana context'i HTTP sunucusunun BaseContext parametresi olarak ayarlıyoruz ve böylece istek kapsamı boyunca bizim için kullanılabilir hale gelecektir.

func main() {
    // Parent context.
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

    srv := &http.Server{
		Addr:    "some-address",
		Handler: "some-handler",
		BaseContext: func(listener net.Listener) context.Context {
			return ctx
		},
	}

    // ...
}

Ana context'i HTTP isteğinden aldığınızdan ve sorguyu çalıştırdığınız yere enjekte ettiğinizden emin olun. Şu anda ben her şeyi işleyicide yapıyorum.

func myHandler(w http.ResponseWriter, r *http.Request) {
    // Grab the parent context.
    ctx := r.Context()
    
    // Create child context derived from the parent context.
    ctx, cancel := context.WithTimeout(ctx, 5 * time.Second)
    defer cancel()

    // Link the child context to the query.
    rows, err := DB.QueryContext(ctx, `some query`)
    // ...
}