Method SetScaleSetDesiredRunnerCount

database/sql/scalesets.go:496–531

(_ context.Context, scaleSetID uint, desiredRunnerCount int)

}

func (s *sqlDatabase) SetScaleSetDesiredRunnerCount(_ context.Context, scaleSetID uint, desiredRunnerCount int) (err error) {
	var scaleSet params.ScaleSet
	var rowsAffected int64
	defer func() {
		if err == nil && scaleSet.ID != 0 && rowsAffected > 0 {
			s.sendNotify(common.ScaleSetEntityType, common.UpdateOperation, scaleSet)
		}
	}()
	if err := s.conn.Transaction(func(tx *gorm.DB) error {
		dbSet, err := s.getScaleSetByID(tx, scaleSetID, "Instances", "Enterprise", "Organization", "Repository")
		if err != nil {
			return fmt.Errorf("error fetching scale set: %w", err)
		}
		result := tx.Model(&dbSet).Updates(map[string]interface{}{
			"desired_runner_count": desiredRunnerCount,
		})
		if result.Error != nil {
			return fmt.Errorf("error saving database entry: %w", result.Error)
		}
		rowsAffected = result.RowsAffected

		// Reload to get the updated value
		dbSet, err = s.getScaleSetByID(tx, scaleSetID, "Instances", "Enterprise", "Organization", "Repository")
		if err != nil {
			return fmt.Errorf("error reloading scale set: %w", err)
		}
		scaleSet, err = s.sqlToCommonScaleSet(dbSet)
		if err != nil {
			return fmt.Errorf("error converting scale set: %w", err)
		}
		return nil
	}); err != nil {
		return fmt.Errorf("error setting desired runner count: %w", err)
	}
	return nil
}