使用 Go 语言在 Google App Engine 中执行原子更新

package main

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

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

// Account 实体
type Account struct {
    Balance   int64
    Rate      int64
    CheckDate time.Time
}

func main() {
    ctx, done, err := aetest.NewContext()
    if err != nil {
        log.Fatal(err)
    }
    defer done()

    // 创建一个 Account 实体
    key := datastore.NewKey(ctx, "Account", "user1", 0, nil)
    account := Account{
        Balance:   100,
        Rate:      1,
        CheckDate: time.Now(),
    }

    // 保存 Account 实体
    _, err = datastore.Put(ctx, key, &account)
    if err != nil {
        log.Fatalf("Failed to put account: %v", err)
    }

    // 定义一个更新 Account 余额的函数
    updateBalance := func(ctx context.Context) error {
        var acc Account
        if err := datastore.Get(ctx, key, &acc); err != nil {
            return err
        }

        // 计算自上次检查以来的秒数
        seconds := time.Since(acc.CheckDate).Seconds()

        // 更新余额
        acc.Balance = acc.Balance + int64(float64(acc.Rate)*seconds)

        // 更新检查日期
        acc.CheckDate = time.Now()

        // 保存更新后的实体
        _, err := datastore.Put(ctx, key, &acc)
        return err
    }

    // 使用事务执行更新操作
    err = datastore.RunInTransaction(ctx, updateBalance, nil)
    if err != nil {
        log.Fatalf("Transaction failed: %v", err)
    }

    // 验证更新结果
    var updatedAccount Account
    if err := datastore.Get(ctx, key, &updatedAccount); err != nil {
        log.Fatalf("Failed to get updated account: %v", err)
    }

    fmt.Printf("Updated Balance: %d\n", updatedAccount.Balance)
}