pipeline_repository.go

package repository

import (
	"context"
	"github.com/cheerego/uci/app/uci-messaging-server/internal/model/pipeline"
	"github.com/cheerego/uci/pkg/orm"
	"github.com/cheerego/uci/pkg/ptr"
	"gorm.io/gorm"
	"time"
)

type PipelineRepository struct {
	orm.BaseRepository[pipeline.Pipeline]
	db *gorm.DB
}

func NewPipelineRepository(db *gorm.DB) *PipelineRepository {
	return &PipelineRepository{
		db:             db,
		BaseRepository: orm.NewBaseRepository[pipeline.Pipeline](db),
	}
}

func (p *PipelineRepository) IncreaseDispatchTimes(ctx context.Context, id uint32) (int64, error) {
	var m pipeline.Pipeline
	tx := orm.FromContext(ctx, p.db).Model(&m).Where(id).Updates(map[string]interface{}{
		"dispatch_times":     gorm.Expr("dispatch_times + ?", 1),
		"last_dispatched_at": time.Now(),
	})
	return tx.RowsAffected, tx.Error

}

func (p *PipelineRepository) UpdateStatus(ctx context.Context, m *pipeline.Pipeline, opts ...pipeline.StatusOption) (int64, error) {
	var selects = make([]string, 0)
	selects = append(selects, "LastStatusChangedAt", "Status")
	for _, opt := range opts {
		selects = append(selects, opt(m))
	}
	m.LastStatusChangedAt = ptr.Ptr(time.Now())
	tx := orm.FromContext(ctx, p.db).Select(selects).Updates(m)
	return tx.RowsAffected, tx.Error
}

func (p *PipelineRepository) FindByUuid(ctx context.Context, uuid string) (*pipeline.Pipeline, error) {
	var m pipeline.Pipeline
	tx := orm.FromContext(ctx, p.db).Model(&m).Where("uuid = ?", uuid).First(&m)
	if tx.Error != nil {
		return nil, tx.Error
	}
	return &m, nil
}

func (p *PipelineRepository) UpdateRawlog(ctx context.Context, id uint32, raw string) (int64, error) {
	var m pipeline.Pipeline
	tx := orm.FromContext(ctx, p.db).Model(&m).Where(id).Updates(map[string]interface{}{
		"raw_log": raw,
	})
	return tx.RowsAffected, tx.Error
}

func (p *PipelineRepository) UpdateEnvs(ctx context.Context, pl *pipeline.Pipeline) (int64, error) {
	tx := orm.FromContext(ctx, p.db).Select("Envs").Updates(pl)
	return tx.RowsAffected, tx.Error
}

func (p *PipelineRepository) Update(ctx context.Context, pl *pipeline.Pipeline) (int64, error) {
	tx := orm.FromContext(ctx, p.db).Updates(pl)
	return tx.RowsAffected, tx.Error
}

func (p *PipelineRepository) FindByStatus(ctx context.Context, s pipeline.Status) ([]*pipeline.Pipeline, error) {
	m := make([]*pipeline.Pipeline, 0)
	err := orm.FromContext(ctx, p.db).Where("status = ?", s).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}