main.go

package main

import (
	"bytes"
	"time"

	log "github.com/sirupsen/logrus"

	"vault_raft_snapshotter/config"
	"vault_raft_snapshotter/snapshot"
)

func main() {
	log.Infoln("Reading configuration...")

	c, err := config.ReadConfig()
	if err != nil {
		log.Fatalln("Configuration could not be found")
	}

	snapshotter, err := snapshot.NewSnapshotter(c)
	if err != nil {
		log.Fatalln("Failed to create snapshotter")
	}

	if snapshotter.TokenExpiration.Before(time.Now()) && (c.Vault.RoleID != "" && c.Vault.SecretID != "") {
		if err = snapshotter.SetClientTokenFromAppRole(c); err != nil {
			log.Fatalf("Unable to login to vault")
		}
	}

	var snapshot bytes.Buffer
	err = snapshotter.API.Sys().RaftSnapshot(&snapshot)

	if err != nil {
		log.Fatalln("Unable to generate snapshot", err.Error())
	}

	now := time.Now().UnixNano()

	if c.Local.Path != "" {
		snapshotPath, err := snapshotter.CreateLocalSnapshot(&snapshot, c, now)
		logSnapshotError("local", snapshotPath, err)
	}

	if c.AWS.Bucket != "" {
		snapshotPath, err := snapshotter.CreateS3Snapshot(&snapshot, c, now)
		logSnapshotError("aws", snapshotPath, err)
	}

	if c.GCP.Bucket != "" {
		snapshotPath, err := snapshotter.CreateGCPSnapshot(&snapshot, c, now)
		logSnapshotError("gcp", snapshotPath, err)
	}

	if c.Azure.ContainerName != "" {
		snapshotPath, err := snapshotter.CreateAzureSnapshot(&snapshot, c, now)
		logSnapshotError("azure", snapshotPath, err)
	}
}

func logSnapshotError(dest, snapshotPath string, err error) {
	if err != nil {
		log.Errorf("Failed to generate %s snapshot to %s: %v\n", dest, snapshotPath, err)
	} else {
		log.Infof("Successfully created %s snapshot to %s\n", dest, snapshotPath)
	}
}