Method Save

app/controlplane/pkg/data/referrer.go:58–123

(ctx context.Context, referrers []*biz.Referrer, workflowID uuid.UUID)

type storedReferrerMap map[string]uuid.UUID

func (r *ReferrerRepo) Save(ctx context.Context, referrers []*biz.Referrer, workflowID uuid.UUID) (err error) {
	ctx, span := otelx.Start(ctx, referrerRepoTracer, "ReferrerRepo.Save")
	defer span.End()

	// find the workflow
	wf, err := r.workflowRepo.FindByID(ctx, workflowID)
	if err != nil {
		return fmt.Errorf("failed to find workflow: %w", err)
	} else if wf == nil {
		return biz.NewErrNotFound("workflow")
	}

	storedMap := make(storedReferrerMap)

	for _, ref := range referrers {
		// Check if it exists already, if not create it
		storedID, err := r.data.DB.Referrer.Create().
			SetDigest(ref.Digest).SetKind(ref.Kind).SetDownloadable(ref.Downloadable).
			SetMetadata(ref.Metadata).SetAnnotations(ref.Annotations).
			AddWorkflowIDs(workflowID).
			OnConflictColumns(
				referrer.FieldDigest, referrer.FieldKind,
			).UpdateNewValues().ID(ctx)
		if err != nil {
			return fmt.Errorf("failed to create referrer: %w", err)
		}

		storedRef, err := r.data.DB.Referrer.Query().Select(referrer.FieldID).Where(referrer.ID(storedID)).First(ctx)
		if err != nil {
			return fmt.Errorf("failed to load referrer: %w", err)
		} else if storedRef == nil {
			return fmt.Errorf("failed to load referrer: %w", err)
		}

		// Store it in the map
		storedMap[ref.MapID()] = storedRef.ID
	}

	// 2 - define the relationship between referrers
	for _, parentRef := range referrers {
		// This is the current item stored in DB
		storedReferrer := storedMap[parentRef.MapID()]
		// Iterate on the items it refer to (references)
		var references []uuid.UUID
		for _, ref := range parentRef.References {
			// amd find it in the DB
			storedReference, ok := storedMap[ref.MapID()]
			if !ok {
				return fmt.Errorf("referrer %v not found", ref)
			}

			references = append(references, storedReference)
		}

		if len(references) == 0 {
			continue
		}