Add job tracking

Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>

database/sql/pools.go

@@ -21,8 +21,8 @@ import (
 	runnerErrors "github.com/cloudbase/garm/errors"
 	"github.com/cloudbase/garm/params"
 
+	"github.com/google/uuid"
 	"github.com/pkg/errors"
-	uuid "github.com/satori/go.uuid"
 	"gorm.io/gorm"
 )
 
@@ -73,7 +73,7 @@ func (s *sqlDatabase) getEntityPool(ctx context.Context, entityType params.PoolT
 		return Pool{}, errors.Wrap(runnerErrors.ErrBadRequest, "missing entity id")
 	}
 
-	u, err := uuid.FromString(poolID)
+	u, err := uuid.Parse(poolID)
 	if err != nil {
 		return Pool{}, errors.Wrap(runnerErrors.ErrBadRequest, "parsing id")
 	}
@@ -112,3 +112,109 @@ func (s *sqlDatabase) getEntityPool(ctx context.Context, entityType params.PoolT
 
 	return pool, nil
 }
+
+func (s *sqlDatabase) listEntityPools(ctx context.Context, entityType params.PoolType, entityID string, preload ...string) ([]Pool, error) {
+	if _, err := uuid.Parse(entityID); err != nil {
+		return nil, errors.Wrap(runnerErrors.ErrBadRequest, "parsing id")
+	}
+
+	q := s.conn
+	if len(preload) > 0 {
+		for _, item := range preload {
+			q = q.Preload(item)
+		}
+	}
+
+	var fieldName string
+	switch entityType {
+	case params.RepositoryPool:
+		fieldName = "repo_id"
+	case params.OrganizationPool:
+		fieldName = "org_id"
+	case params.EnterprisePool:
+		fieldName = "enterprise_id"
+	default:
+		return nil, fmt.Errorf("invalid entityType: %v", entityType)
+	}
+
+	var pools []Pool
+	condition := fmt.Sprintf("%s = ?", fieldName)
+	err := q.Model(&Pool{}).
+		Where(condition, entityID).
+		Omit("extra_specs").
+		Find(&pools).Error
+
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return []Pool{}, nil
+		}
+		return nil, errors.Wrap(err, "fetching pool")
+	}
+
+	return pools, nil
+}
+
+func (s *sqlDatabase) findPoolByTags(id string, poolType params.PoolType, tags []string) ([]params.Pool, error) {
+	if len(tags) == 0 {
+		return nil, runnerErrors.NewBadRequestError("missing tags")
+	}
+	u, err := uuid.Parse(id)
+	if err != nil {
+		return nil, errors.Wrap(runnerErrors.ErrBadRequest, "parsing id")
+	}
+
+	var fieldName string
+	switch poolType {
+	case params.RepositoryPool:
+		fieldName = "repo_id"
+	case params.OrganizationPool:
+		fieldName = "org_id"
+	case params.EnterprisePool:
+		fieldName = "enterprise_id"
+	default:
+		return nil, fmt.Errorf("invalid poolType: %v", poolType)
+	}
+
+	var pools []Pool
+	where := fmt.Sprintf("tags.name in ? and %s = ? and enabled = true", fieldName)
+	q := s.conn.Joins("JOIN pool_tags on pool_tags.pool_id=pools.id").
+		Joins("JOIN tags on tags.id=pool_tags.tag_id").
+		Group("pools.id").
+		Preload("Tags").
+		Having("count(1) = ?", len(tags)).
+		Where(where, tags, u).Find(&pools)
+
+	if q.Error != nil {
+		if errors.Is(q.Error, gorm.ErrRecordNotFound) {
+			return nil, runnerErrors.ErrNotFound
+		}
+		return nil, errors.Wrap(q.Error, "fetching pool")
+	}
+
+	if len(pools) == 0 {
+		return nil, runnerErrors.ErrNotFound
+	}
+
+	ret := make([]params.Pool, len(pools))
+	for idx, val := range pools {
+		ret[idx] = s.sqlToCommonPool(val)
+	}
+
+	return ret, nil
+}
+
+func (s *sqlDatabase) FindPoolsMatchingAllTags(ctx context.Context, entityType params.PoolType, entityID string, tags []string) ([]params.Pool, error) {
+	if len(tags) == 0 {
+		return nil, runnerErrors.NewBadRequestError("missing tags")
+	}
+
+	pools, err := s.findPoolByTags(entityID, entityType, tags)
+	if err != nil {
+		if errors.Is(err, runnerErrors.ErrNotFound) {
+			return []params.Pool{}, nil
+		}
+		return nil, errors.Wrap(err, "fetching pools")
+	}
+
+	return pools, nil
+}