Add "compression" flag to use native CloudSQL export file compression

Author: flashadmin

go.mod

@@ -1,4 +1,4 @@
-module github.com/trufflesecurity/cloudsql-exporter
+module github.com/flashadmin/cloudsql-exporter
 
 go 1.19
 

main.go

@@ -12,8 +12,8 @@ import (
 	"google.golang.org/api/storage/v1"
 	"gopkg.in/alecthomas/kingpin.v2"
 
-	"github.com/trufflesecurity/cloudsql-exporter/pkg/cloudsql"
-	"github.com/trufflesecurity/cloudsql-exporter/pkg/version"
+	"github.com/flashadmin/cloudsql-exporter/pkg/cloudsql"
+	"github.com/flashadmin/cloudsql-exporter/pkg/version"
 )
 
 var (
@@ -22,6 +22,7 @@ var (
 	bucket            = app.Flag("bucket", "Google Cloud Storage bucket name").Required().String()
 	project           = app.Flag("project", "GCP project ID").Required().String()
 	instance          = app.Flag("instance", "Cloud SQL instance name, if not specified all within the project will be enumerated").String()
+	compression       = app.Flag("compression", "Enable compression for exported SQL files").Bool()
 	ensureIamBindings = app.Flag("ensure-iam-bindings", "Ensure that the Cloud SQL service account has the required IAM role binding to export and validate the backup").Bool()
 )
 
@@ -70,7 +71,15 @@ func main() {
 			}
 		}
 
-		err := cloudsql.ExportCloudSQLDatabase(ctx, sqlAdminSvc, databases, *project, string(instance), *bucket, time.Now().Format(time.RFC3339Nano))
+		var objectName string
+
+		if *compression {
+			objectName = time.Now().Format(time.RFC3339Nano) + ".sql.gz"
+		} else {
+			objectName = time.Now().Format(time.RFC3339Nano) + ".sql"
+		}
+
+		err := cloudsql.ExportCloudSQLDatabase(ctx, sqlAdminSvc, databases, *project, string(instance), *bucket, objectName)
 		if err != nil {
 			log.Fatal(err)
 		}

pkg/cloudsql/cloudsql.go

@@ -126,7 +126,7 @@ func ExportCloudSQLDatabase(ctx context.Context, sqlAdminSvc *sqladmin.Service,
 				FileType:  "SQL",
 				Kind:      "sql#exportContext",
 				Databases: []string{database},
-				Uri:       fmt.Sprintf("gs://%s/%s/%s/%s/%s.sql", bucketName, projectID, instanceID, database, objectName),
+				Uri:       fmt.Sprintf("gs://%s/%s/%s/%s/%s", bucketName, projectID, instanceID, database, objectName),
 			},
 		}