// Set puts the specified data into the relevant S3 object location
func (s *S3Storage) Set(key string, data string) *Error {
_, err := s.getService().PutObject(&s3.PutObjectInput{
Bucket: aws.String(os.Getenv("S3_BUCKET")),
Key: aws.String(key),
Body: bytes.NewReader([]byte(data)),
ContentType: aws.String("application/json"),
ServerSideEncryption: aws.String("AES256"),
})
return handleAWSError(err)
}
func (s *S3Storage) getService() *s3.S3 {
if s.service == nil {
config := &aws.Config{
Region: aws.String(os.Getenv("AWS_REGION")),
Endpoint: aws.String(os.Getenv("S3_ENDPOINT")),
}
if os.Getenv("APP_ENV") != "production" {
config.DisableSSL = aws.Bool(true)
config.S3ForcePathStyle = aws.Bool(true)
}
s.service = s3.New(config)
}
return s.service
}
// Get retrieves the specified S3 object
func (s *S3Storage) Get(key string) (string, *Error) {
response, err := s.getService().GetObject(&s3.GetObjectInput{
Bucket: aws.String(os.Getenv("S3_BUCKET")),
Key: aws.String(key),
})
var object []byte
if err != nil {
return string(object), handleAWSError(err)
}
object, readErr := ioutil.ReadAll(response.Body)
if readErr != nil {
return string(object), &Error{Message: readErr.Error()}
}
return string(object), nil
}
