Peernet-Archive/core
1
0
Fork 0
mirror of https://github.com/PeernetOfficial/core.git synced 2026-07-17 02:47:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

New DHT function IsNodeContact to check if node is in local routing table

Browse Source
This commit is contained in:
Kleissner
2021-07-07 02:50:12 +02:00
parent c1ccc3c344
commit 66ebf7c91e
3 changed files with 27 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
dht/Hash Table.go
View File

@@ -74,15 +74,19 @@ func (ht *hashTable) markNodeAsSeen(index int, ID []byte) {
ht.RoutingTable[index] = bucket
}
func (ht *hashTable) doesNodeExistInBucket(bucket int, node []byte) bool {
func (ht *hashTable) doesNodeExistInBucket(bucket int, ID []byte) (node *Node) {
ht.mutex.RLock()
defer ht.mutex.RUnlock()
for _, v := range ht.RoutingTable[bucket] {
if bytes.Compare(v.ID, node) == 0 {
return true
for _, node = range ht.RoutingTable[bucket] {
if bytes.Compare(node.ID, ID) == 0 {
return node
}
}
return false
return nil
}
func (ht *hashTable) doesNodeExist(ID []byte) (node *Node) {
return ht.doesNodeExistInBucket(ht.getBucketIndexFromDifferingBit(ID), ID)
}
// getClosestContacts returns the closest nodes to the target. filterFunc is optional and allows the caller to filter the nodes.
@@ -140,7 +144,7 @@ func (ht *hashTable) insertNode(node *Node, shouldEvict func(nodeOld *Node, node
index := ht.getBucketIndexFromDifferingBit(node.ID)
// If the node already exist, mark it as seen
if ht.doesNodeExistInBucket(index, node.ID) {
if ht.doesNodeExistInBucket(index, node.ID) != nil {
ht.markNodeAsSeen(index, node.ID)
return
}