cluster_backup_strategy.go

package service

import (
	"errors"

	"github.com/KubeOperator/KubeOperator/pkg/db"
	"github.com/KubeOperator/KubeOperator/pkg/dto"
	"github.com/KubeOperator/KubeOperator/pkg/model"
	"github.com/KubeOperator/KubeOperator/pkg/repository"
	"github.com/jinzhu/gorm"
)

var (
	UpdateError = "BACKUP_FILES_NOT_NULL"
)

type CLusterBackupStrategyService interface {
	Get(clusterName string) (*dto.ClusterBackupStrategy, error)
	Save(creation dto.ClusterBackupStrategyRequest) (*dto.ClusterBackupStrategy, error)
}

type cLusterBackupStrategyService struct {
	clusterBackupStrategyRepo   repository.ClusterBackupStrategyRepository
	clusterRepo                 repository.ClusterRepository
	backupAccountService        BackupAccountService
	clusterBackupFileRepository repository.ClusterBackupFileRepository
}

func NewCLusterBackupStrategyService() CLusterBackupStrategyService {
	return &cLusterBackupStrategyService{
		clusterBackupStrategyRepo:   repository.NewClusterBackupStrategyRepository(),
		clusterRepo:                 repository.NewClusterRepository(),
		backupAccountService:        NewBackupAccountService(),
		clusterBackupFileRepository: repository.NewClusterBackupFileRepository(),
	}
}

func (c cLusterBackupStrategyService) Get(clusterName string) (*dto.ClusterBackupStrategy, error) {
	var clusterBackupStrategyDTO dto.ClusterBackupStrategy
	mo, err := c.clusterBackupStrategyRepo.Get(clusterName)
	if err != nil {
		return nil, err
	}

	clusterBackupStrategyDTO = dto.ClusterBackupStrategy{
		ClusterBackupStrategy: *mo,
		BackupAccountName:     mo.BackupAccount.Name,
		ClusterName:           clusterName,
	}
	return &clusterBackupStrategyDTO, nil
}

func (c cLusterBackupStrategyService) Save(creation dto.ClusterBackupStrategyRequest) (*dto.ClusterBackupStrategy, error) {
	backupAccount, err := c.backupAccountService.Get(creation.BackupAccountName)
	if err != nil {
		return nil, err
	}
	cluster, err := c.clusterRepo.Get(creation.ClusterName)
	if err != nil {
		return nil, err
	}
	var id string
	old, err := c.Get(creation.ClusterName)
	if err != nil {
		return nil, err
	} else {
		id = old.ID
		if old.BackupAccountID != backupAccount.ID {
			var backupFiles []model.ClusterBackupFile
			err := db.DB.Where("cluster_backup_strategy_id = ? AND cluster_id = ?", id, cluster.ID).Find(&backupFiles).Error
			if err != nil && !gorm.IsRecordNotFoundError(err) {
				return nil, err
			}
			if len(backupFiles) > 0 {
				return nil, errors.New(UpdateError)
			}
		}
	}
	clusterBackupStrategy := model.ClusterBackupStrategy{
		ID:              id,
		ClusterID:       cluster.ID,
		Cron:            creation.Cron,
		Status:          creation.Status,
		BackupAccountID: backupAccount.ID,
		SaveNum:         creation.SaveNum,
	}

	err = c.clusterBackupStrategyRepo.Save(&clusterBackupStrategy)
	if err != nil {
		return nil, err
	}
	return &dto.ClusterBackupStrategy{ClusterBackupStrategy: clusterBackupStrategy}, err
}