-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsendMessageToSQS.js
More file actions
50 lines (44 loc) · 1.63 KB
/
Copy pathsendMessageToSQS.js
File metadata and controls
50 lines (44 loc) · 1.63 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
const S3 = require('aws-sdk/clients/s3');
const SQS = require('aws-sdk/clients/sqs')
const xlsx = require('xlsx');
function handler(event, context) {
console.log('Received event:', JSON.stringify(event));
bucketName = event.Records[0].s3.bucket.name;
objectKey = event.Records[0].s3.object.key;
const sqs = new SQS();
let s3 = new S3();
s3.getObject({
Bucket: bucketName,
Key: objectKey
}, function (err, data) {
if (err) {
console.log(err, err.stack);
return {status: 500, message: err.stack};
} else {
// convert excel file to json -> data is a excel file
const file = xlsx.read(data.Body, { type: "buffer" })
const json = xlsx.utils.sheet_to_json(file.Sheets[file.SheetNames[0]]);
console.log("jaons",json);
// send message to SQS in a batch of 10
for (let i = 0; i < json.length; i += 10) {
sqs.sendMessageBatch({
QueueUrl: process.env.QUEUE_URL,
Entries: json.slice(i, i + 10).map((item, index) => {
return {
Id: index.toString(),
MessageBody: JSON.stringify(item)
}
})
}, function (err, data) {
if (err) {
console.log(err, err.stack);
} else {
console.log(data);
}
});
}
return {status: 200, message: 'Success'};
}
});
}
module.exports = {handler};