gin框架-上传压缩文件

package Controller

import (
    "archive/zip"
    "fmt"
    "github.com/gin-gonic/gin"
    "go-service/src/Config"
    "io"
    "os"
    "path/filepath"
    "strings"
)

type UploadController struct {
}

var UploadControl = &UploadController{}

func (*UploadController) UploadImage(c *gin.Context) {
    dir := Config.StaticUrl // 配置文件中的静态资源路径
    file, _ := c.FormFile("file")
    filename := RandStr(40) + ".jpg"
    //将文件保存到本地服务器的指定位置
    if err := c.SaveUploadedFile(file, dir+filename); err != nil {
        ResponseError(c, "文件上传失败")
    } else {
        ResponseSuccess(c, filename)
    }
}
// 某些文件的执行依赖于其他文件，比如3D模型，所以需要通过ZIP的方式上传，同时要指定执行文件的名称
// 先将ZIP保存，然后解压，最后删除保存的ZIP
func (*UploadController) UploadZipFile(c *gin.Context) {
    dir := Config.StaticUrl
    file, _ := c.FormFile("file")
    name := c.PostForm("name") // 执行文件名称
    filename := RandStr(40)
    //将文件保存到本地服务器的指定位置
    if err := c.SaveUploadedFile(file, dir+filename+".zip"); err != nil {
        ResponseError(c, "文件上传失败")
    } else {
        det, err := Unzip(filename+".zip", filename, name) // 解压
        if err != nil {
            ResponseError(c, "文件上传失败")
            return
        }
        UploadControl.RemoveFile(filename + ".zip") // 删除zip
        if det == "" {
            ResponseError(c, "文件名称错误")
        } else {
            ResponseSuccess(c, det)
        }
    }
}
// 解压
func Unzip(zipFile string, destDir string, exec string) (string, error) {
    dir := Config.StaticUrl
    dst := ""
    zipReader, err := zip.OpenReader(dir + zipFile)
    if err != nil {
        return dst, err
    }
    defer zipReader.Close()
    for _, f := range zipReader.File {
        fpath := filepath.Join(dir+destDir, f.Name)
        if f.FileInfo().IsDir() {
            os.MkdirAll(fpath, os.ModePerm)
        } else {
            fmt.Println(f.Name, exec)
            if strings.Contains(f.Name, exec) {
                dst = filepath.Join(destDir, f.Name)
            }
            if err = os.MkdirAll(filepath.Dir(fpath), os.ModePerm); err != nil {
                return dst, err
            }

            inFile, err := f.Open()
            if err != nil {
                return dst, err
            }

            outFile, err := os.OpenFile(fpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
            if err != nil {
                return dst, err
            }

            _, err = io.Copy(outFile, inFile)
            if err != nil {
                return dst, err
            }
            outFile.Close()
            inFile.Close()
        }
    }
    return dst, nil
}

func (*UploadController) DownloadFile(c *gin.Context) {
    dir := Config.StaticUrl
    name := c.Query("f")
    arr := strings.Split(name, "/") // 路径默认斜杠，为了兼容Linux
    for _, s := range arr {
        dir = filepath.Join(dir, s)
    }
    c.File(dir)
}

func (*UploadController) RemoveFile(filename string) {
    dir := Config.StaticUrl
    err := os.RemoveAll(filepath.Join(dir, filename))
    if err != nil {
        fmt.Println(err.Error())
    }
}