/
OssService.go
36 lines (29 loc) · 932 Bytes
/
OssService.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
package service
import (
"io"
"sync"
logger "github.com/sirupsen/logrus"
"github.com/yockii/giserver-express/internal/model"
"github.com/yockii/giserver-express/pkg/aliyun"
"github.com/yockii/giserver-express/pkg/database"
)
var OssService = &ossService{ossClients: make(map[database.Int64]*aliyun.OssClient)}
type ossService struct {
ossClients map[database.Int64]*aliyun.OssClient
lock sync.Mutex
}
func (s *ossService) StreamFromStore(store *model.Store, objectKey string) (reader io.Reader, err error) {
client, ok := s.ossClients[store.Id]
if !ok {
s.lock.Lock()
defer s.lock.Unlock()
logger.Debugf("store info: %+v", store)
client, err = aliyun.NewOssClient(store.Endpoint, store.BucketName, store.AccessKeyId, store.AccessKeySecret, store.UseCname == 1)
if err != nil {
logger.Errorln(err)
return nil, err
}
s.ossClients[store.Id] = client
}
return client.ReadStream(objectKey)
}