62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
|
// Copyright (C) 2022 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package storagenodedbtest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"archive/zip"
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
_ "embed"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/zeebo/errs"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
|
||
|
|
"storj.io/storj/storagenode/storagenodedb"
|
||
|
|
)
|
||
|
|
|
||
|
|
//go:generate go run ./gensnapshot testdata/snapshot.zip
|
||
|
|
//go:embed testdata/snapshot.zip
|
||
|
|
var snapshotZip []byte
|
||
|
|
|
||
|
|
// OpenNew opens a new storage node database pre-populated with a newly
|
||
|
|
// initialized and migrated database snapshot.
|
||
|
|
func OpenNew(ctx context.Context, log *zap.Logger, config storagenodedb.Config) (*storagenodedb.DB, error) {
|
||
|
|
db, err := storagenodedb.OpenNew(ctx, log, config)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if err := deploySnapshot(db.DBDirectory()); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func deploySnapshot(storageDir string) error {
|
||
|
|
zipReader, err := zip.NewReader(bytes.NewReader(snapshotZip), int64(len(snapshotZip)))
|
||
|
|
if err != nil {
|
||
|
|
return errs.Wrap(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, f := range zipReader.File {
|
||
|
|
rc, err := f.Open()
|
||
|
|
if err != nil {
|
||
|
|
return errs.Wrap(err)
|
||
|
|
}
|
||
|
|
data, err := io.ReadAll(rc)
|
||
|
|
if err != nil {
|
||
|
|
return errs.Wrap(err)
|
||
|
|
}
|
||
|
|
if err := os.WriteFile(filepath.Join(storageDir, f.Name), data, 0644); err != nil {
|
||
|
|
return errs.Wrap(err)
|
||
|
|
}
|
||
|
|
if err := rc.Close(); err != nil {
|
||
|
|
return errs.Wrap(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|