♻️ Use .pgpass file instead stdin enter password
This commit is contained in:
parent
5a71f8268a
commit
6a520603cb
@ -5,6 +5,7 @@ import (
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"os/user"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
@ -32,8 +33,10 @@ func BackupDb() error {
|
||||
fmt.Sprintf("watchtower_db_backup_%s", time.Now().Format("2006-01-02 15:04:05")),
|
||||
)
|
||||
|
||||
// Reading config
|
||||
var database string
|
||||
var password string
|
||||
var user string
|
||||
var username string
|
||||
var host string
|
||||
var port string
|
||||
|
||||
@ -42,25 +45,54 @@ func BackupDb() error {
|
||||
if strings.HasPrefix(part, "password=") {
|
||||
password = strings.Replace(part, "password=", "", 1)
|
||||
} else if strings.HasPrefix(part, "user=") {
|
||||
user = strings.Replace(part, "user=", "", 1)
|
||||
username = strings.Replace(part, "user=", "", 1)
|
||||
} else if strings.HasPrefix(part, "host=") {
|
||||
host = strings.Replace(part, "host=", "", 1)
|
||||
} else if strings.HasPrefix(part, "port=") {
|
||||
port = strings.Replace(part, "port=", "", 1)
|
||||
} else if strings.HasPrefix(part, "dbname=") {
|
||||
database = strings.Replace(part, "dbname=", "", 1)
|
||||
}
|
||||
}
|
||||
|
||||
// Creating ~/.pgpass
|
||||
usr, err := user.Current()
|
||||
if err != nil {
|
||||
return fmt.Errorf("unable to get current user: %v", err)
|
||||
}
|
||||
|
||||
homeDir := usr.HomeDir
|
||||
pgpassFile := filepath.Join(homeDir, ".pgpass")
|
||||
|
||||
pgpassString := fmt.Sprintf("%s:%s:%s:%s:%s\n", host, port, database, username, password)
|
||||
|
||||
// Open the .pgpass pgpass for writing (create if it doesn't exist)
|
||||
pgpass, err := os.OpenFile(pgpassFile, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600) // Set file permissions to 0600
|
||||
if err != nil {
|
||||
log.Error().Err(err).Str("path", pgpassFile).Msg("Failed to open .pgpass file...")
|
||||
return fmt.Errorf("failed to open .pgpass file: %v", err)
|
||||
}
|
||||
|
||||
if _, err = pgpass.WriteString(pgpassString); err != nil {
|
||||
log.Error().Err(err).Msg("Failed to write to .pgpass file...")
|
||||
return fmt.Errorf("failed to write to .pgpass file: %v", err)
|
||||
} else {
|
||||
pgpass.Close()
|
||||
}
|
||||
|
||||
log.Info().Msg("Wrote to .pgpass file...")
|
||||
|
||||
// Backing up
|
||||
log.Info().
|
||||
Str("password", password).Str("user", user).
|
||||
Str("password", password).Str("user", username).
|
||||
Str("host", host).Str("port", port).
|
||||
Msg("Starting backup database...")
|
||||
|
||||
cmd := exec.Command("pg_dumpall",
|
||||
"-h", host,
|
||||
"-p", port,
|
||||
"-U", user,
|
||||
"-U", username,
|
||||
"-f", outFile,
|
||||
"-W",
|
||||
)
|
||||
cmd.Env = os.Environ()
|
||||
|
||||
@ -68,38 +100,13 @@ func BackupDb() error {
|
||||
cmd.Stdout = &stdout
|
||||
cmd.Stderr = &stderr
|
||||
|
||||
stdin, err := cmd.StdinPipe()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get dumpall stdin: %v", err)
|
||||
}
|
||||
|
||||
start := time.Now()
|
||||
|
||||
if err := cmd.Start(); err != nil {
|
||||
log.Error().
|
||||
Err(err).Str("stdout", stdout.String()).Str("stderr", stderr.String()).
|
||||
Msg("Failed to start backing up the database...")
|
||||
return err
|
||||
}
|
||||
if _, err = stdin.Write([]byte(password + "\n")); err != nil {
|
||||
log.Error().
|
||||
Err(err).Str("stdout", stdout.String()).Str("stderr", stderr.String()).
|
||||
Msg("Failed to passing the password for backuping up the database...")
|
||||
return err
|
||||
}
|
||||
if err = stdin.Close(); err != nil {
|
||||
log.Error().
|
||||
Err(err).Str("stdout", stdout.String()).Str("stderr", stderr.String()).
|
||||
Msg("Failed to end backing up stdin...")
|
||||
return err
|
||||
}
|
||||
if err = cmd.Wait(); err != nil {
|
||||
if err := cmd.Run(); err != nil {
|
||||
log.Error().
|
||||
Err(err).Str("stdout", stdout.String()).Str("stderr", stderr.String()).
|
||||
Msg("Failed to backup the database...")
|
||||
return err
|
||||
}
|
||||
|
||||
took := time.Since(start)
|
||||
|
||||
log.Info().
|
||||
|
Loading…
x
Reference in New Issue
Block a user