/
main.go
102 lines (86 loc) · 1.94 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
// Copyright (C) 2022 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"archive/zip"
"bytes"
"context"
"fmt"
"io"
"os"
"path/filepath"
"github.com/zeebo/errs"
"go.uber.org/zap"
"storj.io/storj/storagenode/storagenodedb"
)
func main() {
outFile := "snapshot.zip"
if len(os.Args) > 1 {
outFile = os.Args[1]
}
if err := run(context.Background(), outFile); err != nil {
fmt.Fprintf(os.Stderr, "Failed to generate snapshot database: %+v\n", err)
os.Exit(1)
}
}
func run(ctx context.Context, outFile string) error {
log, err := zap.NewDevelopment()
if err != nil {
return errs.Wrap(err)
}
tempDir, err := os.MkdirTemp("", "")
if err != nil {
return errs.Wrap(err)
}
defer func() {
if err := os.RemoveAll(tempDir); err != nil {
log.Warn("Could not remove temp directory", zap.Error(err), zap.String("directory", tempDir))
}
}()
cfg := storagenodedb.Config{
Storage: tempDir,
Info: filepath.Join(tempDir, "piecestore.db"),
Info2: filepath.Join(tempDir, "info.db"),
Pieces: tempDir,
TestingDisableWAL: true,
}
db, err := storagenodedb.OpenNew(ctx, log, cfg)
if err != nil {
return errs.Wrap(err)
}
err = db.MigrateToLatest(ctx)
if err != nil {
return errs.Wrap(err)
}
err = db.Close()
if err != nil {
return errs.Wrap(err)
}
matches, err := filepath.Glob(filepath.Join(tempDir, "*.db"))
if err != nil {
return errs.Wrap(err)
}
buf := new(bytes.Buffer)
zipWriter := zip.NewWriter(buf)
for _, match := range matches {
data, err := os.ReadFile(match)
if err != nil {
return errs.Wrap(err)
}
w, err := zipWriter.Create(filepath.Base(match))
if err != nil {
return errs.Wrap(err)
}
_, err = io.Copy(w, bytes.NewReader(data))
if err != nil {
return errs.Wrap(err)
}
}
if err := zipWriter.Close(); err != nil {
return errs.Wrap(err)
}
if err := os.WriteFile(outFile, buf.Bytes(), 0644); err != nil {
return errs.Wrap(err)
}
return nil
}