nlink/storage.go

package main

import (
	"context"
	"errors"
	"io"
	"os"

	"git.wntrmute.dev/kyle/goutils/config"
	"git.wntrmute.dev/kyle/goutils/log"
	"github.com/minio/minio-go/v7"
	"github.com/minio/minio-go/v7/pkg/credentials"
)

var minioClient *minio.Client

func connectMinio() (err error) {
	minioClient, err = minio.New(config.Get("minio_host"), &minio.Options{
		Creds: credentials.NewStaticV4(
			config.Get("minio_access_key"),
			config.Get("minio_secret_key"), ""),
		Secure: true,
	})

	if err != nil {
		log.Errf("while connecting minio: %s", err)
	}

	return err
}

func restoreDatabase(path string) error {
	log.Debugln("restoring database from minio")
	if minioClient == nil {
		if err := connectMinio(); err != nil {
			return err
		}
	}

	if minioClient == nil {
		return errors.New("storage: minio client is not connected")
	}

	obj, err := minioClient.GetObject(
		context.Background(),
		config.Get("minio_bucket"),
		"nlink.db",
		minio.GetObjectOptions{},
	)

	if err != nil {
		return err
	}

	dbFile, err := os.Create(path)
	if err != nil {
		return err
	}
	defer dbFile.Close()

	_, err = io.Copy(dbFile, obj)
	if err != nil {
		return err
	}

	log.Debugln("database restored from minio")
	return nil
}

func saveDatabase(path string) error {
	if minioClient == nil {
		if err := connectMinio(); err != nil {
			return err
		}
	}

	dbFile, err := os.Open(path)
	if err != nil {
		return err
	}
	defer dbFile.Close()

	fi, err := dbFile.Stat()
	if err != nil {
		return err
	}

	_, err = minioClient.PutObject(
		context.Background(),
		config.Get("minio_bucket"),
		"nlink.db",
		dbFile,
		fi.Size(),
		minio.PutObjectOptions{ContentType: "application/octet-stream"},
	)

	if err != nil {
		return err
	}

	log.Debugln("database saved to minio")
	return nil
}