prysm-pulse/beacon-chain/db/http_backup_handler.go
Raul Jordan a468a12ef0
Allow Specifying Output Directory When Creating DB Backups (#7630)
* move flags

* backup db output dir flag

* fix build

* fix up broken backup test

* Radek's feedback

Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
2020-10-26 18:20:54 +00:00

30 lines
737 B
Go

package db
import (
"context"
"fmt"
"net/http"
"github.com/sirupsen/logrus"
)
// BackupHandler for accepting requests to initiate a new database backup.
func BackupHandler(db Database, outputDir string) func(http.ResponseWriter, *http.Request) {
log := logrus.WithField("prefix", "db")
return func(w http.ResponseWriter, _ *http.Request) {
log.Debug("Creating database backup from HTTP webhook.")
if err := db.Backup(context.Background(), outputDir); err != nil {
log.WithError(err).Error("Failed to create backup")
w.WriteHeader(http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusOK)
_, err := fmt.Fprint(w, "OK")
if err != nil {
log.WithError(err).Error("Failed to write OK")
}
}
}