在Go语言中实现GAE Datastore的“IN”查询：通过列表值检索实体

q := datastore.NewQuery("Foo").Filter("CreatorId =", 1).Filter("CreatorId =", 5).Filter("CreatorId =", 23)

package main

import (
    "context"
    "fmt"
    "log"

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

// Foo 定义了Datastore中的实体结构
type Foo struct {
    Id        int64  `datastore:"-"` // "-" 表示该字段不存储在Datastore中，用于本地ID
    Name      string
    CreatorId int64
}

func main() {
    // 假设您已经初始化了Datastore客户端
    // 例如：client, err := datastore.NewClient(ctx, "your-project-id")
    // 这里为了示例方便，我们将模拟Datastore操作
    // 实际应用中需要替换为真实的Datastore客户端
    ctx := context.Background()
    client, err := datastore.NewClient(ctx, "your-project-id") // 替换为您的项目ID
    if err != nil {
        log.Fatalf("Failed to create Datastore client: %v", err)
    }
    defer client.Close()

    // 模拟一些数据写入，以便后续查询
    // 实际应用中这些数据可能已经存在
    keys := []*datastore.Key{
        datastore.IncompleteKey("Foo", nil),
        datastore.IncompleteKey("Foo", nil),
        datastore.IncompleteKey("Foo", nil),
        datastore.IncompleteKey("Foo", nil),
        datastore.IncompleteKey("Foo", nil),
    }
    entities := []*Foo{
        {Name: "Foo A", CreatorId: 1},
        {Name: "Foo B", CreatorId: 5},
        {Name: "Foo C", CreatorId: 23},
        {Name: "Foo D", CreatorId: 10},
        {Name: "Foo E", CreatorId: 1},
    }

    // 批量写入实体
    _, err = client.PutMulti(ctx, keys, entities)
    if err != nil {
        log.Fatalf("Failed to put entities: %v", err)
    }
    fmt.Println("示例实体已写入Datastore。")


    // 目标 CreatorId 列表
    targetCreatorIDs := []int64{1, 5, 23}

    // 用于存储所有查询结果的切片
    var allFooEntities []*Foo

    fmt.Printf("开始查询 CreatorId 在 %v 中的实体...\n", targetCreatorIDs)

    for _, id := range targetCreatorIDs {
        // 为每个 CreatorId 构建一个独立的查询
        q := datastore.NewQuery("Foo").Filter("CreatorId =", id)

        var currentBatchFoos []*Foo
        // 执行查询
        keys, err := client.GetAll(ctx, q, &currentBatchFoos)
        if err != nil {
            log.Printf("查询 CreatorId = %d 时出错: %v", id, err)
            continue // 继续下一个ID的查询
        }

        // 将查询到的实体添加到总结果集中
        for i, foo := range currentBatchFoos {
            foo.Id = keys[i].ID
            allFooEntities = append(allFooEntities, foo)
        }
    }

    // 打印最终结果
    if len(allFooEntities) > 0 {
        fmt.Println("\n查询结果:")
        for _, foo := range allFooEntities {
            fmt.Printf("ID: %d, Name: %s, CreatorId: %d\n", foo.Id, foo.Name, foo.CreatorId)
        }
    } else {
        fmt.Println("\n未找到匹配的实体。")
    }
}