Implement GraphQLEntityProjectionMaker

elide-core/src/main/java/com/yahoo/elide/core/EntityDictionary.java

@@ -246,8 +246,8 @@ public ParseTree getPermissionsForClass(Class<?> resourceClass, Class<? extends
      *         or {@code null} if the permission is not specified on that field
      */
     public ParseTree getPermissionsForField(Class<?> resourceClass,
-            String field,
-            Class<? extends Annotation> annotationClass) {
+                                            String field,
+                                            Class<? extends Annotation> annotationClass) {
         EntityBinding binding = getEntityBinding(resourceClass);
         return binding.entityPermissions.getFieldChecksForPermission(field, annotationClass);
     }

elide-core/src/main/java/com/yahoo/elide/core/PersistentResource.java

@@ -41,12 +41,10 @@
 import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.google.common.base.Preconditions;
 import com.google.common.collect.Sets;
-
 import org.apache.commons.collections4.CollectionUtils;
 
 import lombok.NonNull;
 import lombok.extern.slf4j.Slf4j;
-
 import java.io.Serializable;
 import java.lang.annotation.Annotation;
 import java.util.ArrayList;
@@ -129,7 +127,7 @@ public static <T> PersistentResource<T> createObject(
             RequestScope requestScope,
             Optional<String> uuid) {
 
-        //instead of calling transcation.createObject, create the new object here.
+        //instead of calling transaction.createObject, create the new object here.
         T obj = requestScope.getTransaction().createNewObject(entityClass);
 
         String id = uuid.orElse(null);
@@ -169,8 +167,12 @@ public static <T> PersistentResource<T> createObject(
      * @param id the id
      * @param scope the request scope
      */
-    public PersistentResource(@NonNull T obj, PersistentResource parent,
-                              String id, @NonNull RequestScope scope) {
+    public PersistentResource(
+            @NonNull T obj,
+            PersistentResource parent,
+            String id,
+            @NonNull RequestScope scope
+    ) {
         this.obj = obj;
         this.uuid = Optional.ofNullable(id);
         this.lineage = parent != null ? new ResourceLineage(parent.lineage, parent) : new ResourceLineage();
@@ -212,8 +214,10 @@ public boolean matchesId(String checkId) {
      */
     @SuppressWarnings("resource")
     @NonNull public static <T> PersistentResource<T> loadRecord(
-            EntityProjection projection, String id, RequestScope requestScope)
-            throws InvalidObjectIdentifierException {
+            EntityProjection projection,
+            String id,
+            RequestScope requestScope
+    ) throws InvalidObjectIdentifierException {
         Preconditions.checkNotNull(projection);
         Preconditions.checkNotNull(id);
         Preconditions.checkNotNull(requestScope);
@@ -241,7 +245,12 @@ public boolean matchesId(String checkId) {
             }
         }
 
-        PersistentResource<T> resource = new PersistentResource(obj, null, requestScope.getUUIDFor(obj), requestScope);
+        PersistentResource<T> resource = new PersistentResource<>(
+                (T) obj,
+                null,
+                requestScope.getUUIDFor(obj),
+                requestScope);
+
         // No need to have read access for a newly created object
         if (!requestScope.getNewResources().contains(resource)) {
             resource.checkFieldAwarePermissions(ReadPermission.class);
@@ -1261,11 +1270,10 @@ protected Map<String, Relationship> getRelationships() {
      * @return Relationship mapping
      */
     private Map<String, Relationship> getRelationships(EntityProjection projection) {
-        return getRelationshipsWithRelationshipFunction((relationName) -> {
-            return getRelationCheckedFiltered(projection.getRelationship(relationName)
-                    .orElseThrow(IllegalStateException::new)
-            );
-        });
+        return getRelationshipsWithRelationshipFunction(
+                (relationName) -> getRelationCheckedFiltered(projection.getRelationship(relationName)
+                        .orElseThrow(IllegalStateException::new)
+        ));
     }
 
     /**

elide-core/src/main/java/com/yahoo/elide/core/pagination/Pagination.java

@@ -85,6 +85,21 @@ private Pagination(Map<PaginationKey, Integer> pageData, int defaultMaxPageSize,
         this.defaultPageSize = defaultPageSize;
     }
 
+    /**
+     * Set limit.
+     *
+     * @param perPage page size.
+     */
+    public void setLimit(Integer perPage) {
+        this.limit = perPage;
+        pageData.put(PaginationKey.limit, perPage);
+    }
+
+    public void setOffset(Integer offset) {
+        this.offset = offset;
+        pageData.put(PaginationKey.offset, offset);
+    }
+
     /**
      * TODO - Refactor Pagination.
      * IMPORTANT - This method should only be used for testing until Pagination is refactored.  The

elide-core/src/main/java/com/yahoo/elide/jsonapi/EntityProjectionMaker.java

@@ -338,7 +338,6 @@ private Map<String, EntityProjection> getSparseRelationships(Class<?> entityClas
                 .collect(Collectors.toMap(
                         Function.identity(),
                         (relationshipName) -> {
-
                             FilterExpression filter = scope.getExpressionForRelation(entityClass, relationshipName)
                                     .orElse(null);
 
@@ -367,21 +366,18 @@ private Set<Path> getIncludePaths(Class<?> entityClass) {
                     .flatMap(param -> Arrays.stream(param.split(",")))
                     .map(pathString -> new Path(entityClass, dictionary, pathString))
                     .collect(Collectors.toSet());
-
         }
 
         return new HashSet<>();
     }
 
     private Set<Relationship> toRelationshipSet(Map<String, EntityProjection> relationships) {
         return relationships.entrySet().stream()
-                .map(entry -> {
-                    return Relationship.builder()
-                            .name(entry.getKey())
-                            .alias(entry.getKey())
-                            .projection(entry.getValue())
-                            .build();
-                })
+                .map(entry -> Relationship.builder()
+                        .name(entry.getKey())
+                        .alias(entry.getKey())
+                        .projection(entry.getValue())
+                        .build())
                 .collect(Collectors.toSet());
     }
 }

elide-core/src/main/java/com/yahoo/elide/parsers/state/RecordState.java

@@ -53,7 +53,6 @@ public void handle(StateContext state, SubCollectionReadCollectionContext ctx) {
         } else {
             entityName = dictionary.getJsonAliasFor(paramType);
             entityClass = dictionary.getEntityClass(entityName);
-
         }
         if (entityClass == null) {
             throw new IllegalArgumentException("Unknown type " + entityName);

elide-core/src/main/java/com/yahoo/elide/request/EntityProjection.java

@@ -12,6 +12,7 @@
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
+import lombok.Getter;
 import lombok.NonNull;
 
 import java.util.LinkedHashSet;
@@ -122,9 +123,22 @@ public EntityProjection merge(EntityProjection toMerge) {
      * Customizes the lombok builder to our needs.
      */
     public static class EntityProjectionBuilder {
+        @Getter
+        private Class<?> type;
+
         private Set<Relationship> relationships = new LinkedHashSet<>();
+
         private Set<Attribute> attributes = new LinkedHashSet<>();
 
+        @Getter
+        private FilterExpression filterExpression;
+
+        @Getter
+        private Sorting sorting;
+
+        @Getter
+        private Pagination pagination;
+
         public EntityProjectionBuilder relationships(Set<Relationship> relationships) {
             this.relationships = relationships;
             return this;
@@ -166,5 +180,18 @@ public EntityProjectionBuilder attribute(Attribute attribute) {
             this.attributes.add(attribute);
             return this;
         }
+
+        /**
+         * Get an attribute by alias.
+         *
+         * @param attributeAlias alias to refer to an attribute field
+         * @return found attribute or null
+         */
+        public Attribute getAttributeByAlias(String attributeAlias) {
+            return attributes.stream()
+                    .filter(attribute -> attribute.getAlias().equals(attributeAlias))
+                    .findAny()
+                    .orElse(null);
+        }
     }
 }