mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-11-10 03:03:43 +08:00
opt: Only create actived upload service
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user