mirror of
https://github.com/rancher/k3k.git
synced 2026-05-03 16:06:49 +00:00
ec0e5a4a87
* Support for multi node in shared mode Signed-off-by: galal-hussein <hussein.galal.ahmed.11@gmail.com> * Fixing typo Signed-off-by: galal-hussein <hussein.galal.ahmed.11@gmail.com> * Fixes Signed-off-by: galal-hussein <hussein.galal.ahmed.11@gmail.com> --------- Signed-off-by: galal-hussein <hussein.galal.ahmed.11@gmail.com>
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/rancher/k3k/pkg/apis/k3k.io/v1alpha1"
|
|
"github.com/rancher/k3k/pkg/controller"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
ctrlruntimeclient "sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
|
|
)
|
|
|
|
const (
|
|
configName = "agent-config"
|
|
)
|
|
|
|
type ResourceEnsurer interface {
|
|
EnsureResources(context.Context) error
|
|
}
|
|
|
|
type Config struct {
|
|
cluster *v1alpha1.Cluster
|
|
client ctrlruntimeclient.Client
|
|
scheme *runtime.Scheme
|
|
}
|
|
|
|
func NewConfig(cluster *v1alpha1.Cluster, client ctrlruntimeclient.Client, scheme *runtime.Scheme) *Config {
|
|
return &Config{
|
|
cluster: cluster,
|
|
client: client,
|
|
scheme: scheme,
|
|
}
|
|
}
|
|
|
|
func configSecretName(clusterName string) string {
|
|
return controller.SafeConcatNameWithPrefix(clusterName, configName)
|
|
}
|
|
|
|
func ensureObject(ctx context.Context, cfg *Config, obj ctrlruntimeclient.Object) error {
|
|
log := ctrl.LoggerFrom(ctx)
|
|
|
|
result, err := controllerutil.CreateOrUpdate(ctx, cfg.client, obj, func() error {
|
|
return controllerutil.SetControllerReference(cfg.cluster, obj, cfg.scheme)
|
|
})
|
|
|
|
if result != controllerutil.OperationResultNone {
|
|
key := ctrlruntimeclient.ObjectKeyFromObject(obj)
|
|
log.Info(fmt.Sprintf("ensuring %T", obj), "key", key, "result", result)
|
|
}
|
|
|
|
return err
|
|
}
|