Commit e42dd4159f241dedbefec10fd8c56864ddd890b4 - golang-github-vbauerster-mpb

bar SetRefill: make sure refill doesn't overflow current Vladimir Bauer 2 years ago

2 changed file(s) with 8 addition(s) and 2 deletion(s). Raw diff Collapse all Expand all

-1

bar.go less more

143	143	// operation for example.
144	144	func (b *Bar) SetRefill(amount int64) {
145	145	select {
146		case b.operateState <- func(s *bState) { s.refill = amount }:
	146	case b.operateState <- func(s *bState) {
	147	if amount < s.current {
	148	s.refill = amount
	149	} else {
	150	s.refill = s.current
	151	}
	152	}:
147	153	case <-b.done:
148	154	}
149	155	}

-1

bar_test.go less more

146	146
147	147	bar := p.New(int64(total), mpb.BarStyle().Refiller(refiller), mpb.BarFillerTrim())
148	148
	149	bar.IncrBy(till)
149	150	bar.SetRefill(int64(till))
150		bar.IncrBy(till)
151	151	bar.IncrBy(total - till)
152	152
153	153	p.Wait()