mirror of
https://github.com/rancher/k3k.git
synced 2026-08-20 12:56:16 +00:00
Hussein GalalandGitHub
3abffae92d
* Adding new k3s function to get the config * Adding a new way to get the bootstrap data directly from the server Pod * Gate the bootstrap fetch if external datastore is in use --------- Signed-off-by: galal-hussein <hussein.galal.ahmed.11@gmail.com>
66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
|
|
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
ctrlruntimeclient "sigs.k8s.io/controller-runtime/pkg/client"
|
|
|
|
"github.com/rancher/k3k/pkg/apis/k3k.io/v1beta1"
|
|
"github.com/rancher/k3k/pkg/controller"
|
|
)
|
|
|
|
const (
|
|
configName = "agent-config"
|
|
)
|
|
|
|
type ResourceEnsurer interface {
|
|
EnsureResources(context.Context) error
|
|
}
|
|
|
|
type Config struct {
|
|
cluster *v1beta1.Cluster
|
|
client ctrlruntimeclient.Client
|
|
scheme *runtime.Scheme
|
|
}
|
|
|
|
func NewConfig(cluster *v1beta1.Cluster, client ctrlruntimeclient.Client) *Config {
|
|
return &Config{
|
|
cluster: cluster,
|
|
client: client,
|
|
scheme: client.Scheme(),
|
|
}
|
|
}
|
|
|
|
func configSecretName(clusterName string) string {
|
|
return controller.SafeConcatNameWithPrefix(clusterName, configName)
|
|
}
|
|
|
|
func ensureObject(ctx context.Context, cfg *Config, obj ctrlruntimeclient.Object) error {
|
|
key := ctrlruntimeclient.ObjectKeyFromObject(obj)
|
|
log := ctrl.LoggerFrom(ctx).WithValues("key", key)
|
|
|
|
if err := controllerutil.SetControllerReference(cfg.cluster, obj, cfg.scheme); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := cfg.client.Create(ctx, obj); err != nil {
|
|
if apierrors.IsAlreadyExists(err) {
|
|
log.V(1).Info(fmt.Sprintf("Resource %T already exists, updating.", obj))
|
|
|
|
return cfg.client.Update(ctx, obj)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
log.V(1).Info(fmt.Sprintf("Creating %T.", obj))
|
|
|
|
return nil
|
|
}
|