2020-12-02 00:03:20 +00:00
|
|
|
package watchdog
|
|
|
|
|
|
|
|
import (
|
2021-01-18 17:59:51 +00:00
|
|
|
"errors"
|
2020-12-02 00:03:20 +00:00
|
|
|
"fmt"
|
|
|
|
"log"
|
2020-12-09 15:35:23 +00:00
|
|
|
"math"
|
2021-01-19 19:38:46 +00:00
|
|
|
"os"
|
|
|
|
"path/filepath"
|
2020-12-02 00:03:20 +00:00
|
|
|
"runtime"
|
2020-12-04 13:08:00 +00:00
|
|
|
"runtime/debug"
|
2020-12-02 00:03:20 +00:00
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/elastic/gosigar"
|
|
|
|
"github.com/raulk/clock"
|
|
|
|
)
|
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
// ErrNotSupported is returned when the watchdog does not support the requested
|
|
|
|
// run mode in the current OS/arch.
|
|
|
|
var ErrNotSupported = errors.New("watchdog run mode not supported")
|
|
|
|
|
2020-12-09 15:35:23 +00:00
|
|
|
// PolicyTempDisabled is a marker value for policies to signal that the policy
|
|
|
|
// is temporarily disabled. Use it when all hope is lost to turn around from
|
|
|
|
// significant memory pressure (such as when above an "extreme" watermark).
|
|
|
|
const PolicyTempDisabled uint64 = math.MaxUint64
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// The watchdog is designed to be used as a singleton; global vars are OK for
|
|
|
|
// that reason.
|
|
|
|
var (
|
|
|
|
// Logger is the logger to use. If nil, it will default to a logger that
|
|
|
|
// proxies to a standard logger using the "[watchdog]" prefix.
|
|
|
|
Logger logger = &stdlog{log: log.New(log.Writer(), "[watchdog] ", log.LstdFlags|log.Lmsgprefix)}
|
|
|
|
|
|
|
|
// Clock can be used to inject a mock clock for testing.
|
|
|
|
Clock = clock.New()
|
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
// NotifyGC, if non-nil, will be called when a GC has happened.
|
|
|
|
NotifyGC func() = func() {}
|
2021-01-19 19:38:46 +00:00
|
|
|
|
|
|
|
// HeapdumpThreshold sets the utilization threshold that will trigger a
|
|
|
|
// heap dump to be taken automatically. A zero value disables this feature.
|
|
|
|
// By default, it is disabled.
|
|
|
|
HeapdumpThreshold float64
|
|
|
|
|
|
|
|
// HeapdumpMaxCaptures sets the maximum amount of heapdumps a process will generate.
|
|
|
|
// This limits the amount of episodes that will be captured, in case the
|
|
|
|
// utilization climbs repeatedly over the threshold. By default, it is 10.
|
|
|
|
HeapdumpMaxCaptures = uint(10)
|
|
|
|
|
|
|
|
// HeapdumpDir is the directory where the watchdog will write the heapdump.
|
|
|
|
// It will be created if it doesn't exist upon initialization. An error when
|
|
|
|
// creating the dir will not prevent heapdog initialization; it will just
|
|
|
|
// disable the heapdump capture feature. If zero-valued, the feature is
|
|
|
|
// disabled.
|
|
|
|
//
|
|
|
|
// Heapdumps will be written to path <HeapdumpDir>/<RFC3339Nano formatted timestamp>.heap.
|
|
|
|
HeapdumpDir string
|
2020-12-04 13:08:00 +00:00
|
|
|
)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
var (
|
|
|
|
// ReadMemStats stops the world. But as of go1.9, it should only
|
|
|
|
// take ~25µs to complete.
|
|
|
|
//
|
|
|
|
// Before go1.15, calls to ReadMemStats during an ongoing GC would
|
|
|
|
// block due to the worldsema lock. As of go1.15, this was optimized
|
|
|
|
// and the runtime holds on to worldsema less during GC (only during
|
|
|
|
// sweep termination and mark termination).
|
|
|
|
//
|
|
|
|
// For users using go1.14 and earlier, if this call happens during
|
|
|
|
// GC, it will just block for longer until serviced, but it will not
|
|
|
|
// take longer in itself. No harm done.
|
|
|
|
//
|
|
|
|
// Actual benchmarks
|
|
|
|
// -----------------
|
|
|
|
//
|
|
|
|
// In Go 1.15.5, ReadMem with no ongoing GC takes ~27µs in a MBP 16
|
|
|
|
// i9 busy with another million things. During GC, it takes an
|
|
|
|
// average of less than 175µs per op.
|
|
|
|
//
|
|
|
|
// goos: darwin
|
|
|
|
// goarch: amd64
|
|
|
|
// pkg: github.com/filecoin-project/lotus/api
|
|
|
|
// BenchmarkReadMemStats-16 44530 27523 ns/op
|
|
|
|
// BenchmarkReadMemStats-16 43743 26879 ns/op
|
|
|
|
// BenchmarkReadMemStats-16 45627 26791 ns/op
|
|
|
|
// BenchmarkReadMemStats-16 44538 26219 ns/op
|
|
|
|
// BenchmarkReadMemStats-16 44958 26757 ns/op
|
|
|
|
// BenchmarkReadMemStatsWithGCContention-16 10 183733 p50-ns 211859 p90-ns 211859 p99-ns
|
|
|
|
// BenchmarkReadMemStatsWithGCContention-16 7 198765 p50-ns 314873 p90-ns 314873 p99-ns
|
|
|
|
// BenchmarkReadMemStatsWithGCContention-16 10 195151 p50-ns 311408 p90-ns 311408 p99-ns
|
|
|
|
// BenchmarkReadMemStatsWithGCContention-16 10 217279 p50-ns 295308 p90-ns 295308 p99-ns
|
|
|
|
// BenchmarkReadMemStatsWithGCContention-16 10 167054 p50-ns 327072 p90-ns 327072 p99-ns
|
|
|
|
// PASS
|
|
|
|
//
|
|
|
|
// See: https://github.com/golang/go/issues/19812
|
|
|
|
// See: https://github.com/prometheus/client_golang/issues/403
|
|
|
|
memstatsFn = runtime.ReadMemStats
|
|
|
|
sysmemFn = (*gosigar.Mem).Get
|
|
|
|
)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
|
|
|
var (
|
|
|
|
// ErrAlreadyStarted is returned when the user tries to start the watchdog more than once.
|
|
|
|
ErrAlreadyStarted = fmt.Errorf("singleton memory watchdog was already started")
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
// stateUnstarted represents an unstarted state.
|
2020-12-04 13:08:00 +00:00
|
|
|
stateUnstarted int32 = iota
|
|
|
|
// stateRunning represents an operational state.
|
|
|
|
stateRunning
|
2020-12-02 00:03:20 +00:00
|
|
|
)
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// _watchdog is a global singleton watchdog.
|
|
|
|
var _watchdog struct {
|
2020-12-09 14:54:09 +00:00
|
|
|
lk sync.Mutex
|
|
|
|
state int32
|
2020-12-04 13:08:00 +00:00
|
|
|
|
|
|
|
scope UtilizationType
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2021-01-19 19:38:46 +00:00
|
|
|
hdleft uint // tracks the amount of heapdumps left.
|
|
|
|
hdcurr bool // tracks whether a heapdump has already been taken for this episode.
|
|
|
|
|
2020-12-02 00:03:20 +00:00
|
|
|
closing chan struct{}
|
|
|
|
wg sync.WaitGroup
|
|
|
|
}
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// UtilizationType is the utilization metric in use.
|
|
|
|
type UtilizationType int
|
2020-12-02 00:03:20 +00:00
|
|
|
|
|
|
|
const (
|
2020-12-04 13:08:00 +00:00
|
|
|
// UtilizationSystem specifies that the policy compares against actual used
|
2020-12-02 00:03:20 +00:00
|
|
|
// system memory.
|
2020-12-04 13:08:00 +00:00
|
|
|
UtilizationSystem UtilizationType = iota
|
2021-01-18 17:59:51 +00:00
|
|
|
// UtilizationProcess specifies that the watchdog is using process limits.
|
|
|
|
UtilizationProcess
|
2020-12-04 13:08:00 +00:00
|
|
|
// UtilizationHeap specifies that the policy compares against heap used.
|
|
|
|
UtilizationHeap
|
2020-12-02 00:03:20 +00:00
|
|
|
)
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// PolicyCtor is a policy constructor.
|
|
|
|
type PolicyCtor func(limit uint64) (Policy, error)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// Policy is polled by the watchdog to determine the next utilisation at which
|
|
|
|
// a GC should be forced.
|
2020-12-02 00:03:20 +00:00
|
|
|
type Policy interface {
|
2020-12-04 13:08:00 +00:00
|
|
|
// Evaluate determines when the next GC should take place. It receives the
|
|
|
|
// current usage, and it returns the next usage at which to trigger GC.
|
2020-12-09 15:35:23 +00:00
|
|
|
Evaluate(scope UtilizationType, used uint64) (next uint64)
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// HeapDriven starts a singleton heap-driven watchdog.
|
|
|
|
//
|
|
|
|
// The heap-driven watchdog adjusts GOGC dynamically after every GC, to honour
|
2020-12-09 15:35:23 +00:00
|
|
|
// the policy requirements.
|
2020-12-04 13:08:00 +00:00
|
|
|
//
|
2020-12-09 15:35:23 +00:00
|
|
|
// A zero-valued limit will error.
|
2020-12-04 13:08:00 +00:00
|
|
|
func HeapDriven(limit uint64, policyCtor PolicyCtor) (err error, stopFn func()) {
|
|
|
|
if limit == 0 {
|
|
|
|
return fmt.Errorf("cannot use zero limit for heap-driven watchdog"), nil
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
policy, err := policyCtor(limit)
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("failed to construct policy with limit %d: %w", limit, err), nil
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
if err := start(UtilizationHeap); err != nil {
|
|
|
|
return err, nil
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-09 14:54:56 +00:00
|
|
|
gcTriggered := make(chan struct{}, 16)
|
2020-12-04 13:08:00 +00:00
|
|
|
setupGCSentinel(gcTriggered)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
_watchdog.wg.Add(1)
|
|
|
|
go func() {
|
|
|
|
defer _watchdog.wg.Done()
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// get the initial effective GOGC; guess it's 100 (default), and restore
|
|
|
|
// it to whatever it actually was. This works because SetGCPercent
|
|
|
|
// returns the previous value.
|
|
|
|
originalGOGC := debug.SetGCPercent(debug.SetGCPercent(100))
|
|
|
|
currGOGC := originalGOGC
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
var memstats runtime.MemStats
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-gcTriggered:
|
2021-01-18 17:59:51 +00:00
|
|
|
NotifyGC()
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
case <-_watchdog.closing:
|
|
|
|
return
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// recompute the next trigger.
|
|
|
|
memstatsFn(&memstats)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2021-01-19 19:38:46 +00:00
|
|
|
maybeCaptureHeapdump(memstats.HeapAlloc, limit)
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// heapMarked is the amount of heap that was marked as live by GC.
|
|
|
|
// it is inferred from our current GOGC and the new target picked.
|
|
|
|
heapMarked := uint64(float64(memstats.NextGC) / (1 + float64(currGOGC)/100))
|
|
|
|
if heapMarked == 0 {
|
|
|
|
// this shouldn't happen, but just in case; avoiding a div by 0.
|
2020-12-09 14:56:03 +00:00
|
|
|
Logger.Warnf("heap-driven watchdog: inferred zero heap marked; skipping evaluation")
|
2020-12-04 13:08:00 +00:00
|
|
|
continue
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// evaluate the policy.
|
2020-12-09 15:35:23 +00:00
|
|
|
next := policy.Evaluate(UtilizationHeap, memstats.HeapAlloc)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// calculate how much to set GOGC to honour the next trigger point.
|
2020-12-09 15:35:23 +00:00
|
|
|
// next=PolicyTempDisabled value would make currGOGC extremely high,
|
|
|
|
// greater than originalGOGC, and therefore we'd restore originalGOGC.
|
2020-12-04 13:08:00 +00:00
|
|
|
currGOGC = int(((float64(next) / float64(heapMarked)) - float64(1)) * 100)
|
|
|
|
if currGOGC >= originalGOGC {
|
2020-12-09 14:58:17 +00:00
|
|
|
Logger.Debugf("heap watchdog: requested GOGC percent higher than default; capping at default; requested: %d; default: %d", currGOGC, originalGOGC)
|
2020-12-04 13:08:00 +00:00
|
|
|
currGOGC = originalGOGC
|
|
|
|
} else {
|
|
|
|
if currGOGC < 1 {
|
|
|
|
currGOGC = 1
|
|
|
|
}
|
|
|
|
Logger.Infof("heap watchdog: setting GOGC percent: %d", currGOGC)
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
debug.SetGCPercent(currGOGC)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
memstatsFn(&memstats)
|
|
|
|
Logger.Infof("heap watchdog stats: heap_alloc: %d, heap_marked: %d, next_gc: %d, policy_next_gc: %d, gogc: %d",
|
|
|
|
memstats.HeapAlloc, heapMarked, memstats.NextGC, next, currGOGC)
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
2020-12-04 13:08:00 +00:00
|
|
|
}()
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
return nil, stop
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// SystemDriven starts a singleton system-driven watchdog.
|
|
|
|
//
|
|
|
|
// The system-driven watchdog keeps a threshold, above which GC will be forced.
|
|
|
|
// The watchdog polls the system utilization at the specified frequency. When
|
|
|
|
// the actual utilization exceeds the threshold, a GC is forced.
|
|
|
|
//
|
|
|
|
// This threshold is calculated by querying the policy every time that GC runs,
|
|
|
|
// either triggered by the runtime, or forced by us.
|
|
|
|
func SystemDriven(limit uint64, frequency time.Duration, policyCtor PolicyCtor) (err error, stopFn func()) {
|
|
|
|
if limit == 0 {
|
2021-01-18 17:59:51 +00:00
|
|
|
var sysmem gosigar.Mem
|
|
|
|
if err := sysmemFn(&sysmem); err != nil {
|
|
|
|
return fmt.Errorf("failed to get system memory stats: %w", err), nil
|
2020-12-04 13:08:00 +00:00
|
|
|
}
|
2021-01-18 17:59:51 +00:00
|
|
|
limit = sysmem.Total
|
2020-12-04 13:08:00 +00:00
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
policy, err := policyCtor(limit)
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("failed to construct policy with limit %d: %w", limit, err), nil
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
if err := start(UtilizationSystem); err != nil {
|
|
|
|
return err, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
_watchdog.wg.Add(1)
|
|
|
|
var sysmem gosigar.Mem
|
2021-01-19 19:38:46 +00:00
|
|
|
go pollingWatchdog(policy, frequency, limit, func() (uint64, error) {
|
2021-01-18 17:59:51 +00:00
|
|
|
if err := sysmemFn(&sysmem); err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
return sysmem.ActualUsed, nil
|
|
|
|
})
|
|
|
|
|
|
|
|
return nil, stop
|
|
|
|
}
|
|
|
|
|
|
|
|
// pollingWatchdog starts a polling watchdog with the provided policy, using
|
|
|
|
// the supplied polling frequency. On every tick, it calls usageFn and, if the
|
|
|
|
// usage is greater or equal to the threshold at the time, it forces GC.
|
|
|
|
// usageFn is guaranteed to be called serially, so no locking should be
|
|
|
|
// necessary.
|
2021-01-19 19:38:46 +00:00
|
|
|
func pollingWatchdog(policy Policy, frequency time.Duration, limit uint64, usageFn func() (uint64, error)) {
|
2021-01-18 17:59:51 +00:00
|
|
|
defer _watchdog.wg.Done()
|
2020-12-04 13:08:00 +00:00
|
|
|
|
2020-12-09 14:54:56 +00:00
|
|
|
gcTriggered := make(chan struct{}, 16)
|
2020-12-04 13:08:00 +00:00
|
|
|
setupGCSentinel(gcTriggered)
|
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
var (
|
|
|
|
memstats runtime.MemStats
|
|
|
|
threshold uint64
|
|
|
|
)
|
2020-12-04 13:08:00 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
renewThreshold := func() {
|
|
|
|
// get the current usage.
|
|
|
|
usage, err := usageFn()
|
|
|
|
if err != nil {
|
|
|
|
Logger.Warnf("failed to obtain memory utilization stats; err: %s", err)
|
|
|
|
return
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
2021-01-18 17:59:51 +00:00
|
|
|
// calculate the threshold.
|
|
|
|
threshold = policy.Evaluate(_watchdog.scope, usage)
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
// initialize the threshold.
|
|
|
|
renewThreshold()
|
2020-12-09 15:35:23 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
// initialize an empty timer.
|
|
|
|
timer := Clock.Timer(0)
|
|
|
|
stopTimer := func() {
|
|
|
|
if !timer.Stop() {
|
|
|
|
<-timer.C
|
2020-12-09 16:13:27 +00:00
|
|
|
}
|
2021-01-18 17:59:51 +00:00
|
|
|
}
|
2020-12-09 16:13:27 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
for {
|
|
|
|
timer.Reset(frequency)
|
2020-12-09 16:13:27 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
select {
|
|
|
|
case <-timer.C:
|
|
|
|
// get the current usage.
|
|
|
|
usage, err := usageFn()
|
|
|
|
if err != nil {
|
|
|
|
Logger.Warnf("failed to obtain memory utilizationstats; err: %s", err)
|
|
|
|
continue
|
|
|
|
}
|
2021-01-19 19:38:46 +00:00
|
|
|
|
|
|
|
// evaluate if a heapdump needs to be captured.
|
|
|
|
maybeCaptureHeapdump(usage, limit)
|
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
if usage < threshold {
|
|
|
|
// nothing to do.
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
// trigger GC; this will emit a gcTriggered event which we'll
|
|
|
|
// consume next to readjust the threshold.
|
|
|
|
Logger.Warnf("system-driven watchdog triggering GC; %d/%d bytes (used/threshold)", usage, threshold)
|
|
|
|
forceGC(&memstats)
|
2020-12-04 13:08:00 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
case <-gcTriggered:
|
|
|
|
NotifyGC()
|
2020-12-04 13:08:00 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
renewThreshold()
|
2020-12-04 13:08:00 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
stopTimer()
|
2020-12-09 16:13:27 +00:00
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
case <-_watchdog.closing:
|
|
|
|
stopTimer()
|
|
|
|
return
|
2020-12-02 16:33:00 +00:00
|
|
|
}
|
2020-12-04 13:08:00 +00:00
|
|
|
}
|
|
|
|
}
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// forceGC forces a manual GC.
|
|
|
|
func forceGC(memstats *runtime.MemStats) {
|
|
|
|
Logger.Infof("watchdog is forcing GC")
|
|
|
|
|
|
|
|
// it's safe to assume that the finalizer will attempt to run before
|
|
|
|
// runtime.GC() returns because runtime.GC() waits for the sweep phase to
|
|
|
|
// finish before returning.
|
|
|
|
// finalizers are run in the sweep phase.
|
|
|
|
start := time.Now()
|
|
|
|
runtime.GC()
|
|
|
|
took := time.Since(start)
|
2020-12-02 00:03:20 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
memstatsFn(memstats)
|
|
|
|
Logger.Infof("watchdog-triggered GC finished; took: %s; current heap allocated: %d bytes", took, memstats.HeapAlloc)
|
|
|
|
}
|
2020-12-02 16:33:00 +00:00
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
func setupGCSentinel(gcTriggered chan struct{}) {
|
|
|
|
logger := Logger
|
|
|
|
|
|
|
|
// this non-zero sized struct is used as a sentinel to detect when a GC
|
|
|
|
// run has finished, by setting and resetting a finalizer on it.
|
|
|
|
// it essentially creates a GC notification "flywheel"; every GC will
|
|
|
|
// trigger this finalizer, which will reset itself so it gets notified
|
|
|
|
// of the next GC, breaking the cycle when the watchdog is stopped.
|
|
|
|
type sentinel struct{ a *int }
|
|
|
|
var finalizer func(o *sentinel)
|
|
|
|
finalizer = func(o *sentinel) {
|
2020-12-09 14:54:09 +00:00
|
|
|
_watchdog.lk.Lock()
|
|
|
|
defer _watchdog.lk.Unlock()
|
|
|
|
|
|
|
|
if _watchdog.state != stateRunning {
|
2020-12-04 13:08:00 +00:00
|
|
|
// this GC triggered after the watchdog was stopped; ignore
|
|
|
|
// and do not reset the finalizer.
|
|
|
|
return
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
// reset so it triggers on the next GC.
|
|
|
|
runtime.SetFinalizer(o, finalizer)
|
|
|
|
|
|
|
|
select {
|
|
|
|
case gcTriggered <- struct{}{}:
|
|
|
|
default:
|
|
|
|
logger.Warnf("failed to queue gc trigger; channel backlogged")
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
|
|
|
}
|
2020-12-04 13:08:00 +00:00
|
|
|
|
|
|
|
runtime.SetFinalizer(&sentinel{}, finalizer) // start the flywheel.
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
func start(scope UtilizationType) error {
|
|
|
|
_watchdog.lk.Lock()
|
|
|
|
defer _watchdog.lk.Unlock()
|
|
|
|
|
|
|
|
if _watchdog.state != stateUnstarted {
|
|
|
|
return ErrAlreadyStarted
|
|
|
|
}
|
|
|
|
|
|
|
|
_watchdog.state = stateRunning
|
|
|
|
_watchdog.scope = scope
|
|
|
|
_watchdog.closing = make(chan struct{})
|
2021-01-19 19:38:46 +00:00
|
|
|
|
|
|
|
initHeapdumpCapture()
|
|
|
|
|
2021-01-18 17:59:51 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-12-04 13:08:00 +00:00
|
|
|
func stop() {
|
2020-12-09 14:54:09 +00:00
|
|
|
_watchdog.lk.Lock()
|
|
|
|
defer _watchdog.lk.Unlock()
|
|
|
|
|
|
|
|
if _watchdog.state != stateRunning {
|
2020-12-02 00:03:20 +00:00
|
|
|
return
|
|
|
|
}
|
2020-12-04 13:08:00 +00:00
|
|
|
|
|
|
|
close(_watchdog.closing)
|
|
|
|
_watchdog.wg.Wait()
|
2020-12-09 14:54:09 +00:00
|
|
|
_watchdog.state = stateUnstarted
|
2020-12-02 00:03:20 +00:00
|
|
|
}
|
2021-01-19 19:38:46 +00:00
|
|
|
|
|
|
|
func initHeapdumpCapture() {
|
|
|
|
if HeapdumpDir == "" || HeapdumpThreshold <= 0 {
|
|
|
|
Logger.Debugf("heapdump capture disabled")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if HeapdumpThreshold >= 1 {
|
|
|
|
Logger.Warnf("failed to initialize heapdump capture: threshold must be 0 < t < 1")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if fi, err := os.Stat(HeapdumpDir); os.IsNotExist(err) {
|
|
|
|
if err := os.MkdirAll(HeapdumpDir, 0777); err != nil {
|
|
|
|
Logger.Warnf("failed to initialize heapdump capture: failed to create dir: %s; err: %s", HeapdumpDir, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
} else if err != nil {
|
|
|
|
Logger.Warnf("failed to initialize heapdump capture: failed to stat path: %s; err: %s", HeapdumpDir, err)
|
|
|
|
return
|
|
|
|
} else if !fi.IsDir() {
|
|
|
|
Logger.Warnf("failed to initialize heapdump capture: path exists but is not a directory: %s", HeapdumpDir)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
// all good, set the amount of heapdump captures left.
|
|
|
|
_watchdog.hdleft = HeapdumpMaxCaptures
|
|
|
|
Logger.Infof("initialized heap dump capture; threshold: %f; max captures: %d; dir: %s", HeapdumpThreshold, HeapdumpMaxCaptures, HeapdumpDir)
|
|
|
|
}
|
|
|
|
|
|
|
|
func maybeCaptureHeapdump(usage, limit uint64) {
|
|
|
|
if _watchdog.hdleft <= 0 {
|
|
|
|
// nothing to do; no captures remaining (or captures disabled), or
|
|
|
|
// already captured a heapdump for this episode.
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if float64(usage)/float64(limit) < HeapdumpThreshold {
|
|
|
|
// we are below the threshold, reset the hdcurr flag.
|
|
|
|
_watchdog.hdcurr = false
|
|
|
|
return
|
|
|
|
}
|
|
|
|
// we are above the threshold.
|
|
|
|
if _watchdog.hdcurr {
|
|
|
|
return // we've already captured this episode, skip.
|
|
|
|
}
|
|
|
|
path := filepath.Join(HeapdumpDir, time.Now().Format(time.RFC3339Nano)+".heap")
|
|
|
|
file, err := os.Create(path)
|
|
|
|
if err != nil {
|
|
|
|
Logger.Warnf("failed to write heapdump; failed to create file in path %s; err: %s", path, err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
debug.WriteHeapDump(file.Fd())
|
|
|
|
Logger.Infof("heap dump captured; path: %s", path)
|
|
|
|
_watchdog.hdcurr = true
|
|
|
|
_watchdog.hdleft--
|
|
|
|
}
|