generated from VLADIMIR/template
add Upload and Download files
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
||||
"context"
|
||||
"mime"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"github.com/kzzan/s3kit"
|
||||
)
|
||||
@@ -28,6 +29,17 @@ func NewRustFSStorage(
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
err = client.CreateBucket(
|
||||
ctx,
|
||||
bucket,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &storage{
|
||||
client: client,
|
||||
bucket: bucket,
|
||||
@@ -35,7 +47,8 @@ func NewRustFSStorage(
|
||||
}
|
||||
|
||||
func (s *storage) Put(ctx context.Context, file *File) error {
|
||||
return s.client.PutObjectBytes(ctx, s.bucket, file.Name, file.Data, file.Mime)
|
||||
mime := s.getMimeType(file.Name)
|
||||
return s.client.PutObjectBytes(ctx, s.bucket, file.Name, file.Data, mime)
|
||||
}
|
||||
|
||||
func (s *storage) Get(ctx context.Context, filename string) (*File, error) {
|
||||
|
||||
Reference in New Issue
Block a user