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
|
package instance
import (
"context"
"fmt"
"path/filepath"
"strings"
"github.com/pingcap/tiup/pkg/utils"
)
// DMWorker represent a DM worker instance.
type DMWorker struct {
instance
Process
masters []*DMMaster
}
var _ Instance = &DMWorker{}
// NewDMWorker create a DMWorker instance.
func NewDMWorker(shOpt SharedOptions, binPath string, dir, host, configPath string, id int, port int, masters []*DMMaster) *DMWorker {
if port <= 0 {
port = 8262
}
return &DMWorker{
instance: instance{
BinPath: binPath,
ID: id,
Dir: dir,
Host: host,
Port: utils.MustGetFreePort(host, port, shOpt.PortOffset),
ConfigPath: configPath,
},
masters: masters,
}
}
// MasterAddrs return the master addresses.
func (w *DMWorker) MasterAddrs() []string {
var addrs []string
for _, master := range w.masters {
addrs = append(addrs, utils.JoinHostPort(AdvertiseHost(master.Host), master.StatusPort))
}
return addrs
}
// Name return the name of the instance.
func (w *DMWorker) Name() string {
return fmt.Sprintf("dm-worker-%d", w.ID)
}
// Start starts the instance.
func (w *DMWorker) Start(ctx context.Context) error {
args := []string{
fmt.Sprintf("--name=%s", w.Name()),
fmt.Sprintf("--worker-addr=%s", utils.JoinHostPort(w.Host, w.Port)),
fmt.Sprintf("--advertise-addr=%s", utils.JoinHostPort(AdvertiseHost(w.Host), w.Port)),
fmt.Sprintf("--join=%s", strings.Join(w.MasterAddrs(), ",")),
fmt.Sprintf("--log-file=%s", w.LogFile()),
}
if w.ConfigPath != "" {
args = append(args, fmt.Sprintf("--config=%s", w.ConfigPath))
}
w.Process = &process{cmd: PrepareCommand(ctx, w.BinPath, args, nil, w.Dir)}
logIfErr(w.Process.SetOutputFile(w.LogFile()))
return w.Process.Start()
}
// Component return the component of the instance.
func (w *DMWorker) Component() string {
return "dm-worker"
}
// LogFile return the log file of the instance.
func (w *DMWorker) LogFile() string {
return filepath.Join(w.Dir, "dm-worker.log")
}
|
