Merge pull request #984 from Azanul/bigquery-cost-calculation

Bigquery table cost calculation

providers/gcp/bigquery/tables.go

@@ -13,6 +13,27 @@ import (
 	"google.golang.org/api/option"
 )
 
+const (
+	ACTIVE_STORAGE   = "ACTIVE"
+	LONGTERM_STORAGE = "LONGTERM"
+)
+
+func getTableStorageClass(meta *bigquery.TableMetadata) string {
+	activeThreshold := time.Now().AddDate(0, 0, -90)
+	lastModified := meta.LastModifiedTime
+	if lastModified.After(activeThreshold) {
+		return ACTIVE_STORAGE
+	}
+	return LONGTERM_STORAGE
+}
+
+func getStoragePricingForBigQueryTable() map[string]float64 {
+	return map[string]float64{
+		ACTIVE_STORAGE:   0.02,
+		LONGTERM_STORAGE: 0.01,
+	}
+}
+
 func Tables(ctx context.Context, client providers.ProviderClient) ([]models.Resource, error) {
 	resources := make([]models.Resource, 0)
 
@@ -59,6 +80,9 @@ func Tables(ctx context.Context, client providers.ProviderClient) ([]models.Reso
 				})
 			}
 
+			storageClass := getTableStorageClass(tableMetadata)
+			monthlyCost := float64(tableMetadata.NumBytes) / (1024 * 1024 * 1024) * getStoragePricingForBigQueryTable()[storageClass]
+
 			resources = append(resources, models.Resource{
 				Provider:   "GCP",
 				Account:    client.Name,
@@ -68,6 +92,7 @@ func Tables(ctx context.Context, client providers.ProviderClient) ([]models.Reso
 				Name:       tableMetadata.Name,
 				FetchedAt:  time.Now(),
 				Tags:       tags,
+				Cost:       monthlyCost,
 				Link:       fmt.Sprintf("https://console.cloud.google.com/bigquery?project=%s&page=dataset&p=%s&d=%s", client.GCPClient.Credentials.ProjectID, client.GCPClient.Credentials.ProjectID, dataset.DatasetID),
 			})
 		}