From 38fbc7fa472270630eb8f2837f01c44e6255f84c Mon Sep 17 00:00:00 2001
From: vyzo <vyzo@hackzen.org>
Date: Mon, 9 Mar 2020 13:07:37 +0200
Subject: [PATCH] implement message delivery struct details

---
 score.go | 48 +++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 45 insertions(+), 3 deletions(-)

diff --git a/score.go b/score.go
index 59ab3d8..219c551 100644
--- a/score.go
+++ b/score.go
@@ -148,7 +148,11 @@ type peerScore struct {
 }
 
 type messageDeliveries struct {
-	// TODO
+	records map[string]*deliveryRecord
+
+	// queue for cleaning up old delivery records
+	head *deliveryEntry
+	tail *deliveryEntry
 }
 
 type deliveryRecord struct {
@@ -157,6 +161,12 @@ type deliveryRecord struct {
 	peers     map[peer.ID]struct{}
 }
 
+type deliveryEntry struct {
+	id     string
+	expire time.Time
+	next   *deliveryEntry
+}
+
 // delivery record status
 const (
 	delivery_unknown   = iota // we don't know (yet) if the message is valid
@@ -497,8 +507,40 @@ func (ps *peerScore) DuplicateMessage(msg *Message) {
 
 // message delivery records
 func (d *messageDeliveries) getRecord(id string) *deliveryRecord {
-	// TODO
-	return nil
+	rec, ok := d.records[id]
+	if ok {
+		return rec
+	}
+
+	rec = &deliveryRecord{peers: make(map[peer.ID]struct{})}
+	d.records[id] = rec
+
+	entry := &deliveryEntry{id: id, expire: time.Now().Add(TimeCacheDuration)}
+	if d.tail != nil {
+		d.tail.next = entry
+		d.tail = entry
+	} else {
+		d.head = entry
+		d.tail = entry
+	}
+
+	return rec
+}
+
+func (d *messageDeliveries) gc() {
+	if d.head == nil {
+		return
+	}
+
+	now := time.Now()
+	for d.head != nil && now.After(d.head.expire) {
+		delete(d.records, d.head.id)
+		d.head = d.head.next
+	}
+
+	if d.head == nil {
+		d.tail = nil
+	}
 }
 
 // utilities