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())
}
}
有疑问加站长微信联系(非本文作者)