xps-server-g5/pkg/system/controller/upload_controller.go

package controller

import (
    uuid "github.com/iris-contrib/go.uuid"
    "github.com/kataras/iris/v12/mvc"
    "io"
    "mime/multipart"
    "os"
    "path/filepath"
    "strings"
    "time"
    "xps/cmd/constant"
    "xps/pkg/system/service"
    "xps/pkg/utils"
)

// client version and secret, it should be from client side
const (
    UploadDir string = "E:/dev-x/xps-server-g5/cmd/main/public/upload/"
    UrlBase   string = "/upload/"
)

type FileData struct {
    FileInfo *multipart.FileHeader
    FileData *multipart.File
}

type UploadController struct {
    Base
    Service service.UploadService
}

func (c *UploadController) BeforeActivation(b mvc.BeforeActivation) {
    b.Handle("POST", "/file", "UploadFile")
    b.Handle("POST", "/files", "UploadFiles")
}

// UploadFile :Single Upload
func (c *UploadController) UploadFile() *JsonResult {
    ocId   := c.GetCurOcId()
    userId := c.GetCurUserId()
    f, fh, err := c.Ctx.FormFile("uploadfile")
    if err != nil {
        return ResultErr(err.Error(), nil)
    }
    defer f.Close()

    oldFileName    := fh.Filename
    targetFileName := c.GenerateNewFileName(oldFileName)
    out, err := os.OpenFile(targetFileName, os.O_CREATE|os.O_RDWR|os.O_APPEND, 0666)
    if err != nil {
        return ResultErr(err.Error(), nil)
    }

    defer out.Close()
    _, err = io.Copy(out, f)
    if err != nil {
        return ResultErr(err.Error(), nil)
    }

    targetFile := service.TargetFile{
        FileName: oldFileName,
        FilePath: targetFileName,
    }

    file, err := c.Service.UploadFile(ocId, userId, constant.DOC_DIR_TEMP, targetFile)
    if err != nil {
        return ResultErr(err.Error(), nil)
    } else {
        return ResultOk("获取成功", file)
    }
}

// UploadFiles :Multiple Upload
func (c *UploadController) UploadFiles() *JsonResult {
    _, files, err := utils.UploadMultiFiles(c.Ctx)
    if err != nil {
        return ResultErr(err.Error(), nil)
    } else {
        return ResultOk("获取成功", files)
    }
}

// CreateDateDir Create date directory
func (c *UploadController) CreateDateDir() (srcPath string) {
    folderName := time.Now().Format("20060102")
    // curPath, _ := os.Getwd()
    src := filepath.Join(UploadDir, folderName)
    _, err := os.Stat(srcPath)
    if os.IsNotExist(err) {
        _ = os.Mkdir(src, os.ModePerm)
        _ = os.Chmod(src, os.ModePerm)
    }

    return src
}

// GenerateNewFileName Generate new file name
func (c *UploadController) GenerateNewFileName(oldFileName string) (srcPath string) {
    u1, _ := uuid.NewV4()
    newFileName := u1.String()
    src := c.CreateDateDir()

    dotIndex := strings.LastIndex(oldFileName, ".")
    if dotIndex != -1 && dotIndex != 0 {
        newFileName += oldFileName[dotIndex:]
    }

    src = filepath.Join(src, newFileName)

    return src
}