blob: 2f95761974ad8f1d80f6f63edf808b132181fc37 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Package dontpanic provides function wrappers and supervisors to ensure
// that wrapped code does not panic and cause program crashes.
//
// When should you use this package? Anytime you are running a function or
// goroutine where it isn't obvious whether it can or can't panic. This may
// be a higher risk in long running goroutines and functions or ones that are
// difficult to test completely.
package dontpanic
import (
"time"
sentry "github.com/getsentry/sentry-go"
"gitlab.com/gitlab-org/gitaly/internal/log"
)
// Try will wrap the provided function with a panic recovery. If a panic occurs,
// the recovered panic will be sent to Sentry and logged as an error.
// Returns `true` if no panic and `false` otherwise.
func Try(fn func()) bool { return catchAndLog(fn) }
// Go will run the provided function in a goroutine and recover from any
// panics. If a panic occurs, the recovered panic will be sent to Sentry
// and logged as an error. Go is best used in fire-and-forget goroutines where
// observability is lost.
func Go(fn func()) { go Try(fn) }
var logger = log.Default()
func catchAndLog(fn func()) bool {
var id *sentry.EventID
var recovered interface{}
var normal = true
func() {
defer func() {
recovered = recover()
if recovered != nil {
normal = false
}
if err, ok := recovered.(error); ok {
id = sentry.CaptureException(err)
}
}()
fn()
}()
if id == nil || *id == "" {
return normal
}
logger.WithField("sentry_id", id).Errorf(
"dontpanic: recovered value sent to Sentry: %+v", recovered,
)
return normal
}
// GoForever will keep retrying a function fn in a goroutine forever in the
// background (until the process exits) while recovering from panics. Each
// time a closure panics, the recovered value will be sent to Sentry and
// logged at level error. The provided backoff will delay retries to enable
// "breathing" room to prevent potentially worsening the situation.
func GoForever(backoff time.Duration, fn func()) {
go func() {
for {
if Try(fn) {
continue
}
if backoff <= 0 {
continue
}
logger.Infof("dontpanic: backing off %s before retrying", backoff)
time.Sleep(backoff)
}
}()
}
|