Go语言在GAE Datastore中的并发操作实践

package main

import (
    "fmt"
    "strings"

    "google.golang.org/appengine"
    "google.golang.org/appengine/datastore"
)

// User 定义用户结构体
type User struct {
    Name    string
    Email   string
    // ... 其他用户字段
}

// Entry 定义条目结构体
type Entry struct {
    User string
    Title string
    Content string
    key *datastore.Key // 用于存储Datastore Key
    // ... 其他条目字段
}

// loadUser 并行加载用户及其关联条目
func loadUser(ctx appengine.Context, name string) (*User, []*Entry, error) {
    var u User
    var entries []*Entry
    // 创建一个通道用于接收Goroutine的执行结果（错误信息）
    done := make(chan error)

    // Goroutine 1: 加载用户主要信息
    go func() {
        userKey := datastore.NewKey(ctx, "User", name, 0, nil)
        // datastore.Get是阻塞式调用，但在Goroutine中执行时不会阻塞主Goroutine
        err := datastore.Get(ctx, userKey, &u)
        done <- err // 将错误发送到通道
    }()

    // Goroutine 2: 加载与用户关联的条目
    go func() {
        q := datastore.NewQuery("Entry").Filter("User =", name)
        // datastore.GetAll是阻塞式调用
        keys, err := q.GetAll(ctx, &entries)
        if err == nil {
            // 将获取到的Key赋值给每个Entry
            for i, k := range keys {
                entries[i].key = k
            }
        }
        done <- err // 将错误发送到通道
    }()

    success := true
    var finalErr error
    // 等待两个Goroutine完成，并收集错误
    for i := 0; i < 2 /* 对应上面启动的Goroutine数量 */; i++ {
        if err := <-done; err != nil { // 从通道接收错误
            ctx.Errorf("loadUser: 异步操作错误: %s", err)
            success = false
            if finalErr == nil { // 只记录第一个遇到的错误
                finalErr = err
            }
        }
    }

    if !success {
        return nil, nil, finalErr // 如果有错误，返回nil和错误
    }

    // 可以在这里进行更多操作，例如组合数据等
    return &u, entries, nil
}

func main() {
    // 这是一个模拟App Engine环境的例子，实际运行时ctx由GAE提供
    // ctx := appengine.NewContext(r) // 在GAE处理HTTP请求时获取ctx
    // 为了演示，这里简化ctx的创建
    fmt.Println("此示例代码需要在Google App Engine环境中运行")
    fmt.Println("`appengine.Context`通常由GAE请求处理函数提供")
    // 假设我们有一个名为"Alice"的用户
    // user, entries, err := loadUser(ctx, "Alice")
    // if err != nil {
    //     log.Fatalf("加载用户失败: %v", err)
    // }
    // fmt.Printf("加载用户: %+v\n", user)
    // fmt.Printf("关联条目: %+v\n", entries)
}