Bagaimana untuk Menetapkan Nilai Konteks dengan Selamat dalam Pengendali HTTP dalam Go?

Menetapkan Nilai Konteks dalam Pengendali HTTP

Dalam pakej HTTP Go, fungsi HandleFunc membenarkan pendaftaran fungsi pengendali untuk memproses permintaan HTTP yang masuk. Walaupun menetapkan nilai konteks dalam pengendali adalah amalan biasa, menentukan pendekatan optimum mungkin menimbulkan kebimbangan.

Salah satu pendekatan ialah menggunakan fungsi WithValue daripada pakej konteks untuk menetapkan nilai konteks dalam pengendali. Walau bagaimanapun, adalah penting untuk ambil perhatian bahawa penggunaan berikut boleh menimbulkan isu yang berpotensi:

func setValue(r *http.Request, val string) {
  ctx := context.WithValue(r.Context(), myContext, val)
  *r = *r.WithContext(ctx)
}

Menetapkan *r kepada objek permintaan baharu boleh menyebabkan kod sehingga timbunan menggunakan nilai konteks yang salah. Untuk menangani perkara ini, pendekatan yang disyorkan ialah mencipta salinan cetek permintaan menggunakan WithContext dan mengembalikan penunjuk kepadanya:

func setValue(r *http.Request, val string) *http.Request {
  return r.WithContext(context.WithValue(r.Context(), myContext, val))
}

Dengan mengembalikan penunjuk kepada salinan cetek, anda memastikan bahawa nilai konteks yang betul digunakan sepanjang saluran paip pemprosesan permintaan. Jika pengendali memanggil pengendali lain, hantar permintaan yang diubah suai kepada pengendali seterusnya:

http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  r = setValue(r, "foobar")
  someOtherHandler.ServeHTTP(w, r)
})

Pendekatan ini berkesan menetapkan nilai konteks dalam pengendali HTTP sambil mengelakkan kemungkinan perangkap yang berkaitan dengan mengubah suai objek permintaan asal.

Atas ialah kandungan terperinci Bagaimana untuk Menetapkan Nilai Konteks dengan Selamat dalam Pengendali HTTP dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!