opt: Only create actived upload service

This commit is contained in:
RockYang
2023-09-08 06:46:24 +08:00
parent af9e6f2c46
commit 0a9b325360
3 changed files with 23 additions and 22 deletions

View File

@@ -6,8 +6,7 @@ import (
)
type UploaderManager struct {
active string
uploadServices map[string]Uploader
handler Uploader
}
const Local = "LOCAL"
@@ -15,27 +14,29 @@ const Minio = "MINIO"
const QiNiu = "QINIU"
func NewUploaderManager(config *types.AppConfig) (*UploaderManager, error) {
services := make(map[string]Uploader)
if config.OSS.Minio.AccessKey != "" {
minioService, err := NewMinioService(config)
if err != nil {
return nil, err
}
services[Minio] = minioService
}
if config.OSS.Local.BasePath != "" {
services[Local] = NewLocalStorageService(config)
}
if config.OSS.QiNiu.AccessKey != "" {
services[QiNiu] = NewQiNiuService(config)
}
active := Local
if config.OSS.Active != "" {
active = strings.ToUpper(config.OSS.Active)
}
return &UploaderManager{uploadServices: services, active: active}, nil
var handler Uploader
switch active {
case Local:
handler = NewLocalStorageService(config)
break
case Minio:
service, err := NewMinioService(config)
if err != nil {
return nil, err
}
handler = service
break
case QiNiu:
handler = NewQiNiuService(config)
}
return &UploaderManager{handler: handler}, nil
}
func (m *UploaderManager) GetActiveService() Uploader {
return m.uploadServices[m.active]
func (m *UploaderManager) GetUploadHandler() Uploader {
return m.handler
}