gin 图片上传到本地或者oss

路由层

func registerCommonRouter(v1 *gin.RouterGroup) {

	up := v1.Group("upload")
	{
		up.POST("/upLoadFile", commons.FileUpLoad) //单文件上传
		up.POST("/upLoadMultipartFile", commons.FileUpLoadMultipart) //单文件上传
	}
}

控制器

//单文件上传
/**
@ 单文件上传
@ name="file" 文件名
@ multipart/form-data;
*/
func FileUpLoad(c *gin.Context) {
	err,host, pathRelative := tools.UplaodFile(c)
	if err != nil {
		app.Error(c, 400, err, "")
		return
	}
	app.OK(c, gin.H{
		"host": host,
		"uri":  pathRelative,
		"url":  path.Join(host,pathRelative),
	}, "success")
}


/**
@ 多文件上传
@ name="files[]" 文件名
@ multipart/form-data;
*/
func FileUpLoadMultipart(c *gin.Context) {
	resFileInfo ,err:= tools.UpLoadMultipartFile(c)
	if err != nil {
		app.Error(c, 400, err, "")
		return
	}
	app.OK(c, resFileInfo, "ok")
}

tools里

package tools

import (
	"errors"
	"fmt"
	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"github.com/gin-gonic/gin"
	"mime/multipart"
	"os"
	"path"
	"reflect"
	"strings"
	"time"
)

// @ author haima
type UploadConf struct {
	UpdateVilidateExtString string           `json:"updateVilidateExtString"` //从配置里获取,空为不限制  示例 "jpg,jpeg,png,gif,bmp"
	FileMaxSize             int64            `json:"fileMaxSize"`             //单位Mb 0:不限制大小 / 5
	Drive                   string           `json:"drive"`
	UploadConfLocal         *UploadConfLocal `json:"uploadConfLocal"`
	UploadConfOss           *UploadConfOss   `json:"uploadConfOss"`
	UploadConfQiniu         *UploadConfQiniu `json:"uploadConfQiniu"`
}

// 本地配置
type UploadConfLocal struct {
	Host string `json:"updateVilidateExtString"` //根域名
}

// oss配置
// 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录 https://ram.console.aliyun.com 创建RAM账号。
type UploadConfOss struct {
	Host            string `json:"host"`     //根域名
	Endpoint        string `json:"endpoint"` //根域名
	AccessKeyId     string `json:"accessKeyId"`
	AccessKeySecret string `json:"accessKeySecret"`
	BucketName      string `json:"bucketName"`
}

type UploadConfQiniu struct {
	AccessKey string `json:"accessKey"`
	SecretKey string `json:"secretKey"`
	Bucket    string `json:"bucket"`
	ImgPath   string `json:"imgPath"`
}

func InitUploadConf() *UploadConf {
	return &UploadConf{
		UpdateVilidateExtString: "",    //从配置里获取,空为不限制  示例 "jpg,jpeg,png,gif,bmp"
		FileMaxSize:             5,     //单位Mb 0:不限制大小 / 5
		Drive:                   "local", // local / oss / qiniu
		UploadConfLocal: &UploadConfLocal{
			Host: "http://localhost:8001", //根域名
		},
		//以下oss账号仅限测试。非法操作者。请自重 感谢 壹cup清茶 贡献的测试账号
		UploadConfOss: &UploadConfOss{
			//公司的账号
			Endpoint:        "http://oss-cn-beijing.aliyuncs.com",            //阿里云请求地址 定值,写死
			Host:            "https://test.oss-cn-beijing.aliyuncs.com", //返回的自己的阿里云域名
			AccessKeyId:     "LTAI4FbzKpo9UcoQeXWqVabc",
			AccessKeySecret: "n6xiRIUZhn6dZsnsXYFtA2cZ9jFabc",
			BucketName:      "test",
		},
		//请自行七牛申请对应的 公钥 私钥 bucket 和 域名地址
		UploadConfQiniu: &UploadConfQiniu{
			AccessKey: "",
			SecretKey: "",
			Bucket:    "",
			ImgPath:   "",
		},
	}
}

// 初使化配置
var UploadConfig = new(UploadConf)

func init() {
	UploadConfig = InitUploadConf()
}

//单个文件 上传文件到本地目录里
func UplaodFile(c *gin.Context) (err error, host, pathRelative string) {
	// 单个文件
	file, err := c.FormFile("file")
	if err != nil {
		err = errors.New(fmt.Sprintf("获取数据失败，%v", err))
		return
	}
	return actionUplaodFile(c, file)
}

//多图片返回参
type ResFiles struct {
	HostListStr string     `json:"hostListStr"`
	UriListStr  string     `json:"uriListStr"`
	UrlListStr  string     `json:"urlListStr"`
	FileList    []FileItem `json:"fileList"`
}
type FileItem struct {
	Host string `json:"host"`
	Uri  string `json:"uri"`
	Url  string `json:"url"`
}

//多文件上传
func UpLoadMultipartFile(c *gin.Context) (resFiles ResFiles,err error) {
	form, _ := c.MultipartForm()
	files := form.File["files[]"]
	if len(files) < 1{
		err = errors.New("文件不能为空")
		return
	}
	for _, file := range files {
		err, host, pathRelative := actionUplaodFile(c, file)
		if err != nil {
			continue
		}
		var rItem FileItem
		rItem.Host = host
		rItem.Uri = pathRelative
		rItem.Url = path.Join(host , pathRelative)
		resFiles.FileList = append(resFiles.FileList, rItem)
		if resFiles.HostListStr == "" {
			resFiles.HostListStr = rItem.Host
			resFiles.UriListStr = rItem.Uri
			resFiles.UrlListStr = rItem.Url
		} else {
			resFiles.HostListStr += fmt.Sprintf(",%s", rItem.Host)
			resFiles.UriListStr += fmt.Sprintf(",%s", rItem.Uri)
			resFiles.UrlListStr += fmt.Sprintf(",%s", rItem.Url)
		}
	}
	return resFiles ,nil
}

/**
@ 执行单个文件 上传文件到本地目录里
@ 调用方法 上传图片流,文件名为file   调用tools.UplaodFileToLocal(c)
@ return host 根域名
@ return path 上传后的相对文件路径
@ author haima
*/
func actionUplaodFile(c *gin.Context, file *multipart.FileHeader) (err error, host, pathRelative string) {

	// 判断上传文件的大小
	if UploadConfig.FileMaxSize != 0 {
		fsize := file.Size //上传文件的大小
		if fsize > UploadConfig.FileMaxSize*1024*1024 {
			err = errors.New(fmt.Sprintf("只能上传小于 %dMb 的文件 ", UploadConfig.FileMaxSize))

			return
		}
	}

	//获取上传文件后缀
	extString := strings.ToUpper(Ext(file.Filename))
	if extString == "" {
		err = errors.New(fmt.Sprintf("上传失败，文件类型不支持，只能上传 %s 格式的。", UploadConfig.UpdateVilidateExtString))
		return
	}

	//验证文件类型
	if len(UploadConfig.UpdateVilidateExtString) > 0 {
		VilidateExtStrSliceTmp := strings.Split(strings.ToUpper(UploadConfig.UpdateVilidateExtString), ",")
		VilidateExtStrSlice := make([]string, 0)
		for _, v := range VilidateExtStrSliceTmp {
			VilidateExtStrSlice = append(VilidateExtStrSlice, fmt.Sprintf(".%s", v))
		}

		//验证文件类型
		//extString .JPG
		//VilidateExtStrSlice  []string{".JPG",".JPEG",".PNG",".GIF",".BMP"}
		if !ContainArray(extString, VilidateExtStrSlice) {
			err = errors.New(fmt.Sprintf("上传失败，文件类型不支持，只能上传 %s 格式的。", UploadConfig.UpdateVilidateExtString))
			return
		}
	}
	//上传的文件名
	file.Filename = fmt.Sprintf("%s%s", time.Now().Format("20060102150405"), file.Filename) // 文件名格式 自己可以改 建议保证唯一性 20060102150405test.xlsx
	switch UploadConfig.Drive {
	case "local":
		return uploadfileToLocal(c, file, file.Filename)
	case "oss":
		return uploadfileToOss(c, file, file.Filename)
	case "qiniu":
		return nil, UploadConfig.UploadConfQiniu.ImgPath, "qiniu 开发中...."
	default:
		err = errors.New("只支持上传到本地,oss,七牛")
		return
	}
}

func uploadfileToLocal(c *gin.Context, file *multipart.FileHeader, filename string) (err error, host, pathRelative string) {
	host = UploadConfig.UploadConfLocal.Host
	filepath := path.Join("static/uploadfile",time.Now().Format("20060102"))
	//上传到的路径
	pathRelative = path.Join(filepath, filename) //路径+文件名上传
	//如果没有filepath文件目录就创建一个
	if _, err := os.Stat(filepath); err != nil {
		if !os.IsExist(err) {
			os.MkdirAll(filepath, os.ModePerm)
		}
	}
	// 上传文件到指定的目录
	err = c.SaveUploadedFile(file, pathRelative)
	if err != nil {
		err = errors.New(fmt.Sprintf("上传失败，%v", err))
		return
	}
	return
}

func uploadfileToOss(c *gin.Context, file *multipart.FileHeader, filename string) (err error, host, ossPathFileName string) {
	host = UploadConfig.UploadConfOss.Host
	err, _, localPathFileName := uploadfileToLocal(c, file, filename)
	if err != nil {
		err = errors.New(fmt.Sprintf("上传失败，%v", err))
	}

	ossPath := path.Join("upload" ,time.Now().Format("20060102"))
	ossPathFileName = path.Join(ossPath , file.Filename)
	// 创建OSSClient实例。
	client, err := oss.New(UploadConfig.UploadConfOss.Endpoint, UploadConfig.UploadConfOss.AccessKeyId, UploadConfig.UploadConfOss.AccessKeySecret)
	if err != nil {
		os.Remove(localPathFileName)
		err = errors.New(fmt.Sprintf("文件上传服务器失败. err:%s", err.Error()))
		return
	}
	// 获取存储空间。
	bucket, err := client.Bucket(UploadConfig.UploadConfOss.BucketName)
	if err != nil {
		os.Remove(localPathFileName)
		err = errors.New(fmt.Sprintf("文件上传云端失败. err:%s", err.Error()))
		return
	}
	// 上传文件。
	err = bucket.PutObjectFromFile(ossPathFileName, localPathFileName)
	if err != nil {
		os.Remove(localPathFileName)
		err = errors.New(fmt.Sprintf("文件上传云端失败. err:%s", err.Error()))
		return
	}
	err = os.Remove(localPathFileName)  //如果本地不想删除,可以注释了
	if err != nil {
		fmt.Println(err)
	}
	return
}

//获取文件的扩展名
func Ext(path string) string {
	for i := len(path) - 1; i >= 0 && path[i] != '/'; i-- {
		if path[i] == '.' {
			return path[i:]
		}
	}
	return ""
}

//Contain 判断obj是否在target中，target支持的类型array,slice,map   false:不在 true:在
func ContainArray(obj interface{}, target interface{}) bool {
	targetValue := reflect.ValueOf(target)
	switch reflect.TypeOf(target).Kind() {
	case reflect.Slice, reflect.Array:
		for i := 0; i < targetValue.Len(); i++ {
			if targetValue.Index(i).Interface() == obj {
				return true
			}
		}
	case reflect.Map:
		if targetValue.MapIndex(reflect.ValueOf(obj)).IsValid() {
			return true
		}
	}

	return false
}