67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package mediaStorage
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/minio/minio-go/v7"
|
|
log "github.com/sirupsen/logrus"
|
|
"io"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Service struct {
|
|
client *minio.Client
|
|
domain string
|
|
endpoint string
|
|
}
|
|
|
|
func newService(client *minio.Client, domain, endpoint string) *Service {
|
|
return &Service{
|
|
client: client,
|
|
domain: domain,
|
|
}
|
|
}
|
|
|
|
func (s *Service) CheckBucketExists(bucketName string) (bool, error) {
|
|
ctx := context.Background()
|
|
exists, err := s.client.BucketExists(ctx, bucketName)
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Media storage | Failed to check bucket existence")
|
|
return exists, err
|
|
}
|
|
log.Infof("Media storage | Bucket %s exists", bucketName)
|
|
return exists, nil
|
|
}
|
|
|
|
func (s *Service) Upload(ctx context.Context, bucket, object string, reader io.Reader, size int64) error {
|
|
_, err := s.client.PutObject(ctx, bucket, object, reader, size, minio.PutObjectOptions{ContentType: "image/jpeg"})
|
|
return err
|
|
}
|
|
|
|
func (s *Service) Get(ctx context.Context, bucket, object string, expires time.Duration, params url.Values) (string, error) {
|
|
presigned, err := s.client.PresignedGetObject(ctx, bucket, object, expires, params)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
link := presigned.String()
|
|
if s.domain != "" {
|
|
link = strings.Replace(link, fmt.Sprintf("http://%s", s.endpoint), s.domain, 1)
|
|
}
|
|
|
|
return link, nil
|
|
}
|
|
|
|
func (s *Service) Delete(ctx context.Context, bucket, object string) error {
|
|
return s.client.RemoveObject(ctx, bucket, object, minio.RemoveObjectOptions{})
|
|
}
|
|
|
|
func (s *Service) GetObjectEtag(ctx context.Context, bucketName, object string) (string, error) {
|
|
info, err := s.client.StatObject(ctx, bucketName, object, minio.StatObjectOptions{})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return info.ETag, nil
|
|
}
|