• 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())
        }
    }
     
  • 相关阅读:
    intellij idea
    this.getClass().getResource(String) 路径问题
    org.hibernate.AssertionFailure: null id 错误
    Hibernate 映射
    关于idea 在创建maven 骨架较慢问题解决
    常用base.css
    form表单样式
    ul li自适应居中导航
    table-cell实现未知宽高图片,文本水平垂直居中在div
    多行文字水平垂直居中在div
  • 原文地址:https://www.cnblogs.com/peteremperor/p/16330248.html
Copyright © 2020-2023  润新知