mirror of
https://github.com/zrepl/zrepl.git
synced 2024-12-22 23:20:51 +01:00
31 lines
578 B
Go
31 lines
578 B
Go
package frameconn
|
|
|
|
import "sync"
|
|
|
|
type shutdownFSM struct {
|
|
mtx sync.Mutex
|
|
state shutdownFSMState
|
|
}
|
|
|
|
type shutdownFSMState uint32
|
|
|
|
const (
|
|
// zero value is important
|
|
shutdownStateOpen shutdownFSMState = iota
|
|
shutdownStateBegin
|
|
)
|
|
|
|
func (f *shutdownFSM) Begin() (thisCallStartedShutdown bool) {
|
|
f.mtx.Lock()
|
|
defer f.mtx.Unlock()
|
|
thisCallStartedShutdown = f.state != shutdownStateOpen
|
|
f.state = shutdownStateBegin
|
|
return thisCallStartedShutdown
|
|
}
|
|
|
|
func (f *shutdownFSM) IsShuttingDown() bool {
|
|
f.mtx.Lock()
|
|
defer f.mtx.Unlock()
|
|
return f.state != shutdownStateOpen
|
|
}
|