mashibing-deployment/test/framework/installer.go

package framework

import (
	"k8s.io/apimachinery/pkg/util/validation/field"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"sync"
)

// 使用步骤：
// 1. installer ：= NewInstaller（config）
// 2. installer.Install()

type Installer struct {
	Steps []Install `json:"steps"`

	config *Config
	once   sync.Once
}

func NewInstaller(config *Config) *Installer {
	return &Installer{config: config}
}

// 加载配置到 installer 对象
func (i *Installer) init() error {
	var err error
	// 只能执行一次
	i.once.Do(func() {
		// 安全检查，config不能为空
		if i.config == nil || i.config.Sub("install") == nil {
			err = field.Invalid(
				field.NewPath("install"),
				nil,
				"Not inital config object")
			return
		}

		installer := new(Installer)
		if e := i.config.Sub("install").Unmarshal(installer); e != nil {
			err = e
			return
		}
		if installer.Steps != nil {
			i.Steps = installer.Steps
		}
	})

	return err
}

func (i *Installer) Install() error {
	// 1. 初始化，加载配置
	if err := i.init(); err != nil {
		return err
	}

	// 2. 判断执行的队列是否存在
	if len(i.Steps) == 0 {
		return nil
	}

	// 3. 遍历这个队列，执行 validate
	root := field.NewPath("install")
	for index, inst := range i.Steps {
		fld := root.Index(index)
		if err := (&inst).validate(fld); err != nil {
			return err
		}
	}

	// 4. 遍历这个队列，执行 install
	for _, inst := range i.Steps {
		inst.config = i.config
		if err := (&inst).install(); err != nil {
			return err
		}
	}
	return nil
}

type Install struct {
	Name       string   `json:"name"`
	Cmd        string   `json:"cmd"`
	Args       []string `json:"args"`
	Path       string   `json:"path"`
	IgnoreFail bool     `json:"ignoreFail"`

	config *Config `json:"-"`
}

func (i *Install) validate(root *field.Path) error {
	errs := field.ErrorList{}
	// 1. 验证 name 字段
	if strings.TrimSpace(i.Name) == "" {
		errs = append(errs, field.Invalid(root.Child("name"), i.Name, "Cannot be empty"))
	}

	// 2. 验证 cmd 字段
	if strings.TrimSpace(i.Cmd) == "" {
		errs = append(errs, field.Invalid(root.Child("cmd"), i.Cmd, "Cannot be empty"))
	}

	// 3. 检查并设置 path 默认值
	if strings.TrimSpace(i.Path) == "" {
		i.Path = "."
	}

	return errs.ToAggregate()
}

func (i *Install) install() error {
	// 1. 获取当前路径
	currentDir, err := os.Getwd()
	if err != nil {
		return err
	}

	absPath := i.Path
	if !filepath.IsAbs(absPath) {
		if absPath, err = filepath.Abs(absPath); err != nil {
			return err
		}
	}
	// 2. 切换路径
	// 2.1  判断路径是否相同
	if currentDir != absPath {
		// 2.1.1 不同责切换路径
		if err := os.Chdir(absPath); err != nil {
			return err
		}
	}

	// 3. defer 推出前回到之前的路径
	defer func() { _ = os.Chdir(currentDir) }()

	// 4. 执行命令
	cmd := exec.Command(i.Cmd, i.Args...)
	cmd.Stdout = i.config.Stdout
	cmd.Stderr = i.config.Stderr
	return cmd.Run()
}