Absinthe项目中使用Dataloader优化GraphQL查询性能-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00939/article/details/148527241

Absinthe项目中使用Dataloader优化GraphQL查询性能

absinthe The GraphQL toolkit for Elixir 项目地址: https://gitcode.com/gh_mirrors/ab/absinthe

为什么需要Dataloader

在构建GraphQL API时，我们经常会遇到N+1查询问题。当查询一个用户及其所有帖子时，传统的实现方式会导致先查询1次获取用户列表，然后对每个用户再查询1次获取帖子列表。这种低效的查询方式会严重影响系统性能。

Absinthe项目通过集成Dataloader库，为我们提供了一种优雅的解决方案。Dataloader的核心思想是批量加载关联数据，将多个单独的查询合并为一次批量查询，从而显著提高查询效率。

基础配置

1. 添加依赖

首先需要在项目的mix.exs文件中添加dataloader依赖：

defp deps do
  [
    {:dataloader, "~> 1.0.7"}
    # 其他依赖...
  ]
end

2. 配置上下文

在业务上下文模块中(如lib/blog/content.ex)，设置Dataloader数据源：

def data(), do: Dataloader.Ecto.new(Repo, query: &query/2)

def query(queryable, params) do
  queryable
end

这里的query/2函数允许我们根据传入参数对查询进行定制化处理。

3. 配置Schema

在GraphQL Schema中(lib/blog_web/schema.ex)，需要设置Dataloader：

defmodule BlogWeb.Schema do
  use Absinthe.Schema

  def context(ctx) do
    loader =
      Dataloader.new()
      |> Dataloader.add_source(Content, Content.data())

    Map.put(ctx, :loader, loader)
  end

  def plugins do
    [Absinthe.Middleware.Dataloader | Absinthe.Plugin.defaults()]
  end
end

使用Dataloader解析字段

配置完成后，我们可以简化字段解析器的实现。例如用户帖子列表字段：

@desc "A user of the blog"
object :user do
  field :id, :id
  field :name, :string
  field :contacts, list_of(:contact)
  field :posts, list_of(:post) do
    arg :date, :date
    resolve dataloader(Content)
  end
end

这种方式比传统的手动解析器简洁得多，且自动支持批量加载。

高级用法

自定义查询逻辑

我们可以通过模式匹配为不同的查询条件定制不同的加载策略：

def query(query, %{has_admin_rights: true}), do: query

def query(query, _), do: from(a in query, select_merge: %{street_number: nil})

手动控制加载过程

对于更复杂的场景，可以手动控制加载过程：

field :posts, list_of(:post), resolve: fn user, args, %{context: %{loader: loader}} ->
  loader
  |> Dataloader.load(Blog, :posts, user)
  |> on_load(fn loader ->
    {:ok, Dataloader.get(loader, Blog, :posts, user)}
  end)
end