diff --git a/pkg/cli/cmds/server.go b/pkg/cli/cmds/server.go index 56a7374c29be..31b6c9bf2a85 100644 --- a/pkg/cli/cmds/server.go +++ b/pkg/cli/cmds/server.go @@ -81,7 +81,51 @@ type Server struct { EtcdS3Folder string } -var ServerConfig Server +var ( + ServerConfig Server + ClusterCIDR = cli.StringSliceFlag{ + Name: "cluster-cidr", + Usage: "(networking) IPv4/IPv6 network CIDRs to use for pod IPs (default: 10.42.0.0/16)", + Value: &ServerConfig.ClusterCIDR, + } + ServiceCIDR = cli.StringSliceFlag{ + Name: "service-cidr", + Usage: "(networking) IPv4/IPv6 network CIDRs to use for service IPs (default: 10.43.0.0/16)", + Value: &ServerConfig.ServiceCIDR, + } + ServiceNodePortRange = cli.StringFlag{ + Name: "service-node-port-range", + Usage: "(networking) Port range to reserve for services with NodePort visibility", + Destination: &ServerConfig.ServiceNodePortRange, + Value: "30000-32767", + } + ClusterDNS = cli.StringSliceFlag{ + Name: "cluster-dns", + Usage: "(networking) IPv4 Cluster IP for coredns service. Should be in your service-cidr range (default: 10.43.0.10)", + Value: &ServerConfig.ClusterDNS, + } + ClusterDomain = cli.StringFlag{ + Name: "cluster-domain", + Usage: "(networking) Cluster Domain", + Destination: &ServerConfig.ClusterDomain, + Value: "cluster.local", + } + ExtraAPIArgs = cli.StringSliceFlag{ + Name: "kube-apiserver-arg", + Usage: "(flags) Customized flag for kube-apiserver process", + Value: &ServerConfig.ExtraAPIArgs, + } + ExtraSchedulerArgs = cli.StringSliceFlag{ + Name: "kube-scheduler-arg", + Usage: "(flags) Customized flag for kube-scheduler process", + Value: &ServerConfig.ExtraSchedulerArgs, + } + ExtraControllerArgs = cli.StringSliceFlag{ + Name: "kube-controller-manager-arg", + Usage: "(flags) Customized flag for kube-controller-manager process", + Value: &ServerConfig.ExtraControllerArgs, + } +) func NewServerCommand(action func(*cli.Context) error) cli.Command { return cli.Command{ @@ -128,33 +172,11 @@ func NewServerCommand(action func(*cli.Context) error) cli.Command { Usage: "(data) Folder to hold state default /var/lib/rancher/" + version.Program + " or ${HOME}/.rancher/" + version.Program + " if not root", Destination: &ServerConfig.DataDir, }, - cli.StringSliceFlag{ - Name: "cluster-cidr", - Usage: "(networking) IPv4/IPv6 network CIDRs to use for pod IPs (default: 10.42.0.0/16)", - Value: &ServerConfig.ClusterCIDR, - }, - cli.StringSliceFlag{ - Name: "service-cidr", - Usage: "(networking) IPv4/IPv6 network CIDRs to use for service IPs (default: 10.43.0.0/16)", - Value: &ServerConfig.ServiceCIDR, - }, - cli.StringFlag{ - Name: "service-node-port-range", - Usage: "(networking) Port range to reserve for services with NodePort visibility", - Destination: &ServerConfig.ServiceNodePortRange, - Value: "30000-32767", - }, - cli.StringSliceFlag{ - Name: "cluster-dns", - Usage: "(networking) IPv4 Cluster IP for coredns service. Should be in your service-cidr range (default: 10.43.0.10)", - Value: &ServerConfig.ClusterDNS, - }, - cli.StringFlag{ - Name: "cluster-domain", - Usage: "(networking) Cluster Domain", - Destination: &ServerConfig.ClusterDomain, - Value: "cluster.local", - }, + ClusterCIDR, + ServiceCIDR, + ServiceNodePortRange, + ClusterDNS, + ClusterDomain, cli.StringFlag{ Name: "flannel-backend", Usage: "(networking) One of 'none', 'vxlan', 'ipsec', 'host-gw', or 'wireguard'", @@ -185,21 +207,9 @@ func NewServerCommand(action func(*cli.Context) error) cli.Command { Destination: &ServerConfig.KubeConfigMode, EnvVar: version.ProgramUpper + "_KUBECONFIG_MODE", }, - cli.StringSliceFlag{ - Name: "kube-apiserver-arg", - Usage: "(flags) Customized flag for kube-apiserver process", - Value: &ServerConfig.ExtraAPIArgs, - }, - cli.StringSliceFlag{ - Name: "kube-scheduler-arg", - Usage: "(flags) Customized flag for kube-scheduler process", - Value: &ServerConfig.ExtraSchedulerArgs, - }, - cli.StringSliceFlag{ - Name: "kube-controller-manager-arg", - Usage: "(flags) Customized flag for kube-controller-manager process", - Value: &ServerConfig.ExtraControllerArgs, - }, + ExtraAPIArgs, + ExtraControllerArgs, + ExtraSchedulerArgs, cli.StringSliceFlag{ Name: "kube-cloud-controller-manager-arg", Usage: "(flags) Customized flag for kube-cloud-controller-manager process", diff --git a/pkg/etcd/etcd.go b/pkg/etcd/etcd.go index fa6f41f50521..03f1a47136f7 100644 --- a/pkg/etcd/etcd.go +++ b/pkg/etcd/etcd.go @@ -210,7 +210,7 @@ func (e *ETCD) Reset(ctx context.Context, rebootstrap func() error) error { return err } logrus.Infof("Retrieving etcd snapshot %s from S3", e.config.ClusterResetRestorePath) - if err := e.s3.download(ctx); err != nil { + if err := e.s3.Download(ctx); err != nil { return err } logrus.Infof("S3 download complete for %s", e.config.ClusterResetRestorePath) @@ -960,7 +960,7 @@ func (e *ETCD) listSnapshots(ctx context.Context, snapshotDir string) ([]Snapsho // if it hasn't yet been initialized. func (e *ETCD) initS3IfNil(ctx context.Context) error { if e.s3 == nil { - s3, err := newS3(ctx, e.config) + s3, err := NewS3(ctx, e.config) if err != nil { return err } diff --git a/pkg/etcd/s3.go b/pkg/etcd/s3.go index c93aec86cac7..cc35cb4ac89c 100644 --- a/pkg/etcd/s3.go +++ b/pkg/etcd/s3.go @@ -34,7 +34,7 @@ type s3 struct { // newS3 creates a new value of type s3 pointer with a // copy of the config.Control pointer and initializes // a new Minio client. -func newS3(ctx context.Context, config *config.Control) (*s3, error) { +func NewS3(ctx context.Context, config *config.Control) (*s3, error) { tr := http.DefaultTransport if config.EtcdS3EndpointCA != "" { trCA, err := setTransportCA(tr, config.EtcdS3EndpointCA, config.EtcdS3SkipSSLVerify) @@ -109,7 +109,7 @@ func (s *s3) upload(ctx context.Context, snapshot string) error { // download downloads the given snapshot from the configured S3 // compatible backend. -func (s *s3) download(ctx context.Context) error { +func (s *s3) Download(ctx context.Context) error { var remotePath string if s.config.EtcdS3Folder != "" { remotePath = filepath.Join(s.config.EtcdS3Folder, s.config.ClusterResetRestorePath)