mirror of
https://git.vectorsigma.ru/public/k3s.git
synced 2026-07-28 22:19:34 +00:00
068a01351a
Also cleans up some of the server.Context factory creation stuff to eliminate unused code paths and avoid registering the Helm controller when helm is disabled.
As of fe465cc832 we no longer call NewContext outside pkg/server, so the isServer bool flag to use the supervisor kubeconfig and create an event recorder is unnecessary.
Signed-off-by: Brad Davidson <brad.davidson@rancher.com>
117 lines
3.3 KiB
Go
117 lines
3.3 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
k3scrds "github.com/k3s-io/api/pkg/crds"
|
|
"github.com/k3s-io/api/pkg/generated/controllers/k3s.cattle.io"
|
|
helmcrds "github.com/k3s-io/helm-controller/pkg/crds"
|
|
"github.com/k3s-io/helm-controller/pkg/generated/controllers/helm.cattle.io"
|
|
"github.com/k3s-io/k3s/pkg/util"
|
|
"github.com/k3s-io/k3s/pkg/version"
|
|
"github.com/rancher/wrangler/v3/pkg/crd"
|
|
"github.com/rancher/wrangler/v3/pkg/generated/controllers/apps"
|
|
"github.com/rancher/wrangler/v3/pkg/generated/controllers/batch"
|
|
"github.com/rancher/wrangler/v3/pkg/generated/controllers/core"
|
|
"github.com/rancher/wrangler/v3/pkg/generated/controllers/discovery"
|
|
"github.com/rancher/wrangler/v3/pkg/generated/controllers/rbac"
|
|
"github.com/rancher/wrangler/v3/pkg/start"
|
|
apiextv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
|
|
apiext "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/tools/record"
|
|
"k8s.io/client-go/util/retry"
|
|
)
|
|
|
|
type Context struct {
|
|
K3s *k3s.Factory
|
|
Helm *helm.Factory
|
|
Batch *batch.Factory
|
|
Apps *apps.Factory
|
|
Auth *rbac.Factory
|
|
Core *core.Factory
|
|
Discovery *discovery.Factory
|
|
|
|
Event record.EventRecorder
|
|
K8s kubernetes.Interface
|
|
Ext apiext.Interface
|
|
}
|
|
|
|
func (c *Context) Start(ctx context.Context) error {
|
|
starters := []start.Starter{
|
|
c.K3s, c.Apps, c.Auth, c.Batch, c.Core, c.Discovery,
|
|
}
|
|
if c.Helm != nil {
|
|
starters = append(starters, c.Helm)
|
|
}
|
|
|
|
return start.All(ctx, 5, starters...)
|
|
}
|
|
|
|
func NewContext(ctx context.Context, config *Config) (*Context, error) {
|
|
restConfig, err := util.GetRESTConfig(config.ControlConfig.Runtime.KubeConfigSupervisor)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
restConfig.UserAgent = util.GetUserAgent(version.Program + "-supervisor")
|
|
|
|
k8s, err := kubernetes.NewForConfig(restConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ext, err := apiext.NewForConfig(restConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var hf *helm.Factory
|
|
if !config.ControlConfig.DisableHelmController {
|
|
hf = helm.NewFactoryFromConfigOrDie(restConfig)
|
|
}
|
|
|
|
c := &Context{
|
|
K3s: k3s.NewFactoryFromConfigOrDie(restConfig),
|
|
Auth: rbac.NewFactoryFromConfigOrDie(restConfig),
|
|
Apps: apps.NewFactoryFromConfigOrDie(restConfig),
|
|
Batch: batch.NewFactoryFromConfigOrDie(restConfig),
|
|
Core: core.NewFactoryFromConfigOrDie(restConfig),
|
|
Discovery: discovery.NewFactoryFromConfigOrDie(restConfig),
|
|
Helm: hf,
|
|
|
|
Event: util.BuildControllerEventRecorder(k8s, version.Program+"-supervisor", metav1.NamespaceAll),
|
|
K8s: k8s,
|
|
Ext: ext,
|
|
}
|
|
|
|
if err := c.registerCRDs(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return c, nil
|
|
}
|
|
|
|
type crdLister func() ([]*apiextv1.CustomResourceDefinition, error)
|
|
|
|
func (c *Context) registerCRDs(ctx context.Context) error {
|
|
listers := []crdLister{k3scrds.List}
|
|
if c.Helm != nil {
|
|
listers = append(listers, helmcrds.List)
|
|
}
|
|
|
|
crds := []*apiextv1.CustomResourceDefinition{}
|
|
for _, list := range listers {
|
|
l, err := list()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
crds = append(crds, l...)
|
|
}
|
|
|
|
return retry.RetryOnConflict(retry.DefaultRetry, func() error {
|
|
return crd.BatchCreateCRDs(ctx, c.Ext.ApiextensionsV1().CustomResourceDefinitions(), nil, time.Minute, crds)
|
|
})
|
|
}
|