queue.go

// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package ids

import (
	"container/list"
)

// QueueSet is a set of IDs stored in fifo order
type QueueSet struct {
	idList *list.List
}

func (qs *QueueSet) init() {
	if qs.idList == nil {
		qs.idList = list.New()
	}
}

// SetHead ...
func (qs *QueueSet) SetHead(id ID) {
	qs.init()

	for qs.idList.Len() > 0 {
		element := qs.idList.Front()
		head := element.Value.(ID)
		if head.Equals(id) {
			return
		}
		qs.idList.Remove(element)
	}

	qs.idList.PushFront(id)
}

// Append ...
func (qs *QueueSet) Append(id ID) {
	qs.init()

	qs.idList.PushBack(id)
}

// GetTail ...
func (qs *QueueSet) GetTail() ID {
	qs.init()

	if qs.idList.Len() == 0 {
		return ID{}
	}
	return qs.idList.Back().Value.(ID)
}