mirror of
https://github.com/matrix-org/dendrite
synced 2024-12-25 08:53:49 +01:00
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package process
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
|
|
"github.com/getsentry/sentry-go"
|
|
"github.com/sirupsen/logrus"
|
|
"go.uber.org/atomic"
|
|
)
|
|
|
|
type ProcessContext struct {
|
|
wg *sync.WaitGroup // used to wait for components to shutdown
|
|
ctx context.Context // cancelled when Stop is called
|
|
shutdown context.CancelFunc // shut down Dendrite
|
|
degraded atomic.Bool
|
|
}
|
|
|
|
func NewProcessContext() *ProcessContext {
|
|
ctx, shutdown := context.WithCancel(context.Background())
|
|
return &ProcessContext{
|
|
ctx: ctx,
|
|
shutdown: shutdown,
|
|
wg: &sync.WaitGroup{},
|
|
}
|
|
}
|
|
|
|
func (b *ProcessContext) Context() context.Context {
|
|
return context.WithValue(b.ctx, "scope", "process") // nolint:staticcheck
|
|
}
|
|
|
|
func (b *ProcessContext) ComponentStarted() {
|
|
b.wg.Add(1)
|
|
}
|
|
|
|
func (b *ProcessContext) ComponentFinished() {
|
|
b.wg.Done()
|
|
}
|
|
|
|
func (b *ProcessContext) ShutdownDendrite() {
|
|
b.shutdown()
|
|
}
|
|
|
|
func (b *ProcessContext) WaitForShutdown() <-chan struct{} {
|
|
return b.ctx.Done()
|
|
}
|
|
|
|
func (b *ProcessContext) WaitForComponentsToFinish() {
|
|
b.wg.Wait()
|
|
}
|
|
|
|
func (b *ProcessContext) Degraded() {
|
|
if b.degraded.CAS(false, true) {
|
|
logrus.Warn("Dendrite is running in a degraded state")
|
|
sentry.CaptureException(fmt.Errorf("Process is running in a degraded state"))
|
|
}
|
|
}
|
|
|
|
func (b *ProcessContext) IsDegraded() bool {
|
|
return b.degraded.Load()
|
|
}
|