Merge pull request #1 from iamrare/add-more-functionality

Add more functionality

Author: iamrare

Dockerfile

@@ -1,4 +1,4 @@
-FROM golang:1.12-stretch as build
+FROM golang:1.14.0-stretch as build
 LABEL maintainer="Infinity Works"
 
 ENV GO111MODULE=on
@@ -10,7 +10,7 @@ RUN go mod download \
     && go test ./... \
     && CGO_ENABLED=0 GOOS=linux go build -o /bin/main
 
-FROM alpine:3.10
+FROM alpine:3.11.3
 
 RUN apk --no-cache add ca-certificates \
      && addgroup exporter \

exporter/gather.go

@@ -35,6 +35,10 @@ func (e *Exporter) gatherData() ([]*Datum, *RateLimits, error) {
 			if strings.Contains(response.url, "/repos/") {
 				getReleases(e, response.url, &d.Releases)
 			}
+			// Get PRs
+			if strings.Contains(response.url, "/repos/") {
+				getPRs(e, response.url, &d.Pulls)
+			}
 			json.Unmarshal(response.body, &d)
 			data = append(data, d)
 		}
@@ -111,6 +115,20 @@ func getReleases(e *Exporter, url string, data *[]Release) {
 	json.Unmarshal(releasesResponse[0].body, &data)
 }
 
+func getPRs(e *Exporter, url string, data *[]Pull) {
+	i := strings.Index(url, "?")
+	baseURL := url[:i]
+	pullsURL := baseURL + "/pulls"
+	pullsResponse, err := asyncHTTPGets([]string{pullsURL}, e.APIToken)
+
+	if err != nil {
+		log.Errorf("Unable to obtain pull requests from API, Error: %s", err)
+	}
+	fmt.Println(&data)
+
+	json.Unmarshal(pullsResponse[0].body, &data)
+}
+
 // isArray simply looks for key details that determine if the JSON response is an array or not.
 func isArray(body []byte) bool {
 

exporter/metrics.go

@@ -18,6 +18,11 @@ func AddMetrics() map[string]*prometheus.Desc {
 		"Total number of open issues for given repository",
 		[]string{"repo", "user", "private", "fork", "archived", "license", "language"}, nil,
 	)
+	APIMetrics["PullRequestCount"] = prometheus.NewDesc(
+                prometheus.BuildFQName("github", "repo", "pull_request_count"),
+                "Total number of pull requests for given repository",
+                []string{}, nil,
+        )
 	APIMetrics["Watchers"] = prometheus.NewDesc(
 		prometheus.BuildFQName("github", "repo", "watchers"),
 		"Total number of watchers/subscribers for given repository",
@@ -64,7 +69,6 @@ func (e *Exporter) processMetrics(data []*Datum, rates *RateLimits, ch chan<- pr
 	for _, x := range data {
 		ch <- prometheus.MustNewConstMetric(e.APIMetrics["Stars"], prometheus.GaugeValue, x.Stars, x.Name, x.Owner.Login, strconv.FormatBool(x.Private), strconv.FormatBool(x.Fork), strconv.FormatBool(x.Archived), x.License.Key, x.Language)
 		ch <- prometheus.MustNewConstMetric(e.APIMetrics["Forks"], prometheus.GaugeValue, x.Forks, x.Name, x.Owner.Login, strconv.FormatBool(x.Private), strconv.FormatBool(x.Fork), strconv.FormatBool(x.Archived), x.License.Key, x.Language)
-		ch <- prometheus.MustNewConstMetric(e.APIMetrics["OpenIssues"], prometheus.GaugeValue, x.OpenIssues, x.Name, x.Owner.Login, strconv.FormatBool(x.Private), strconv.FormatBool(x.Fork), strconv.FormatBool(x.Archived), x.License.Key, x.Language)
 		ch <- prometheus.MustNewConstMetric(e.APIMetrics["Watchers"], prometheus.GaugeValue, x.Watchers, x.Name, x.Owner.Login, strconv.FormatBool(x.Private), strconv.FormatBool(x.Fork), strconv.FormatBool(x.Archived), x.License.Key, x.Language)
 		ch <- prometheus.MustNewConstMetric(e.APIMetrics["Size"], prometheus.GaugeValue, x.Size, x.Name, x.Owner.Login, strconv.FormatBool(x.Private), strconv.FormatBool(x.Fork), strconv.FormatBool(x.Archived), x.License.Key, x.Language)
 
@@ -73,6 +77,15 @@ func (e *Exporter) processMetrics(data []*Datum, rates *RateLimits, ch chan<- pr
 				ch <- prometheus.MustNewConstMetric(e.APIMetrics["ReleaseDownloads"], prometheus.GaugeValue, float64(asset.Downloads), x.Name, x.Owner.Login, release.Name, asset.Name, asset.CreatedAt)
 			}
 		}
+		prCount := 0
+		for _, _ = range x.Pulls {
+			prCount += 1
+		}
+		// issueCount = x.OpenIssue - prCount
+		ch <- prometheus.MustNewConstMetric(e.APIMetrics["OpenIssues"], prometheus.GaugeValue, (x.OpenIssues - float64(prCount)), x.Name, x.Owner.Login, strconv.FormatBool(x.Private), strconv.FormatBool(x.Fork), strconv.FormatBool(x.Archived), x.License.Key, x.Language)
+
+		// prCount
+		ch <- prometheus.MustNewConstMetric(e.APIMetrics["PullRequestCount"], prometheus.GaugeValue, float64(prCount))
 	}
 
 	// Set Rate limit stats

exporter/structs.go

@@ -38,13 +38,21 @@ type Datum struct {
 	Watchers   float64 `json:"subscribers_count"`
 	Size       float64 `json:"size"`
 	Releases   []Release
+	Pulls      []Pull
 }
 
 type Release struct {
 	Name   string  `json:"name"`
 	Assets []Asset `json:"assets"`
 }
 
+type Pull struct {
+	Url  string `json:"url"`
+	User struct {
+		Login string `json:"login"`
+	} `json:"user"`
+}
+
 type Asset struct {
 	Name      string `json:"name"`
 	Size      int64  `json:"size"`