71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package attachment
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"time"
|
|
|
|
"heckel.io/ntfy/v2/log"
|
|
"heckel.io/ntfy/v2/s3"
|
|
)
|
|
|
|
const (
|
|
tagS3Backend = "s3_backend"
|
|
deleteBatchSize = 1000
|
|
)
|
|
|
|
type s3Backend struct {
|
|
client *s3.Client
|
|
}
|
|
|
|
var _ backend = (*s3Backend)(nil)
|
|
|
|
func newS3Backend(client *s3.Client) *s3Backend {
|
|
return &s3Backend{client: client}
|
|
}
|
|
|
|
func (b *s3Backend) Put(id string, in io.Reader) error {
|
|
return b.client.PutObject(context.Background(), id, in)
|
|
}
|
|
|
|
func (b *s3Backend) Get(id string) (io.ReadCloser, int64, error) {
|
|
return b.client.GetObject(context.Background(), id)
|
|
}
|
|
|
|
func (b *s3Backend) List() ([]object, error) {
|
|
objects, err := b.client.ListAllObjects(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result := make([]object, 0, len(objects))
|
|
for _, obj := range objects {
|
|
result = append(result, object{
|
|
ID: obj.Key,
|
|
Size: obj.Size,
|
|
LastModified: obj.LastModified,
|
|
})
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (b *s3Backend) Delete(ids ...string) error {
|
|
// S3 DeleteObjects supports up to 1000 keys per call
|
|
for i := 0; i < len(ids); i += deleteBatchSize {
|
|
end := i + deleteBatchSize
|
|
if end > len(ids) {
|
|
end = len(ids)
|
|
}
|
|
batch := ids[i:end]
|
|
for _, id := range batch {
|
|
log.Tag(tagS3Backend).Field("message_id", id).Debug("Deleting attachment from S3")
|
|
}
|
|
if err := b.client.DeleteObjects(context.Background(), batch); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b *s3Backend) DeleteIncomplete(cutoff time.Time) error {
|
|
return b.client.AbortIncompleteUploads(context.Background(), cutoff)
|
|
}
|