GolangHTTP文件上传与下载示例

Request

ResponseWriter

package main  import (     "fmt"     "io"     "net/http"     "os"     "path/filepath"     "time" )  // uploadHandler 处理文件上传请求 func uploadHandler(w http.ResponseWriter, r *http.Request) {     if r.Method != "POST" {         http.Error(w, "只支持POST方法", http.StatusMethodNotAllowed)         return     }      // 限制上传文件大小，例如10MB     r.ParseMultipartForm(10 << 20) // 10MB     if r.MultipartForm == nil {         http.Error(w, "无法解析多部分表单", http.StatusBadRequest)         return     }      // 获取文件     file, handler, err := r.FormFile("uploadFile") // "uploadFile" 是表单中文件字段的名称     if err != nil {         fmt.printf("获取文件失败: %vn", err)         http.Error(w, "获取上传文件失败", http.StatusBadRequest)         return     }     defer file.Close()      // 检查文件类型（简单示例，实际应用中应更严格）     // 例如，只允许图片或pdf     ext := filepath.Ext(handler.Filename)     if ext != ".jpg" && ext != ".png" && ext != ".pdf" {         http.Error(w, "不支持的文件类型，只允许JPG, PNG, PDF", http.StatusBadRequest)         return     }      // 创建目标文件     // 为了避免文件名冲突，可以加上时间戳或UUID     newFileName := fmt.Sprintf("%d%s", time.Now().UnixNano(), ext)     dstPath := filepath.Join("./uploads", newFileName) // 假设存在一个名为 'uploads' 的目录     dst, err := os.Create(dstPath)     if err != nil {         fmt.Printf("创建目标文件失败: %vn", err)         http.Error(w, "服务器无法创建文件", http.StatusinternalServerError)         return     }     defer dst.Close()      // 将上传文件内容拷贝到目标文件     if _, err := io.Copy(dst, file); err != nil {         fmt.Printf("拷贝文件内容失败: %vn", err)         http.Error(w, "保存文件失败", http.StatusInternalServerError)         return     }      fmt.Fprintf(w, "文件 '%s' 上传成功，保存为 '%s'！", handler.Filename, newFileName) }  // downloadHandler 处理文件下载请求 func downloadHandler(w http.ResponseWriter, r *http.Request) {     // 从URL路径中获取文件名，例如 /download/myimage.jpg     fileName := filepath.Base(r.URL.Path) // 只取文件名部分，防止路径遍历攻击     filePath := filepath.Join("./uploads", fileName)      // 检查文件是否存在     _, err := os.Stat(filePath)     if os.IsNotExist(err) {         http.Error(w, "文件不存在", http.StatusNotFound)         return     } else if err != nil {         http.Error(w, "无法访问文件", http.StatusInternalServerError)         return     }      // 设置响应头，告诉浏览器这是一个下载文件     w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename="%s"", fileName))     w.Header().Set("Content-Type", "application/octet-stream") // 常见的二进制流类型     // 也可以根据文件扩展名设置更具体的Content-Type，例如 image/jpeg, application/pdf     // mimeType := mime.TypeByExtension(filepath.Ext(fileName))     // if mimeType != "" {     //  w.Header().Set("Content-Type", mimeType)     // }      // 打开并传输文件     http.ServeFile(w, r, filePath) // net/http 包提供了便捷的 ServeFile 函数 }  func main() {     // 确保上传目录存在     if _, err := os.Stat("./uploads"); os.IsNotExist(err) {         os.Mkdir("./uploads", 0755)     }      http.HandleFunc("/upload", uploadHandler)     http.HandleFunc("/download/", downloadHandler) // 注意这里的斜杠，表示匹配 /download/ 后的所有路径     fmt.Println("服务器正在监听 :8080...")     err := http.ListenAndServe(":8080", nil)     if err != nil {         fmt.Printf("服务器启动失败: %vn", err)     } }

curl -X POST -F "uploadFile=@/path/to/your/image.jpg" http://localhost:8080/upload

/path/to/your/image.jpg

curl -O http://localhost:8080/download/1678888888888888888.jpg # 替换成你上传后的文件名

-O

io.CopyBuffer

Article Forge

行业文案AI写作软件，可自动为特定主题或行业生成内容

22

查看详情

defer

http.StatusBadRequest

http.StatusInternalServerError

http.StatusNotFound

r.FormFile

os.Create

defer file.Close()

"Internal Server Error"

log.Printf

logrus

zap

fmt.Printf("创建目标文件失败: %vn", err)

大家都在看：