mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-20 08:11:08 +00:00
31 lines
589 B
Go
31 lines
589 B
Go
package bbgo
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
const IsolationContextKey = "bbgo"
|
|
|
|
var defaultIsolation = NewIsolation()
|
|
|
|
type Isolation struct {
|
|
gracefulShutdown GracefulShutdown
|
|
}
|
|
|
|
func NewIsolation() *Isolation {
|
|
return &Isolation{}
|
|
}
|
|
|
|
func NewIsolationFromContext(ctx context.Context) *Isolation {
|
|
isolatedContext, ok := ctx.Value(IsolationContextKey).(*Isolation)
|
|
if ok {
|
|
return isolatedContext
|
|
}
|
|
|
|
return defaultIsolation
|
|
}
|
|
|
|
func NewContextWithIsolation(parent context.Context, isolation *Isolation) context.Context {
|
|
return context.WithValue(parent, IsolationContextKey, isolation)
|
|
}
|