PipelineConfigRepository.go

/*
 * Copyright (c) 2020 Devtron Labs
 *
 * 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.
 *
 */

package chartConfig

import (
	"github.com/devtron-labs/devtron/internal/sql/models"
	"github.com/devtron-labs/devtron/internal/sql/repository/pipelineConfig"
	"github.com/go-pg/pg"
	"github.com/juju/errors"
)

type PipelineStrategy struct {
	tableName  struct{}                          `sql:"pipeline_strategy" pg:",discard_unknown_columns"`
	Id         int                               `sql:"id,pk"`
	PipelineId int                               `sql:"pipeline_id"`
	Strategy   pipelineConfig.DeploymentTemplate `sql:"strategy,notnull"`
	Config     string                            `sql:"config"`
	Default    bool                              `sql:"default,notnull"`
	Deleted    bool                              `sql:"deleted,notnull"`
	models.AuditLog
}

type PipelineConfigRepository interface {
	Save(pipelineStrategy *PipelineStrategy, tx *pg.Tx) error
	Update(pipelineStrategy *PipelineStrategy, tx *pg.Tx) error
	FindById(id int) (chart *PipelineStrategy, err error)
	FindByStrategy(strategy pipelineConfig.DeploymentTemplate) (pipelineStrategy *PipelineStrategy, err error)
	FindByStrategyAndPipelineId(strategy pipelineConfig.DeploymentTemplate, pipelineId int) (pipelineStrategy *PipelineStrategy, err error)
	GetAllStrategyByPipelineId(pipelineId int) ([]*PipelineStrategy, error)
	GetDefaultStrategyByPipelineId(pipelineId int) (pipelineStrategy *PipelineStrategy, err error)
	Delete(pipelineStrategy *PipelineStrategy, tx *pg.Tx) error
}

type PipelineConfigRepositoryImpl struct {
	dbConnection *pg.DB
}

func NewPipelineConfigRepository(dbConnection *pg.DB) *PipelineConfigRepositoryImpl {
	return &PipelineConfigRepositoryImpl{dbConnection: dbConnection}
}

func (impl PipelineConfigRepositoryImpl) Save(pipelineStrategy *PipelineStrategy, tx *pg.Tx) error {
	return tx.Insert(pipelineStrategy)
}

func (impl PipelineConfigRepositoryImpl) Update(pipelineStrategy *PipelineStrategy, tx *pg.Tx) error {
	_, err := impl.dbConnection.Model(pipelineStrategy).WherePK().UpdateNotNull()
	return err
}

func (impl PipelineConfigRepositoryImpl) FindById(id int) (pipelineStrategy *PipelineStrategy, err error) {
	pipelineStrategy = &PipelineStrategy{}
	err = impl.dbConnection.Model(pipelineStrategy).
		Where("id = ?", id).Select()
	return pipelineStrategy, err
}

func (impl PipelineConfigRepositoryImpl) FindByStrategy(strategy pipelineConfig.DeploymentTemplate) (pipelineStrategy *PipelineStrategy, err error) {
	pipelineStrategy = &PipelineStrategy{}
	err = impl.dbConnection.Model(pipelineStrategy).
		Where("strategy = ?", strategy).Select()
	return pipelineStrategy, err
}

func (impl PipelineConfigRepositoryImpl) FindByStrategyAndPipelineId(strategy pipelineConfig.DeploymentTemplate, pipelineId int) (pipelineStrategy *PipelineStrategy, err error) {
	pipelineStrategy = &PipelineStrategy{}
	err = impl.dbConnection.Model(pipelineStrategy).
		Where("strategy = ?", strategy).
		Where("pipeline_id = ?", pipelineId).Select()
	return pipelineStrategy, err
}

//it will return for multiple pipeline config for pipeline, per pipeline single pipeline config(blue green, canary)
func (impl PipelineConfigRepositoryImpl) GetAllStrategyByPipelineId(pipelineId int) ([]*PipelineStrategy, error) {
	var pipelineStrategies []*PipelineStrategy
	err := impl.dbConnection.
		Model(&pipelineStrategies).
		Where("pipeline_id = ?", pipelineId).
		Where("deleted = ?", false).
		Select()
	if pg.ErrNoRows == err {
		return nil, errors.NotFoundf(err.Error())
	}
	return pipelineStrategies, err
}

//it will return single latest pipeline config for requested pipeline
func (impl PipelineConfigRepositoryImpl) GetDefaultStrategyByPipelineId(pipelineId int) (pipelineStrategy *PipelineStrategy, err error) {
	pipelineStrategy = &PipelineStrategy{}
	err = impl.dbConnection.
		Model(pipelineStrategy).
		Where("pipeline_strategy.pipeline_id = ?", pipelineId).
		Where("pipeline_strategy.default = ?", true).
		Where("pipeline_strategy.deleted = ?", false).
		Select()
	if pg.ErrNoRows == err {
		return nil, errors.NotFoundf(err.Error())
	}
	return pipelineStrategy, err
}

func (impl PipelineConfigRepositoryImpl) Delete(pipelineStrategy *PipelineStrategy, tx *pg.Tx) error {
	return tx.Delete(pipelineStrategy)
}