Go语言学习：使用mgo插入文档后如何返回新创建的文档？

嘿，我之前也踩过这个坑！用mgo的时候确实容易搞混Insert和Upsert的返回逻辑，下面分两种场景给你讲清楚怎么实现返回新创建的文档：

一、单纯插入新文档（Insert） #

如果你的需求是创建全新的文档，不需要处理已存在的情况，直接用Insert方法就完美解决——这个方法会自动把MongoDB生成的_id填充到你传入的结构体里，插入成功后直接返回这个结构体就行。

首先得确保你的Book结构体正确映射MongoDB字段，尤其是_id字段：

import "gopkg.in/mgo.v2/bson"

type Book struct {
    ID     bson.ObjectId `bson:"_id,omitempty" json:"id"` // omitempty表示未设置时MongoDB自动生成ID
    Title  string        `bson:"title" json:"title"`
    Author string        `bson:"author" json:"author"`
    // 其他自定义字段按需添加
}

然后是API的具体处理逻辑：

func createBookHandler(w http.ResponseWriter, r *http.Request) {
    // 解析请求体到Book结构体
    var newBook Book
    if err := json.NewDecoder(r.Body).Decode(&newBook); err != nil {
        http.Error(w, "请求参数解析失败: "+err.Error(), http.StatusBadRequest)
        return
    }

    // 获取MongoDB集合（这里需要你提前初始化好Mongo会话）
    coll := yourMongoSession.DB("your_database_name").C("books")

    // 执行插入操作
    if err := coll.Insert(&newBook); err != nil {
        http.Error(w, "插入文档失败: "+err.Error(), http.StatusInternalServerError)
        return
    }

    // 重点：此时newBook已经包含MongoDB自动生成的ID，直接返回给用户即可
    w.Header().Set("Content-Type", "application/json")
    if err := json.NewEncoder(w).Encode(newBook); err != nil {
        http.Error(w, "返回结果编码失败: "+err.Error(), http.StatusInternalServerError)
    }
}

二、Upsert操作（不存在则插入，存在则更新） #

如果你的需求是Upsert，那Upsert方法返回的是*mgo.ChangeInfo（只包含更新/插入的行数等元数据），不会自动返回文档。这时候需要在Upsert完成后，根据你的查询条件去查询对应的文档，再返回给用户。

比如你用Title作为唯一标识进行Upsert：

func upsertBookHandler(w http.ResponseWriter, r *http.Request) {
    var bookData Book
    if err := json.NewDecoder(r.Body).Decode(&bookData); err != nil {
        http.Error(w, "请求参数解析失败: "+err.Error(), http.StatusBadRequest)
        return
    }

    coll := yourMongoSession.DB("your_database_name").C("books")

    // 定义Upsert的查询条件（这里用Title作为唯一键）
    selector := bson.M{"title": bookData.Title}
    // 执行Upsert操作
    changeInfo, err := coll.Upsert(selector, &bookData)
    if err != nil {
        http.Error(w, "Upsert操作失败: "+err.Error(), http.StatusInternalServerError)
        return
    }

    // 查询最终的文档（无论是新插入的还是更新后的）
    var finalBook Book
    if err := coll.Find(selector).One(&finalBook); err != nil {
        http.Error(w, "查询文档失败: "+err.Error(), http.StatusInternalServerError)
        return
    }

    // 返回查询到的完整文档
    w.Header().Set("Content-Type", "application/json")
    if err := json.NewEncoder(w).Encode(finalBook); err != nil {
        http.Error(w, "返回结果编码失败: "+err.Error(), http.StatusInternalServerError)
    }
}

如果是用ID作为Upsert条件，建议用UpsertId，之后直接通过FindId查询，效率会更高：

// 假设bookData已经设置了ID
changeInfo, err := coll.UpsertId(bookData.ID, &bookData)
if err != nil { /* 错误处理逻辑 */ }

var finalBook Book
if err := coll.FindId(bookData.ID).One(&finalBook); err != nil { /* 错误处理逻辑 */ }

小提醒 #

• 一定要确保Book结构体的bson标签和MongoDB的字段完全匹配，否则会出现字段缺失或者无法正确映射的问题。
• 如果你的_id不是用bson.ObjectId，而是自定义的类型，记得同步调整结构体的标签和查询逻辑。

内容的提问来源于stack exchange，提问作者WagnerMatosUK