Documentation ¶
Overview ¶
Package ThreadManager provides a utility for performing organized clean shutdown and quick shutdown of long running groups of threads, such as networking threads, background threads, or resources like databases.
The OnStop and AfterStop functions are helpers which enable shutdown code to be inlined with resource allocation, similar to defer. The difference is that `OnStop` and `AfterStop` will be called following tg.Stop, instead of when the parent function goes out of scope.
Index ¶
Constants ¶
This section is empty.
Variables ¶
var ErrStopped = errors.New("ThreadManager already stopped")
ErrStopped is returned by ThreadManager methods if Stop has already been called.
Functions ¶
This section is empty.
Types ¶
type ThreadManager ¶
type ThreadManager struct {
// contains filtered or unexported fields
}
A ThreadManager is a one-time-use object to manage the life cycle of a group of threads. It is a sync.WaitGroup that provides functions for coordinating actions and shutting down threads. After Stop() is called, the thread group is no longer useful.
It is safe to call Add(), Done(), and Stop() concurrently.
func (*ThreadManager) Add ¶
func (tg *ThreadManager) Add() error
Add increments the thread group counter.
func (*ThreadManager) AfterStop ¶
func (tg *ThreadManager) AfterStop(fn func() error) error
AfterStop ensures that a function will be called after Stop() has been called and after all running routines have called Done(). The functions will be called in reverse order to how they were added, similar to defer. If Stop() has already been called, the input function will be called immediately, and a composition of ErrStopped and the error from calling fn will be returned.
The primary use of AfterStop is to allow code that opens and closes resources to be positioned next to each other. The purpose is similar to `defer`, except for resources that outlive the function which creates them.
func (*ThreadManager) Done ¶
func (tg *ThreadManager) Done()
Done decrements the thread group counter.
func (*ThreadManager) OnStop ¶
func (tg *ThreadManager) OnStop(fn func() error) error
OnStop ensures that a function will be called after Stop() has been called, and before blocking until all running routines have called Done(). It is safe to use OnStop to coordinate the closing of long-running threads. The OnStop functions will be called in the reverse order in which they were added, similar to defer. If Stop() has already been called, the input function will be called immediately, and a composition of ErrStopped and the error from calling fn will be returned.
func (*ThreadManager) Stop ¶
func (tg *ThreadManager) Stop() error
Stop will close the stop channel of the thread group, then call all 'OnStop' functions in reverse order, then will wait until the thread group counter reaches zero, then will call all of the 'AfterStop' functions in reverse order.
The errors returned by the OnStop and AfterStop functions will be composed into a single error.
func (*ThreadManager) StopChan ¶
func (tg *ThreadManager) StopChan() <-chan struct{}
StopChan provides read-only access to the ThreadGroup's stopChan. Callers should select on StopChan in order to interrupt long-running reads (such as time.After).