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())
	}
}