upstream-corteza/server/compose/envoy/record_getter.go

package envoy

import (
	"context"

	"github.com/cortezaproject/corteza/server/compose/dalutils"
	"github.com/cortezaproject/corteza/server/compose/types"
	"github.com/cortezaproject/corteza/server/pkg/dal"
	"github.com/cortezaproject/corteza/server/pkg/envoyx"
	"github.com/spf13/cast"
)

type (
	// recordGetter is a utility struct to resolve record references from
	// different parts of the system such as the dep graph and the database
	recordGetter struct {
		// Bits to get data from the dep graph
		relDatasource *RecordDatasource

		// Bits to get data from the database
		dalSvc     dal.FullService
		relModule  types.Module
		baseFilter types.RecordFilter

		// resolved refs, including misses; imports resolve the same handful
		// of refs over and over so this saves a query per value
		cache map[string]uint64
	}
)

func makeRecordGetter(dalSvc dal.FullService, tt envoyx.Traverser, n *envoyx.Node, modRef, dsRef envoyx.Ref) (g *recordGetter) {
	g = &recordGetter{
		dalSvc: dalSvc,
		cache:  make(map[string]uint64),
	}

	// Resolve from dep graph
	auxDs := tt.ParentForRef(n, dsRef)
	if auxDs != nil {
		g.relDatasource = auxDs.Datasource.(*RecordDatasource)
	}

	// Resolve from the database
	mod := g.getRefMod(tt, n, modRef)
	if mod != nil {
		g.baseFilter = types.RecordFilter{
			ModuleID:    mod.ID,
			NamespaceID: mod.NamespaceID,
		}
	}

	return
}

// resolve resolves the provided reference into a record ID; 0 if can't be resolved
func (g *recordGetter) resolve(ctx context.Context, ref any) (out uint64, err error) {
	key := cast.ToString(ref)
	if out, ok := g.cache[key]; ok {
		return out, nil
	}
	defer func() {
		if err == nil {
			g.cache[key] = out
		}
	}()

	// Try to get from datasource
	if g.relDatasource != nil {
		out, err = g.getDS(ref)
		if err != nil {
			return
		}
	}
	if out > 0 {
		return
	}

	// Fallback to the store
	out, err = g.getDB(ctx, ref)
	if err != nil {
		return
	}
	if out > 0 {
		return
	}
	return
}

func (g *recordGetter) getDS(ref any) (out uint64, err error) {
	return g.relDatasource.ResolveRef(ref)
}

func (g *recordGetter) getDB(ctx context.Context, ref any) (out uint64, err error) {
	// @note the old version only resolved IDs so that's what we're doing here also
	// @todo consider expanding this
	id, err := cast.ToUint64E(ref)
	if err != nil {
		return 0, nil
	}

	aux, err := dalutils.ComposeRecordsFind(ctx, g.dalSvc, &g.relModule, id)
	if err != nil {
		return
	}

	out = aux.ID
	return
}

func (g *recordGetter) getRefMod(tt envoyx.Traverser, n *envoyx.Node, ref envoyx.Ref) (refMod *types.Module) {
	aux := tt.ParentForRef(n, ref)
	if aux == nil {
		return nil
	}

	return aux.Resource.(*types.Module)
}