mirror of
https://github.com/kubesphere/kubekey.git
synced 2025-12-30 15:12:54 +00:00
117 lines
3.4 KiB
Go
117 lines
3.4 KiB
Go
/*
|
|
Copyright 2021 The KubeSphere Authors.
|
|
|
|
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,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/kubesphere/kubekey/pkg/common"
|
|
"github.com/kubesphere/kubekey/pkg/core/connector"
|
|
"github.com/pkg/errors"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strconv"
|
|
"unicode"
|
|
)
|
|
|
|
type ModifyConfig struct {
|
|
common.KubeAction
|
|
}
|
|
|
|
func (m *ModifyConfig) Execute(runtime connector.Runtime) error {
|
|
fp, _ := filepath.Abs(m.KubeConf.Arg.FilePath)
|
|
cmd0 := fmt.Sprintf("cat %s | grep %s | wc -l", fp, m.KubeConf.Arg.NodeName)
|
|
nodeNameNum, err0 := exec.Command("/bin/sh", "-c", cmd0).CombinedOutput()
|
|
if err0 != nil {
|
|
return errors.Wrap(err0, "Failed to get node num")
|
|
}
|
|
|
|
host := &connector.BaseHost{Name: m.KubeConf.Arg.NodeName}
|
|
if string(nodeNameNum) == "2\n" {
|
|
cmd := fmt.Sprintf("sed -i /%s/d %s", m.KubeConf.Arg.NodeName, fp)
|
|
_ = exec.Command("/bin/sh", "-c", cmd).Run()
|
|
runtime.DeleteHost(host)
|
|
} else if string(nodeNameNum) == "1\n" {
|
|
cmd := fmt.Sprintf("sed -i /%s/d %s", m.KubeConf.Arg.NodeName, fp)
|
|
_ = exec.Command("/bin/sh", "-c", cmd).Run()
|
|
runtime.DeleteHost(host)
|
|
|
|
var newNodeName []string
|
|
for i := 0; i < len(runtime.GetHostsByRole(common.Worker)); i++ {
|
|
name := runtime.GetHostsByRole(common.Worker)[i].GetName()
|
|
if m.KubeConf.Arg.NodeName == name {
|
|
continue
|
|
} else {
|
|
newNodeName = append(newNodeName, name)
|
|
}
|
|
}
|
|
var connNodeName []string
|
|
for j := 0; j < len(newNodeName); j++ {
|
|
t := j
|
|
nodename1 := newNodeName[t]
|
|
for t+1 < len(newNodeName) && IsAdjoin(newNodeName[t], newNodeName[t+1]) {
|
|
t++
|
|
}
|
|
if t == j {
|
|
connNodeName = append(connNodeName, nodename1)
|
|
} else {
|
|
connNodeName = append(connNodeName, Merge(nodename1, newNodeName[t]))
|
|
j = t
|
|
}
|
|
}
|
|
cmd1 := fmt.Sprintf("sed -i -n '1,/worker/p;/controlPlaneEndpoint/,$p' %s", fp)
|
|
_ = exec.Command("/bin/sh", "-c", cmd1).Run()
|
|
for k := 0; k < len(connNodeName); k++ {
|
|
workPar := connNodeName[k]
|
|
workPar1 := fmt.Sprintf("%s", workPar)
|
|
cmd2 := fmt.Sprintf("sed -i '/worker/a\\ \\ \\ \\ \\- %s' %s", workPar1, fp)
|
|
_ = exec.Command("/bin/sh", "-c", cmd2).Run()
|
|
}
|
|
} else {
|
|
return errors.New("Please check the node name in the config-sample.yaml or only support to delete worker")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Merge(name1, name2 string) (endName string) {
|
|
par1, par2 := SplitNum(name1)
|
|
_, par4 := SplitNum(name2)
|
|
endName = fmt.Sprintf("%s[%s:%s]", par1, strconv.Itoa(par2), strconv.Itoa(par4))
|
|
return endName
|
|
}
|
|
func IsAdjoin(name1, name2 string) bool {
|
|
IsAd := false
|
|
par1, par2 := SplitNum(name1)
|
|
par3, par4 := SplitNum(name2)
|
|
if par1 == par3 && par4 == par2+1 {
|
|
IsAd = true
|
|
}
|
|
return IsAd
|
|
}
|
|
|
|
func SplitNum(nodeName string) (name string, num int) {
|
|
nodeLen := len(nodeName)
|
|
i := nodeLen - 1
|
|
for ; nodeLen > 0; i-- {
|
|
if !unicode.IsDigit(rune(nodeName[i])) {
|
|
num, _ := strconv.Atoi(nodeName[i+1:])
|
|
name := nodeName[:i+1]
|
|
return name, num
|
|
}
|
|
}
|
|
return "", 0
|
|
}
|