Use the same database prefix as the existing rows when preloading via…

… a lateral join
absinthe-graphql · Oct 13, 2020 · 1c354b3 · 1c354b3
1 parent 3ea0f68
commit 1c354b3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/dataloader/ecto.ex b/lib/dataloader/ecto.ex
@@ -673,9 +673,12 @@ if Code.ensure_loaded?(Ecto) do
 
       def preload_lateral([], _assoc, _query, _opts), do: []
 
-      def preload_lateral([%schema{} | _] = structs, assoc, query, repo, repo_opts) do
+      def preload_lateral([%schema{} = struct | _] = structs, assoc, query, repo, repo_opts) do
         [pk] = schema.__schema__(:primary_key)
 
+        # Carry the database prefix across from already-loaded records if not already set
+        repo_opts = Keyword.put_new(repo_opts, :prefix, struct.__meta__.prefix)
+
         assocs = expand_assocs(schema, [assoc])
 
         inner_query =