mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-10-14 00:59:33 +00:00
3c88184b38
* [DEV-169] add to BlockDAG TipHashes() and HighestTipHash() and remove Tiphashes() and SelectedTipHash from VirtualBlock * [DEV-169] move highest node logic to separate method of blockset
111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
// Copyright (c) 2017 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package blockdag
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/daglabs/btcd/wire"
|
|
)
|
|
|
|
// VirtualBlock is a virtual block whose parents are the tips of the DAG.
|
|
type VirtualBlock struct {
|
|
mtx sync.Mutex
|
|
phantomK uint32
|
|
utxoSet *fullUTXOSet
|
|
blockNode
|
|
}
|
|
|
|
// newVirtualBlock creates and returns a new VirtualBlock.
|
|
func newVirtualBlock(tips blockSet, phantomK uint32) *VirtualBlock {
|
|
// The mutex is intentionally not held since this is a constructor.
|
|
var virtual VirtualBlock
|
|
virtual.phantomK = phantomK
|
|
virtual.utxoSet = NewFullUTXOSet()
|
|
virtual.setTips(tips)
|
|
|
|
return &virtual
|
|
}
|
|
|
|
// clone creates and returns a clone of the virtual block.
|
|
func (v *VirtualBlock) clone() *VirtualBlock {
|
|
return &VirtualBlock{
|
|
phantomK: v.phantomK,
|
|
utxoSet: v.utxoSet.clone().(*fullUTXOSet),
|
|
blockNode: v.blockNode,
|
|
}
|
|
}
|
|
|
|
// setTips replaces the tips of the virtual block with the blocks in the
|
|
// given blockSet. This only differs from the exported version in that it
|
|
// is up to the caller to ensure the lock is held.
|
|
//
|
|
// This function MUST be called with the view mutex locked (for writes).
|
|
func (v *VirtualBlock) setTips(tips blockSet) {
|
|
v.blockNode = *newBlockNode(nil, tips, v.phantomK)
|
|
}
|
|
|
|
// SetTips replaces the tips of the virtual block with the blocks in the
|
|
// given blockSet.
|
|
//
|
|
// This function is safe for concurrent access.
|
|
func (v *VirtualBlock) SetTips(tips blockSet) {
|
|
v.mtx.Lock()
|
|
v.setTips(tips)
|
|
v.mtx.Unlock()
|
|
}
|
|
|
|
// addTip adds the given tip to the set of tips in the virtual block.
|
|
// All former tips that happen to be the given tips parents are removed
|
|
// from the set. This only differs from the exported version in that it
|
|
// is up to the caller to ensure the lock is held.
|
|
//
|
|
// This function MUST be called with the view mutex locked (for writes).
|
|
func (v *VirtualBlock) addTip(newTip *blockNode) {
|
|
updatedTips := v.tips().clone()
|
|
for _, parent := range newTip.parents {
|
|
updatedTips.remove(parent)
|
|
}
|
|
|
|
updatedTips.add(newTip)
|
|
v.setTips(updatedTips)
|
|
}
|
|
|
|
// AddTip adds the given tip to the set of tips in the virtual block.
|
|
// All former tips that happen to be the given tip's parents are removed
|
|
// from the set.
|
|
//
|
|
// This function is safe for concurrent access.
|
|
func (v *VirtualBlock) AddTip(newTip *blockNode) {
|
|
v.mtx.Lock()
|
|
v.addTip(newTip)
|
|
v.mtx.Unlock()
|
|
}
|
|
|
|
// tips returns the current tip block nodes for the DAG. It will return
|
|
// an empty blockSet if there is no tip.
|
|
//
|
|
// This function is safe for concurrent access.
|
|
func (v *VirtualBlock) tips() blockSet {
|
|
return v.parents
|
|
}
|
|
|
|
// SelectedTip returns the current selected tip for the DAG.
|
|
// It will return nil if there is no tip.
|
|
//
|
|
// This function is safe for concurrent access.
|
|
func (v *VirtualBlock) SelectedTip() *blockNode {
|
|
return v.selectedParent
|
|
}
|
|
|
|
// GetUTXOEntry returns the requested unspent transaction output. The returned
|
|
// instance must be treated as immutable since it is shared by all callers.
|
|
//
|
|
// This function is safe for concurrent access. However, the returned entry (if
|
|
// any) is NOT.
|
|
func (v *VirtualBlock) GetUTXOEntry(outPoint wire.OutPoint) (*UTXOEntry, bool) {
|
|
return v.utxoSet.get(outPoint)
|
|
}
|