26 lines
536 B
Go
26 lines
536 B
Go
package helpers
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.uber.org/fx"
|
|
)
|
|
|
|
// MetricsCtx is a context wrapper with metrics
|
|
type MetricsCtx context.Context
|
|
|
|
// LifecycleCtx creates a context which will be cancelled when lifecycle stops
|
|
//
|
|
// This is a hack which we need because most of our services use contexts in a
|
|
// wrong way
|
|
func LifecycleCtx(mctx MetricsCtx, lc fx.Lifecycle) context.Context {
|
|
ctx, cancel := context.WithCancel(mctx)
|
|
lc.Append(fx.Hook{
|
|
OnStop: func(_ context.Context) error {
|
|
cancel()
|
|
return nil
|
|
},
|
|
})
|
|
return ctx
|
|
}
|