mirror of https://github.com/go-sonic/sonic.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"mime/multipart"
|
|
|
|
"github.com/go-sonic/sonic/consts"
|
|
"github.com/go-sonic/sonic/model/dto"
|
|
storageimpl "github.com/go-sonic/sonic/service/storage/impl"
|
|
)
|
|
|
|
type FileStorage interface {
|
|
Upload(ctx context.Context, file *multipart.FileHeader) (*dto.AttachmentDTO, error)
|
|
Delete(ctx context.Context, fileKey string) error
|
|
GetAttachmentType() consts.AttachmentType
|
|
GetFilePath(ctx context.Context, relativePath string) (string, error)
|
|
}
|
|
|
|
type FileStorageComposite interface {
|
|
GetFileStorage(storageType consts.AttachmentType) FileStorage
|
|
}
|
|
type fileStorageComposite struct {
|
|
localStorage *storageimpl.LocalFileStorage
|
|
minio *storageimpl.MinIO
|
|
aliyunOSS *storageimpl.Aliyun
|
|
}
|
|
|
|
func NewFileStorageComposite(localStorage *storageimpl.LocalFileStorage, minio *storageimpl.MinIO, aliyun *storageimpl.Aliyun) FileStorageComposite {
|
|
return &fileStorageComposite{
|
|
localStorage: localStorage,
|
|
minio: minio,
|
|
aliyunOSS: aliyun,
|
|
}
|
|
}
|
|
|
|
func (f *fileStorageComposite) GetFileStorage(storageType consts.AttachmentType) FileStorage {
|
|
switch storageType {
|
|
case consts.AttachmentTypeLocal:
|
|
return f.localStorage
|
|
case consts.AttachmentTypeMinIO:
|
|
return f.minio
|
|
case consts.AttachmentTypeAliOSS:
|
|
return f.aliyunOSS
|
|
default:
|
|
panic("Unsupported file storage")
|
|
}
|
|
}
|