70 lines
1.2 KiB
Go
70 lines
1.2 KiB
Go
package wrapper
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/benbjohnson/clock"
|
|
)
|
|
|
|
// ChargeTimer measures charging time between start and stop events
|
|
type ChargeTimer struct {
|
|
sync.Mutex
|
|
clck clock.Clock
|
|
|
|
charging bool
|
|
start time.Time
|
|
duration time.Duration
|
|
}
|
|
|
|
// NewChargeTimer creates ChargeTimer for tracking duration between
|
|
// start and stop events
|
|
func NewChargeTimer() *ChargeTimer {
|
|
return &ChargeTimer{
|
|
clck: clock.New(),
|
|
}
|
|
}
|
|
|
|
// StartCharge signals charge timer start
|
|
func (m *ChargeTimer) StartCharge(continued bool) {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
|
|
m.start = m.clck.Now()
|
|
|
|
if continued {
|
|
m.charging = true
|
|
} else {
|
|
m.duration = 0
|
|
}
|
|
}
|
|
|
|
// StopCharge signals charge timer stop
|
|
func (m *ChargeTimer) StopCharge() {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
|
|
m.charging = false
|
|
m.duration += m.clck.Since(m.start)
|
|
}
|
|
|
|
var _ ChargeResetter = (*ChargeTimer)(nil)
|
|
|
|
// ChargeResetter resets the charging session
|
|
func (m *ChargeTimer) ResetCharge() {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
|
|
m.duration = 0
|
|
}
|
|
|
|
// ChargeDuration implements the api.ChargeTimer interface
|
|
func (m *ChargeTimer) ChargeDuration() (time.Duration, error) {
|
|
m.Lock()
|
|
defer m.Unlock()
|
|
|
|
if m.charging {
|
|
return m.duration + m.clck.Since(m.start), nil
|
|
}
|
|
return m.duration, nil
|
|
}
|