types: remove EntityLoader interface

authorize.go

@@ -18,7 +18,7 @@ const (
 
 // IsAuthorized uses the combination of the PolicySet and Entities to determine
 // if the given Request to determine Decision and Diagnostic.
-func (p PolicySet) IsAuthorized(entities EntityLoader, req Request) (Decision, Diagnostic) {
+func (p PolicySet) IsAuthorized(entities types.EntityMap, req Request) (Decision, Diagnostic) {
 	env := eval.Env{
 		Entities:  entities,
 		Principal: req.Principal,

internal/eval/evalers.go

@@ -22,7 +22,7 @@ func zeroValue() types.Value {
 }
 
 type Env struct {
-	Entities                    types.EntityLoader
+	Entities                    types.EntityMap
 	Principal, Action, Resource types.Value
 	Context                     types.Value
 }

types.go

@@ -31,7 +31,6 @@ type String = types.String
 
 // Other Cedar types
 
-type EntityLoader = types.EntityLoader
 type Entities = types.EntityMap
 type Entity = types.Entity
 type EntityType = types.EntityType

types/entity_map.go

@@ -8,11 +8,6 @@ import (
 	"golang.org/x/exp/maps"
 )
 
-// EntityLoader defines the interface for loading entities from an entity store.
-type EntityLoader interface {
-	Load(EntityUID) (Entity, bool)
-}
-
 // An EntityMap is a collection of all the entities that are needed to evaluate
 // authorization requests.  The key is an EntityUID which uniquely identifies
 // the Entity (it must be the same as the UID within the Entity itself.)

x/exp/batch/batch.go

@@ -102,7 +102,7 @@ var errInvalidPart = fmt.Errorf("invalid part")
 //   - It will error in case there are unused variables.
 //
 // The result passed to the callback must be used / cloned immediately and not modified.
-func Authorize(ctx context.Context, ps *cedar.PolicySet, entities types.EntityLoader, request Request, cb Callback) error {
+func Authorize(ctx context.Context, ps *cedar.PolicySet, entities types.EntityMap, request Request, cb Callback) error {
 	be := &batchEvaler{}
 	var found mapset.MapSet[types.String]
 	findVariables(&found, request.Principal)