TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2025-02-16 21:16:30 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #2758 from nspcc-dev/check-inflight-tx-invs

Browse source 
network: check inv against currently processed transactions
This commit is contained in:
Roman Khimov 2022-10-24 14:16:33 +07:00 • committed by GitHub
commit f42b8e78fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkg/network/server.go
View file

@ -111,7 +111,7 @@ type (
txCallback func(*transaction.Transaction)
txCbEnabled atomic.Bool
txInLock sync.Mutex
txInLock sync.RWMutex
txin chan *transaction.Transaction
txInMap map[util.Uint256]struct{}
@ -760,7 +760,12 @@ func (s *Server) handlePong(p Peer, pong *payload.Ping) error {
func (s *Server) handleInvCmd(p Peer, inv *payload.Inventory) error {
var reqHashes = inv.Hashes[:0]
var typExists = map[payload.InventoryType]func(util.Uint256) bool{
payload.TXType: s.mempool.ContainsKey,
payload.TXType: func(h util.Uint256) bool {
s.txInLock.RLock()
_, ok := s.txInMap[h]
s.txInLock.RUnlock()
return ok || s.mempool.ContainsKey(h)
},
payload.BlockType: s.chain.HasBlock,
payload.ExtensibleType: func(h util.Uint256) bool {
cp := s.extensiblePool.Get(h)