mirror of https://github.com/k3s-io/k3s
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package apiaddresses
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/k3s-io/k3s/pkg/daemons/config"
|
|
"github.com/k3s-io/k3s/pkg/etcd"
|
|
"github.com/k3s-io/k3s/pkg/util"
|
|
"github.com/k3s-io/k3s/pkg/version"
|
|
controllerv1 "github.com/rancher/wrangler/pkg/generated/controllers/core/v1"
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
v1 "k8s.io/api/core/v1"
|
|
)
|
|
|
|
type EndpointsControllerGetter func() controllerv1.EndpointsController
|
|
|
|
func Register(ctx context.Context, runtime *config.ControlRuntime, endpoints controllerv1.EndpointsController) error {
|
|
h := &handler{
|
|
endpointsController: endpoints,
|
|
runtime: runtime,
|
|
ctx: ctx,
|
|
}
|
|
endpoints.OnChange(ctx, version.Program+"-apiserver-lb-controller", h.sync)
|
|
|
|
cl, err := etcd.GetClient(h.ctx, h.runtime)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
h.etcdClient = cl
|
|
|
|
go func() {
|
|
<-ctx.Done()
|
|
h.etcdClient.Close()
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
type handler struct {
|
|
endpointsController controllerv1.EndpointsController
|
|
runtime *config.ControlRuntime
|
|
ctx context.Context
|
|
etcdClient *clientv3.Client
|
|
}
|
|
|
|
// This controller will update the version.program/apiaddresses etcd key with a list of
|
|
// api addresses endpoints found in the kubernetes service in the default namespace
|
|
func (h *handler) sync(key string, endpoint *v1.Endpoints) (*v1.Endpoints, error) {
|
|
if endpoint != nil &&
|
|
endpoint.Namespace == "default" &&
|
|
endpoint.Name == "kubernetes" {
|
|
w := &bytes.Buffer{}
|
|
if err := json.NewEncoder(w).Encode(util.GetAddresses(endpoint)); err != nil {
|
|
return nil, err
|
|
}
|
|
_, err := h.etcdClient.Put(h.ctx, etcd.AddressKey, w.String())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return endpoint, nil
|
|
}
|