Codebase list golang-github-vbauerster-mpb / c954111 progress_posix.go
c954111

Tree @c954111 (Download .tar.gz)

progress_posix.go @c954111raw · history · blame

// +build !windows

package mpb

import (
	"fmt"
	"os"
	"os/signal"
	"runtime"
	"syscall"
	"time"

	"github.com/vbauerster/mpb/cwriter"
)

func (p *Progress) serve(s *pState) {
	winch := make(chan os.Signal, 1)
	signal.Notify(winch, syscall.SIGWINCH)

	var numP, numA int
	var timer *time.Timer
	var resumeTicker <-chan time.Time
	resumeDelay := 320 * time.Millisecond

	for {
		select {
		case op := <-p.operateState:
			op(s)
		case <-s.ticker.C:
			if s.bHeap.Len() == 0 {
				runtime.Gosched()
				break
			}
			if s.heapUpdated {
				numP = s.bHeap.maxNumP()
				numA = s.bHeap.maxNumA()
				s.heapUpdated = false
			}
			tw, _, _ := cwriter.TermSize()
			err := s.writeAndFlush(tw, numP, numA)
			if err != nil {
				fmt.Fprintln(os.Stderr, err)
			}
		case <-winch:
			tw, _, _ := cwriter.TermSize()
			err := s.writeAndFlush(tw-tw/8, numP, numA)
			if err != nil {
				fmt.Fprintln(os.Stderr, err)
			}
			if timer != nil && timer.Reset(resumeDelay) {
				break
			}
			s.ticker.Stop()
			timer = time.NewTimer(resumeDelay)
			resumeTicker = timer.C
		case <-resumeTicker:
			s.ticker = time.NewTicker(s.rr)
			resumeTicker = nil
		case <-s.cancel:
			s.ticker.Stop()
			s.cancel = nil
			// don't return here, p.Stop() must be called eventually
		case <-p.quit:
			if s.cancel != nil {
				s.ticker.Stop()
			}
			if s.shutdownNotifier != nil {
				close(s.shutdownNotifier)
			}
			signal.Stop(winch)
			close(p.quit)
			return
		}
	}
}