mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-11-26 23:46:08 +00:00
3bfd50a3d9
Just some name changes, put in a stand in emission amount, and started copying the algo from Karlsen. Not release worthy yet. Therefore Dev branch exists now. Also, for now this is for research purposes only. I got no clue what to build on top of Kaspa yet. Help would be appreciated for ideas and implementations.
119 lines
4.6 KiB
Go
119 lines
4.6 KiB
Go
package flowcontext
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/zoomy-network/zoomyd/util/mstime"
|
|
|
|
"github.com/zoomy-network/zoomyd/domain/consensus/model/externalapi"
|
|
|
|
"github.com/zoomy-network/zoomyd/domain"
|
|
|
|
peerpkg "github.com/zoomy-network/zoomyd/app/protocol/peer"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/config"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/network/addressmanager"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/network/connmanager"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/network/netadapter"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/network/netadapter/id"
|
|
)
|
|
|
|
// OnNewBlockTemplateHandler is a handler function that's triggered when a new block template is available
|
|
type OnNewBlockTemplateHandler func() error
|
|
|
|
// OnPruningPointUTXOSetOverrideHandler is a handle function that's triggered whenever the UTXO set
|
|
// resets due to pruning point change via IBD.
|
|
type OnPruningPointUTXOSetOverrideHandler func() error
|
|
|
|
// OnTransactionAddedToMempoolHandler is a handler function that's triggered
|
|
// when a transaction is added to the mempool
|
|
type OnTransactionAddedToMempoolHandler func()
|
|
|
|
// FlowContext holds state that is relevant to more than one flow or one peer, and allows communication between
|
|
// different flows that can be associated to different peers.
|
|
type FlowContext struct {
|
|
cfg *config.Config
|
|
netAdapter *netadapter.NetAdapter
|
|
domain domain.Domain
|
|
addressManager *addressmanager.AddressManager
|
|
connectionManager *connmanager.ConnectionManager
|
|
|
|
timeStarted int64
|
|
|
|
onNewBlockTemplateHandler OnNewBlockTemplateHandler
|
|
onPruningPointUTXOSetOverrideHandler OnPruningPointUTXOSetOverrideHandler
|
|
onTransactionAddedToMempoolHandler OnTransactionAddedToMempoolHandler
|
|
|
|
lastRebroadcastTime time.Time
|
|
sharedRequestedTransactions *SharedRequestedTransactions
|
|
|
|
sharedRequestedBlocks *SharedRequestedBlocks
|
|
|
|
ibdPeer *peerpkg.Peer
|
|
ibdPeerMutex sync.RWMutex
|
|
|
|
peers map[id.ID]*peerpkg.Peer
|
|
peersMutex sync.RWMutex
|
|
|
|
orphans map[externalapi.DomainHash]*externalapi.DomainBlock
|
|
orphansMutex sync.RWMutex
|
|
|
|
transactionIDsToPropagate []*externalapi.DomainTransactionID
|
|
lastTransactionIDPropagationTime time.Time
|
|
transactionIDPropagationLock sync.Mutex
|
|
|
|
shutdownChan chan struct{}
|
|
}
|
|
|
|
// New returns a new instance of FlowContext.
|
|
func New(cfg *config.Config, domain domain.Domain, addressManager *addressmanager.AddressManager,
|
|
netAdapter *netadapter.NetAdapter, connectionManager *connmanager.ConnectionManager) *FlowContext {
|
|
|
|
return &FlowContext{
|
|
cfg: cfg,
|
|
netAdapter: netAdapter,
|
|
domain: domain,
|
|
addressManager: addressManager,
|
|
connectionManager: connectionManager,
|
|
sharedRequestedTransactions: NewSharedRequestedTransactions(),
|
|
sharedRequestedBlocks: NewSharedRequestedBlocks(),
|
|
peers: make(map[id.ID]*peerpkg.Peer),
|
|
orphans: make(map[externalapi.DomainHash]*externalapi.DomainBlock),
|
|
timeStarted: mstime.Now().UnixMilliseconds(),
|
|
transactionIDsToPropagate: []*externalapi.DomainTransactionID{},
|
|
lastTransactionIDPropagationTime: time.Now(),
|
|
shutdownChan: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
// Close signals to all flows the the protocol manager is closed.
|
|
func (f *FlowContext) Close() {
|
|
close(f.shutdownChan)
|
|
}
|
|
|
|
// ShutdownChan is a chan where flows can subscribe to shutdown
|
|
// event.
|
|
func (f *FlowContext) ShutdownChan() <-chan struct{} {
|
|
return f.shutdownChan
|
|
}
|
|
|
|
// IsNearlySynced returns whether current consensus is considered synced or close to being synced.
|
|
func (f *FlowContext) IsNearlySynced() (bool, error) {
|
|
return f.Domain().Consensus().IsNearlySynced()
|
|
}
|
|
|
|
// SetOnNewBlockTemplateHandler sets the onNewBlockTemplateHandler handler
|
|
func (f *FlowContext) SetOnNewBlockTemplateHandler(onNewBlockTemplateHandler OnNewBlockTemplateHandler) {
|
|
f.onNewBlockTemplateHandler = onNewBlockTemplateHandler
|
|
}
|
|
|
|
// SetOnPruningPointUTXOSetOverrideHandler sets the onPruningPointUTXOSetOverrideHandler handler
|
|
func (f *FlowContext) SetOnPruningPointUTXOSetOverrideHandler(onPruningPointUTXOSetOverrideHandler OnPruningPointUTXOSetOverrideHandler) {
|
|
f.onPruningPointUTXOSetOverrideHandler = onPruningPointUTXOSetOverrideHandler
|
|
}
|
|
|
|
// SetOnTransactionAddedToMempoolHandler sets the onTransactionAddedToMempool handler
|
|
func (f *FlowContext) SetOnTransactionAddedToMempoolHandler(onTransactionAddedToMempoolHandler OnTransactionAddedToMempoolHandler) {
|
|
f.onTransactionAddedToMempoolHandler = onTransactionAddedToMempoolHandler
|
|
}
|