mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 15:26:42 +00:00
d14809694f
* [NOD-1223] Delete unused files/packages. * [NOD-1223] Move signal and limits to the os package. * [NOD-1223] Put database and dbaccess into the db package. * [NOD-1223] Fold the logs package into the logger package. * [NOD-1223] Rename domainmessage to appmessage. * [NOD-1223] Rename to/from DomainMessage to AppMessage. * [NOD-1223] Move appmessage to the app packge. * [NOD-1223] Move protocol to the app packge. * [NOD-1223] Move the network package to the infrastructure packge. * [NOD-1223] Rename cmd to executables. * [NOD-1223] Fix go.doc in the logger package.
47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package blockrelay
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
)
|
|
|
|
// SharedRequestedBlocks is a data structure that is shared between peers that
|
|
// holds the hashes of all the requested blocks to prevent redundant requests.
|
|
type SharedRequestedBlocks struct {
|
|
blocks map[daghash.Hash]struct{}
|
|
sync.Mutex
|
|
}
|
|
|
|
func (s *SharedRequestedBlocks) remove(hash *daghash.Hash) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
delete(s.blocks, *hash)
|
|
}
|
|
|
|
func (s *SharedRequestedBlocks) removeSet(blockHashes map[daghash.Hash]struct{}) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
for hash := range blockHashes {
|
|
delete(s.blocks, hash)
|
|
}
|
|
}
|
|
|
|
func (s *SharedRequestedBlocks) addIfNotExists(hash *daghash.Hash) (exists bool) {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
_, ok := s.blocks[*hash]
|
|
if ok {
|
|
return true
|
|
}
|
|
s.blocks[*hash] = struct{}{}
|
|
return false
|
|
}
|
|
|
|
// NewSharedRequestedBlocks returns a new instance of SharedRequestedBlocks.
|
|
func NewSharedRequestedBlocks() *SharedRequestedBlocks {
|
|
return &SharedRequestedBlocks{
|
|
blocks: make(map[daghash.Hash]struct{}),
|
|
}
|
|
}
|