女主宣言

越来越多的项目中都能看到GraphQL的身影，不知道大家在项目中有没有使用过GraphQL呢？今天给大家分享一下使用Go，来创建基础GraphQL API，供大家参考学习。

PS：丰富的一线技术、多元化的表现形式，尽在“HULK一线技术杂谈”，点关注哦！

 引言

随着时间推移，我们在越来越多的项目中，都可以发现GraphQL的身影。

从几个方面看，包括图形化的数据、多个分布式团队和高度版本化的api，以及关于类型安全和文档的问题。GraphQL看起来非常适合许多不同的应用程序。

本文的目标不是去介绍GraphQL的基础知识，而是在实际场景中看到它的实际操作。当计划将现有的REST API移动到GraphQL时，需要引入一个转换层，以实现平稳过渡。

在本篇文章中，我们将使用jsonplaceholder作为我们用GraphQL包装的API。这里有几个用于graphQL的库，在本篇文章示例中，将使用graphql-go和graphql-go-handler。

我们的目标是从jsonplaceholder获取文章和评论，最后以一种通过ID获取文章的方式，如果API使用者希望获取评论，则通过GraphQL将评论嵌套到文章中。

让我们开始吧。

 开始实现

首先，我们为文章和评论定义数据模型：

type Post struct {
    UserID int    `json:"userId"`
    ID     int    `json:"id"`
    Title  string `json:"title"`
    Body   string `json:"body"`
}

type Comment struct {
    PostID int    `json:"postId"`
    ID     int    `json:"id"`
    Name   string `json:"name"`
    Email  string `json:"email"`
    Body   string `json:"body"`
}

我们定义 fetchPostByiD(id)函数，它用来调用http://jsonplaceholder.typicode.com/posts/${id} 并将生成的JSON转换成Post。当然，还有一个fetchCommentByPostId(post.ID)助手函数，它对注释做同样的事情，从http://jsonplaceholder.typico/posts/${id}/注释中获取数据，并将其转换为[]Comment。

然后我们继续创建graphQL模式。我们从定义queryType开始，它是模式的根：

func createQueryType(postType *graphql.Object) graphql.ObjectConfig {
    return graphql.ObjectConfig{Name: "QueryType", Fields: graphql.Fields{
        "post": &graphql.Field{
            Type: postType,
            Args: graphql.FieldConfigArgument{
                "id": &graphql.ArgumentConfig{
                    Type: graphql.NewNonNull(graphql.Int),
                },
            },
            Resolve: func(p graphql.ResolveParams) (interface{}, error) {
                id := p.Args["id"]
                v, _ := id.(int)
                log.Printf("fetching post with id: %d", v)
                return fetchPostByiD(v)
            },
        },
    }}
}

根查询类型只有一个字段 —— post。这个字段是由postType定义的，我们稍后会看到。它只接受一个名为id的参数。

文章通过从p.Args中解析出id，并将它传递给fetchPostsByID，返回已获取和转换后的Post以及错误信息。

接下来，我们定义postType，这很有趣。我们基本上只是把post字段从数据模型映射到graphQL类型，但我们也添加了comments字段。如果客户端显式地想要取回它们，那么评论的Resolve函数才会被调用。

为了解决评论，我们使用p.Source来访问这个查询的“父”。它为我们提供了一个Post实例*Post——已获取的文章。使用这篇文章的id，我们可以获取评论：

func createPostType(commentType *graphql.Object) *graphql.Object {
    return graphql.NewObject(graphql.ObjectConfig{
        Name: "Post",
        Fields: graphql.Fields{
            "userId": &graphql.Field{
                Type: graphql.NewNonNull(graphql.Int),
            },
            "id": &graphql.Field{
                Type: graphql.NewNonNull(graphql.Int),
            },
            "title": &graphql.Field{
                Type: graphql.String,
            },
            "body": &graphql.Field{
                Type: graphql.String,
            },
            "comments": &graphql.Field{
                Type: graphql.NewList(commentType),
                Resolve: func(p graphql.ResolveParams) (interface{}, error) {
                    post, _ := p.Source.(*Post)
                    log.Printf("fetching comments of post with id: %d", post.ID)
                    return fetchCommentsByPostID(post.ID)
                },
            },
        },
    })
}

在模式中唯一要定义的类型是commentType，这很无聊，因为它只将数据模型的字段映射到graphQL类型：

func createCommentType() *graphql.Object {
    return graphql.NewObject(graphql.ObjectConfig{
        Name: "Comment",
        Fields: graphql.Fields{
            "postid": &graphql.Field{
                Type: graphql.NewNonNull(graphql.Int),
            },
            "id": &graphql.Field{
                Type: graphql.NewNonNull(graphql.Int),
            },
            "name": &graphql.Field{
                Type: graphql.String,
            },
            "email": &graphql.Field{
                Type: graphql.String,
            },
            "body": &graphql.Field{
                Type: graphql.String,
            },
        },
    })
}

好了，我们的模式被定义了，剩下的就是把它们放在一起。

我们实例化一个graphQL模式并将其传递给graphql-go-handler，它是一个http中间件，它帮助我们处理graphQL查询。然后我们简单地启动一个http服务器，它的返回处理程序被路由到/graphql。

这像这样：

func main() {
    schema, err := graphql.NewSchema(graphql.SchemaConfig{
        Query: graphql.NewObject(
            createQueryType(
                createPostType(
                    createCommentType(),
                ),
            ),
        ),
    })
    if err != nil {
        log.Fatalf("failed to create schema, error: %v", err)
    }
    handler := gqlhandler.New(&gqlhandler.Config{
        Schema: &schema,
    })
    http.Handle("/graphql", handler)
    log.Println("Server started at http://localhost:3000/graphql")
    log.Fatal(http.ListenAndServe(":3000", nil))
}

好了，以上就是全部实现！

启动服务器之后，我们可以使用GraphiQL查询一个具有特定id的文章，指定我们感兴趣的字段：

query {
    post(id: 5) {
        userId
        id
        body
        title
        comments {
            id
            email
            name
        }
    }
}

获取到以下返回信息：

{
    "data": {
        "post": {
             "userId": 1,
             "id": 5,
             "title": "...",
             "body": "...",      
             "comments": [
                 {
                     "id": 21,          
                     "email": "...",          
                     "name": "..."
                 },
                 ...
             ]
        }
    }
}

如果我们在查询中去掉了评论，那么就不会发出获取评论的请求，我们只是简单地将选定的帖子作为响应。

完整的示例代码地址 https://github.com/zupzup/example-go-graphql。

总结

这个例子展示了如何使用一个简单的Go层将现有的REST API转换成GraphQL。我们使用的这个库，graphql-go工作得很好，提供了可靠的文档和良好的示例。

当然，有更简洁、更有趣的方法来定义这样的模式，但是由于技术等有限，我选择了这个解决方案，它的重点是清晰性。希望大家举一反三，自己结合实际去编写适合项目的实现。

HULK一线技术杂谈

由360云平台团队打造的技术分享公众号，内容涉及云计算、数据库、大数据、监控、泛前端、自动化测试等众多技术领域，通过夯实的技术积累和丰富的一线实战经验，为你带来最有料的技术分享