Rename to UploadFile().

This commit is contained in:
Yang Luo
2021-08-08 15:56:36 +08:00
parent c55fa4f452
commit a5783598ff
2 changed files with 4 additions and 4 deletions

View File

@ -256,7 +256,7 @@ func (c *ApiController) UploadAvatar() {
} }
dist, _ := base64.StdEncoding.DecodeString(avatarBase64[index+1:]) dist, _ := base64.StdEncoding.DecodeString(avatarBase64[index+1:])
fileUrl, err := object.UploadAvatar(provider, user.GetId(), dist) fileUrl, err := object.UploadFile(provider, "avatar", user.GetId(), dist)
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseError(err.Error())
return return

View File

@ -23,7 +23,7 @@ import (
"github.com/casbin/casdoor/util" "github.com/casbin/casdoor/util"
) )
func UploadAvatar(provider *Provider, username string, avatar []byte) (string, error) { func UploadFile(provider *Provider, folder string, subFolder string, fileBytes []byte) (string, error) {
storageProvider := storage.GetStorageProvider(provider.Type, provider.ClientId, provider.ClientSecret, provider.RegionId, provider.Bucket, provider.Endpoint) storageProvider := storage.GetStorageProvider(provider.Type, provider.ClientId, provider.ClientSecret, provider.RegionId, provider.Bucket, provider.Endpoint)
if storageProvider == nil { if storageProvider == nil {
return "", fmt.Errorf("the provider type: %s is not supported", provider.Type) return "", fmt.Errorf("the provider type: %s is not supported", provider.Type)
@ -34,8 +34,8 @@ func UploadAvatar(provider *Provider, username string, avatar []byte) (string, e
UpdateProvider(provider.GetId(), provider) UpdateProvider(provider.GetId(), provider)
} }
path := fmt.Sprintf("%s/%s.png", util.UrlJoin(util.GetUrlPath(provider.Domain), "/avatar"), username) path := fmt.Sprintf("%s/%s.png", util.UrlJoin(util.GetUrlPath(provider.Domain), folder), subFolder)
_, err := storageProvider.Put(path, bytes.NewReader(avatar)) _, err := storageProvider.Put(path, bytes.NewReader(fileBytes))
if err != nil { if err != nil {
return "", err return "", err
} }