基于gin框架的文件上传(逐行解析)

记录一下使用gin框架完成一个文件上传的功能，一下是实现该功能的代码，适合小白，代码都有逐行解释！

app.go:

package routerimport ("chat/service""github.com/gin-gonic/gin"
)func Router() *gin.Engine {r := gin.Default()//上传文件，使用post请求r.POST("/attach/upload", service.Upload)return r
}

attach.go:

package serviceimport ("chat/utils""fmt""github.com/gin-gonic/gin""io""math/rand""net/http""os""strings""time"
)func Upload(c *gin.Context) {UploadLocal(c)
}// 上传文件到本地
func UploadLocal(c *gin.Context) {w := c.Writer //获取响应的gin上下文写入器req := c.Request//获取当前请求对象//从请求中通过FormFile方方法获取指定名称("file")的文件//srcFile为获取到的源文件//head为文件的头部信息//err为可能出现的错误srcFile, head, err := req.FormFile("file") if err != nil {//使用utils.RespFail()函数响应写入失败信息和错误内容utils.RespFail(w, err.Error())}
//定义默认的文件后缀为".png"suffix := ".png"//从head中获取文件信息head有以下常用变量//head.Header 文件的头部信息，包含一些与文件相关的元数据，比如文件的类型、大小//head.Size 文件的大小//head.Filename 文件的名字ofilName := head.Filename //定义一个字符串切片tem，使用strings。Split()将文件名ofilName按照(.)进行分割再装入这个切片当中，tem := strings.Split(ofilName, ".")//如果分割后的结果长度大于1，表示有后缀if len(tem) > 1 {// 将后缀更新为分割后的最后一部分（该切片的最后一个元素）suffix = "." + tem[len(tem)-1]}// 根据当前时间的 Unix 时间戳、随机生成的 32 位整数和后缀生成文件名fileName := fmt.Sprintf("%d%04d%s", time.Now().Unix(), rand.Int31(), suffix)//使用os.Create()在指定路径("./files/"+fileName)创建新的文件dstFile, err := os.Create("./files/" + fileName)if err != nil {//使用utils.RespFail()函数响应写入失败信息和错误内容utils.RespFail(w, err.Error())}// 使用 io.Copy 方法将源文件的内容拷贝到目标文件中// err 为可能出现的拷贝过程中的错误_, err = io.Copy(dstFile, srcFile)if err != nil {//使用utils.RespFail()函数响应写入失败信息和错误内容utils.RespFail(w, err.Error())}//构建文件的完整的URLurl := "./files/" + fileName// 使用 utils.RespOK 函数向响应写入成功信息、文件的 URL 以及成功消息utils.RespOK(w, url, "发送成功")
}

resp.go:主要用处理请求响应的各种状态

package utilsimport (// 导入相关包"encoding/json""fmt""net/http"
)type H struct {// 定义结构体 H，包含代码、消息、数据、行数和总数等字段Code  intMsg   stringData  interface{}Rows  interface{}Total interface{}
}// Resp 函数，用于响应数据
func Resp(w http.ResponseWriter, code int, data interface{}, msg string) {// 设置响应头的内容类型为 JSONw.Header().Set("Content-Type", "application/json")// 设置响应状态码为 200（OK）w.WriteHeader(http.StatusOK)h := H{// 初始化结构体 H 的字段Code: code,Data: data,Msg:  msg,}// 对结构体 H 进行 JSON 序列化操作ret, err := json.Marshal(h)if err!= nil {// 处理序列化错误fmt.Println(err)}w.Write(ret)
}// RespList 函数，用于响应列表数据
func RespList(w http.ResponseWriter, code int, data interface{}, total interface{}) {w.Header().Set("Content-Type", "application/json")w.WriteHeader(http.StatusOK)h := H{// 初始化结构体 H 的字段Code:  code,Rows:  data,Total: total,}ret, err := json.Marshal(h)if err!= nil {// 处理序列化错误fmt.Println(err)}w.Write(ret)
}// RespFail 函数，用于响应失败信息
func RespFail(w http.ResponseWriter, msg string) {// 调用 Resp 函数，设置代码为-1，表示失败Resp(w, -1, nil, msg)
}// RespOK 函数，用于响应成功信息
func RespOK(w http.ResponseWriter, data interface{}, msg string) {// 调用 Resp 函数，设置代码为 0，表示成功Resp(w, 0, data, msg)
}// RespOKList 函数，用于响应成功的列表信息
func RespOKList(w http.ResponseWriter, data interface{}, total int) {// 调用 RespList 函数，设置代码为 0，表示成功RespList(w, 0, data, total)
}

main.go:

package mainimport "chat/router"func main() {r := router.Router()r.Run(":8081")
}

写完了：测试一下

使用postman提交一个文件参数

查看是否上传成功

可以看到文件上传成功了！