mirror of
https://github.com/Luzifer/mondash.git
synced 2024-09-19 08:52:58 +00:00
Knut Ahlers
f3c31476b4
* Update dependencies * Improve code quality * Apply linter advices * Add gzip compression to all requests * Rework frontend * Apply bootswatch theme * Hide historic details when not requested * Remove debugging header * Apply auto-migration of meta fields * Fix broken "last update" time * Pre-sort metrics for frontend / API * Add tooltip with absolute time * Some design fixes * Add tooltip with absolute date to last ok * Implement filters * Apply eslint --fix * Remove unused var * Remove remains of old template engine * Update baked in assets * Update Dockerfile for new version Signed-off-by: Knut Ahlers <knut@ahlers.me>
100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"net/url"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/awserr"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
)
|
|
|
|
// S3Storage is a storage adapter storing the data into single S3 files
|
|
type S3Storage struct {
|
|
s3connection *s3.S3
|
|
bucket string
|
|
prefix string
|
|
}
|
|
|
|
// NewS3Storage instanciates a new S3Storage
|
|
func NewS3Storage(uri *url.URL) *S3Storage {
|
|
s3connection := s3.New(session.Must(session.NewSession()))
|
|
return &S3Storage{
|
|
s3connection: s3connection,
|
|
bucket: uri.Host,
|
|
prefix: uri.Path,
|
|
}
|
|
}
|
|
|
|
// Put writes the given data to S3
|
|
func (s *S3Storage) Put(dashboardID string, data []byte) error {
|
|
_, err := s.s3connection.PutObject(&s3.PutObjectInput{
|
|
Bucket: aws.String(s.bucket),
|
|
ContentType: aws.String("application/json"),
|
|
Key: s.getKey(dashboardID),
|
|
Body: bytes.NewReader(data),
|
|
ACL: aws.String("private"),
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
// Get loads the data for the given dashboard from S3
|
|
func (s *S3Storage) Get(dashboardID string) ([]byte, error) {
|
|
res, err := s.s3connection.GetObject(&s3.GetObjectInput{
|
|
Bucket: aws.String(s.bucket),
|
|
Key: s.getKey(dashboardID),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer res.Body.Close()
|
|
|
|
data, err := ioutil.ReadAll(res.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
// Delete deletes the given dashboard from S3
|
|
func (s *S3Storage) Delete(dashboardID string) error {
|
|
_, err := s.s3connection.DeleteObject(&s3.DeleteObjectInput{
|
|
Bucket: aws.String(s.bucket),
|
|
Key: s.getKey(dashboardID),
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
// Exists checks for the existence of the given dashboard
|
|
func (s *S3Storage) Exists(dashboardID string) (bool, error) {
|
|
_, err := s.s3connection.HeadObject(&s3.HeadObjectInput{
|
|
Bucket: aws.String(s.bucket),
|
|
Key: s.getKey(dashboardID),
|
|
})
|
|
|
|
if err != nil {
|
|
if awsErr, ok := err.(awserr.Error); ok {
|
|
if strings.Contains(awsErr.Error(), "status code: 404") {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
return false, err
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
func (s S3Storage) getKey(dashboardID string) *string {
|
|
p := path.Join(s.prefix, dashboardID)
|
|
return aws.String(strings.TrimLeft(p, "/"))
|
|
}
|