From 09f9f4e8e4fdc2aafb65bbc31a88551341dded80 Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Mon, 25 May 2020 20:09:57 -0500
Subject: [PATCH] use heap.Fix instead of heap.Remove + heap.Push when updating
 queues, this is theoretically faster

---
 src/yggdrasil/packetqueue.go | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/yggdrasil/packetqueue.go b/src/yggdrasil/packetqueue.go
index 358aaeb1..6273e6c8 100644
--- a/src/yggdrasil/packetqueue.go
+++ b/src/yggdrasil/packetqueue.go
@@ -38,12 +38,16 @@ func (q *packetQueue) drop() bool {
 			longestIdx = idx
 		}
 	}
-	stream := heap.Remove(q, longestIdx).(pqStream)
+	stream := q.streams[longestIdx]
 	info := stream.infos[0]
 	if len(stream.infos) > 1 {
 		stream.infos = stream.infos[1:]
 		stream.size -= uint64(len(info.packet))
-		heap.Push(q, stream)
+		q.streams[longestIdx] = stream
+		q.size -= uint64(len(info.packet))
+		heap.Fix(q, longestIdx)
+	} else {
+		heap.Remove(q, longestIdx)
 	}
 	pool_putBytes(info.packet)
 	return true
@@ -67,12 +71,16 @@ func (q *packetQueue) push(packet []byte) {
 
 func (q *packetQueue) pop() ([]byte, bool) {
 	if q.size > 0 {
-		stream := heap.Pop(q).(pqStream)
+		stream := q.streams[0]
 		info := stream.infos[0]
 		if len(stream.infos) > 1 {
 			stream.infos = stream.infos[1:]
 			stream.size -= uint64(len(info.packet))
-			heap.Push(q, stream)
+			q.streams[0] = stream
+			q.size -= uint64(len(info.packet))
+			heap.Fix(q, 0)
+		} else {
+			heap.Remove(q, 0)
 		}
 		return info.packet, true
 	}