k3s/pkg/server/context.go

70 lines
2.0 KiB
Go
Raw Normal View History

2019-05-09 22:05:51 +00:00
package server
import (
"context"
"github.com/rancher/helm-controller/pkg/generated/controllers/helm.cattle.io"
2019-05-09 22:05:51 +00:00
"github.com/rancher/k3s/pkg/generated/controllers/k3s.cattle.io"
"github.com/rancher/wrangler-api/pkg/generated/controllers/apps"
"github.com/rancher/wrangler-api/pkg/generated/controllers/batch"
"github.com/rancher/wrangler-api/pkg/generated/controllers/core"
"github.com/rancher/wrangler-api/pkg/generated/controllers/rbac"
2019-05-09 22:05:51 +00:00
"github.com/rancher/wrangler/pkg/apply"
"github.com/rancher/wrangler/pkg/crd"
"github.com/rancher/wrangler/pkg/start"
2019-08-27 04:41:20 +00:00
"k8s.io/client-go/kubernetes"
2019-05-09 22:05:51 +00:00
"k8s.io/client-go/rest"
2019-08-27 04:41:20 +00:00
"k8s.io/client-go/tools/clientcmd"
2019-05-09 22:05:51 +00:00
)
type Context struct {
K3s *k3s.Factory
Helm *helm.Factory
Batch *batch.Factory
2019-05-09 22:05:51 +00:00
Apps *apps.Factory
Auth *rbac.Factory
2019-05-09 22:05:51 +00:00
Core *core.Factory
K8s kubernetes.Interface
Apply apply.Apply
}
func (c *Context) Start(ctx context.Context) error {
return start.All(ctx, 5, c.K3s, c.Helm, c.Apps, c.Auth, c.Batch, c.Core)
2019-05-09 22:05:51 +00:00
}
func newContext(ctx context.Context, cfg string) (*Context, error) {
restConfig, err := clientcmd.BuildConfigFromFlags("", cfg)
if err != nil {
return nil, err
}
if err := crds(ctx, restConfig); err != nil {
return nil, err
}
k8s := kubernetes.NewForConfigOrDie(restConfig)
return &Context{
K3s: k3s.NewFactoryFromConfigOrDie(restConfig),
Helm: helm.NewFactoryFromConfigOrDie(restConfig),
2019-05-09 22:05:51 +00:00
K8s: k8s,
Auth: rbac.NewFactoryFromConfigOrDie(restConfig),
2019-05-09 22:05:51 +00:00
Apps: apps.NewFactoryFromConfigOrDie(restConfig),
Batch: batch.NewFactoryFromConfigOrDie(restConfig),
2019-05-09 22:05:51 +00:00
Core: core.NewFactoryFromConfigOrDie(restConfig),
2019-12-12 01:23:55 +00:00
Apply: apply.New(k8s, apply.NewClientFactory(restConfig)).WithDynamicLookup(),
2019-05-09 22:05:51 +00:00
}, nil
}
func crds(ctx context.Context, config *rest.Config) error {
factory, err := crd.NewFactoryFromClient(config)
if err != nil {
return err
}
factory.BatchCreateCRDs(ctx, crd.NamespacedTypes(
"Addon.k3s.cattle.io/v1",
"HelmChart.helm.cattle.io/v1")...)
2019-05-09 22:05:51 +00:00
return factory.BatchWait()
}