mirror of
https://gitee.com/johng/gf.git
synced 2024-12-04 13:18:01 +08:00
124 lines
3.8 KiB
Go
124 lines
3.8 KiB
Go
// Copyright 2019 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package ghttp
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/gogf/gf/internal/intlog"
|
|
"github.com/gogf/gf/os/gfile"
|
|
"github.com/gogf/gf/os/gtime"
|
|
"github.com/gogf/gf/util/grand"
|
|
"io"
|
|
"mime/multipart"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// UploadFile wraps the multipart uploading file with more and convenient features.
|
|
type UploadFile struct {
|
|
*multipart.FileHeader
|
|
}
|
|
|
|
// UploadFiles is array type for *UploadFile.
|
|
type UploadFiles []*UploadFile
|
|
|
|
// Save saves the single uploading file to directory path and returns the saved file name.
|
|
//
|
|
// The parameter <dirPath> should be a directory path or it returns error.
|
|
//
|
|
// Note that it will OVERWRITE the target file if there's already a same name file exist.
|
|
func (f *UploadFile) Save(dirPath string, randomlyRename ...bool) (filename string, err error) {
|
|
if f == nil {
|
|
return "", errors.New("file is empty, maybe you retrieve it from invalid field name or form enctype")
|
|
}
|
|
if !gfile.Exists(dirPath) {
|
|
if err = gfile.Mkdir(dirPath); err != nil {
|
|
return
|
|
}
|
|
} else if !gfile.IsDir(dirPath) {
|
|
return "", errors.New(`parameter "dirPath" should be a directory path`)
|
|
}
|
|
|
|
file, err := f.Open()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer file.Close()
|
|
|
|
name := gfile.Basename(f.Filename)
|
|
if len(randomlyRename) > 0 && randomlyRename[0] {
|
|
name = strings.ToLower(strconv.FormatInt(gtime.TimestampNano(), 36) + grand.S(6))
|
|
name = name + gfile.Ext(f.Filename)
|
|
}
|
|
filePath := gfile.Join(dirPath, name)
|
|
newFile, err := gfile.Create(filePath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer newFile.Close()
|
|
intlog.Printf(`save upload file: %s`, filePath)
|
|
if _, err := io.Copy(newFile, file); err != nil {
|
|
return "", err
|
|
}
|
|
return gfile.Basename(filePath), nil
|
|
}
|
|
|
|
// Save saves all uploading files to specified directory path and returns the saved file names.
|
|
//
|
|
// The parameter <dirPath> should be a directory path or it returns error.
|
|
//
|
|
// The parameter <randomlyRename> specifies whether randomly renames all the file names.
|
|
func (fs UploadFiles) Save(dirPath string, randomlyRename ...bool) (filenames []string, err error) {
|
|
if len(fs) == 0 {
|
|
return nil, errors.New("file array is empty, maybe you retrieve it from invalid field name or form enctype")
|
|
}
|
|
for _, f := range fs {
|
|
if filename, err := f.Save(dirPath, randomlyRename...); err != nil {
|
|
return filenames, err
|
|
} else {
|
|
filenames = append(filenames, filename)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// GetUploadFile retrieves and returns the uploading file with specified form name.
|
|
// This function is used for retrieving single uploading file object, which is
|
|
// uploaded using multipart form content type.
|
|
//
|
|
// It returns nil if retrieving failed or no form file with given name posted.
|
|
//
|
|
// Note that the <name> is the file field name of the multipart form from client.
|
|
func (r *Request) GetUploadFile(name string) *UploadFile {
|
|
uploadFiles := r.GetUploadFiles(name)
|
|
if len(uploadFiles) > 0 {
|
|
return uploadFiles[0]
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetUploadFiles retrieves and returns multiple uploading files with specified form name.
|
|
// This function is used for retrieving multiple uploading file objects, which are
|
|
// uploaded using multipart form content type.
|
|
//
|
|
// It returns nil if retrieving failed or no form file with given name posted.
|
|
//
|
|
// Note that the <name> is the file field name of the multipart form from client.
|
|
func (r *Request) GetUploadFiles(name string) UploadFiles {
|
|
multipartFiles := r.GetMultipartFiles(name)
|
|
if len(multipartFiles) > 0 {
|
|
uploadFiles := make(UploadFiles, len(multipartFiles))
|
|
for k, v := range multipartFiles {
|
|
uploadFiles[k] = &UploadFile{
|
|
FileHeader: v,
|
|
}
|
|
}
|
|
return uploadFiles
|
|
}
|
|
return nil
|
|
}
|