Get the issue counts in one query

Signed-off-by: Andrew Thornton <[email protected]>

Author: zeripath

models/models.go

@@ -272,7 +272,8 @@ type Statistic struct {
 	Counter struct {
 		User, Org, PublicKey,
 		Repo, Watch, Star, Action, Access,
-		IssueClosed, IssueOpen, Comment, Oauth, Follow,
+		Issue, IssueClosed, IssueOpen,
+		Comment, Oauth, Follow,
 		Mirror, Release, LoginSource, Webhook,
 		Milestone, Label, HookTask,
 		Team, UpdateTask, Attachment int64
@@ -289,8 +290,24 @@ func GetStatistic() (stats Statistic) {
 	stats.Counter.Star, _ = x.Count(new(Star))
 	stats.Counter.Action, _ = x.Count(new(Action))
 	stats.Counter.Access, _ = x.Count(new(Access))
-	stats.Counter.IssueClosed, _ = x.Where("is_closed=?", true).Count(new(Issue))
-	stats.Counter.IssueOpen, _ = x.Where("is_closed=?", false).Count(new(Issue))
+
+	type IssueCount struct {
+		Count    int64
+		IsClosed bool
+	}
+	issueCounts := []IssueCount{}
+
+	_ = x.Select("COUNT(*) AS count, is_closed").Table("issue").GroupBy("is_closed").Find(&issueCounts)
+	for _, c := range issueCounts {
+		if c.IsClosed {
+			stats.Counter.IssueClosed = c.Count
+		} else {
+			stats.Counter.IssueOpen = c.Count
+		}
+	}
+
+	stats.Counter.Issue = stats.Counter.IssueClosed + stats.Counter.IssueOpen
+
 	stats.Counter.Comment, _ = x.Count(new(Comment))
 	stats.Counter.Oauth = 0
 	stats.Counter.Follow, _ = x.Count(new(Follow))

modules/metrics/collector.go

@@ -21,6 +21,7 @@ type Collector struct {
 	Comments      *prometheus.Desc
 	Follows       *prometheus.Desc
 	HookTasks     *prometheus.Desc
+	Issues        *prometheus.Desc
 	IssuesOpen    *prometheus.Desc
 	IssuesClosed  *prometheus.Desc
 	Labels        *prometheus.Desc
@@ -73,6 +74,11 @@ func NewCollector() Collector {
 			"Number of HookTasks",
 			nil, nil,
 		),
+		Issues: prometheus.NewDesc(
+			namespace+"issues",
+			"Number of Issues",
+			nil, nil,
+		),
 		IssuesOpen: prometheus.NewDesc(
 			namespace+"issues_open",
 			"Number of open Issues",
@@ -170,6 +176,7 @@ func (c Collector) Describe(ch chan<- *prometheus.Desc) {
 	ch <- c.Comments
 	ch <- c.Follows
 	ch <- c.HookTasks
+	ch <- c.Issues
 	ch <- c.IssuesOpen
 	ch <- c.IssuesClosed
 	ch <- c.Labels
@@ -223,6 +230,11 @@ func (c Collector) Collect(ch chan<- prometheus.Metric) {
 		prometheus.GaugeValue,
 		float64(stats.Counter.HookTask),
 	)
+	ch <- prometheus.MustNewConstMetric(
+		c.Issues,
+		prometheus.GaugeValue,
+		float64(stats.Counter.Issue),
+	)
 	ch <- prometheus.MustNewConstMetric(
 		c.IssuesClosed,
 		prometheus.GaugeValue,