Filter non Free plan zones from graphql metrics (#70)

Co-authored-by: Laurynas Četyrkinas <[email protected]>
lablabs · Jul 27, 2022 · fa0fe1b · fa0fe1b
1 parent 44a539f
commit fa0fe1b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/cloudflare.go b/cloudflare.go
@@ -598,3 +598,12 @@ func extractZoneIDs(zones []cloudflare.Zone) []string {
 
 	return IDs
 }
+
+func filterNonFreePlanZones(zones []cloudflare.Zone) (filteredZones []cloudflare.Zone) {
+	for _, z := range zones {
+		if z.Plan.ZonePlanCommon.ID != "0feeeeeeeeeeeeeeeeeeeeeeeeeeeeee" {
+			filteredZones = append(filteredZones, z)
+		}
+	}
+	return
+}
diff --git a/prometheus.go b/prometheus.go
@@ -228,7 +228,11 @@ func fetchZoneColocationAnalytics(zones []cloudflare.Zone, wg *sync.WaitGroup) {
 	if cfgFreeTier {
 		return
 	}
-	zoneIDs := extractZoneIDs(zones)
+
+	zoneIDs := extractZoneIDs(filterNonFreePlanZones(zones))
+	if len(zoneIDs) == 0 {
+		return
+	}
 
 	r, err := fetchColoTotals(zoneIDs)
 	if err != nil {
@@ -256,7 +260,10 @@ func fetchZoneAnalytics(zones []cloudflare.Zone, wg *sync.WaitGroup) {
 		return
 	}
 
-	zoneIDs := extractZoneIDs(zones)
+	zoneIDs := extractZoneIDs(filterNonFreePlanZones(zones))
+	if len(zoneIDs) == 0 {
+		return
+	}
 
 	r, err := fetchZoneTotals(zoneIDs)
 	if err != nil {
@@ -389,7 +396,10 @@ func fetchLoadBalancerAnalytics(zones []cloudflare.Zone, wg *sync.WaitGroup) {
 		return
 	}
 
-	zoneIDs := extractZoneIDs(zones)
+	zoneIDs := extractZoneIDs(filterNonFreePlanZones(zones))
+	if len(zoneIDs) == 0 {
+		return
+	}
 
 	l, err := fetchLoadBalancerTotals(zoneIDs)
 	if err != nil {