golang-文件上传

爱吃豆包 · · 2206 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

package handler

import (
    "encoding/json"
    "io"
    "io/ioutil"
    "log"
    "net/http"
    "os"
)

// UploadHandler 上传接口
func UploadHandler(w http.ResponseWriter, r *http.Request) {
    if r.Method == "GET" {
        // 返回上传html页面
        data, err := ioutil.ReadFile("./static/view/index.html")
        if err != nil {
            log.Printf("UploadHandler -> {%s}", err)
            _, _ = io.WriteString(w, "服务器错误")
            return
        }
         _, _ = io.WriteString(w, string(data))
    } else if r.Method == "POST" {
        // 接收文件流
        file, fileHeader, err := r.FormFile("file")
        if err != nil {
            log.Printf("UploadHandler: 上传文件出错 -> {%s}", err)
            _, _ = io.WriteString(w, "上传文件出错！")
            return
        }
        defer file.Close()

        headerByte, _ := json.Marshal(fileHeader.Header)
        log.Printf("当前文件：Filename - >{%s}, Size -> {%v}, FileHeader -> {%s}", fileHeader.Filename, fileHeader.Size, string(headerByte))

        newFile, err := os.Create("/temp/" + fileHeader.Filename)
        if err != nil {
            log.Printf("UploadHandler: 创建文件失败！")
            _, _ = io.WriteString(w, "服务器错误！")
            return
        }
        defer newFile.Close()

        // 复制文件到目标目录
        _, errCopy := io.Copy(newFile, file)
        if errCopy != nil {
            log.Printf("UploadHandler: 文件复制失败！ -> {%s}", err)
            _, _ = io.WriteString(w, "服务器错误！")
            return
        }

        // 成功响应

        // 重定向到这个请求
        http.Redirect(w, r, "/file/upload/suc", http.StatusFound)

    }
}

// UploadSuccessHandler 响应
func UploadSuccessHandler(w http.ResponseWriter, r *http.Request) {
    _, _ = io.WriteString(w, "Upload finished！")
}

测试主函数

package main

import (
    "filestore-server/handler"
    "fmt"
    "net/http"
)

func main() {

    //  http.HandleFunc 指定路由规则

    // 上传文件
    http.HandleFunc("/file/upload", handler.UploadHandler)
    // 上传成功响应，/file/upload 请求处理成功后，会重定向到这个该请求
    http.HandleFunc("/file/upload/suc", handler.UploadSuccessHandler)


    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        fmt.Printf("启动服务器失败: %s", err.Error())
    }
}

有疑问加站长微信联系（非本文作者）