|
1 | | -// Package main is the implementation of the CNPG-i PostgreSQL sidecar |
| 1 | +// Package main is the entrypoint of operator plugin |
2 | 2 | package main |
3 | 3 |
|
4 | 4 | import ( |
5 | | - "errors" |
| 5 | + "fmt" |
6 | 6 | "os" |
7 | 7 |
|
8 | | - cnpgv1 "github.com/cloudnative-pg/cloudnative-pg/api/v1" |
9 | | - "k8s.io/apimachinery/pkg/fields" |
10 | | - "k8s.io/apimachinery/pkg/runtime" |
11 | | - utilruntime "k8s.io/apimachinery/pkg/util/runtime" |
12 | | - clientgoscheme "k8s.io/client-go/kubernetes/scheme" |
13 | | - ctrl "sigs.k8s.io/controller-runtime" |
14 | | - "sigs.k8s.io/controller-runtime/pkg/cache" |
15 | | - "sigs.k8s.io/controller-runtime/pkg/client" |
| 8 | + "github.com/cloudnative-pg/machinery/pkg/log" |
| 9 | + "github.com/spf13/cobra" |
| 10 | + "github.com/spf13/viper" |
16 | 11 |
|
17 | | - barmancloudv1 "github.com/cloudnative-pg/plugin-barman-cloud/api/v1" |
18 | 12 | "github.com/cloudnative-pg/plugin-barman-cloud/internal/cnpgi/instance" |
19 | 13 | ) |
20 | 14 |
|
21 | | -var ( |
22 | | - scheme = runtime.NewScheme() |
23 | | - setupLog = ctrl.Log.WithName("setup") |
24 | | -) |
25 | | - |
26 | | -func init() { |
27 | | - utilruntime.Must(clientgoscheme.AddToScheme(scheme)) |
28 | | - |
29 | | - utilruntime.Must(barmancloudv1.AddToScheme(scheme)) |
30 | | - // +kubebuilder:scaffold:scheme |
31 | | -} |
32 | | - |
33 | 15 | func main() { |
34 | | - setupLog.Info("Starting barman cloud instance plugin") |
35 | | - namespace := mustGetEnv("NAMESPACE") |
36 | | - boName := mustGetEnv("BARMAN_OBJECT_NAME") |
37 | | - clusterName := mustGetEnv("CLUSTER_NAME") |
38 | | - instanceName := mustGetEnv("INSTANCE_NAME") |
39 | | - |
40 | | - mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{ |
41 | | - Scheme: scheme, |
42 | | - Cache: cache.Options{ |
43 | | - ByObject: map[client.Object]cache.ByObject{ |
44 | | - &barmancloudv1.ObjectStore{}: { |
45 | | - Field: fields.OneTermEqualSelector("metadata.name", boName), |
46 | | - Namespaces: map[string]cache.Config{ |
47 | | - namespace: {}, |
48 | | - }, |
49 | | - }, |
50 | | - &cnpgv1.Cluster{}: { |
51 | | - Field: fields.OneTermEqualSelector("metadata.name", clusterName), |
52 | | - Namespaces: map[string]cache.Config{ |
53 | | - namespace: {}, |
54 | | - }, |
55 | | - }, |
56 | | - }, |
| 16 | + cobra.EnableTraverseRunHooks = true |
| 17 | + |
| 18 | + logFlags := &log.Flags{} |
| 19 | + rootCmd := &cobra.Command{ |
| 20 | + Use: "instance", |
| 21 | + Short: "Starts the Barman Cloud CNPG-i sidecar plugin", |
| 22 | + PersistentPreRunE: func(_ *cobra.Command, _ []string) error { |
| 23 | + logFlags.ConfigureLogging() |
| 24 | + return nil |
57 | 25 | }, |
58 | | - }) |
59 | | - if err != nil { |
60 | | - setupLog.Error(err, "unable to start manager") |
61 | | - os.Exit(1) |
62 | | - } |
63 | | - |
64 | | - if err := mgr.Add(&instance.CNPGI{ |
65 | | - Client: mgr.GetClient(), |
66 | | - ClusterObjectKey: client.ObjectKey{ |
67 | | - Namespace: namespace, |
68 | | - Name: clusterName, |
69 | | - }, |
70 | | - BarmanObjectKey: client.ObjectKey{ |
71 | | - Namespace: namespace, |
72 | | - Name: boName, |
| 26 | + RunE: func(cmd *cobra.Command, _ []string) error { |
| 27 | + requiredSettings := []string{ |
| 28 | + "namespace", |
| 29 | + "barman-object-name", |
| 30 | + "cluster-name", |
| 31 | + "pod-name", |
| 32 | + "spool-directory", |
| 33 | + } |
| 34 | + |
| 35 | + for _, k := range requiredSettings { |
| 36 | + if len(viper.GetString(k)) == 0 { |
| 37 | + return fmt.Errorf("missing required %s setting", k) |
| 38 | + } |
| 39 | + } |
| 40 | + |
| 41 | + return instance.Start(cmd.Context()) |
73 | 42 | }, |
74 | | - InstanceName: instanceName, |
75 | | - // TODO: improve |
76 | | - PGDataPath: mustGetEnv("PGDATA"), |
77 | | - PGWALPath: mustGetEnv("PGWAL"), |
78 | | - SpoolDirectory: mustGetEnv("SPOOL_DIRECTORY"), |
79 | | - ServerCertPath: mustGetEnv("SERVER_CERT"), |
80 | | - ServerKeyPath: mustGetEnv("SERVER_KEY"), |
81 | | - ClientCertPath: mustGetEnv("CLIENT_CERT"), |
82 | | - ServerAddress: mustGetEnv("SERVER_ADDRESS"), |
83 | | - PluginPath: mustGetEnv("PLUGIN_PATH"), |
84 | | - }); err != nil { |
85 | | - setupLog.Error(err, "unable to create CNPGI runnable") |
86 | | - os.Exit(1) |
87 | 43 | } |
88 | 44 |
|
89 | | - if err := mgr.Start(ctrl.SetupSignalHandler()); err != nil { |
90 | | - setupLog.Error(err, "problem running manager") |
91 | | - os.Exit(1) |
92 | | - } |
93 | | -} |
| 45 | + logFlags.AddFlags(rootCmd.PersistentFlags()) |
| 46 | + |
| 47 | + _ = viper.BindEnv("namespace", "NAMESPACE") |
| 48 | + _ = viper.BindEnv("barman-object-name", "BARMAN_OBJECT_NAME") |
| 49 | + _ = viper.BindEnv("cluster-name", "CLUSTER_NAME") |
| 50 | + _ = viper.BindEnv("pod-name", "POD_NAME") |
| 51 | + _ = viper.BindEnv("pgdata", "PGDATA") |
| 52 | + _ = viper.BindEnv("spool-directory", "SPOOL_DIRECTORY") |
94 | 53 |
|
95 | | -func mustGetEnv(envName string) string { |
96 | | - value := os.Getenv(envName) |
97 | | - if value == "" { |
98 | | - setupLog.Error( |
99 | | - errors.New("missing required env variable"), |
100 | | - "while fetching env variables", |
101 | | - "name", |
102 | | - envName, |
103 | | - ) |
| 54 | + if err := rootCmd.Execute(); err != nil { |
| 55 | + fmt.Println(err) |
104 | 56 | os.Exit(1) |
105 | 57 | } |
106 | | - return value |
107 | 58 | } |
0 commit comments