The core protocol of WoopChain
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
woop/internal/utils/timer.go

64 lines
1.4 KiB

package utils
import (
"time"
)
// TimeoutState indicates the state of Timeout class
type TimeoutState int
// Enum for different TimeoutState
const (
Active TimeoutState = iota
Inactive
Expired
)
// Timeout is the implementation of timeout
type Timeout struct {
state TimeoutState
d time.Duration
start time.Time
}
// NewTimeout creates a new timeout class
func NewTimeout(d time.Duration) *Timeout {
timeout := Timeout{state: Inactive, d: d, start: time.Now()}
return &timeout
}
// Start starts the timeout clock
func (timeout *Timeout) Start() {
timeout.state = Active
timeout.start = time.Now()
}
// Stop stops the timeout clock
func (timeout *Timeout) Stop() {
timeout.state = Inactive
timeout.start = time.Now()
}
// CheckExpire checks whether the timeout is reached/expired
func (timeout *Timeout) CheckExpire() bool {
if timeout.state == Active && time.Since(timeout.start) > timeout.d {
timeout.state = Expired
}
if timeout.state == Expired {
return true
}
return false
}
// Duration returns the duration period of timeout
func (timeout *Timeout) Duration() time.Duration {
return timeout.d
}
// IsActive checks whether timeout clock is active;
// A timeout is active means it's not stopped caused by stop
// and also not expired with time elapses longer than duration from start
func (timeout *Timeout) IsActive() bool {
return timeout.state == Active
}