mirror of
https://github.com/k3s-io/k3s.git
synced 2026-02-03 20:39:49 -05:00
c73e9187bb
Disables k8s generic api server systemd ready notification and send our own ready notification after server available and kubeconfig available.
129 lines
3.2 KiB
Go
129 lines
3.2 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
systemd "github.com/coreos/go-systemd/daemon"
|
|
"github.com/docker/docker/pkg/reexec"
|
|
"github.com/natefinch/lumberjack"
|
|
"github.com/rancher/k3s/pkg/agent"
|
|
"github.com/rancher/k3s/pkg/cli/cmds"
|
|
"github.com/rancher/k3s/pkg/server"
|
|
"github.com/rancher/norman/signal"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/urfave/cli"
|
|
"k8s.io/apimachinery/pkg/util/net"
|
|
genericapiserver "k8s.io/apiserver/pkg/server"
|
|
|
|
_ "github.com/mattn/go-sqlite3" // ensure we have sqlite
|
|
)
|
|
|
|
func setupLogging(app *cli.Context) {
|
|
if !app.GlobalBool("debug") {
|
|
flag.Set("stderrthreshold", "3")
|
|
flag.Set("alsologtostderr", "false")
|
|
flag.Set("logtostderr", "false")
|
|
}
|
|
}
|
|
|
|
func runWithLogging(app *cli.Context, cfg *cmds.Server) error {
|
|
l := &lumberjack.Logger{
|
|
Filename: cfg.Log,
|
|
MaxSize: 50,
|
|
MaxBackups: 3,
|
|
MaxAge: 28,
|
|
Compress: true,
|
|
}
|
|
|
|
args := append([]string{"k3s"}, os.Args[1:]...)
|
|
cmd := reexec.Command(args...)
|
|
cmd.Env = os.Environ()
|
|
cmd.Env = append(cmd.Env, "_RIO_REEXEC_=true")
|
|
cmd.Stderr = l
|
|
cmd.Stdout = l
|
|
cmd.Stdin = os.Stdin
|
|
return cmd.Run()
|
|
}
|
|
|
|
func Run(app *cli.Context) error {
|
|
genericapiserver.NotifySystemD = false
|
|
return run(app, &cmds.ServerConfig)
|
|
}
|
|
|
|
func run(app *cli.Context, cfg *cmds.Server) error {
|
|
if cfg.Log != "" && os.Getenv("_RIO_REEXEC_") == "" {
|
|
return runWithLogging(app, cfg)
|
|
}
|
|
|
|
setupLogging(app)
|
|
|
|
if !cfg.DisableAgent && os.Getuid() != 0 {
|
|
return fmt.Errorf("must run as root unless --disable-agent is specified")
|
|
}
|
|
|
|
serverConfig := server.Config{}
|
|
serverConfig.ControlConfig.ClusterSecret = cfg.ClusterSecret
|
|
serverConfig.ControlConfig.DataDir = cfg.DataDir
|
|
serverConfig.ControlConfig.KubeConfigOutput = cfg.KubeConfigOutput
|
|
serverConfig.ControlConfig.KubeConfigMode = cfg.KubeConfigMode
|
|
serverConfig.TLSConfig.HTTPSPort = cfg.HTTPSPort
|
|
serverConfig.TLSConfig.HTTPPort = cfg.HTTPPort
|
|
serverConfig.TLSConfig.KnownIPs = knownIPs()
|
|
|
|
// TODO: support etcd
|
|
serverConfig.ControlConfig.NoLeaderElect = true
|
|
|
|
for _, noDeploy := range app.StringSlice("no-deploy") {
|
|
if noDeploy == "servicelb" {
|
|
serverConfig.DisableServiceLB = true
|
|
continue
|
|
}
|
|
|
|
if !strings.HasSuffix(noDeploy, ".yaml") {
|
|
noDeploy = noDeploy + ".yaml"
|
|
}
|
|
serverConfig.ControlConfig.Skips = append(serverConfig.ControlConfig.Skips, noDeploy)
|
|
}
|
|
|
|
logrus.Info("Starting k3s ", app.App.Version)
|
|
ctx := signal.SigTermCancelContext(context.Background())
|
|
certs, err := server.StartServer(ctx, &serverConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logrus.Info("k3s is up and running")
|
|
go systemd.SdNotify(false, "READY=1")
|
|
|
|
if cfg.DisableAgent {
|
|
<-ctx.Done()
|
|
return nil
|
|
}
|
|
|
|
url := fmt.Sprintf("https://localhost:%d", serverConfig.TLSConfig.HTTPSPort)
|
|
token := server.FormatToken(serverConfig.ControlConfig.Runtime.NodeToken, certs)
|
|
|
|
agentConfig := cmds.AgentConfig
|
|
agentConfig.Debug = app.GlobalBool("bool")
|
|
agentConfig.DataDir = filepath.Dir(serverConfig.ControlConfig.DataDir)
|
|
agentConfig.ServerURL = url
|
|
agentConfig.Token = token
|
|
|
|
return agent.Run(ctx, agentConfig)
|
|
}
|
|
|
|
func knownIPs() []string {
|
|
ips := []string{
|
|
"127.0.0.1",
|
|
}
|
|
ip, err := net.ChooseHostInterface()
|
|
if err == nil {
|
|
ips = append(ips, ip.String())
|
|
}
|
|
return ips
|
|
}
|