Mengendalikan pembatalan konteks rentas lapisan

Kandungan soalan

Saya sedang membangunkan API bahagian belakang berlapis dengan pengawal, perkhidmatan, repositori dsb.

Konteks

Setiap kaedah dalam lapisan ini mengambil context.Context sebagai parameter pertamanya yang mengandungi konteks permintaan. Ini mudah kerana sebarang kaedah boleh mengakses pelbagai data berkaitan permintaan (seperti correlationID, dll.)

Setiap konteks permintaan mempunyai tamat masa yang ditetapkan oleh TimeoutMiddleware di bawah:

func TimeoutMiddleware(timeoutFn func(*gin.Context) time.Duration) gin.HandlerFunc {
    return func(c *gin.Context) {
        timeout := timeoutFn(c)
        ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
        defer cancel()

        c.Request = c.Request.WithContext(ctx)

        c.Next()
    }
}

func TimeoutFn(c *gin.Context) time.Duration {
    return conf.HTTPRouter.DefaultContextTimeout
}

Cabaran

Ideanya adalah untuk menghentikan sebarang operasi yang sedang berjalan dengan anggun apabila konteks permintaan tamat. Berdasarkan pemahaman saya tentang konteks dan konkurensi (sangat sedikit), saya membina fungsi pembantu ini:

package helpers

import "context"

// Checks for context cancellation and returns ctx.Err() if canceled.
func HandleContextCancel(ctx context.Context) error {
    select {
    case <-ctx.Done(): // If the context is canceled
        return ctx.Err() // return immediately with the canceled error.

    default: 
        return nil // Continue with the normal processing.
    }
}

Secara teorinya, jika saya mahu menghentikan sebarang operasi secepat mungkin , saya perlu untuk memanggil fungsi ini pada permulaan

setiap kaedah dalam aplikasi, seperti ini:

func DoSomething(ctx context.Context, ...) resterrors.RestErr {
if err := helpers.HandleContextCancel(ctx); err != nil {
    return resterrors.NewRequestTimeoutError(
        fmt.Errorf("DoSomething: %w", err),
    )
}
// ...

Selain itu, saya tahu bahawa dalam repositori yang mengakses pangkalan data, kebanyakan fungsi memerlukan Query、QueryRow、Execkonteks

, seperti

..., seperti ini: internal_server_error

rows, err := pgclient.GetSession().Query(ctx, query, param1, ...)

Jadi setiap kali ralat berlaku dalam baris di atas, saya perlu menyemak sama ada ralat itu bukan disebabkan oleh pembatalan konteks dan bukannya hanya kembali

dan mesej ralat seperti ini: HandleContextCancel 函数看起来有点多余，您对此有何看法？

正确答案

您不需要 HandleContextCancel

rows, err := pgclient.GetSession().Query(ctx, query, param1, ...)
if err != nil {
   return helpers.MapRepoError("DoSomething: Query Error:", err)
}

func MapRepoError(location string, err error) resterrors.RestErr {
    if errors.Is(err, context.DeadlineExceeded) {
        return resterrors.NewRequestTimeoutError(
            fmt.Errorf("%s request_timeout", location),
        )
    }
    return resterrors.NewInternalServerError(
        fmt.Errorf("%s %w", location, err),
    )
}

Menggunakan fungsi HandleContextCancel nampaknya agak berlebihan, apakah pendapat anda tentang perkara ini?

Unwrap() error

Jawapan betul🎜🎜Anda tidak memerlukan fungsi HandleContextCancel, anda hanya boleh melakukan perkara berikut: 🎜

if ctx.Err()!=nil {
  // Context timed out or canceled. Return
  return ctx.Err()
}

🎜Jika fungsi pengendalian ralat anda yang lain membalut ralat ini dengan betul (iaitu ia melaksanakan kaedah 🎜), maka anda boleh menyemak di peringkat atas sama ada ralat itu mengandungi ralat tamat masa/pembatalan dan memutuskan jenis ralat yang anda ingin kembalikan tidak perlu Lakukan ini untuk setiap peringkat 🎜.

Atas ialah kandungan terperinci Mengendalikan pembatalan konteks rentas lapisan. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!