mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-14 01:56:44 +00:00
42e50e6dc2
* [NOD-1191] Convert wire protocol to 100% protobuf * [NOD-1191] Simplify wire interface and remove redundant messages * [NOD-1191] Map all proto to wire conversions * [NOD-1203] Create netadapter outside of protocol manager * [NOD-1191] Fix nil errors * [NOD-1191] Fix comments * [NOD-1191] Add converter interface * [NOD-1191] Add missing GetBlockLocator message * [NOD-1191] Change message names that starts with 'get' to 'request' * [NOD-1191] Change message commands values * [NOD-1191] Remove redundant methods * [NOD-1191] Rename message constructors * [NOD-1191] Change message commands to use iota * [NOD-1191] Add missing outputs to protobuf conversion * [NOD-1191] Make block header a required field * [NOD-1191] Rename variables * [NOD-1212] Fix test names * [NOD-1191] Rename flow names * [NOD-1191] Fix infinite loop
71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
package flowcontext
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/mempool"
|
|
"github.com/kaspanet/kaspad/protocol/flows/relaytransactions"
|
|
"github.com/kaspanet/kaspad/util"
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
"github.com/kaspanet/kaspad/wire"
|
|
"github.com/pkg/errors"
|
|
"time"
|
|
)
|
|
|
|
// AddTransaction adds transaction to the mempool and propagates it.
|
|
func (f *FlowContext) AddTransaction(tx *util.Tx) error {
|
|
f.transactionsToRebroadcastLock.Lock()
|
|
defer f.transactionsToRebroadcastLock.Unlock()
|
|
|
|
transactionsAcceptedToMempool, err := f.txPool.ProcessTransaction(tx, false, 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(transactionsAcceptedToMempool) > 1 {
|
|
return errors.New("got more than one accepted transactions when no orphans were allowed")
|
|
}
|
|
|
|
f.transactionsToRebroadcast[*tx.ID()] = tx
|
|
inv := wire.NewMsgInvTransaction([]*daghash.TxID{tx.ID()})
|
|
return f.Broadcast(inv)
|
|
}
|
|
|
|
func (f *FlowContext) updateTransactionsToRebroadcast(block *util.Block) {
|
|
f.transactionsToRebroadcastLock.Lock()
|
|
defer f.transactionsToRebroadcastLock.Unlock()
|
|
// Note: if the block is red, its transactions won't be rebroadcasted
|
|
// anymore, although they are not included in the UTXO set.
|
|
// This is probably ok, since red blocks are quite rare.
|
|
for _, tx := range block.Transactions() {
|
|
delete(f.transactionsToRebroadcast, *tx.ID())
|
|
}
|
|
}
|
|
|
|
func (f *FlowContext) shouldRebroadcastTransactions() bool {
|
|
const rebroadcastInterval = 30 * time.Second
|
|
return time.Since(f.lastRebroadcastTime) > rebroadcastInterval
|
|
}
|
|
|
|
func (f *FlowContext) txIDsToRebroadcast() []*daghash.TxID {
|
|
f.transactionsToRebroadcastLock.Lock()
|
|
defer f.transactionsToRebroadcastLock.Unlock()
|
|
|
|
txIDs := make([]*daghash.TxID, len(f.transactionsToRebroadcast))
|
|
i := 0
|
|
for _, tx := range f.transactionsToRebroadcast {
|
|
txIDs[i] = tx.ID()
|
|
i++
|
|
}
|
|
return txIDs
|
|
}
|
|
|
|
// SharedRequestedTransactions returns a *relaytransactions.SharedRequestedTransactions for sharing
|
|
// data about requested transactions between different peers.
|
|
func (f *FlowContext) SharedRequestedTransactions() *relaytransactions.SharedRequestedTransactions {
|
|
return f.sharedRequestedTransactions
|
|
}
|
|
|
|
// TxPool returns the transaction pool associated to the manager.
|
|
func (f *FlowContext) TxPool() *mempool.TxPool {
|
|
return f.txPool
|
|
}
|