forked from wal-g/wal-g
/
segment_config_maker.go
69 lines (56 loc) · 1.72 KB
/
segment_config_maker.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
package greenplum
import (
"encoding/json"
"fmt"
"io"
"os"
"github.com/T0n0T/wal-g/utility"
"github.com/greenplum-db/gp-common-go-libs/cluster"
)
type SegConfigMaker interface {
Make(SegmentMetadata) (cluster.SegConfig, error)
}
type RestoreCfgSegMaker struct {
restoreCfg ClusterRestoreConfig
}
func NewRestoreCfgSegMaker(restoreConfigReader io.Reader) (SegConfigMaker, error) {
restoreCfgBytes, err := io.ReadAll(restoreConfigReader)
if err != nil {
return nil, err
}
var restoreCfg ClusterRestoreConfig
err = json.Unmarshal(restoreCfgBytes, &restoreCfg)
if err != nil {
return nil, fmt.Errorf("failed to unmarshal the provided restore config: %v", err)
}
return &RestoreCfgSegMaker{restoreCfg}, nil
}
func (c *RestoreCfgSegMaker) Make(metadata SegmentMetadata) (cluster.SegConfig, error) {
segmentCfg := metadata.ToSegConfig()
segRestoreCfg, ok := c.restoreCfg.Segments[metadata.ContentID]
if !ok {
return cluster.SegConfig{},
fmt.Errorf(
"could not find content ID %d in the provided restore configuration",
metadata.ContentID)
}
segmentCfg.Hostname = segRestoreCfg.Hostname
segmentCfg.Port = segRestoreCfg.Port
segmentCfg.DataDir = segRestoreCfg.DataDir
return segmentCfg, nil
}
type InPlaceSegMaker struct{}
func (c *InPlaceSegMaker) Make(metadata SegmentMetadata) (cluster.SegConfig, error) {
return metadata.ToSegConfig(), nil
}
func NewSegConfigMaker(restoreCfgPath string, inPlaceRestore bool) (SegConfigMaker, error) {
if inPlaceRestore {
return &InPlaceSegMaker{}, nil
}
file, err := os.Open(restoreCfgPath)
if err != nil {
return nil, fmt.Errorf("failed to open the provided restore config file: %v", err)
}
defer utility.LoggedClose(file, "")
return NewRestoreCfgSegMaker(file)
}