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
|
package instance
import (
"context"
"fmt"
"path/filepath"
"strings"
"github.com/pingcap/tiup/pkg/utils"
)
// DMMaster represent a DM master instance.
type DMMaster struct {
instance
Process
initEndpoints []*DMMaster
}
var _ Instance = &DMMaster{}
// NewDMMaster create a new DMMaster instance.
func NewDMMaster(shOpt SharedOptions, binPath string, dir, host, configPath string, id int, port int) *DMMaster {
if port <= 0 {
port = 8261
}
return &DMMaster{
instance: instance{
BinPath: binPath,
ID: id,
Dir: dir,
Host: host,
Port: utils.MustGetFreePort(host, 8291, shOpt.PortOffset),
// Similar like PD's client port, here use StatusPort for Master Port.
StatusPort: utils.MustGetFreePort(host, port, shOpt.PortOffset),
ConfigPath: configPath,
},
}
}
// Name return the name of the instance.
func (m *DMMaster) Name() string {
return fmt.Sprintf("dm-master-%d", m.ID)
}
// Start starts the instance.
func (m *DMMaster) Start(ctx context.Context) error {
args := []string{
fmt.Sprintf("--name=%s", m.Name()),
fmt.Sprintf("--master-addr=http://%s", utils.JoinHostPort(m.Host, m.StatusPort)),
fmt.Sprintf("--advertise-addr=http://%s", utils.JoinHostPort(AdvertiseHost(m.Host), m.StatusPort)),
fmt.Sprintf("--peer-urls=http://%s", utils.JoinHostPort(m.Host, m.Port)),
fmt.Sprintf("--advertise-peer-urls=http://%s", utils.JoinHostPort(AdvertiseHost(m.Host), m.Port)),
fmt.Sprintf("--log-file=%s", m.LogFile()),
}
endpoints := make([]string, 0)
for _, master := range m.initEndpoints {
endpoints = append(endpoints, fmt.Sprintf("%s=http://%s", master.Name(), utils.JoinHostPort(master.Host, master.Port)))
}
args = append(args, fmt.Sprintf("--initial-cluster=%s", strings.Join(endpoints, ",")))
if m.ConfigPath != "" {
args = append(args, fmt.Sprintf("--config=%s", m.ConfigPath))
}
m.Process = &process{cmd: PrepareCommand(ctx, m.BinPath, args, nil, m.Dir)}
logIfErr(m.Process.SetOutputFile(m.LogFile()))
return m.Process.Start()
}
// SetInitEndpoints set the initial endpoints for the DM master.
func (m *DMMaster) SetInitEndpoints(endpoints []*DMMaster) {
m.initEndpoints = endpoints
}
// Component return the component of the instance.
func (m *DMMaster) Component() string {
return "dm-master"
}
// LogFile return the log file path of the instance.
func (m *DMMaster) LogFile() string {
return filepath.Join(m.Dir, "dm-master.log")
}
// Addr return the address of the instance.
func (m *DMMaster) Addr() string {
return utils.JoinHostPort(m.Host, m.StatusPort)
}
|
