feat: provide config option for yurtadm (#1547)

Signed-off-by: Liang Deng <[email protected]>

pkg/yurtadm/cmd/join/join.go

@@ -42,6 +42,7 @@ import (
 )
 
 type joinOptions struct {
+	cfgPath                  string
 	token                    string
 	nodeType                 string
 	nodeName                 string
@@ -112,6 +113,9 @@ func NewCmdJoin(in io.Reader, out io.Writer, outErr io.Writer) *cobra.Command {
 
 // addJoinConfigFlags adds join flags bound to the config to the specified flagset
 func addJoinConfigFlags(flagSet *flag.FlagSet, joinOptions *joinOptions) {
+	flagSet.StringVar(
+		&joinOptions.cfgPath, yurtconstants.CfgPath, "", "Path to a joinConfiguration file.",
+	)
 	flagSet.StringVar(
 		&joinOptions.token, yurtconstants.TokenStr, "",
 		"Use this token for both discovery-token and tls-bootstrap-token when those values are not provided.",
@@ -207,6 +211,7 @@ func (nodeJoiner *nodeJoiner) Run() error {
 }
 
 type joinData struct {
+	cfgPath                  string
 	joinNodeData             *joindata.NodeRegistration
 	apiServerEndpoint        string
 	token                    string
@@ -271,6 +276,7 @@ func newJoinData(args []string, opt *joinOptions) (*joinData, error) {
 	}
 
 	data := &joinData{
+		cfgPath:               opt.cfgPath,
 		apiServerEndpoint:     apiServerEndpoint,
 		token:                 opt.token,
 		tlsBootstrapCfg:       nil,
@@ -366,6 +372,11 @@ func newJoinData(args []string, opt *joinOptions) (*joinData, error) {
 	return data, nil
 }
 
+// CfgPath returns path to a joinConfiguration file.
+func (j *joinData) CfgPath() string {
+	return j.cfgPath
+}
+
 // ServerAddr returns the public address of kube-apiserver.
 func (j *joinData) ServerAddr() string {
 	return j.apiServerEndpoint

pkg/yurtadm/cmd/join/joindata/data.go

@@ -31,6 +31,7 @@ type NodeRegistration struct {
 }
 
 type YurtJoinData interface {
+	CfgPath() string
 	ServerAddr() string
 	JoinToken() string
 	PauseImage() string

pkg/yurtadm/cmd/join/phases/joinnode.go

@@ -29,7 +29,12 @@ import (
 
 // RunJoinNode executes the node join process.
 func RunJoinNode(data joindata.YurtJoinData, out io.Writer, outErr io.Writer) error {
-	kubeadmJoinConfigFilePath := filepath.Join(constants.KubeletWorkdir, constants.KubeadmJoinConfigFileName)
+	var kubeadmJoinConfigFilePath string
+	if data.CfgPath() != "" {
+		kubeadmJoinConfigFilePath = data.CfgPath()
+	} else {
+		kubeadmJoinConfigFilePath = filepath.Join(constants.KubeletWorkdir, constants.KubeadmJoinConfigFileName)
+	}
 	kubeadmCmd := exec.Command("kubeadm", "join", fmt.Sprintf("--config=%s", kubeadmJoinConfigFilePath))
 	kubeadmCmd.Stdout = out
 	kubeadmCmd.Stderr = outErr

pkg/yurtadm/cmd/join/phases/prepare.go

@@ -76,8 +76,10 @@ func RunPrepare(data joindata.YurtJoinData) error {
 	if err := yurtadmutil.SetDiscoveryConfig(data); err != nil {
 		return err
 	}
-	if err := yurtadmutil.SetKubeadmJoinConfig(data); err != nil {
-		return err
+	if data.CfgPath() == "" {
+		if err := yurtadmutil.SetKubeadmJoinConfig(data); err != nil {
+			return err
+		}
 	}
 	return nil
 }

pkg/yurtadm/constants/constants.go

@@ -92,7 +92,9 @@ const (
 	Organizations = "organizations"
 	// PauseImage flag sets the pause image for worker node.
 	PauseImage = "pause-image"
-	// TokenStr flags sets both the discovery-token and the tls-bootstrap-token when those values are not provided
+	// CfgPath flag sets the path to a JoinConfiguration file.
+	CfgPath = "config"
+	// TokenStr flag sets both the discovery-token and the tls-bootstrap-token when those values are not provided
 	TokenStr = "token"
 	// TokenDiscoveryCAHash flag instruct kubeadm to validate that the root CA public key matches this hash (for token-based discovery)
 	TokenDiscoveryCAHash = "discovery-token-ca-cert-hash"

pkg/yurtadm/util/yurthub/yurthub_test.go

@@ -235,6 +235,10 @@ type testData struct {
 	joinNodeData *joindata.NodeRegistration
 }
 
+func (j *testData) CfgPath() string {
+	return ""
+}
+
 func (j *testData) ServerAddr() string {
 	return ""
 }