91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
package fs
|
|
|
|
import (
|
|
"ZhenTuLocalPassiveAdapter/config"
|
|
"ZhenTuLocalPassiveAdapter/dto"
|
|
"ZhenTuLocalPassiveAdapter/util"
|
|
"context"
|
|
"fmt"
|
|
"github.com/aws/aws-sdk-go-v2/credentials"
|
|
"path"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
)
|
|
|
|
type S3Adapter struct {
|
|
StorageConfig config.StorageConfig
|
|
s3Client *s3.Client
|
|
}
|
|
|
|
func (s *S3Adapter) getClient() (*s3.Client, error) {
|
|
if s.s3Client == nil {
|
|
const defaultRegion = "us-east-1"
|
|
resolver := aws.EndpointResolverFunc(func(service, region string) (aws.Endpoint, error) {
|
|
return aws.Endpoint{
|
|
PartitionID: "aws",
|
|
URL: s.StorageConfig.S3.Endpoint, // or where ever you ran minio
|
|
SigningRegion: defaultRegion,
|
|
HostnameImmutable: true,
|
|
}, nil
|
|
})
|
|
creds := credentials.NewStaticCredentialsProvider(s.StorageConfig.S3.AkId, s.StorageConfig.S3.AkSec, "")
|
|
cfg := aws.Config{
|
|
Credentials: creds,
|
|
Region: defaultRegion,
|
|
EndpointResolver: resolver,
|
|
}
|
|
s.s3Client = s3.NewFromConfig(cfg)
|
|
}
|
|
return s.s3Client, nil
|
|
}
|
|
|
|
func (s *S3Adapter) GetFileList(dirPath string, relDt time.Time) ([]dto.File, error) {
|
|
if s.StorageConfig.S3.Bucket == "" {
|
|
return nil, fmt.Errorf("未配置S3存储桶")
|
|
}
|
|
|
|
listObjectsInput := &s3.ListObjectsV2Input{
|
|
Bucket: aws.String(s.StorageConfig.S3.Bucket),
|
|
Prefix: aws.String(path.Join(s.StorageConfig.S3.Prefix, dirPath)),
|
|
}
|
|
|
|
client, err := s.getClient()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := client.ListObjectsV2(context.TODO(), listObjectsInput)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var fileList []dto.File
|
|
for _, object := range result.Contents {
|
|
key := *object.Key
|
|
if !util.IsVideoFile(path.Base(key)) {
|
|
continue
|
|
}
|
|
startTime, stopTime, err := util.ParseStartStopTime(path.Base(key), relDt)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
if startTime.Equal(stopTime) || stopTime.IsZero() {
|
|
stopTime = stopTime.Add(time.Second * time.Duration(config.Config.Record.Duration))
|
|
}
|
|
fileList = append(fileList, dto.File{
|
|
BasePath: s.StorageConfig.S3.Bucket,
|
|
Name: path.Base(key),
|
|
Path: path.Dir(key),
|
|
Url: key,
|
|
StartTime: startTime,
|
|
EndTime: stopTime,
|
|
})
|
|
}
|
|
sort.Slice(fileList, func(i, j int) bool {
|
|
return fileList[i].StartTime.Before(fileList[j].StartTime)
|
|
})
|
|
return fileList, nil
|
|
}
|