mirror of
https://github.com/rancher/rke.git
synced 2025-04-27 19:25:44 +00:00
75 lines
2.4 KiB
Go
75 lines
2.4 KiB
Go
package cluster
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path"
|
|
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/rancher/rke/docker"
|
|
"github.com/rancher/rke/hosts"
|
|
"github.com/rancher/rke/log"
|
|
v3 "github.com/rancher/rke/types"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
ContainerName = "file-deployer"
|
|
ServiceName = "file-deploy"
|
|
ConfigEnv = "FILE_DEPLOY"
|
|
)
|
|
|
|
func deployFile(ctx context.Context, uniqueHosts []*hosts.Host, alpineImage string, prsMap map[string]v3.PrivateRegistry, fileName, fileContents string) error {
|
|
for _, host := range uniqueHosts {
|
|
log.Infof(ctx, "[%s] Deploying file [%s] to node [%s]", ServiceName, fileName, host.Address)
|
|
if err := doDeployFile(ctx, host, fileName, fileContents, alpineImage, prsMap); err != nil {
|
|
return fmt.Errorf("[%s] Failed to deploy file [%s] on node [%s]: %v", ServiceName, fileName, host.Address, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func doDeployFile(ctx context.Context, host *hosts.Host, fileName, fileContents, alpineImage string, prsMap map[string]v3.PrivateRegistry) error {
|
|
// remove existing container. Only way it's still here is if previous deployment failed
|
|
if err := docker.DoRemoveContainer(ctx, host.DClient, ContainerName, host.Address); err != nil {
|
|
return err
|
|
}
|
|
var cmd, containerEnv []string
|
|
|
|
// fileContents determines if a file is placed or removed
|
|
// exception to this is the cloud-config file, as it is valid being empty (for example, when only specifying the aws cloudprovider and no additional config)
|
|
if fileContents != "" || fileName == cloudConfigFileName {
|
|
containerEnv = []string{ConfigEnv + "=" + fileContents}
|
|
cmd = []string{
|
|
"sh",
|
|
"-c",
|
|
fmt.Sprintf("t=$(mktemp); echo -e \"$%s\" > $t && mv $t %s && chmod 600 %s", ConfigEnv, fileName, fileName),
|
|
}
|
|
} else {
|
|
cmd = []string{
|
|
"sh",
|
|
"-c",
|
|
fmt.Sprintf("rm -f %s", fileName),
|
|
}
|
|
}
|
|
|
|
imageCfg := &container.Config{
|
|
Image: alpineImage,
|
|
Cmd: cmd,
|
|
Env: containerEnv,
|
|
}
|
|
hostCfg := &container.HostConfig{
|
|
Binds: []string{
|
|
fmt.Sprintf("%s:/etc/kubernetes:z", path.Join(host.PrefixPath, "/etc/kubernetes")),
|
|
},
|
|
}
|
|
if err := docker.DoRunOnetimeContainer(ctx, host.DClient, imageCfg, hostCfg, ContainerName, host.Address, ServiceName, prsMap); err != nil {
|
|
return err
|
|
}
|
|
if err := docker.DoRemoveContainer(ctx, host.DClient, ContainerName, host.Address); err != nil {
|
|
return err
|
|
}
|
|
logrus.Debugf("[%s] Successfully deployed file [%s] on node [%s]", ServiceName, fileName, host.Address)
|
|
return nil
|
|
}
|