Merge 4cdd126 into 12ae8e7

brotherlogic · Jan 2, 2019 · 3757b99 · 3757b99
2 parents 12ae8e7 + 4cdd126
commit 3757b99
Show file tree

Hide file tree

Showing 3 changed files with 88 additions and 25 deletions.
diff --git a/proto/wantslist.pb.go b/proto/wantslist.pb.go
diff --git a/proto/wantslist.proto b/proto/wantslist.proto
@@ -8,7 +8,15 @@ message Config {
 
 message WantListEntry {
   int32 want = 1;
-  int32 index = 2; 
+  int32 index = 2;
+
+  enum Status {
+    UNPROCESSED = 0;
+    WANTED = 1;
+    IN_COLLECTION = 2;
+  }
+
+  Status status = 3;
 }
 
 message WantList {

diff --git a/wantslist.go b/wantslist.go
@@ -73,8 +73,26 @@ func (s *Server) Mote(ctx context.Context, master bool) error {
 
 // GetState gets the state of the server
 func (s *Server) GetState() []*pbg.State {
+	unprocCount := int64(0)
+	wantedCount := int64(0)
+	unknownCount := int64(0)
+	for _, list := range s.config.Lists {
+		for _, entry := range list.Wants {
+			switch entry.Status {
+			case pb.WantListEntry_UNPROCESSED:
+				unprocCount++
+			case pb.WantListEntry_WANTED:
+				wantedCount++
+			default:
+				unknownCount++
+			}
+		}
+	}
 	return []*pbg.State{
 		&pbg.State{Key: "lists", Value: int64(len(s.config.Lists))},
+		&pbg.State{Key: "unproc", Value: unprocCount},
+		&pbg.State{Key: "wanted", Value: wantedCount},
+		&pbg.State{Key: "unknown", Value: unknownCount},
 	}
 }