model: change flags array to bitmask

Using a list of integers is not optimal. Use a bit mask instead.

Signed-off-by: Robin Jarry <robin@jarry.cc>
Acked-by: Tim Culverhouse <tim@timculverhouse.com>

1 files changed, 2 insertions, 8 deletions

diff --git a/worker/lib/sort.go b/worker/lib/sort.go
index 1a1bb47f..2828999c 100644
--- a/worker/lib/sort.go
+++ b/worker/lib/sort.go
@@ -90,18 +90,12 @@ func sortAddresses(messageInfos []*models.MessageInfo, criterion *types.SortCrit
 }
 
 func sortFlags(messageInfos []*models.MessageInfo, criterion *types.SortCriterion,
-	testFlag models.Flag,
+	testFlag models.Flags,
 ) {
 	var slice []*boolStore
 	for _, msgInfo := range messageInfos {
-		flagPresent := false
-		for _, flag := range msgInfo.Flags {
-			if flag == testFlag {
-				flagPresent = true
-			}
-		}
 		slice = append(slice, &boolStore{
-			Value:   flagPresent,
+			Value:   msgInfo.Flags.Has(testFlag),
 			MsgInfo: msgInfo,
 		})
 	}