You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
611 B
Go
34 lines
611 B
Go
package rescue
|
|
|
|
import (
|
|
"context"
|
|
"runtime/debug"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
// Recover is used with defer to do cleanup on panics.
|
|
// Use it like:
|
|
//
|
|
// defer Recover(func() {})
|
|
func Recover(cleanups ...func()) {
|
|
for _, cleanup := range cleanups {
|
|
cleanup()
|
|
}
|
|
|
|
if p := recover(); p != nil {
|
|
logx.ErrorStack(p)
|
|
}
|
|
}
|
|
|
|
// RecoverCtx is used with defer to do cleanup on panics.
|
|
func RecoverCtx(ctx context.Context, cleanups ...func()) {
|
|
for _, cleanup := range cleanups {
|
|
cleanup()
|
|
}
|
|
|
|
if p := recover(); p != nil {
|
|
logx.WithContext(ctx).Errorf("%+v\n%s", p, debug.Stack())
|
|
}
|
|
}
|