plugin-barman-cloud/internal/cmd/restore/main.go
Armando Ruocco a19f89b57f
fix(restore): bind custom CNPG group and version env vars
The restore command was missing the CUSTOM_CNPG_GROUP and
CUSTOM_CNPG_VERSION environment variable bindings that other
commands (e.g. instance) already had.

Signed-off-by: Armando Ruocco <armando.ruocco@enterprisedb.com>

refactor(restore): use shared scheme generation for custom CNPG groups

Extract generateScheme() from the instance package into
common.GenerateScheme() and use it from both instance and
restore managers. The restore manager previously used a
hardcoded init() that ignored CUSTOM_CNPG_GROUP and
CUSTOM_CNPG_VERSION, causing restore jobs to fail when
running under a non-standard CNPG API group.

Signed-off-by: Marco Nenciarini <marco.nenciarini@enterprisedb.com>
2026-04-13 11:31:13 +02:00

65 lines
1.7 KiB
Go

/*
Copyright © contributors to CloudNativePG, established as
CloudNativePG a Series of LF Projects, LLC.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
SPDX-License-Identifier: Apache-2.0
*/
// Package restore is the entrypoint of restore capabilities
package restore
import (
"fmt"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"github.com/cloudnative-pg/plugin-barman-cloud/internal/cnpgi/restore"
)
// NewCmd creates the "restore" subcommand
func NewCmd() *cobra.Command {
cobra.EnableTraverseRunHooks = true
cmd := &cobra.Command{
Use: "restore",
Short: "Starts the Barman Cloud CNPG-I sidecar plugin",
RunE: func(cmd *cobra.Command, _ []string) error {
requiredSettings := []string{
"namespace",
"pod-name",
"spool-directory",
}
for _, k := range requiredSettings {
if len(viper.GetString(k)) == 0 {
return fmt.Errorf("missing required %s setting", k)
}
}
return restore.Start(cmd.Context())
},
}
_ = viper.BindEnv("namespace", "NAMESPACE")
_ = viper.BindEnv("pod-name", "POD_NAME")
_ = viper.BindEnv("pgdata", "PGDATA")
_ = viper.BindEnv("spool-directory", "SPOOL_DIRECTORY")
_ = viper.BindEnv("custom-cnpg-group", "CUSTOM_CNPG_GROUP")
_ = viper.BindEnv("custom-cnpg-version", "CUSTOM_CNPG_VERSION")
return cmd
}