31 lines
933 B
Go
31 lines
933 B
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
apierrors "git.yuntuo.hk/k2safe/go-omni/errors"
|
|
"git.yuntuo.hk/k2safe/go-omni/httpx"
|
|
)
|
|
|
|
// ContextErrors classifies unhandled context errors using project catalog keys.
|
|
// Empty keys leave that class unchanged. Explicit API errors always take
|
|
// precedence, even if they wrap a context failure. Causes stay available to logs.
|
|
func ContextErrors(canceledKey, deadlineKey string) httpx.Middleware {
|
|
return func(next httpx.Handler) httpx.Handler {
|
|
return func(c *httpx.Context) error {
|
|
err := next(c)
|
|
var public *apierrors.APIError
|
|
if err == nil || errors.As(err, &public) {
|
|
return err
|
|
}
|
|
if canceledKey != "" && errors.Is(err, context.Canceled) {
|
|
return apierrors.ByCode(canceledKey).WithCause(err)
|
|
}
|
|
if deadlineKey != "" && errors.Is(err, context.DeadlineExceeded) {
|
|
return apierrors.ByCode(deadlineKey).WithCause(err)
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
}
|