91 lines
2 KiB
Go
91 lines
2 KiB
Go
package trb
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/FAU-CDI/wisski-distillery/internal/wisski/ingredient"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/wisski/ingredient/barrel"
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/logging"
|
|
)
|
|
|
|
type TRB struct {
|
|
ingredient.Base
|
|
|
|
dependencies struct {
|
|
Barrel *barrel.Barrel
|
|
}
|
|
}
|
|
|
|
func (trb *TRB) DoSomething(ctx context.Context, out io.Writer, allowEmptyRepository bool) (err error) {
|
|
|
|
// stop instance, restart when done
|
|
logging.LogMessage(out, "Shutting down instance")
|
|
if err := trb.dependencies.Barrel.Stack().Down(ctx, out); err != nil {
|
|
return err
|
|
}
|
|
|
|
defer func() {
|
|
logging.LogMessage(out, "Restarting instance")
|
|
e := trb.dependencies.Barrel.Stack().Up(ctx, out)
|
|
if err == nil {
|
|
err = e
|
|
}
|
|
}()
|
|
|
|
// make the backup
|
|
logging.LogMessage(out, "Dumping triplestore")
|
|
path, err := trb.makeBackup(ctx, allowEmptyRepository)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("Wrote %q\n", path)
|
|
|
|
logging.LogMessage(out, "Purging triplestore")
|
|
if err := trb.Malt.TS.Purge(ctx, trb.Instance, trb.Domain()); err != nil {
|
|
return err
|
|
}
|
|
|
|
logging.LogMessage(out, "Provising triplestore")
|
|
if err := trb.Malt.TS.Provision(ctx, trb.Instance, trb.Domain()); err != nil {
|
|
return err
|
|
}
|
|
|
|
logging.LogMessage(out, "Loading dump file")
|
|
content, err := os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer content.Close()
|
|
|
|
logging.LogMessage(out, "Restoring triplestore")
|
|
if err := trb.Malt.TS.RestoreDB(ctx, trb.GraphDBRepository, content); err != nil {
|
|
return err
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
var errBackupEmpty = errors.New("no data contained in backup file (is the repository empty?)")
|
|
|
|
func (trb *TRB) makeBackup(ctx context.Context, allowEmptyRepository bool) (path string, err error) {
|
|
f, err := os.CreateTemp("", "")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer f.Close()
|
|
|
|
count, err := trb.Malt.TS.SnapshotDB(ctx, f, trb.GraphDBRepository)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if count == 0 && !allowEmptyRepository {
|
|
return "", errBackupEmpty
|
|
}
|
|
|
|
return f.Name(), nil
|
|
}
|