pauser.go

package go_engine_io_parser

import "sync"

type pauserStatus int

const (
	statusNormal pauserStatus = iota
	statusPausing
	statusPaused
)

type pauser struct {
	lock sync.Mutex
	c    *sync.Cond

	worker int
	status pauserStatus

	pausing chan struct{}
	paused  chan struct{}
}

func newPauser() *pauser {
	ret := &pauser{
		pausing: make(chan struct{}),
		paused:  make(chan struct{}),
		status:  statusNormal,
	}
	ret.c = sync.NewCond(&ret.lock)
	return ret
}

func (p *pauser) Pause() bool {
	p.lock.Lock()
	defer p.lock.Unlock()

	switch p.status {
	case statusPaused:
		return false
	case statusNormal:
		close(p.pausing)
		p.status = statusPausing
	}

	for p.worker != 0 {
		p.c.Wait()
	}

	if p.status == statusPaused {
		return false
	}
	close(p.paused)
	p.status = statusPaused
	p.c.Broadcast()

	return true
}

func (p *pauser) Resume() {
	p.lock.Lock()
	defer p.lock.Unlock()

	p.status = statusNormal
	p.paused = make(chan struct{})
	p.pausing = make(chan struct{})
}

func (p *pauser) Working() bool {
	p.lock.Lock()
	defer p.lock.Unlock()

	if p.status == statusPaused {
		return false
	}
	p.worker++
	return true
}

func (p *pauser) Done() {
	p.lock.Lock()
	defer p.lock.Unlock()

	if p.status == statusPaused || p.worker == 0 {
		return
	}
	p.worker--
	p.c.Broadcast()
}

func (p *pauser) PausingTrigger() <-chan struct{} {
	p.lock.Lock()
	defer p.lock.Unlock()

	return p.pausing
}

func (p *pauser) PausedTrigger() <-chan struct{} {
	p.lock.Lock()
	defer p.lock.Unlock()

	return p.paused
}