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.
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package flowcontext
|
|
|
|
import (
|
|
"github.com/zoomy-network/zoomyd/app/appmessage"
|
|
"github.com/zoomy-network/zoomyd/app/protocol/common"
|
|
peerpkg "github.com/zoomy-network/zoomyd/app/protocol/peer"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/network/connmanager"
|
|
"github.com/zoomy-network/zoomyd/infrastructure/network/netadapter"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// NetAdapter returns the net adapter that is associated to the flow context.
|
|
func (f *FlowContext) NetAdapter() *netadapter.NetAdapter {
|
|
return f.netAdapter
|
|
}
|
|
|
|
// ConnectionManager returns the connection manager that is associated to the flow context.
|
|
func (f *FlowContext) ConnectionManager() *connmanager.ConnectionManager {
|
|
return f.connectionManager
|
|
}
|
|
|
|
// AddToPeers marks this peer as ready and adds it to the ready peers list.
|
|
func (f *FlowContext) AddToPeers(peer *peerpkg.Peer) error {
|
|
f.peersMutex.Lock()
|
|
defer f.peersMutex.Unlock()
|
|
|
|
if _, ok := f.peers[*peer.ID()]; ok {
|
|
return errors.Wrapf(common.ErrPeerWithSameIDExists, "peer with ID %s already exists", peer.ID())
|
|
}
|
|
|
|
f.peers[*peer.ID()] = peer
|
|
|
|
return nil
|
|
}
|
|
|
|
// RemoveFromPeers remove this peer from the peers list.
|
|
func (f *FlowContext) RemoveFromPeers(peer *peerpkg.Peer) {
|
|
f.peersMutex.Lock()
|
|
defer f.peersMutex.Unlock()
|
|
|
|
delete(f.peers, *peer.ID())
|
|
}
|
|
|
|
// readyPeerConnections returns the NetConnections of all the ready peers.
|
|
func (f *FlowContext) readyPeerConnections() []*netadapter.NetConnection {
|
|
f.peersMutex.RLock()
|
|
defer f.peersMutex.RUnlock()
|
|
peerConnections := make([]*netadapter.NetConnection, len(f.peers))
|
|
i := 0
|
|
for _, peer := range f.peers {
|
|
peerConnections[i] = peer.Connection()
|
|
i++
|
|
}
|
|
return peerConnections
|
|
}
|
|
|
|
// Broadcast broadcast the given message to all the ready peers.
|
|
func (f *FlowContext) Broadcast(message appmessage.Message) error {
|
|
return f.netAdapter.P2PBroadcast(f.readyPeerConnections(), message)
|
|
}
|
|
|
|
// Peers returns the currently active peers
|
|
func (f *FlowContext) Peers() []*peerpkg.Peer {
|
|
f.peersMutex.RLock()
|
|
defer f.peersMutex.RUnlock()
|
|
|
|
peers := make([]*peerpkg.Peer, len(f.peers))
|
|
i := 0
|
|
for _, peer := range f.peers {
|
|
peers[i] = peer
|
|
i++
|
|
}
|
|
return peers
|
|
}
|
|
|
|
// HasPeers returns whether there are currently active peers
|
|
func (f *FlowContext) HasPeers() bool {
|
|
f.peersMutex.RLock()
|
|
defer f.peersMutex.RUnlock()
|
|
return len(f.peers) > 0
|
|
}
|