refactor(backend, transport): add pooled packet

Author: vasyl-ks

backend/pkg/transport/packet/data/decoder.go

@@ -35,6 +35,7 @@ func (decoder *Decoder) Decode(id abstraction.PacketId, reader io.Reader) (abstr
 		return nil, ErrUnexpectedId{Id: id}
 	}
 
+	//packet := GetPacket(id)
 	packet := NewPacket(id)
 	for _, value := range descriptor {
 		val, err := value.Decode(decoder.endianness, reader)

backend/pkg/transport/packet/data/packet.go

@@ -1,6 +1,7 @@
 package data
 
 import (
+	"sync"
 	"time"
 
 	"github.com/HyperloopUPV-H8/h9-backend/pkg/abstraction"
@@ -27,6 +28,16 @@ func NewPacket(id abstraction.PacketId) *Packet {
 	}
 }
 
+var packetPool = sync.Pool{
+	New: func() any {
+		return &Packet{
+			values:  make(map[ValueName]Value),
+			enabled: make(map[ValueName]bool),
+		}
+	},
+}
+
+
 // NewPacketWithValues creates a new data packet with the given values
 func NewPacketWithValues(id abstraction.PacketId, values map[ValueName]Value, enabled map[ValueName]bool) *Packet {
 	return &Packet{
@@ -62,3 +73,35 @@ func (packet *Packet) SetTimestamp(timestamp time.Time) *Packet {
 	packet.timestamp = timestamp
 	return packet
 }
+
+func (packet *Packet) Reset() {
+	clear(packet.values)
+	clear(packet.enabled)
+	packet.id = 0
+	packet.timestamp = time.Time{}
+}
+
+func GetPacket(id abstraction.PacketId) *Packet {
+	p := packetPool.Get().(*Packet)
+	if p.values == nil {
+		p.values = make(map[ValueName]Value)
+	} else {
+		clear(p.values)
+	}
+	if p.enabled == nil {
+		p.enabled = make(map[ValueName]bool)
+	} else {
+		clear(p.enabled)
+	}
+	p.id = id
+	p.timestamp = time.Now()
+	return p
+}
+
+func ReleasePacket(p *Packet) {
+	if p == nil {
+		return
+	}
+	p.Reset()
+	packetPool.Put(p)
+}

backend/pkg/transport/transport.go

@@ -259,6 +259,10 @@ func (transport *Transport) readLoopTCPConn(conn net.Conn, logger zerolog.Logger
 
 			logger.Trace().Type("type", packet).Msg("packet")
 			transport.api.Notification(NewPacketNotification(packet, from, to, time.Now()))
+
+			//if dataPacket, ok := packet.(*data.Packet); ok {
+			//	data.ReleasePacket(dataPacket)
+			//}
 		}
 	}()
 }
@@ -464,6 +468,10 @@ func (transport *Transport) handleUDPPacket(udpPacket udp.Packet) {
 
 	// Send notification
 	transport.api.Notification(NewPacketNotification(packet, srcAddr, dstAddr, udpPacket.Timestamp))
+
+	//if dataPacket, ok := packet.(*data.Packet); ok {
+	//	data.ReleasePacket(dataPacket)
+	//s}
 }
 
 // handleConversation is called when the sniffer detects a new conversation and handles its specific packets
@@ -488,6 +496,10 @@ func (transport *Transport) handleConversation(socket network.Socket, reader io.
 
 			// Send notification
 			transport.api.Notification(NewPacketNotification(packet, srcAddr, dstAddr, time.Now()))
+
+			//if dataPacket, ok := packet.(*data.Packet); ok {
+			//	data.ReleasePacket(dataPacket)
+			//}
 		}
 	}()
 }