等待 AWS SNS 将回调发布到 return 调用方法的值

Question

我正在尝试在用户请求重设密码时发送短信。我想等待发送消息以提醒用户是否成功。我目前正在尝试按如下方式进行：

async function sendResetPasswordTextMessage(req, res) {
    let result = {};

    let phoneNumber = req.body.phoneNumber;                

    if (phoneNumber === undefined) {                       
        return sendInvalidParametersMessage(res);          
    }                                                      

    phoneNumber = phoneNumber.toString();                  

    const userProfile = await models.UserProfile.findOne({ 
        where: {                                           
            phoneNumber: phoneNumber                       
        }                                                  
    });                                                    
    ************************** RELEVANT CODE TO ISSUE *************************
    if (userProfile) {
        const message = "Your username is:\n" + userProfile.username;
        const sent = await AWSSNSClient.sendMessage(message, phoneNumber);

        if (!sent) {
            result.error = setTitleAndMessage("Error", "An error occurred");
        } else {
            result.success = setTitleAndMessage("Success", "Message sent"); 
        }
    }
    return res.send(result);
    ***************************************************************************
}

在我的另一个 class AWSSNSClient 中，我有以下 sendMessage 函数：

function sendMessage(message, phoneNumber) {
    const params = { 
        Message: message, 
        MessageStructure: "string", 
        PhoneNumber: "+1" + phoneNumber
    };
    let sent = false;
    sns.publish(params, function(err, data) {
        if (err) {
            console.log(err, err.stack); // an error occurred
        }
        else {
            sent = true;
        }
    });

    return sent;
}

我不知道如何让 sendMessage 在 return 之前等待 sns.publish 到 return。我已尝试将其设为异步方法并在 sns.publish 上添加等待，但函数在发送前 returns 仍设置为 true。

我知道消息发送没有错误，因为我正在接收它们并且没有打印控制台日志。

Answer 1

您可以简单地为此使用回调。像这样修改您的 sendMessge

function sendMessage(message, phoneNumber, cb) {
    const params = { 
        Message: message, 
        MessageStructure: "string", 
        PhoneNumber: "+1" + phoneNumber
    };
    sns.publish(params, cb);
}

然后在您的主文件中，您可以像这样提供 callback

if (userProfile) {
  const message = "Your username is:\n" + userProfile.username;
  AWSSNSClient.sendMessage(message, phoneNumber, (err, data) => {
    if (err) {
      result.error = setTitleAndMessage("Error", "An error occurred");
    }
    else {
      result.success = setTitleAndMessage("Success", "Message sent");
    }
    res.send(result);
  });
}

Answer 2

今天通过 Google 试图自己解决这个问题而偶然发现了这个 - 我现在正在使用的简短答案：

You can now do this with Async/Await — and Call the AWS service (SNS for example) with a .promise() extension to tell aws-sdk to use the promise-ified version of that service function (SNS) instead of the call back based version.

这里唯一需要注意的是包含函数还必须是异步的才能使用 await 语法。

例如：

let snsResult = await sns.publish({
    Message: snsPayload,
    MessageStructure: 'json',
    TargetArn: endPointArn
}, async function (err, data) {
    if (err) {
        console.log("SNS Push Failed:");
        console.log(err.stack);
        return;
    }
    console.log('SNS push suceeded: ' + data);
    return data;
}).promise();

The important part is the .promise() on the end there. Full docs on using aws-sdk in an async / promise based manner can be found here: https://docs.aws.amazon.com/sdk-for-javascript/v2/developer-guide/using-promises.html

为了运行另一个 aws-sdk 任务，您可以类似地向该函数添加 await 和 .promise() 扩展（假设可用）。

对于运行进入此线程并且实际上只是想简单地将多个 aws-sdk 承诺推送到一个数组并等待整个数组完成（不考虑先执行哪个承诺）的任何人我结果是这样的：

let snsPromises = [] // declare array to hold promises
let snsResult = await sns.publish({
    Message: snsPayload,
    MessageStructure: 'json',
    TargetArn: endPointArn
}, async function (err, data) {
    if (err) {
        console.log("Search Push Failed:");
        console.log(err.stack);
        return;
    }
    console.log('Search push suceeded: ' + data);
    return data;
}).promise();

snsPromises.push(snsResult)
await Promise.all(snsPromises)

希望能帮助那些像我一样通过 google 偶然发现这个问题的人！

Answer 3

此处右更新API，2018 年 8 月，Necevil 回复发送短信两次。

// using config.env
AWS.config.region = 'eu-west-1';
AWS.config.update({
  accessKeyId: process.env.AMAZON_SMS_ID,
  secretAccessKey: process.env.AMAZON_SMS_TOKEN,
});

// parameters 
let params = {
   Message: contentSMS,  // here your sms
   PhoneNumber: mobile,  // here the cellphone
 };


 const snsResult = await sns.publish(params, async (err, data) => {
    if (err) {
       console.log("ERROR", err.stack);
    }
    console.log('SNS ok: ' , JSON.stringify (data));
  });

Answer 4

stackdave 真的会等吗？

Necevil "Search push suceeded will get logged twice" 因为您通过传递回调和使用承诺来混合调用操作。您应该只使用一种获取结果的方法

let snsResult = await sns.publish({
    Message: snsPayload,
    MessageStructure: 'json',
    TargetArn: endPointArn}).promise()

会成功的

Answer 5

如果您遇到发送重复 SNS 消息的问题，我利用 AWS 中的示例解决了这个问题：

// Load the AWS SDK for Node.js
var AWS = require('aws-sdk');
// Set region
AWS.config.update({region: 'REGION'});

// Create publish parameters
var params = {
  Message: 'MESSAGE_TEXT', /* required */
  TopicArn: 'TOPIC_ARN'
};

// Create promise and SNS service object
var publishTextPromise = new AWS.SNS({apiVersion: '2010-03-31'}).publish(params).promise();

// Handle promise's fulfilled/rejected states
publishTextPromise.then(
  function(data) {
    console.log("Message ${params.Message} send sent to the topic ${params.TopicArn}");
    console.log("MessageID is " + data.MessageId);
  }).catch(
    function(err) {
    console.error(err, err.stack);
  });

通过使用传统的 .then() 我能够消除上面评论中提到的重复消息错误。

Answer 6

你可以创建一个使用 promise 方法的异步函数

async function sendMessage(message, phoneNumber){
     const params = { 
        Message: message, 
        PhoneNumber: phoneNumber
    };
  
 return new Promise((resolve, reject) => {
    SNS.publish(params, (err, data) => {
      if (err) {
         console.log("Search Push Failed:");
        console.log(err.stack);
        return reject(err);
      } else {
            console.log('Search push suceeded:' + phoneNumber);
        return resolve(data);
      }
    })
    
    });
  
}

然后你可以调用

var  s=  await sendMessage(message,phoneNumber);

等待 AWS SNS 将回调发布到 return 调用方法的值

Wait for AWS SNS publish callback to return a value to calling method

javascript

node.js

amazon-sns

async-await

aws-sdk-js