mirrors_public/schollz_croc
1
1
Fork 0
mirror of https://github.com/schollz/croc.git synced 2025-10-11 13:21:00 +02:00
Code Activity

sender is more accurate bar

Browse source
This commit is contained in:
Zack Scholl 2019-04-11 07:46:57 -07:00
parent 5e7839a282
commit 67abc97c6a
1 changed files with 14 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/webrtc/pkg/session/sender/sender.go
View file

@ -60,6 +60,8 @@ type Session struct {
// Stats/infos
readingStats *stats.Stats
bar *progressbar.ProgressBar
fileSize int64
firstByte bool
}
// New creates a new sender session
@ -177,13 +179,8 @@ func (s *Session) readFile(pathToFile string) error {
return err
}
stat, _ := f.Stat()
s.bar = progressbar.NewOptions64(
stat.Size(),
progressbar.OptionSetRenderBlankState(true),
progressbar.OptionSetBytes64(stat.Size()),
progressbar.OptionSetWriter(os.Stderr),
progressbar.OptionThrottle(1/60*time.Second),
)
s.fileSize = stat.Size()
s.firstByte = true
log.Debugf("Starting to read data from '%s'", pathToFile)
s.readingStats.Start()
defer func() {
@ -248,6 +245,16 @@ func (s *Session) onBufferedAmountLow() func() {
return
}
s.sess.NetworkStats.AddBytes(uint64(cur.n))
if s.firstByte {
s.firstByte = false
s.bar = progressbar.NewOptions64(
s.fileSize,
progressbar.OptionSetRenderBlankState(true),
progressbar.OptionSetBytes64(s.fileSize),
progressbar.OptionSetWriter(os.Stderr),
progressbar.OptionThrottle(1/60*time.Second),
)
}
s.bar.Add(cur.n)
s.msgToBeSent = s.msgToBeSent[1:]
}