mirror of
https://github.com/samuelncui/yatm.git
synced 2026-01-03 11:45:21 +00:00
fix: change tape error
This commit is contained in:
@@ -1,8 +1,55 @@
|
||||
package executor
|
||||
|
||||
import (
|
||||
"sync/atomic"
|
||||
"time"
|
||||
)
|
||||
|
||||
const SpeedLen = 30
|
||||
|
||||
type speedEvent struct {
|
||||
bytes int64
|
||||
time time.Time
|
||||
}
|
||||
|
||||
type progress struct {
|
||||
speed int64
|
||||
speedEvents []speedEvent
|
||||
speedLen int
|
||||
speedIdx int
|
||||
|
||||
startTime time.Time
|
||||
speed int64
|
||||
|
||||
totalBytes, totalFiles int64
|
||||
bytes, files int64
|
||||
}
|
||||
|
||||
func newProgress() *progress {
|
||||
return &progress{speedEvents: make([]speedEvent, SpeedLen), speedLen: SpeedLen, startTime: time.Now()}
|
||||
}
|
||||
|
||||
func (p *progress) setBytes(bytes int64) {
|
||||
atomic.StoreInt64(&p.bytes, bytes)
|
||||
now := time.Now()
|
||||
|
||||
p.speedEvents[p.speedIdx] = speedEvent{bytes: bytes, time: now}
|
||||
for earliest := p.speedIdx; ; {
|
||||
earliest++
|
||||
if earliest >= p.speedLen {
|
||||
earliest = 0
|
||||
}
|
||||
if earliest == p.speedIdx {
|
||||
break
|
||||
}
|
||||
|
||||
if !p.speedEvents[earliest].time.IsZero() {
|
||||
p.speed = (bytes - p.speedEvents[earliest].bytes) * 1e9 / now.Sub(p.speedEvents[earliest].time).Nanoseconds()
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
p.speedIdx++
|
||||
if p.speedIdx >= p.speedLen {
|
||||
p.speedIdx = 0
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user