105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
package backupsession
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
"context"
|
|
"os"
|
|
"strconv"
|
|
"path/filepath"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/log/zap"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
formolv1alpha1 "github.com/desmo999r/formol/api/v1alpha1"
|
|
"github.com/go-logr/logr"
|
|
)
|
|
|
|
var (
|
|
config *rest.Config
|
|
scheme *runtime.Scheme
|
|
cl client.Client
|
|
logger logr.Logger
|
|
)
|
|
|
|
func init() {
|
|
logger = zap.New(zap.UseDevMode(true))
|
|
log := logger.WithName("InitBackupSession")
|
|
ctrl.SetLogger(logger)
|
|
config, err := rest.InClusterConfig()
|
|
if err != nil {
|
|
config, err = clientcmd.BuildConfigFromFlags("", filepath.Join(os.Getenv("HOME"), ".kube", "config",))
|
|
if err != nil {
|
|
log.Error(err, "unable to get config")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
scheme = runtime.NewScheme()
|
|
_ = formolv1alpha1.AddToScheme(scheme)
|
|
_ = clientgoscheme.AddToScheme(scheme)
|
|
cl, err = client.New(config, client.Options{Scheme: scheme})
|
|
if err != nil {
|
|
log.Error(err, "unable to get client")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func DeleteBackupSession(name string, namespace string) error {
|
|
log := logger.WithName("CreateBackupSession")
|
|
log.V(0).Info("CreateBackupSession called")
|
|
backupSession := &formolv1alpha1.BackupSession{}
|
|
if err := cl.Get(context.TODO(), client.ObjectKey{
|
|
Namespace: namespace,
|
|
Name: name,
|
|
}, backupSession); err != nil {
|
|
log.Error(err, "unable to get backupsession", "backupsession", name)
|
|
return err
|
|
}
|
|
if err := cl.Delete(context.TODO(), backupSession); err != nil {
|
|
log.Error(err, "unable to delete backupsession", "backupsession", name)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func CreateBackupSession(name string, namespace string) {
|
|
log := logger.WithName("CreateBackupSession")
|
|
log.V(0).Info("CreateBackupSession called")
|
|
backupConfList := &formolv1alpha1.BackupConfigurationList{}
|
|
if err := cl.List(context.TODO(), backupConfList, client.InNamespace(namespace)); err != nil {
|
|
log.Error(err, "unable to get backupconf")
|
|
os.Exit(1)
|
|
}
|
|
backupConf := &formolv1alpha1.BackupConfiguration{}
|
|
for _, bc := range backupConfList.Items {
|
|
if bc.Name == name {
|
|
*backupConf = bc
|
|
}
|
|
}
|
|
|
|
backupSession := &formolv1alpha1.BackupSession{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: strings.Join([]string{"backupsession",name,strconv.FormatInt(time.Now().Unix(), 10)}, "-"),
|
|
Namespace: namespace,
|
|
},
|
|
Spec: formolv1alpha1.BackupSessionSpec{
|
|
Ref: formolv1alpha1.Ref{
|
|
Name: name,
|
|
},
|
|
},
|
|
Status: formolv1alpha1.BackupSessionStatus{},
|
|
}
|
|
if err := ctrl.SetControllerReference(backupConf, backupSession, scheme); err != nil {
|
|
log.Error(err, "unable to set controller reference")
|
|
os.Exit(1)
|
|
}
|
|
if err := cl.Create(context.TODO(), backupSession); err != nil {
|
|
log.Error(err, "unable to create backupsession")
|
|
os.Exit(1)
|
|
}
|
|
}
|