mirror of
https://github.com/rancher/rke.git
synced 2025-04-28 03:31:24 +00:00
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/rancher/rke/docker"
|
|
"github.com/rancher/rke/hosts"
|
|
"github.com/rancher/rke/log"
|
|
"github.com/rancher/types/apis/management.cattle.io/v3"
|
|
)
|
|
|
|
const (
|
|
ETCDRole = "etcd"
|
|
ControlRole = "controlplane"
|
|
WorkerRole = "worker"
|
|
|
|
SidekickServiceName = "sidekick"
|
|
RBACAuthorizationMode = "rbac"
|
|
|
|
KubeAPIContainerName = "kube-api"
|
|
KubeletContainerName = "kubelet"
|
|
KubeproxyContainerName = "kube-proxy"
|
|
KubeControllerContainerName = "kube-controller"
|
|
SchedulerContainerName = "scheduler"
|
|
EtcdContainerName = "etcd"
|
|
NginxProxyContainerName = "nginx-proxy"
|
|
SidekickContainerName = "service-sidekick"
|
|
|
|
KubeAPIPort = 6443
|
|
SchedulerPort = 10251
|
|
KubeControllerPort = 10252
|
|
KubeletPort = 10250
|
|
KubeproxyPort = 10256
|
|
)
|
|
|
|
func GetKubernetesServiceIP(serviceClusterRange string) (net.IP, error) {
|
|
ip, ipnet, err := net.ParseCIDR(serviceClusterRange)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Failed to get kubernetes service IP from Kube API option [service_cluster_ip_range]: %v", err)
|
|
}
|
|
ip = ip.Mask(ipnet.Mask)
|
|
for j := len(ip) - 1; j >= 0; j-- {
|
|
ip[j]++
|
|
if ip[j] > 0 {
|
|
break
|
|
}
|
|
}
|
|
return ip, nil
|
|
}
|
|
|
|
func buildSidekickConfig(sidekickImage string) (*container.Config, *container.HostConfig) {
|
|
imageCfg := &container.Config{
|
|
Image: sidekickImage,
|
|
}
|
|
hostCfg := &container.HostConfig{
|
|
NetworkMode: "none",
|
|
}
|
|
return imageCfg, hostCfg
|
|
}
|
|
|
|
func runSidekick(ctx context.Context, host *hosts.Host, sidekickImage string, prsMap map[string]v3.PrivateRegistry) error {
|
|
isRunning, err := docker.IsContainerRunning(ctx, host.DClient, host.Address, SidekickContainerName, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isRunning {
|
|
log.Infof(ctx, "[%s] Sidekick container already created on host [%s]", SidekickServiceName, host.Address)
|
|
return nil
|
|
}
|
|
imageCfg, hostCfg := buildSidekickConfig(sidekickImage)
|
|
if err := docker.UseLocalOrPull(ctx, host.DClient, host.Address, sidekickImage, SidekickServiceName, prsMap); err != nil {
|
|
return err
|
|
}
|
|
if _, err := docker.CreateContiner(ctx, host.DClient, host.Address, SidekickContainerName, imageCfg, hostCfg); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func removeSidekick(ctx context.Context, host *hosts.Host) error {
|
|
return docker.DoRemoveContainer(ctx, host.DClient, SidekickContainerName, host.Address)
|
|
}
|