mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-24 07:45:15 +00:00
19 lines
386 B
Go
19 lines
386 B
Go
|
// Code generated by "callbackgen -type Graceful"; DO NOT EDIT.
|
||
|
|
|
||
|
|
package bbgo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (g *Graceful) OnShutdown(cb func(ctx context.Context, wg *sync.WaitGroup)) {
|
||
|
|
g.shutdownCallbacks = append(g.shutdownCallbacks, cb)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *Graceful) EmitShutdown(ctx context.Context, wg *sync.WaitGroup) {
|
||
|
|
for _, cb := range g.shutdownCallbacks {
|
||
|
|
cb(ctx, wg)
|
||
|
|
}
|
||
|
|
}
|