From 3ecaa4a0cdac43b8d34ce77c7bbadc9243ea9579 Mon Sep 17 00:00:00 2001
From: Maarten van Vliet <maartenvanvliet@users.noreply.github.com>
Date: Tue, 31 Mar 2020 17:45:42 +0200
Subject: [PATCH] Add `dataloader/2` function to helpers (#898)

Adds a convenience `dataloader/2` function that I've needed a couple of
times. It's useful in cases where the resource can be inferred from the
field name but you still want to pass in extra opts such as `args` or
`callback` to the dataloader function.
---
 lib/absinthe/resolution/helpers.ex | 40 +++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/lib/absinthe/resolution/helpers.ex b/lib/absinthe/resolution/helpers.ex
index 3be08828ac..f98781f10c 100644
--- a/lib/absinthe/resolution/helpers.ex
+++ b/lib/absinthe/resolution/helpers.ex
@@ -159,9 +159,47 @@ defmodule Absinthe.Resolution.Helpers do
     """
     @spec dataloader(Dataloader.source_name()) :: dataloader_key_fun()
     def dataloader(source) do
+      dataloader(source, [])
+    end
+
+    @doc """
+    Resolve a field with a dataloader source.
+
+    This function is not imported by default. To make it available in your module do
+
+    ```
+    import Absinthe.Resolution.Helpers
+    ```
+
+    Same as `dataloader/3`, but it infers the resource name from the field name. For `opts` see
+    `dataloader/3` on what options can be passed in.
+
+    ## Examples
+
+    ```
+    object :user do
+      field :posts, list_of(:post),
+        resolve: dataloader(Blog, args: %{deleted: false})
+
+      field :organization, :organization do
+        resolve dataloader(Accounts, use_parent: false)
+      end
+
+      field(:account_active, non_null(:boolean), resolve: dataloader(
+          Accounts, callback: fn account, _parent, _args ->
+            {:ok, account.active}
+          end
+        )
+      )
+    end
+    ```
+
+    """
+    @spec dataloader(Dataloader.source_name(), [dataloader_opt]) :: dataloader_key_fun()
+    def dataloader(source, opts) when is_list(opts) do
       fn parent, args, %{context: %{loader: loader}} = res ->
         resource = res.definition.schema_node.identifier
-        do_dataloader(loader, source, resource, args, parent, [])
+        do_dataloader(loader, source, resource, args, parent, opts)
       end
     end