Go语言在App Engine Datastore中高效更新实体教程

package main

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

// User 结构体定义
type User struct {
    Name  string
    Email string
    // ID 字段用于存储Datastore的IntID。
    // `datastore:"-"` 标签指示Datastore忽略此字段的存储和检索。
    ID    int64 `datastore:"-"` 
}

// IsNew 方法判断实体是否为新实体。
// 如果ID为0，则认为是新实体（尚未被Datastore分配ID）。
func (u *User) IsNew() bool {
    return u.ID == 0
}

import (
    "net/http"
    "google.golang.org/appengine"
    "google.golang.org/appengine/datastore"
    "html/template"
)

// saveUser 示例函数：保存（创建或更新）用户
func saveUser(w http.ResponseWriter, r *http.Request, user *User) error {
    c := appengine.NewContext(r)

    var k *datastore.Key
    if user.IsNew() {
        // 对于新实体，创建不完整键。Datastore将自动生成ID。
        k = datastore.NewIncompleteKey(c, "User", nil)
    } else {
        // 对于现有实体，创建完整键，使用已知的IntID。
        // 注意：这里的Kind必须与查询时的Kind一致。
        k = datastore.NewKey(c, "User", "", user.ID, nil)
    }

    // 执行Put操作。Put会返回实际使用的键和可能的错误。
    finalKey, err := datastore.Put(c, k, user)
    if err != nil {
        return err
    }

    // 如果是新实体，或者为了确保ID与Datastore一致，
    // 将Datastore生成的IntID赋值回User对象的ID字段。
    user.ID = finalKey.IntID()
    return nil
}

// subscribe 处理用户订阅（创建或更新）请求
func subscribe(w http.ResponseWriter, r *http.Request) {
    user := User{
        Name:  r.FormValue("username"),
        Email: r.FormValue("useremail"),
        // 如果是从表单提交，且可能包含ID，需要在这里解析并赋值。
        // 例如：user.ID, _ = strconv.ParseInt(r.FormValue("id"), 10, 64)
    }

    // 假设我们有一个机制来查找现有用户，如果找到，则更新其ID
    // 否则，user.ID 仍为0，被视为新用户。
    // 这里简化为直接调用saveUser
    err := saveUser(w, r, &user)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    template.Must(template.ParseFiles("confirmation.html")).Execute(w, nil)
}

// checkusers 查询所有用户并显示
func checkusers(w http.ResponseWriter, r *http.Request) {
    c := appengine.NewContext(r)

    qUsers := datastore.NewQuery("User")

    var users []User // 用于存储查询到的用户实体

    // GetAll 返回实体切片和对应的键切片
    keys, err := qUsers.GetAll(c, &users)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 遍历键切片和实体切片，将键的IntID赋值给对应的用户实体
    for i := 0; i < len(users); i++ {
        users[i].ID = keys[i].IntID()
    }

    template.Must(template.ParseFiles("users.html")).Execute(w, users)
}