From 611dfa3b1ce9298760e4202c28cf9bc005e5f469 Mon Sep 17 00:00:00 2001
From: Kleissner <peter@kleissner.investments>
Date: Sun, 25 Apr 2021 03:44:52 +0200
Subject: [PATCH] RefreshBuckets: Use self ID on bucket 0.

---
 dht/DHT Lite.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/dht/DHT Lite.go b/dht/DHT Lite.go
index 4d1cdad..20804fb 100644
--- a/dht/DHT Lite.go	
+++ b/dht/DHT Lite.go	
@@ -241,6 +241,12 @@ func (dht *DHT) RefreshBuckets(target int) {
 	for bucket, total := range dht.ht.getTotalNodesPerBucket() {
 		if target == 0 || total < target {
 			nodeR := dht.ht.getRandomIDFromBucket(bucket)
+
+			// Refreshing closest bucket? Use self ID instead of random one.
+			if bucket == 0 {
+				nodeR = dht.ht.Self.ID
+			}
+
 			dht.FindNode(nodeR)
 		}
 	}