cerc-io/lotus
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

worker: Fix progress bars

Browse Source
This commit is contained in:
Łukasz Magiera 2020-01-23 15:38:36 +01:00
parent 558e4c7669
commit f5540195de
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cmd/lotus-seal-worker/transfer.go
View File

@ -14,6 +14,13 @@ import (
"github.com/filecoin-project/lotus/lib/tarutil" "github.com/filecoin-project/lotus/lib/tarutil"
) )
func (w *worker) sizeForType(typ string) int64 {
size := int64(w.sb.SectorSize())
if typ == "cache" {
size *= 10
}
return size
}
func (w *worker) fetch(typ string, sectorID uint64) error { func (w *worker) fetch(typ string, sectorID uint64) error {
outname := filepath.Join(w.repo, typ, w.sb.SectorName(sectorID)) outname := filepath.Join(w.repo, typ, w.sb.SectorName(sectorID))
@ -37,7 +44,7 @@ func (w *worker) fetch(typ string, sectorID uint64) error {
return xerrors.Errorf("non-200 code: %d", resp.StatusCode) return xerrors.Errorf("non-200 code: %d", resp.StatusCode)
} }
bar := pb.New64(resp.ContentLength) bar := pb.New64(w.sizeForType(typ))
bar.ShowPercent = true bar.ShowPercent = true
bar.ShowSpeed = true bar.ShowSpeed = true
bar.Units = pb.U_BYTES bar.Units = pb.U_BYTES
@ -88,7 +95,7 @@ func (w *worker) push(typ string, sectorID uint64) error {
return xerrors.Errorf("opening push reader: %w", err) return xerrors.Errorf("opening push reader: %w", err)
} }
bar := pb.New64(0) bar := pb.New64(w.sizeForType(typ))
bar.ShowPercent = true bar.ShowPercent = true
bar.ShowSpeed = true bar.ShowSpeed = true
bar.ShowCounters = true bar.ShowCounters = true