-
Notifications
You must be signed in to change notification settings - Fork 309
/
import.go
128 lines (108 loc) · 3.27 KB
/
import.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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// Copyright 2020 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
package command
import (
"context"
"fmt"
"os"
"github.com/fatih/color"
"github.com/pingcap/errors"
"github.com/pingcap/tiup/components/dm/ansible"
"github.com/pingcap/tiup/pkg/cliutil"
cansible "github.com/pingcap/tiup/pkg/cluster/ansible"
"github.com/pingcap/tiup/pkg/cluster/ctxt"
"github.com/pingcap/tiup/pkg/cluster/manager"
tiuputils "github.com/pingcap/tiup/pkg/utils"
"github.com/spf13/cobra"
"gopkg.in/yaml.v2"
)
func newImportCmd() *cobra.Command {
var ansibleDir string
var inventoryFileName string
var rename string
var clusterVersion string
cmd := &cobra.Command{
Use: "import",
Short: "Import an exist DM 1.0 cluster from dm-ansible and re-deploy 2.0 version",
RunE: func(cmd *cobra.Command, args []string) error {
if err := supportVersion(clusterVersion); err != nil {
return err
}
importer, err := ansible.NewImporter(ansibleDir, inventoryFileName, gOpt.SSHType, gOpt.SSHTimeout)
if err != nil {
return err
}
ctx := ctxt.New(context.Background())
clusterName, meta, err := importer.ImportFromAnsibleDir(ctx)
if err != nil {
return err
}
if rename != "" {
clusterName = rename
}
err = importer.ScpSourceToMaster(ctx, meta.Topology)
if err != nil {
return err
}
data, err := yaml.Marshal(meta.Topology)
if err != nil {
return errors.AddStack(err)
}
f, err := os.CreateTemp("", "tiup-*")
if err != nil {
return errors.AddStack(err)
}
_, err = f.Write(data)
if err != nil {
return errors.AddStack(err)
}
fmt.Println(color.HiYellowString("Will use the following topology to deploy a DM cluster: "))
fmt.Println(string(data))
if !skipConfirm {
err = cliutil.PromptForConfirmOrAbortError(
color.HiYellowString("Using the Topology to deploy DM %s cluster %s, Please Stop the DM cluster from ansible side first.\nDo you want to continue? [y/N]: ",
clusterVersion,
clusterName,
))
if err != nil {
return err
}
}
err = cm.Deploy(
clusterName,
clusterVersion,
f.Name(),
manager.DeployOptions{
IdentityFile: cansible.SSHKeyPath(),
User: tiuputils.CurrentUser(),
},
nil,
skipConfirm,
gOpt,
)
if err != nil {
return err
}
return nil
},
}
cmd.Flags().StringVarP(&ansibleDir, "dir", "d", "./", "The path to DM-Ansible directory")
cmd.Flags().StringVar(&inventoryFileName, "inventory", cansible.AnsibleInventoryFile, "The name of inventory file")
cmd.Flags().StringVarP(&rename, "rename", "r", "", "Rename the imported cluster to `NAME`")
cmd.Flags().StringVarP(&clusterVersion, "cluster-version", "v", "", "cluster version of DM to deploy (required)")
err := cmd.MarkFlagRequired("cluster-version")
if err != nil { // if no this flag
panic(err)
}
return cmd
}