mirror of
https://github.com/cloudnative-pg/plugin-barman-cloud.git
synced 2026-01-12 05:33:11 +01:00
f936bf128f
This patch allows the barman-cloud plugin to work with operator being structurally identical with CNPG but with a different API group. It does that by using lenient decoding of the passed CNPG resources and by injecting the detected GVK to the sidecar, that uses it to properly encode and decode the Kubernetes resources. Signed-off-by: Leonardo Cecchi <leonardo.cecchi@enterprisedb.com>
101 lines
3.0 KiB
Go
101 lines
3.0 KiB
Go
package instance
|
|
|
|
import (
|
|
"context"
|
|
"path"
|
|
|
|
cnpgv1 "github.com/cloudnative-pg/cloudnative-pg/api/v1"
|
|
"github.com/spf13/viper"
|
|
corev1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
|
|
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/log"
|
|
"sigs.k8s.io/controller-runtime/pkg/scheme"
|
|
|
|
barmancloudv1 "github.com/cloudnative-pg/plugin-barman-cloud/api/v1"
|
|
extendedclient "github.com/cloudnative-pg/plugin-barman-cloud/internal/cnpgi/instance/internal/client"
|
|
)
|
|
|
|
// Start starts the sidecar informers and CNPG-i server
|
|
func Start(ctx context.Context) error {
|
|
scheme := generateScheme(ctx)
|
|
|
|
setupLog := log.FromContext(ctx)
|
|
setupLog.Info("Starting barman cloud instance plugin")
|
|
podName := viper.GetString("pod-name")
|
|
|
|
controllerOptions := ctrl.Options{
|
|
Scheme: scheme,
|
|
Client: client.Options{
|
|
Cache: &client.CacheOptions{
|
|
DisableFor: []client.Object{
|
|
&corev1.Secret{},
|
|
&barmancloudv1.ObjectStore{},
|
|
&cnpgv1.Cluster{},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), controllerOptions)
|
|
if err != nil {
|
|
setupLog.Error(err, "unable to start manager")
|
|
return err
|
|
}
|
|
|
|
if err := mgr.Add(&CNPGI{
|
|
Client: extendedclient.NewExtendedClient(mgr.GetClient()),
|
|
InstanceName: podName,
|
|
// TODO: improve
|
|
PGDataPath: viper.GetString("pgdata"),
|
|
PGWALPath: path.Join(viper.GetString("pgdata"), "pg_wal"),
|
|
SpoolDirectory: viper.GetString("spool-directory"),
|
|
PluginPath: viper.GetString("plugin-path"),
|
|
}); err != nil {
|
|
setupLog.Error(err, "unable to create CNPGI runnable")
|
|
return err
|
|
}
|
|
|
|
if err := mgr.Start(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// generateScheme creates a runtime.Scheme object with all the
|
|
// definition needed to support the sidecar. This allows
|
|
// the plugin to be used in every CNPG-based operator.
|
|
func generateScheme(ctx context.Context) *runtime.Scheme {
|
|
result := runtime.NewScheme()
|
|
|
|
utilruntime.Must(barmancloudv1.AddToScheme(result))
|
|
utilruntime.Must(clientgoscheme.AddToScheme(result))
|
|
|
|
cnpgGroup := viper.GetString("custom-cnpg-group")
|
|
cnpgVersion := viper.GetString("custom-cnpg-version")
|
|
if len(cnpgGroup) == 0 {
|
|
cnpgGroup = cnpgv1.SchemeGroupVersion.Group
|
|
}
|
|
if len(cnpgVersion) == 0 {
|
|
cnpgVersion = cnpgv1.SchemeGroupVersion.Version
|
|
}
|
|
|
|
// Proceed with custom registration of the CNPG scheme
|
|
schemeGroupVersion := schema.GroupVersion{Group: cnpgGroup, Version: cnpgVersion}
|
|
schemeBuilder := &scheme.Builder{GroupVersion: schemeGroupVersion}
|
|
schemeBuilder.Register(&cnpgv1.Cluster{}, &cnpgv1.ClusterList{})
|
|
schemeBuilder.Register(&cnpgv1.Backup{}, &cnpgv1.BackupList{})
|
|
schemeBuilder.Register(&cnpgv1.ScheduledBackup{}, &cnpgv1.ScheduledBackupList{})
|
|
utilruntime.Must(schemeBuilder.AddToScheme(result))
|
|
|
|
schemeLog := log.FromContext(ctx)
|
|
schemeLog.Info("CNPG types registration", "schemeGroupVersion", schemeGroupVersion)
|
|
|
|
return result
|
|
}
|