llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-07-02 11:20:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve UploadFile() API.

Browse Source
This commit is contained in:
Yang Luo
2021-08-14 15:58:12 +08:00
parent b7e0a4fe4e
commit 004282f970
5 changed files with 48 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
object/storage.go
View File

@ -17,15 +17,13 @@ package object
import (
"bytes"
"fmt"
"io"
"mime/multipart"
"strings"
"github.com/casbin/casdoor/storage"
"github.com/casbin/casdoor/util"
)
func UploadFile(provider *Provider, folder string, subFolder string, file multipart.File, suffix string) (string, error) {
func UploadFile(provider *Provider, fullFilePath string, fileBuffer *bytes.Buffer) (string, error) {
storageProvider := storage.GetStorageProvider(provider.Type, provider.ClientId, provider.ClientSecret, provider.RegionId, provider.Bucket, provider.Endpoint)
if storageProvider == nil {
return "", fmt.Errorf("the provider type: %s is not supported", provider.Type)
@ -36,12 +34,8 @@ func UploadFile(provider *Provider, folder string, subFolder string, file multip
UpdateProvider(provider.GetId(), provider)
}
path := fmt.Sprintf("%s/%s.%s", util.UrlJoin(util.GetUrlPath(provider.Domain), folder), subFolder, suffix)
fileBuf := bytes.NewBuffer(nil)
if _, err := io.Copy(fileBuf, file); err != nil {
return "", err
}
_, err := storageProvider.Put(path, fileBuf)
path := util.UrlJoin(util.GetUrlPath(provider.Domain), fullFilePath)
_, err := storageProvider.Put(path, fileBuffer)
if err != nil {
return "", err
}