Support optimized select for top-level query

src/Pagination/PaginateDirective.php

@@ -2,17 +2,20 @@
 
 namespace Nuwave\Lighthouse\Pagination;
 
+use GraphQL\Error\Error;
 use GraphQL\Language\AST\FieldDefinitionNode;
 use GraphQL\Language\AST\ObjectTypeDefinitionNode;
 use Illuminate\Contracts\Pagination\Paginator;
 use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
 use Illuminate\Database\Eloquent\Relations\Relation;
 use Illuminate\Database\Query\Builder as QueryBuilder;
+use Illuminate\Support\Arr;
 use Laravel\Scout\Builder as ScoutBuilder;
 use Nuwave\Lighthouse\Execution\ResolveInfo;
 use Nuwave\Lighthouse\Schema\AST\DocumentAST;
 use Nuwave\Lighthouse\Schema\Directives\BaseDirective;
 use Nuwave\Lighthouse\Schema\Values\FieldValue;
+use Nuwave\Lighthouse\Select\SelectHelper;
 use Nuwave\Lighthouse\Support\Contracts\FieldManipulator;
 use Nuwave\Lighthouse\Support\Contracts\FieldResolver;
 use Nuwave\Lighthouse\Support\Contracts\GraphQLContext;
@@ -156,6 +159,45 @@ public function resolveField(FieldValue $fieldValue): FieldValue
                 $this->directiveArgValue('scopes', [])
             );
 
+            if (config('lighthouse.optimized_selects')) {
+                if ($query instanceof EloquentBuilder) {
+                    $fieldSelection = $resolveInfo->getFieldSelection(2);
+
+                    if (($hasData = Arr::has($fieldSelection, 'data')) || Arr::has($fieldSelection, 'edges')) {
+                        $data = $hasData
+                            ? $fieldSelection['data']
+                            : $fieldSelection['edges']['node'];
+
+                        /** @var array<int, string> $fieldSelection */
+                        $fieldSelection = array_keys($data);
+
+                        $model = $query->getModel();
+
+                        $selectColumns = SelectHelper::getSelectColumns(
+                            $this->definitionNode,
+                            $fieldSelection,
+                            get_class($model)
+                        );
+
+                        if (empty($selectColumns)) {
+                            throw new Error('The select column is empty.');
+                        }
+
+                        $query = $query->select($selectColumns);
+
+                        /** @var string|string[] $keyName */
+                        $keyName = $model->getKeyName();
+                        if (is_string($keyName)) {
+                            $keyName = [$keyName];
+                        }
+
+                        foreach ($keyName as $name) {
+                            $query->orderBy($name);
+                        }
+                    }
+                }
+            }
+
             $paginationArgs = PaginationArgs::extractArgs($args, $this->paginationType(), $this->paginateMaxCount());
 
             $paginationArgs->type = $this->optimalPaginationType($resolveInfo);

src/Pagination/PaginationManipulator.php

@@ -4,6 +4,7 @@
 
 use GraphQL\Language\AST\FieldDefinitionNode;
 use GraphQL\Language\AST\NamedTypeNode;
+use GraphQL\Language\AST\Node;
 use GraphQL\Language\AST\NonNullTypeNode;
 use GraphQL\Language\AST\ObjectTypeDefinitionNode;
 use GraphQL\Language\AST\TypeNode;
@@ -15,6 +16,8 @@
 use Nuwave\Lighthouse\Schema\AST\DocumentAST;
 use Nuwave\Lighthouse\Schema\Directives\ModelDirective;
 
+use function Safe\preg_replace;
+
 class PaginationManipulator
 {
     /**
@@ -273,4 +276,14 @@ protected function paginationResultType(string $typeName): TypeNode
 
         return $typeNode;
     }
+
+    public static function getReturnTypeName(Node $fieldDefinition): ?string
+    {
+        if (! ASTHelper::directiveDefinition($fieldDefinition, 'paginate')) {
+            return null;
+        }
+        $fieldTypeName = ASTHelper::getUnderlyingTypeName($fieldDefinition);
+
+        return preg_replace('/(Connection|SimplePaginator|Paginator)$/', '', $fieldTypeName);
+    }
 }

src/Schema/Directives/AllDirective.php

@@ -2,16 +2,19 @@
 
 namespace Nuwave\Lighthouse\Schema\Directives;
 
+use GraphQL\Error\Error;
 use GraphQL\Language\AST\FieldDefinitionNode;
 use GraphQL\Language\AST\ObjectTypeDefinitionNode;
 use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
 use Illuminate\Database\Eloquent\Relations\Relation;
 use Illuminate\Database\Query\Builder as QueryBuilder;
+use Illuminate\Database\Query\Expression;
 use Illuminate\Support\Collection;
 use Laravel\Scout\Builder as ScoutBuilder;
 use Nuwave\Lighthouse\Execution\ResolveInfo;
 use Nuwave\Lighthouse\Schema\AST\DocumentAST;
 use Nuwave\Lighthouse\Schema\Values\FieldValue;
+use Nuwave\Lighthouse\Select\SelectHelper;
 use Nuwave\Lighthouse\Support\Contracts\FieldManipulator;
 use Nuwave\Lighthouse\Support\Contracts\FieldResolver;
 use Nuwave\Lighthouse\Support\Contracts\GraphQLContext;
@@ -63,12 +66,57 @@ public function resolveField(FieldValue $fieldValue): FieldValue
                 $query = $this->getModelClass()::query();
             }
 
-            return $resolveInfo
+            $builder = $resolveInfo
                 ->enhanceBuilder(
                     $query,
                     $this->directiveArgValue('scopes', [])
-                )
-                ->get();
+                );
+
+            if (config('lighthouse.optimized_selects')) {
+                if ($builder instanceof EloquentBuilder) {
+                    $fieldSelection = array_keys($resolveInfo->getFieldSelection(1));
+
+                    $model = $builder->getModel();
+
+                    $selectColumns = SelectHelper::getSelectColumns(
+                        $this->definitionNode,
+                        $fieldSelection,
+                        get_class($model)
+                    );
+
+                    if (empty($selectColumns)) {
+                        throw new Error('The select column is empty.');
+                    }
+
+                    $query = $builder->getQuery();
+
+                    if (null !== $query->columns) {
+                        $bindings = $query->getRawBindings();
+
+                        $expressions = array_filter($query->columns, function ($column) {
+                            return $column instanceof Expression;
+                        });
+
+                        $builder = $builder->select(array_unique(array_merge($selectColumns, $expressions)));
+
+                        $builder = $builder->addBinding($bindings['select'], 'select');
+                    } else {
+                        $builder = $builder->select($selectColumns);
+                    }
+
+                    /** @var string|string[] $keyName */
+                    $keyName = $model->getKeyName();
+                    if (is_string($keyName)) {
+                        $keyName = [$keyName];
+                    }
+
+                    foreach ($keyName as $name) {
+                        $query->orderBy($name);
+                    }
+                }
+            }
+
+            return $builder->get();
         });
 
         return $fieldValue;

src/Schema/Directives/FindDirective.php

@@ -6,6 +6,7 @@
 use Illuminate\Database\Eloquent\Model;
 use Nuwave\Lighthouse\Execution\ResolveInfo;
 use Nuwave\Lighthouse\Schema\Values\FieldValue;
+use Nuwave\Lighthouse\Select\SelectHelper;
 use Nuwave\Lighthouse\Support\Contracts\FieldResolver;
 use Nuwave\Lighthouse\Support\Contracts\GraphQLContext;
 
@@ -35,12 +36,41 @@ public static function definition(): string
     public function resolveField(FieldValue $fieldValue): FieldValue
     {
         $fieldValue->setResolver(function ($root, array $args, GraphQLContext $context, ResolveInfo $resolveInfo): ?Model {
-            $results = $resolveInfo
+            $builder = $resolveInfo
                 ->enhanceBuilder(
                     $this->getModelClass()::query(),
                     $this->directiveArgValue('scopes', [])
-                )
-                ->get();
+                );
+
+            if (config('lighthouse.optimized_selects')) {
+                $fieldSelection = array_keys($resolveInfo->getFieldSelection(1));
+
+                $selectColumns = SelectHelper::getSelectColumns(
+                    $this->definitionNode,
+                    $fieldSelection,
+                    $this->getModelClass()
+                );
+
+                if (empty($selectColumns)) {
+                    throw new Error('The select column is empty.');
+                }
+
+                $builder = $builder->select($selectColumns);
+
+                $model = $builder->getModel();
+
+                /** @var string|string[] $keyName */
+                $keyName = $model->getKeyName();
+                if (is_string($keyName)) {
+                    $keyName = [$keyName];
+                }
+
+                foreach ($keyName as $name) {
+                    $builder->orderBy($name);
+                }
+            }
+
+            $results = $builder->get();
 
             if ($results->count() > 1) {
                 throw new Error('The query returned more than one result.');

src/Schema/Directives/SelectDirective.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace Nuwave\Lighthouse\Schema\Directives;
+
+class SelectDirective extends BaseDirective
+{
+    public static function definition(): string
+    {
+        return /** @lang GraphQL */ <<<'GRAPHQL'
+"""
+Specify the SQL column dependencies of this field.
+"""
+directive @select(
+  """
+  SQL column names to include in the `SELECT` part of the query.
+  """
+  columns: [String!]!
+) on FIELD_DEFINITION
+GRAPHQL;
+    }
+}