Add Prometheus exporter

1 files changed, 46 insertions, 3 deletions

diff --git a/storage/feed.go b/storage/feed.go
index 55ce6e83..697ef095 100644
--- a/storage/feed.go
+++ b/storage/feed.go
@@ -76,6 +76,37 @@ func (s *Storage) AnotherFeedURLExists(userID, feedID int64, feedURL string) boo
 	return result
 }
 
+// CountAllFeeds returns the number of feeds in the database.
+func (s *Storage) CountAllFeeds() map[string]int64 {
+	rows, err := s.db.Query(`SELECT disabled, count(*) FROM feeds GROUP BY disabled`)
+	if err != nil {
+		return nil
+	}
+	defer rows.Close()
+
+	results := make(map[string]int64)
+	results["enabled"] = 0
+	results["disabled"] = 0
+
+	for rows.Next() {
+		var disabled bool
+		var count int64
+
+		if err := rows.Scan(&disabled, &count); err != nil {
+			continue
+		}
+
+		if disabled {
+			results["disabled"] = count
+		} else {
+			results["enabled"] = count
+		}
+	}
+
+	results["total"] = results["disabled"] + results["enabled"]
+	return results
+}
+
 // CountFeeds returns the number of feeds that belongs to the given user.
 func (s *Storage) CountFeeds(userID int64) int {
 	var result int
@@ -87,9 +118,9 @@ func (s *Storage) CountFeeds(userID int64) int {
 	return result
 }
 
-// CountErrorFeeds returns the number of feeds with parse errors that belong to the given user.
-func (s *Storage) CountErrorFeeds(userID int64) int {
-	query := `SELECT count(*) FROM feeds WHERE user_id=$1 AND parsing_error_count>=$2`
+// CountUserFeedsWithErrors returns the number of feeds with parsing errors that belong to the given user.
+func (s *Storage) CountUserFeedsWithErrors(userID int64) int {
+	query := `SELECT count(*) FROM feeds WHERE user_id=$1 AND parsing_error_count >= $2`
 	var result int
 	err := s.db.QueryRow(query, userID, maxParsingError).Scan(&result)
 	if err != nil {
@@ -99,6 +130,18 @@ func (s *Storage) CountErrorFeeds(userID int64) int {
 	return result
 }
 
+// CountAllFeedsWithErrors returns the number of feeds with parsing errors.
+func (s *Storage) CountAllFeedsWithErrors() int {
+	query := `SELECT count(*) FROM feeds WHERE parsing_error_count >= $1`
+	var result int
+	err := s.db.QueryRow(query, maxParsingError).Scan(&result)
+	if err != nil {
+		return 0
+	}
+
+	return result
+}
+
 // Feeds returns all feeds that belongs to the given user.
 func (s *Storage) Feeds(userID int64) (model.Feeds, error) {
 	return s.fetchFeeds(feedListQuery, "", userID)