2018-03-09 15:55:25 +00:00
|
|
|
package core
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/binary"
|
|
|
|
"io"
|
|
|
|
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/util"
|
|
|
|
)
|
|
|
|
|
|
|
|
// A HeaderHashList represents a list of header hashes.
|
2018-03-14 09:36:59 +00:00
|
|
|
// This datastructure in not routine safe and should be
|
|
|
|
// used under some kind of protection against race conditions.
|
2018-03-09 15:55:25 +00:00
|
|
|
type HeaderHashList struct {
|
|
|
|
hashes []util.Uint256
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewHeaderHashList return a new pointer to a HeaderHashList.
|
|
|
|
func NewHeaderHashList(hashes ...util.Uint256) *HeaderHashList {
|
|
|
|
return &HeaderHashList{
|
|
|
|
hashes: hashes,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Add appends the given hash to the list of hashes.
|
|
|
|
func (l *HeaderHashList) Add(h util.Uint256) {
|
|
|
|
l.hashes = append(l.hashes, h)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Len return the length of the underlying hashes slice.
|
|
|
|
func (l *HeaderHashList) Len() int {
|
|
|
|
return len(l.hashes)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Get returns the hash by the given index.
|
|
|
|
func (l *HeaderHashList) Get(i int) util.Uint256 {
|
2018-03-14 09:36:59 +00:00
|
|
|
if l.Len() <= i {
|
2018-03-09 15:55:25 +00:00
|
|
|
return util.Uint256{}
|
|
|
|
}
|
|
|
|
return l.hashes[i]
|
|
|
|
}
|
|
|
|
|
|
|
|
// Last return the last hash in the HeaderHashList.
|
|
|
|
func (l *HeaderHashList) Last() util.Uint256 {
|
|
|
|
return l.hashes[l.Len()-1]
|
|
|
|
}
|
|
|
|
|
|
|
|
// Slice return a subslice of the underlying hashes.
|
|
|
|
// Subsliced from start to end.
|
|
|
|
// Example:
|
|
|
|
// headers := headerList.Slice(0, 2000)
|
|
|
|
func (l *HeaderHashList) Slice(start, end int) []util.Uint256 {
|
|
|
|
return l.hashes[start:end]
|
|
|
|
}
|
|
|
|
|
|
|
|
// WriteTo will write n underlying hashes to the given io.Writer
|
|
|
|
// starting from start.
|
|
|
|
func (l *HeaderHashList) Write(w io.Writer, start, n int) error {
|
|
|
|
if err := util.WriteVarUint(w, uint64(n)); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
hashes := l.Slice(start, start+n)
|
|
|
|
for _, hash := range hashes {
|
|
|
|
if err := binary.Write(w, binary.LittleEndian, hash); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|