[AWS] Simple Queue Service - Phần 1

Mục Đích

• Call api tới AWS Simple Queue Service(SQS) thông qua sdk

Code Mẫu

Chuẩn bị

• cài đặt sdk aws, run:

go get -u github.com/aws/aws-sdk-go/

• cấu hình access-key and secret-key, nếu bạn không biết thì đọc tham khảo link: https://viblo.asia/p/cau-hinh-aws-credential-zOQJwYPxVMP

Tạo một session

• tạo một session để tái sử dụng trong những function khác
• profile được dùng là default, để biết profile đang được cấu hình, run trên linux/mac

cat ~/.aws/credentials 

• function để tạo 1 session

func GetSession() *session.Session { sess, err := session.NewSessionWithOptions(session.Options{ Profile: "default", Config: aws.Config{ Region: aws.String("us-west-1"), }, }) if err != nil { panic(err) } return sess
}

Cách tạo ra 1 queue

• sử dụng function CreateQueue trong aws-sdk để tạo 1 queue cho việc test.
• có một số parameter quan trọng cần lưu ý:
  • queueName: name của queue mà bạn muốn tạo
  • DelaySeconds: thời gian message bạn muốn giữ lại trước khi gửi đi,
  • VisibilityTimeout: thời gian một message trước khi expire.
• function create 1 queue

func CreateQueue(sess *session.Session, queueName string) (*sqs.CreateQueueOutput, error) { sqsClient := sqs.New(sess) result, err := sqsClient.CreateQueue(&sqs.CreateQueueInput{ QueueName: &queueName, Attributes: map[string]*string{ "DelaySeconds": aws.String("0"), "VisibilityTimeout": aws.String("60"), }, }) if err != nil { return nil, err } return result, nil
}

Get URL của 1 queue

• Tất cả các apis điều yêu cầu url của queue, vì thế sử dụng GetQueueURL để truy vấn url của queue-name
• function get url

func GetQueueURL(sess *session.Session, queue string) (*sqs.GetQueueUrlOutput, error) { sqsClient := sqs.New(sess) result, err := sqsClient.GetQueueUrl(&sqs.GetQueueUrlInput{ QueueName: &queue, }) if err != nil { return nil, err } return result, nil
}

Gửi message đến queue

• chúng ta sẽ sử dụng hàm SendMessage từ aws-sdk để gửi message đến queue
• một vài paramters quan trọng cần chú ý trong lúc gửi message:
  • QueueUrl: url của queue muốn gửi message.
  • MessageBody: body được gửi đến queue, có thể là json-string hoặc string
• function gửi message

func SendMessage(sess *session.Session, queueUrl string, messageBody string) (*sqs.SendMessageOutput, error) { sqsClient := sqs.New(sess) sendOut, err := sqsClient.SendMessage(&sqs.SendMessageInput{ QueueUrl: &queueUrl, MessageBody: aws.String(messageBody), }) if err != nil { return nil, err } return sendOut, nil
}

Nhận message từ queue

• chúng ta sẽ sử dụng hàm ReceiveMessage từ aws-sdk để lấy message từ queue
• một vài paramters quan trọng cần được chú ý trong lúc nhận message:
  • QueueUrl: url của queue muốn nhận message.
  • MaxNumberOfMessages: tổng số message có thể nhận được.
• function receive messsage:

func GetMessages(sess *session.Session, queueUrl string, maxMessages int) (*sqs.ReceiveMessageOutput, error) { sqsClient := sqs.New(sess) msgResult, err := sqsClient.ReceiveMessage(&sqs.ReceiveMessageInput{ QueueUrl: &queueUrl, MaxNumberOfMessages: aws.Int64(int64(maxMessages)), }) if err != nil { return nil, err } return msgResult, nil
}

Xoá message trong queue

• khi nhận message từ queue, message sẽ không tự động xoá ra khỏi queue.
• consumer khác có thể nhận message sau thời gian VisibilityTimeout hết hạn.
• để đảm bảo rằng không bị trùng message thì cần phải xoá.
• chúng ta sẽ sử dụng hàm DeleteMessage từ aws-sdk để xoá message từ queue
• cần cung cấp ReceiptHandle trong thành phần của method
• function xoá message:

func DeleteMessage(sess *session.Session, queueUrl string, messageHandle *string) error { sqsClient := sqs.New(sess) _, err := sqsClient.DeleteMessage(&sqs.DeleteMessageInput{ QueueUrl: &queueUrl, ReceiptHandle: messageHandle, }) return err
}

Xoá tất cả message trong queue

• chúng ta sẽ sử dụng hàm PurgeQueue từ aws-sdk để xoá tất cả message trong queue
• một vài paramters quan trọng cần được chú ý trong lúc xoá tất cả message:
• function xoá tất cả message:

func PurgeQueue(sess *session.Session, queueUrl string) error { sqsClient := sqs.New(sess) _, err := sqsClient.PurgeQueue(&sqs.PurgeQueueInput{ QueueUrl: aws.String(queueUrl), }) return err
}

Ref

• fullsource:
  • https://github.com/ducnpdev/open-dev/blob/master/aws/sqs/sqs.go
  • https://github.com/ducnpdev/open-dev/blob/master/aws/sqs/README_VI.md
• aws sqs document: https://docs.aws.amazon.com/sqs/index.html

Contact

• facebook: https://www.facebook.com/phucducdev/
• gmail: ducnp09081998@gmail.com
• linkedin: https://www.linkedin.com/in/phucducktpm/
• hashnode: https://hashnode.com/@OpenDev
• telegram: https://t.me/OpenDevGolang