1
0
mirror of https://github.com/rancher/rke.git synced 2025-04-28 03:31:24 +00:00
rke/services/kubecontroller.go
2017-10-29 11:45:21 +02:00

80 lines
2.5 KiB
Go

package services
import (
"context"
"fmt"
"github.com/Sirupsen/logrus"
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/container"
"github.com/rancher/rke/hosts"
)
type KubeController struct {
Version string `yaml:"version"`
Image string `yaml:"image"`
}
func runKubeController(host hosts.Host, kubeControllerService KubeController) error {
isRunning, err := IsContainerRunning(host, KubeControllerContainerName)
if err != nil {
return err
}
if isRunning {
logrus.Infof("[ControlPlane] Kube-Controller is already running on host [%s]", host.Hostname)
return nil
}
err = runKubeControllerContainer(host, kubeControllerService)
if err != nil {
return err
}
return nil
}
func runKubeControllerContainer(host hosts.Host, kubeControllerService KubeController) error {
logrus.Debugf("[ControlPlane] Pulling Kube Controller Image on host [%s]", host.Hostname)
err := PullImage(host, kubeControllerService.Image+":"+kubeControllerService.Version)
if err != nil {
return err
}
logrus.Infof("[ControlPlane] Successfully pulled Kube Controller image on host [%s]", host.Hostname)
err = doRunKubeController(host, kubeControllerService)
if err != nil {
return err
}
logrus.Infof("[ControlPlane] Successfully ran Kube Controller container on host [%s]", host.Hostname)
return nil
}
func doRunKubeController(host hosts.Host, kubeControllerService KubeController) error {
imageCfg := &container.Config{
Image: kubeControllerService.Image + ":" + kubeControllerService.Version,
Cmd: []string{"/hyperkube",
"controller-manager",
"--address=0.0.0.0",
"--cloud-provider=",
"--master=http://" + host.IP + ":8080",
"--enable-hostpath-provisioner=false",
"--node-monitor-grace-period=40s",
"--pod-eviction-timeout=5m0s",
"--v=2",
"--allocate-node-cidrs=true",
"--cluster-cidr=10.233.64.0/18",
"--service-cluster-ip-range=10.233.0.0/18"},
}
hostCfg := &container.HostConfig{
RestartPolicy: container.RestartPolicy{Name: "always"},
}
resp, err := host.DClient.ContainerCreate(context.Background(), imageCfg, hostCfg, nil, KubeControllerContainerName)
if err != nil {
return fmt.Errorf("Failed to create Kube Controller container on host [%s]: %v", host.Hostname, err)
}
if err := host.DClient.ContainerStart(context.Background(), resp.ID, types.ContainerStartOptions{}); err != nil {
return fmt.Errorf("Failed to start Kube Controller container on host [%s]: %v", host.Hostname, err)
}
logrus.Debugf("[ControlPlane] Successfully started Kube Controller container: %s", resp.ID)
return nil
}