async-graphql/docs/zh-CN/src/define_union.md
2020-10-01 11:23:24 +08:00

1.7 KiB

联合(Union)

联合的定义和接口非常像,但不允许定义字段。这两个类型的实现原理也差不多,对于Async-graphql来说,联合类型是接口类型的子集。

下面把接口定义的例子做一个小小的修改,去掉字段的定义。

use async_graphql::*;

struct Circle {
    radius: f32,
}

#[Object]
impl Circle {
    async fn area(&self) -> f32 {
        std::f32::consts::PI * self.radius * self.radius
    }

    async fn scale(&self, s: f32) -> Shape {
        Circle { radius: self.radius * s }.into()
    }
}

struct Square {
    width: f32,
}

#[Object]
impl Square {
    async fn area(&self) -> f32 {
        self.width * self.width
    }

    async fn scale(&self, s: f32) -> Shape {
        Square { width: self.width * s }.into()
    }
}

#[derive(Union)]
enum Shape {
    Circle(Circle),
    Square(Square),
}

展平嵌套联合

GraphQL的有个限制是Union类型内不能包含其它联合类型。 所有成员必须为Object。 位置支持嵌套Union,我们可以用#graphql(flatten),是它们合并到上级Union类型。

#[derive(async_graphql::Union)]
pub enum TopLevelUnion {
    A(A),

    // 除非我们使用`flatten`属性,否则将无法编译
    #[graphql(flatten)]
    B(B),
}

#[derive(async_graphql::SimpleObject)]
pub struct A {
    // ...
}

#[derive(async_graphql::Union)]
pub enum B {
    C(C),
    D(D),
}

#[derive(async_graphql::SimpleObject)]
pub struct C {
    // ...
}

#[derive(async_graphql::SimpleObject)]
pub struct D {
    // ...
}

上面的示例将顶级Union转换为以下等效形式:

#[derive(async_graphql::Union)]
pub enum TopLevelUnion {
    A(A),
    C(C),
    D(D),
}