3455f491ca
This commit adds and passes context around to (almost) every function. This allows cancelling (almost) every function call globally.
39 lines
1.2 KiB
Go
39 lines
1.2 KiB
Go
package triplestore
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/FAU-CDI/wisski-distillery/internal/dis/component"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/models"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func (Triplestore) SnapshotNeedsRunning() bool { return false }
|
|
|
|
func (Triplestore) SnapshotName() string { return "triplestore" }
|
|
|
|
func (ts *Triplestore) Snapshot(wisski models.Instance, scontext component.StagingContext) error {
|
|
return scontext.AddDirectory(".", func(ctx context.Context) error {
|
|
return scontext.AddFile(wisski.GraphDBRepository+".nq", func(ctx context.Context, file io.Writer) error {
|
|
_, err := ts.SnapshotDB(ctx, file, wisski.GraphDBRepository)
|
|
return err
|
|
})
|
|
})
|
|
}
|
|
|
|
var errTSBackupWrongStatusCode = errors.New("Triplestore.Backup: Wrong status code")
|
|
|
|
// SnapshotDB snapshots the provided repository into dst
|
|
func (ts Triplestore) SnapshotDB(ctx context.Context, dst io.Writer, repo string) (int64, error) {
|
|
res, err := ts.OpenRaw(ctx, "GET", "/repositories/"+repo+"/statements?infer=false", nil, "", "application/n-quads")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if res.StatusCode != http.StatusOK {
|
|
return 0, errTSBackupWrongStatusCode
|
|
}
|
|
defer res.Body.Close()
|
|
return io.Copy(dst, res.Body)
|
|
}
|