gqlgen go，通过实现解析器来优化数据库访问

本篇文章给大家分享《gqlgen go，通过实现解析器来优化数据库访问》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

我在解决特定情况时遇到一些问题，导致性能下降。 我很确定这是可以做到的，但我不知道如何去做。

以下是暴露问题的示例架构：

type answer{
    answerid: string!
    text: string!
    topic: topic!
}

type topic {
    topicid: string!
    name: string!
    level: int!
}

extend type query {
    answer(answerid: string!): answer!
    answers: [answer!]! 
}

我已遵循文档，特别是这部分 https://gqlgen.com/getting-started/#dont-eagerly-fetch-the-user 根据我的架构，它生成以下解析器：

func (r *queryresolver) answer(ctx context.context, answerid string) (*models.answer, error) {
...
#single query which retrives single record of answer from db.
#fills a model answer with the id and the text
#proceeds by calling the topic resolver
...
}

func (r *queryresolver) answers(ctx context.context) ([]*models.answer, error) {
...
#single query which retrives list of answers from db
#fills a list of model answer with the id and the text
-->#for each element of that list, it calls the topic resolver
...
}


func (r *answerresolver) topic(ctx context.context, obj *models.answer) (*models.topic, error) {
...
#single query which retrives single record of topic from db
#return a model topic with id, name and level
...
}

当使用 answerid 参数调用 answer 查询时，会触发 answer 解析器，它会解析 text 属性并调用 topic 解析器。 topic 解析器按预期工作，检索 topic 并将其合并到 answer 内并返回。

当在没有 answerid 参数的情况下调用 answers 查询时，会触发 answer 解析器，它会使用单个查询检索 answers 列表。 然后，对于该 list 的每个元素，它调用 topic 解析器。 topic 检索 topic 并将其合并到单个 answer 中并返回。

结果在两种情况下都没有问题，但是如果我要求大量答案，则 answers 查询会成为性能问题。 对于每个答案，都会触发 topic 解析器并执行查询以检索单个记录。

例如。如果我有 2 个答案 --> 1 个查询 [answer0, answer1]，则 1 个查询 topic0 和 1 个查询 topic1

例如。 10 个答案 --> 1 代表 [answer0, ..., answer9]，然后每个 topicn 10 个

我想获得一些 topic 数组解析器，例如

func (r *answersResolver) Topics(ctx context.Context, obj *[]models.Answer) (*[]models.Topic, error) {
...
#Single Query which retrives list of Topics from DB
#Return a list of model Topic with id, name and level
...
}

我希望返回数组的每个元素都与 answers 数组的相应元素合并。

有可能以某种方式实现吗？我在哪里可以找到这种方法的示例？ 谢谢

正确答案

可以使用 dataloader 解决该问题 (docs)

我必须为 topics 实现以下数据源：

package dataloader

import (
    "github.com/graph-gophers/dataloader"
)

type ctxKey string

const (
    loadersKey = ctxKey("dataloaders")
)


type TopicReader struct {
    conn *sql.DB
}

func (t *TopicReader) GetTopics(ctx context.Context, keys dataloader.Keys) []*dataloader.Result {
    topicIDs := make([]string, len(keys))
    for ix, key := range keys {
        topicIDs[ix] = key.String()
    }
    res := u.db.Exec(
        r.Conn,
        "SELECT id, name, level
        FROM topics
        WHERE id IN (?" + strings.Repeat(",?", len(topicIDs-1)) + ")",
        topicIDs...,
    )
    defer res.Close()

    output := make([]*dataloader.Result, len(keys))
    for index, _ := range keys {
            output[index] = &dataloader.Result{Data: res[index], Error: nil}
    }
    return output
}

type Loaders struct {
    TopicLoader *dataloader.Loader
}


func NewLoaders(conn *sql.DB) *Loaders {
    topicReader := &TopicReader{conn: conn}
    loaders := &Loaders{
        TopicLoader: dataloader.NewBatchedLoader(t.GetTopics),
    }
    return loaders
}

func Middleware(loaders *Loaders, next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        nextCtx := context.WithValue(r.Context(), loadersKey, loaders)
        r = r.WithContext(nextCtx)
        next.ServeHTTP(w, r)
    })
}

func For(ctx context.Context) *Loaders {
    return ctx.Value(loadersKey).(*Loaders)
}

func GetTopic(ctx context.Context, topicID string) (*model.Topic, error) {
    loaders := For(ctx)
    thunk := loaders.TopicLoader.Load(ctx, dataloader.StringKey(topicID))
    result, err := thunk()
    if err != nil {
        return nil, err
    }
    return result.(*model.Topic), nil
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~