深入理解Go语言在GAE多租户环境下的数据存储事务机制

package main

import (
    "context"
    "fmt"
    "log"
    "time"

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

// User 定义一个用户实体
type User struct {
    Name      string
    Email     string
    LoginCount int
    UpdatedAt time.Time
}

// updateLoginCountInTransaction 示例：在事务中更新用户登录次数
// 该函数必须是幂等的
func updateLoginCountInTransaction(ctx context.Context, userID string) error {
    // 确保在事务中操作的是特定命名空间下的实体
    // ctx 已经通过 appengine.NewContextWithNamespace 设置了命名空间
    key := datastore.NewKey(ctx, "User", userID, 0, nil)

    _, err := datastore.RunInTransaction(ctx, func(tx *datastore.Transaction) error {
        user := new(User)
        err := tx.Get(key, user)
        if err != nil && err != datastore.ErrNoSuchEntity {
            return fmt.Errorf("failed to get user: %v", err)
        }

        if err == datastore.ErrNoSuchEntity {
            // 用户不存在，创建新用户
            user.Name = userID // 假设userID就是用户名
            user.Email = userID + "@example.com"
            user.LoginCount = 1
        } else {
            // 用户存在，更新登录次数
            user.LoginCount++
        }
        user.UpdatedAt = time.Now()

        _, err = tx.Put(key, user)
        if err != nil {
            return fmt.Errorf("failed to put user in transaction: %v", err)
        }
        return nil
    }, nil) // 最后一个参数为Options，nil表示使用默认选项

    if err != nil {
        return fmt.Errorf("transaction failed: %v", err)
    }
    return nil
}

func main() {
    // 模拟App Engine环境上下文
    // 在实际GAE应用中，ctx会由HTTP请求提供
    // 这里我们手动设置一个命名空间为 "tenant-A"
    ctx := appengine.NewContext(context.Background())
    ctx, err := appengine.Namespace(ctx, "tenant-A")
    if err != nil {
        log.Fatalf("Failed to set namespace: %v", err)
    }

    userID := "user123"

    // 第一次调用
    log.Printf("Tenant-A: Attempting to update login count for %s (first time)", userID)
    if err := updateLoginCountInTransaction(ctx, userID); err != nil {
        log.Printf("Error: %v", err)
    } else {
        log.Printf("Tenant-A: Login count updated successfully for %s.", userID)
    }

    // 第二次调用，模拟重试或再次登录
    log.Printf("Tenant-A: Attempting to update login count for %s (second time)", userID)
    if err := updateLoginCountInTransaction(ctx, userID); err != nil {
        log.Printf("Error: %v", err)
    } else {
        log.Printf("Tenant-A: Login count updated successfully for %s.", userID)
    }

    // 切换到另一个命名空间 "tenant-B"
    ctxB, err := appengine.Namespace(appengine.NewContext(context.Background()), "tenant-B")
    if err != nil {
        log.Fatalf("Failed to set namespace for tenant-B: %v", err)
    }

    // 在 tenant-B 中操作相同的 userID
    log.Printf("Tenant-B: Attempting to update login count for %s (first time)", userID)
    if err := updateLoginCountInTransaction(ctxB, userID); err != nil {
        log.Printf("Error: %v", err)
    } else {
        log.Printf("Tenant-B: Login count updated successfully for %s.", userID)
    }
}