bbgo_origin/pkg/bbgo/graceful_callbacks.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)
	}
}