Skip to content
Snippets Groups Projects
Commit 809fb167 authored by Daniel Naude's avatar Daniel Naude
Browse files

rename session to client

parent cca23cda
No related branches found
No related tags found
1 merge request!48Migrate to aws sdk for go v2
...@@ -30,13 +30,13 @@ func ValidateSQSEndpoints(endpoints map[string]interface{}) (map[string]interfac ...@@ -30,13 +30,13 @@ func ValidateSQSEndpoints(endpoints map[string]interface{}) (map[string]interfac
return endpoints, nil return endpoints, nil
} }
func GetRecord(s3Session *s3.SessionWithHelpers, bucket string, message events.SQSMessage, recordType reflect.Type) (interface{}, error) { func GetRecord(s3Client *s3.ClientWithHelpers, bucket string, message events.SQSMessage, recordType reflect.Type) (interface{}, error) {
recordValuePtr := reflect.New(recordType) recordValuePtr := reflect.New(recordType)
// Check if message body should be retrieved from S3 // Check if message body should be retrieved from S3
if messageAttribute, ok := message.MessageAttributes[sqs.SQSMessageOnS3Key]; ok && messageAttribute.StringValue != nil && *messageAttribute.StringValue == "true" { if messageAttribute, ok := message.MessageAttributes[sqs.SQSMessageOnS3Key]; ok && messageAttribute.StringValue != nil && *messageAttribute.StringValue == "true" {
messageBytes, err := sqs.RetrieveMessageFromS3(s3Session, bucket, message.Body) messageBytes, err := sqs.RetrieveMessageFromS3(s3Client, bucket, message.Body)
if err != nil { if err != nil {
return nil, errors.Wrapf(err, "failed to get sqs message body from s3") return nil, errors.Wrapf(err, "failed to get sqs message body from s3")
} }
......
...@@ -162,10 +162,10 @@ func SendSQSMessage(msgr Messenger, objectToSend interface{}, currentRequestID * ...@@ -162,10 +162,10 @@ func SendSQSMessage(msgr Messenger, objectToSend interface{}, currentRequestID *
return nil return nil
} }
func uploadMessageToS3(session *s3.ClientWithHelpers, bucket string, name string, messageBytes []byte) error { func uploadMessageToS3(client *s3.ClientWithHelpers, bucket string, name string, messageBytes []byte) error {
// Upload message // Upload message
expiry := 24 * 7 * time.Hour // 3 days expiry := 24 * 7 * time.Hour // 3 days
_, err := session.UploadWithSettings(messageBytes, bucket, name, s3.S3UploadSettings{ _, err := client.UploadWithSettings(messageBytes, bucket, name, s3.S3UploadSettings{
ExpiryDuration: &expiry, ExpiryDuration: &expiry,
}) })
if err != nil { if err != nil {
...@@ -175,9 +175,9 @@ func uploadMessageToS3(session *s3.ClientWithHelpers, bucket string, name string ...@@ -175,9 +175,9 @@ func uploadMessageToS3(session *s3.ClientWithHelpers, bucket string, name string
return nil return nil
} }
func RetrieveMessageFromS3(session *s3.ClientWithHelpers, bucket string, filename string) ([]byte, error) { func RetrieveMessageFromS3(client *s3.ClientWithHelpers, bucket string, filename string) ([]byte, error) {
// get the file contents // get the file contents
rawObject, err := session.GetObject(bucket, filename, false) rawObject, err := client.GetObject(bucket, filename, false)
if err != nil { if err != nil {
return []byte{}, err return []byte{}, err
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment