enhance join arguments generation logic using template

Signed-off-by: 欧阳钦华10079130 <[email protected]>

change JoinArgsData to joinArgsData, improve template readability

Signed-off-by: redhatlinux10 <[email protected]>

fix extra = type

Signed-off-by: redhatlinux10 <[email protected]>

Author: bruceauyeung

cmd/kubeadm/app/cmd/init.go

@@ -17,10 +17,11 @@ limitations under the License.
 package cmd
 
 import (
+	"bytes"
 	"fmt"
+	"html/template"
 	"io"
 	"io/ioutil"
-	"strings"
 
 	"github.com/renstrom/dedent"
 	"github.com/spf13/cobra"
@@ -36,6 +37,18 @@ import (
 	netutil "k8s.io/kubernetes/pkg/util/net"
 )
 
+const (
+	joinArgsTemplateLiteral = `--token={{.Cfg.Secrets.GivenToken -}}
+		{{if ne .Cfg.API.BindPort .DefaultAPIBindPort -}}
+		{{" --api-port="}}{{.Cfg.API.BindPort -}}
+		{{end -}}
+		{{if ne .Cfg.Discovery.BindPort .DefaultDiscoveryBindPort -}}
+		{{" --discovery-port="}}{{.Cfg.Discovery.BindPort -}}
+		{{end -}}
+		{{" "}}{{index .Cfg.API.AdvertiseAddresses 0 -}}
+`
+)
+
 var (
 	initDoneMsgf = dedent.Dedent(`
 		Kubernetes master initialised successfully!
@@ -186,6 +199,13 @@ func NewInit(cfgPath string, cfg *kubeadmapi.MasterConfiguration, skipPreFlight
 	return &Init{cfg: cfg}, nil
 }
 
+// joinArgsData denotes a data object which is needed by function generateJoinArgs to generate kubeadm join arguments.
+type joinArgsData struct {
+	Cfg                      *kubeadmapi.MasterConfiguration
+	DefaultAPIBindPort       uint
+	DefaultDiscoveryBindPort uint
+}
+
 // Run executes master node provisioning, including certificates, needed static pod manifests, etc.
 func (i *Init) Run(out io.Writer) error {
 	if err := kubemaster.CreateTokenAuthFile(&i.cfg.Secrets); err != nil {
@@ -239,16 +259,21 @@ func (i *Init) Run(out io.Writer) error {
 		return err
 	}
 
-	// TODO(phase1+) we could probably use templates for this logic, and reference struct fields directly etc
-	joinArgs := []string{fmt.Sprintf("--token=%s", i.cfg.Secrets.GivenToken)}
-	if i.cfg.API.BindPort != kubeadmapi.DefaultAPIBindPort {
-		joinArgs = append(joinArgs, fmt.Sprintf("--api-port=%d", i.cfg.API.BindPort))
-	}
-	if i.cfg.Discovery.BindPort != kubeadmapi.DefaultDiscoveryBindPort {
-		joinArgs = append(joinArgs, fmt.Sprintf("--discovery-port=%d", i.cfg.Discovery.BindPort))
+	data := joinArgsData{i.cfg, kubeadmapi.DefaultAPIBindPort, kubeadmapi.DefaultDiscoveryBindPort}
+	if joinArgs, err := generateJoinArgs(data); err != nil {
+		return err
+	} else {
+		fmt.Fprintf(out, initDoneMsgf, joinArgs)
 	}
-	joinArgs = append(joinArgs, i.cfg.API.AdvertiseAddresses[0])
-	fmt.Fprintf(out, initDoneMsgf, strings.Join(joinArgs, " "))
-
 	return nil
 }
+
+// generateJoinArgs generates kubeadm join arguments
+func generateJoinArgs(data joinArgsData) (string, error) {
+	joinArgsTemplate := template.Must(template.New("joinArgsTemplate").Parse(joinArgsTemplateLiteral))
+	var b bytes.Buffer
+	if err := joinArgsTemplate.Execute(&b, data); err != nil {
+		return "", err
+	}
+	return b.String(), nil
+}