methylmaus/backup-tool
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor upload into 's3helper' package

Browse Source
This commit is contained in:
Alex Csengery
2023-08-29 21:00:49 -04:00
parent cab35ddfdd
commit a0081fab99
2 changed files with 75 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
main.go
View File

@@ -1,75 +1,19 @@
package main package main
import ( import (
"backup-tool/s3helper"
"context" "context"
"github.com/joho/godotenv"
"log" "log"
"os" "os"
"github.com/minio/minio-go/v7" "github.com/joho/godotenv"
"github.com/minio/minio-go/v7/pkg/credentials"
)
const (
ENDPOINT = "s3.us-west-004.backblazeb2.com"
BUCKETNAME = "ducimon-db-backups"
) )
func main() { func main() {
ctx := context.Background()
accessKeyID, secretAccessKey := getCredentials() accessKeyID, secretAccessKey := getCredentials()
client := createClient(accessKeyID, secretAccessKey) s3helper.UploadFile(context.Background(), accessKeyID, secretAccessKey)
verifyBucket(client, ctx)
absolutePath, basename := validateUploadFile()
_, err := client.FPutObject(ctx, BUCKETNAME, "test", absolutePath, minio.PutObjectOptions{})
if err != nil {
log.Fatalln(err)
}
log.Printf("upload of %s complete\n", basename)
}
func validateUploadFile() (string, string) {
if len(os.Args) < 2 {
log.Fatalln("upload file not specified")
}
file, err := os.Stat(os.Args[1])
if err != nil {
log.Fatalln(err)
}
if file.IsDir() {
log.Fatalln("upload of directories is not supported")
}
return os.Args[1], file.Name()
}
func verifyBucket(client *minio.Client, ctx context.Context) {
exists, err := client.BucketExists(ctx, BUCKETNAME)
if err != nil {
log.Fatalln(err)
}
if !exists {
log.Fatalf("bucket %s does not exist\n", BUCKETNAME)
}
}
func createClient(accessKeyID string, secretAccessKey string) *minio.Client {
// Initialize minio client object.
minioClient, err := minio.New(ENDPOINT, &minio.Options{
Creds: credentials.NewStaticV4(accessKeyID, secretAccessKey, ""),
Secure: true,
})
if err != nil {
log.Fatalln(err)
}
return minioClient
} }
func getCredentials() (string, string) { func getCredentials() (string, string) {

72
s3helper/s3helper.go Normal file
View File

@@ -0,0 +1,72 @@
package s3helper
import (
"context"
"log"
"os"
"github.com/minio/minio-go/v7"
"github.com/minio/minio-go/v7/pkg/credentials"
)
const (
ENDPOINT = "s3.us-west-004.backblazeb2.com"
BUCKETNAME = "ducimon-db-backups"
)
func UploadFile(ctx context.Context, accessKeyID, secretAccessKey) error {
client := createClient(accessKeyID, secretAccessKey)
verifyBucket(client, ctx)
absolutePath, basename := validateUploadFile()
_, err := client.FPutObject(ctx, BUCKETNAME, "test", absolutePath, minio.PutObjectOptions{})
if err != nil {
log.Fatalln(err)
}
log.Printf("upload of %s complete\n", basename)
return nil
}
func createClient(accessKeyID string, secretAccessKey string) *minio.Client {
// Initialize minio client object.
minioClient, err := minio.New(ENDPOINT, &minio.Options{
Creds: credentials.NewStaticV4(accessKeyID, secretAccessKey, ""),
Secure: true,
})
if err != nil {
log.Fatalln(err)
}
return minioClient
}
func verifyBucket(client *minio.Client, ctx context.Context) {
exists, err := client.BucketExists(ctx, BUCKETNAME)
if err != nil {
log.Fatalln(err)
}
if !exists {
log.Fatalf("bucket %s does not exist\n", BUCKETNAME)
}
}
func validateUploadFile() (string, string) {
if len(os.Args) < 2 {
log.Fatalln("upload file not specified")
}
file, err := os.Stat(os.Args[1])
if err != nil {
log.Fatalln(err)
}
if file.IsDir() {
log.Fatalln("upload of directories is not supported")
}
return os.Args[1], file.Name()
}