mirror of https://github.com/hashicorp/consul
77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
/*
|
|
Copyright (c) 2014 VMware, Inc. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package progress
|
|
|
|
type scaledReport struct {
|
|
Report
|
|
n int
|
|
i int
|
|
}
|
|
|
|
func (r scaledReport) Percentage() float32 {
|
|
b := 100 * float32(r.i) / float32(r.n)
|
|
return b + (r.Report.Percentage() / float32(r.n))
|
|
}
|
|
|
|
type scaleOne struct {
|
|
s Sinker
|
|
n int
|
|
i int
|
|
}
|
|
|
|
func (s scaleOne) Sink() chan<- Report {
|
|
upstream := make(chan Report)
|
|
downstream := s.s.Sink()
|
|
go s.loop(upstream, downstream)
|
|
return upstream
|
|
}
|
|
|
|
func (s scaleOne) loop(upstream <-chan Report, downstream chan<- Report) {
|
|
defer close(downstream)
|
|
|
|
for r := range upstream {
|
|
downstream <- scaledReport{
|
|
Report: r,
|
|
n: s.n,
|
|
i: s.i,
|
|
}
|
|
}
|
|
}
|
|
|
|
type scaleMany struct {
|
|
s Sinker
|
|
n int
|
|
i int
|
|
}
|
|
|
|
func Scale(s Sinker, n int) Sinker {
|
|
return &scaleMany{
|
|
s: s,
|
|
n: n,
|
|
}
|
|
}
|
|
|
|
func (s *scaleMany) Sink() chan<- Report {
|
|
if s.i == s.n {
|
|
s.n++
|
|
}
|
|
|
|
ch := scaleOne{s: s.s, n: s.n, i: s.i}.Sink()
|
|
s.i++
|
|
return ch
|
|
}
|