Strawberry GraphQL 中自定义 Info 类的正确配置方式

在 GraphQL 开发中，Strawberry 是一个强大的 Python 库，它提供了灵活的方式来定制 GraphQL 查询的执行上下文。本文将详细介绍如何正确地在 Strawberry GraphQL 中配置自定义的 Info 类。

Info 类的作用

在 Strawberry GraphQL 中，Info 类是一个重要的组件，它包含了当前 GraphQL 查询执行的上下文信息。开发者可以通过继承 Info 类来扩展其功能，添加自定义属性和方法，以满足特定的业务需求。

常见错误配置

许多开发者会尝试直接将自定义的 Info 类传递给 strawberry.Schema 的构造函数，如下所示：

schema = strawberry.Schema(query=Query, info_class=CustomInfo)

这种配置方式在旧版本的 Strawberry 中可能有效，但在当前版本中已经不再适用。这种错误的配置会导致运行时错误，因为 Strawberry 的 API 已经发生了变化。

正确的配置方法

正确的做法是通过 StrawberryConfig 来配置自定义的 Info 类。StrawberryConfig 是 Strawberry 提供的配置容器，专门用于集中管理各种 Schema 级别的配置选项。

schema = strawberry.Schema(
    query=Query,
    config=StrawberryConfig(info_class=CustomInfo)
)

自定义 Info 类的实现示例

下面是一个完整的自定义 Info 类实现示例，展示了如何扩展 Info 类以访问 HTTP 响应头：

from strawberry.types import Info
from strawberry import StrawberryConfig
from somewhere import Headers  # 根据实际项目导入 Headers 类型

class CustomInfo(Info):
    @property
    def response_headers(self) -> Headers:
        """获取当前 HTTP 响应的头部信息"""
        return self.context["response"].headers

# 创建 Schema 时使用自定义的 Info 类
schema = strawberry.Schema(
    query=Query,
    config=StrawberryConfig(info_class=CustomInfo)
)

为什么需要这种配置方式

这种通过 StrawberryConfig 的配置方式有以下几个优点：

1. 一致性：所有 Schema 配置都通过统一的接口管理
2. 可扩展性：未来添加新的配置选项不会破坏现有代码
3. 明确性：清晰地表明了配置的意图和范围

实际应用场景

自定义 Info 类在以下场景中特别有用：

1. 需要在解析器中访问特定的上下文信息
2. 想要为解析器提供便捷的辅助方法
3. 需要统一处理某些上下文数据的访问逻辑

总结

在 Strawberry GraphQL 中正确配置自定义 Info 类需要注意使用 StrawberryConfig 作为配置容器。这种方式不仅符合当前版本的 API 设计，也为未来的扩展提供了良好的基础。开发者应该避免直接传递 info_class 参数给 Schema 构造函数，而是采用推荐的配置方式。