ui/progress: Load both values in a single Lock/Unlock

We always need both values, except in a test, so we don't need to lock
twice and risk scheduling in between.

Also, removed the resetting in Done. This copied a mutex, which isn't
allowed. Static analyzers tend to trip over that.
This commit is contained in:
greatroar
2022-10-25 07:41:44 +02:00
parent b57d42905c
commit 2dafda9164
2 changed files with 14 additions and 18 deletions

View File

@@ -78,32 +78,25 @@ func (c *Counter) Done() {
c.tick.Stop()
}
close(c.stop)
<-c.stopped // Wait for last progress report.
*c = Counter{} // Prevent reuse.
<-c.stopped // Wait for last progress report.
}
// Get the current Counter value. This method is concurrency-safe.
func (c *Counter) Get() uint64 {
// Get returns the current value and the maximum of c.
// This method is concurrency-safe.
func (c *Counter) Get() (v, max uint64) {
c.valueMutex.Lock()
v := c.value
v, max = c.value, c.max
c.valueMutex.Unlock()
return v
}
func (c *Counter) getMax() uint64 {
c.valueMutex.Lock()
max := c.max
c.valueMutex.Unlock()
return max
return v, max
}
func (c *Counter) run() {
defer close(c.stopped)
defer func() {
// Must be a func so that time.Since isn't called at defer time.
c.report(c.Get(), c.getMax(), time.Since(c.start), true)
v, max := c.Get()
c.report(v, max, time.Since(c.start), true)
}()
var tick <-chan time.Time
@@ -123,6 +116,7 @@ func (c *Counter) run() {
return
}
c.report(c.Get(), c.getMax(), now.Sub(c.start), false)
v, max := c.Get()
c.report(v, max, now.Sub(c.start), false)
}
}