Go CRUD后端Post接口存储图片时MongoDB连接被强制关闭

问题根源分析 #

你遇到的Connection was forcibly closed by a peer错误，核心原因是代码中的未捕获panic和请求头配置错误，同时存在几处潜在问题会触发异常：

1. 请求头配置错误 #

• Content-Type设置无效：代码中写的"writer.FormDataContentType()"是字符串字面量，并非调用方法，正确做法是使用请求自身的Content-Type或直接指定为multipart/form-data
• CORS头名称错误：应为Access-Control-Allow-Methods，而非Allow-Control-Allow-Methods

2. 未捕获panic导致连接终止 #

Post和AddPost函数中直接使用panic(err)，当出现ObjectID解析失败、MongoDB操作异常等情况时，Go HTTP服务器会直接终止当前连接，返回连接关闭错误，而非合法的HTTP响应。

3. 模型字段标签不匹配 #

Credentials结构体的Email字段bson标签为_email，通常MongoDB字段名应为email，这会导致数据存储/查询异常，间接引发错误。

4. Context使用不规范 #

AddPost中调用UpdateOne使用context.Background()，未传递请求上下文，无法在请求超时或取消时正确终止MongoDB操作。

修复后的代码示例 #

修复后的Post函数 #

func Post(w http.ResponseWriter, r *http.Request) {
    // 修正CORS和Content-Type头
    w.Header().Set("Access-Control-Allow-Methods", "POST")
    w.Header().Set("Content-Type", "application/json")

    // 解析multipart表单，最大内存限制32MB
    err := r.ParseMultipartForm(32 << 20)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }

    params := mux.Vars(r)
    category := r.FormValue("category")
    description := r.FormValue("description")

    file, _, err := r.FormFile("file")
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    defer file.Close()

    fileBytes, err := io.ReadAll(file)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    data := models.Post{
        Image:       fileBytes,
        Category:    category,
        Description: description,
    }

    // 传递请求上下文，处理控制器返回的错误
    err = controllers.AddPost(r.Context(), params["id"], data)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    json.NewEncoder(w).Encode(map[string]string{"user_id": params["id"]})
}

修复后的AddPost函数 #

func AddPost(ctx context.Context, user string, model models.Post) error {
    exists, err := Checkuserexists(user)
    if err != nil {
        return err
    }
    if !exists {
        return fmt.Errorf("user %s does not exist", user)
    }

    id, err := primitive.ObjectIDFromHex(user)
    if err != nil {
        return fmt.Errorf("invalid user ID: %w", err)
    }

    filter := bson.M{"_id": id}
    update := bson.M{"$push": bson.M{"Posts": model}}
    _, err = Mongocollection.UpdateOne(ctx, filter, update)
    if err != nil {
        return fmt.Errorf("failed to update post: %w", err)
    }

    return nil
}

修复后的模型代码 #

type Credentials struct {
    ID       primitive.ObjectID `json:"_id,omitempty" bson:"_id,omitempty"`
    Email    string             `json:"email" validate:"email, required" bson:"email"` // 修正bson标签
    Password string             `json:"-"`
    Posts    []Post             `json:"posts,omitempty" bson:"posts,omitempty"`
}

type Post struct {
    ID          primitive.ObjectID `json:"id,omitempty" bson:"_id,omitempty"`
    Image       []byte             `json:"image,omitempty" bson:"image"`
    Category    string             `json:"category,omitempty" bson:"category"`
    Description string             `json:"description" bson:"description"`
}

额外优化建议 #

• 禁止在HTTP handler中使用panic，所有错误需通过http.Error返回给客户端，避免服务器崩溃
• 大文件上传建议使用流式处理，替代io.ReadAll一次性读取到内存，防止内存溢出
• 为MongoDB操作设置超时上下文，例如ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second); defer cancel()
• 增加日志记录，将错误信息写入日志系统，便于后续排查问题

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