piece.go

// Completion ...
func (p *Piece) Completion() storage.Completion {
	p.mu.Lock()
	defer p.mu.Unlock()

	return storage.Completion{
		Complete: p.Completed,
		Ok:       true,
	}
}

// MarkComplete ...
func (p *Piece) MarkComplete() error {
	p.mu.Lock()
	defer p.mu.Unlock()

	log.Debugf("Complete: %#v", p.Index)

	if !p.Active || p.Size != p.Length || p.Length == 0 {
		return errors.New("piece is not complete")
	}

	p.Completed = true

	return nil
}

// MarkNotComplete ...
func (p *Piece) MarkNotComplete() error {
	p.mu.Lock()
	defer p.mu.Unlock()

	log.Debugf("NotComplete: %#v", p.Index)
	p.Completed = false
	return nil
}