checklist: add Limit and Len methods

godeep · Oct 6, 2018 · 75e8285 · 75e8285
1 parent 630b149
commit 75e8285
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/checklist.go b/checklist.go
@@ -68,3 +68,17 @@ func (c *Checklist) Prune() {
 	}
 	c.Pairs = result
 }
+
+// Limit ensures maximum length of pairs, removing the pairs with least priority
+// if needed.
+func (c *Checklist) Limit(max int) {
+	if len(c.Pairs) <= max {
+		return
+	}
+	c.Pairs = c.Pairs[:max]
+}
+
+// Len returns pairs count.
+func (c *Checklist) Len() int {
+	return len(c.Pairs)
+}
diff --git a/checklist_test.go b/checklist_test.go
@@ -97,3 +97,33 @@ func TestChecklist_Prune(t *testing.T) {
 		t.Error("unexpected result length")
 	}
 }
+
+func TestChecklist_Limit(t *testing.T) {
+	c := Checklist{
+		Pairs: Pairs{
+			{
+				Priority: 100,
+			},
+			{
+				Priority: 99,
+			},
+			{
+				Priority: 98,
+			},
+			{
+				Priority: 97,
+			},
+			{
+				Priority: 96,
+			},
+		},
+	}
+	c.Limit(10)
+	if c.Len() != 5 {
+		t.Error("unexpected length")
+	}
+	c.Limit(3)
+	if c.Len() != 3 {
+		t.Error("unexpected length ")
+	}
+}