design_quantity/web/api/upload_api.go

/*
 * @description:
 * @Author: CP
 * @Date: 2021-03-31 15:06:21
 * @FilePath: \design_quantity\web\api\upload_api.go
 */
package api

import (
	"bytes"
	"fmt"
	"io"
	"log"
	"os"
	"time"

	"github.com/kataras/iris/v12"
	"go.mod/conf"
	"go.mod/services"
	"go.mod/web/utils"
	"go.mod/web/viewmodels"
)

type UploadApi struct {
	//框架-web应用上下文环境
	Ctx iris.Context
	// // 需要用的service
	ServiceContract services.ContractService
}

// Upload ： 处理文件上传
func (c *UploadApi) Post() {
	errCode := 0
	defer func() {
		if errCode < 0 {
			c.Ctx.JSON(iris.Map{"code": -1, "msg": "上传失败"})

		} else {
			c.Ctx.JSON(iris.Map{"code": 0, "msg": "上传成功"})
		}
		return
	}()

	file, head, err := c.Ctx.FormFile("file")
	if err != nil {
		log.Printf("Failed to get form data, err:%s\n", err.Error())
		errCode = -1
		return
	}
	defer file.Close()

	buf := bytes.NewBuffer(nil)
	if _, err := io.Copy(buf, file); err != nil {
		log.Printf("Failed to get file data, err:%s\n", err.Error())
		errCode = -2
		return
	}

	fmt.Println(head.Header)

	//  文件元信息-todo
	// FileName := head.Filename
	FileSize := int64(len(buf.Bytes()))
	UploadAt := time.Now().Format("2006-01-02 15:04:05")

	Location := conf.MergeLocalRootDir + UploadAt + head.Filename // 存储地址
	newFile, err := os.Create(Location)
	if err != nil {
		log.Printf("Failed to create file, err:%s\n", err.Error())
		errCode = -3
		return
	}
	defer newFile.Close()

	nByte, err := newFile.Write(buf.Bytes())
	if int64(nByte) != FileSize || err != nil {
		log.Printf("Failed to save data into file, writtenSize:%d, err:%s\n", nByte, err.Error())
		errCode = -4
		return
	}

	treeVM := &viewmodels.TreeSectionContract{}
	err = c.Ctx.ReadForm(treeVM)
	if err != nil {
		log.Println("folder-ValidRule-ReadForm转换异常, error=", err)
		errCode = -6
		return
	}

	id, err := utils.GetDecryptId(treeVM.Id)
	if err != nil {
		errCode = -6
		return
	}

	err = c.ServiceContract.SaveUpload(Location, head.Filename, id)
	if err != nil {
		errCode = -6
		return
	}

}