App Engine Go 一对多关系教程详解

本文档介绍了如何在 Google App Engine 的 Go 语言环境中实现一对多关系。由于 App Engine Datastore 的限制，无法直接在实体中存储大量的关联键。本文将探讨两种实现方法，重点介绍通过在子实体中存储父实体键的方式，并提供示例代码演示如何进行查询。

App Engine Datastore 的限制

在 Google App Engine 的 Go 语言环境中，Datastore 对实体字段的类型有所限制。具体来说，允许的类型包括：

• 有符号整数 (int, int8, int16, int32, int64)
• bool
• string
• float32 和 float64
• 以上预定义类型的基础类型
• *datastore.Key
• appengine.BlobKey
• []byte (最大 1MB)
• 以上类型的切片 (最大 100 个元素)

由于切片长度的限制，直接在父实体中存储大量子实体的键可能不可行。因此，我们需要寻找其他方法来实现一对多关系。

实现一对多关系的两种方法

1. 父实体存储子实体键的切片： 这种方法是将子实体的键存储在父实体的切片中。例如，Comment 结构体可以包含一个 []*datastore.Key 类型的字段，用于存储关联的 Vote 实体键。但是，这种方法受限于 Datastore 对切片长度的 100 个元素的限制。对于任何可能拥有超过 100 个子实体的父实体，这种方法都不可行。

2. 子实体存储父实体键： 这种方法是在子实体中存储指向父实体的键。例如，Vote 结构体可以包含一个 *datastore.Key 类型的字段 CommentKey，用于存储关联的 Comment 实体键。这种方法避免了切片长度的限制，更适合处理大量关联实体的情况。

使用子实体存储父实体键实现一对多关系

我们推荐使用第二种方法，即在子实体中存储父实体键。以下示例代码演示了如何使用这种方法来实现 Comment 和 Vote 之间的一对多关系。

首先，定义 Comment 和 Vote 结构体：

import (
    "time"

    "cloud.google.com/go/datastore"
)

type Comment struct {
    Author  string
    Content string
    Date    time.Time
}

type Vote struct {
    User       string
    Score      int
    CommentKey *datastore.Key
}

然后，演示如何查询与特定 Comment 关联的所有 Vote：

import (
    "context"
    "fmt"
    "log"

    "cloud.google.com/go/datastore"
    "google.golang.org/api/iterator"
)

func GetVotesForComment(ctx context.Context, client *datastore.Client, commentKey *datastore.Key) ([]Vote, error) {
    var votes []Vote

    q := datastore.NewQuery("Vote").Filter("CommentKey =", commentKey)
    it := client.Run(ctx, q)
    for {
        var vote Vote
        key, err := it.Next(&vote)
        if err == iterator.Done {
            break
        }
        if err != nil {
            return nil, fmt.Errorf("failed to fetch next Vote: %v", err)
        }
        vote.CommentKey = key // Store the key for potential later use

        votes = append(votes, vote)
    }

    return votes, nil
}

func main() {
    ctx := context.Background()
    projectID := "your-project-id" // Replace with your Google Cloud project ID

    client, err := datastore.NewClient(ctx, projectID)
    if err != nil {
        log.Fatalf("Failed to create client: %v", err)
    }
    defer client.Close()

    // Create a dummy comment
    comment := Comment{
        Author:  "Test Author",
        Content: "Test Content",
        Date:    time.Now(),
    }

    // Save the comment to Datastore
    commentKey := datastore.NameKey("Comment", "test-comment", nil)
    commentKey, err = client.Put(ctx, commentKey, &comment)
    if err != nil {
        log.Fatalf("Failed to save comment: %v", err)
    }

    // Create some dummy votes associated with the comment
    vote1 := Vote{
        User:       "User1",
        Score:      5,
        CommentKey: commentKey,
    }
    vote2 := Vote{
        User:       "User2",
        Score:      3,
        CommentKey: commentKey,
    }

    // Save the votes to Datastore
    _, err = client.Put(ctx, datastore.IncompleteKey("Vote", commentKey), &vote1)
    if err != nil {
        log.Fatalf("Failed to save vote1: %v", err)
    }
    _, err = client.Put(ctx, datastore.IncompleteKey("Vote", commentKey), &vote2)
    if err != nil {
        log.Fatalf("Failed to save vote2: %v", err)
    }

    // Retrieve all votes associated with the comment
    votes, err := GetVotesForComment(ctx, client, commentKey)
    if err != nil {
        log.Fatalf("Failed to retrieve votes: %v", err)
    }

    // Print the retrieved votes
    fmt.Println("Votes for comment:")
    for _, vote := range votes {
        fmt.Printf("  User: %s, Score: %d\n", vote.User, vote.Score)
    }
}

代码解释：

1. GetVotesForComment 函数：

   • 接收 context.Context、*datastore.Client 和 *datastore.Key 作为参数，其中 commentKey 是要查询的 Comment 实体的键。
   • 创建一个新的 Datastore 查询，筛选条件是 CommentKey 等于传入的 commentKey。
   • 使用 client.Run 执行查询，并遍历结果。
   • 将每个 Vote 实体添加到 votes 切片中。
   • 返回 votes 切片和可能发生的错误。

2. main 函数：

   • 创建 Datastore 客户端。
   • 创建一个示例 Comment 实体，并将其保存到 Datastore 中。
   • 创建两个示例 Vote 实体，并将它们的 CommentKey 设置为 Comment 实体的键。
   • 将 Vote 实体保存到 Datastore 中。
   • 调用 GetVotesForComment 函数来检索与 Comment 实体关联的所有 Vote 实体。
   • 打印检索到的 Vote 实体的信息。

注意事项：

• 请确保将 your-project-id 替换为您的 Google Cloud 项目 ID。
• 在生产环境中，您需要处理可能发生的错误，并实现适当的重试机制。
• 根据您的实际需求，您可能需要添加额外的索引来优化查询性能。

总结

本文介绍了如何在 Google App Engine 的 Go 语言环境中实现一对多关系。由于 Datastore 的限制，建议在子实体中存储父实体的键。通过示例代码，演示了如何查询与特定父实体关联的所有子实体。希望本文能够帮助您在 App Engine 应用中有效地管理一对多关系。