如何使用promises多次调用dynamodb扫描

时间:2017-06-20 07:50:41

标签: javascript promise amazon-dynamodb

我收到来自dynamodb.scan()的承诺。如果响应包含LastEvaluatedKey,则解析promise,则必须再次调用dynamodb.scan()。当响应不包含LastEvaluatedKey对象时,表的所有记录都已扫描并且查询完成。我不知道提前调用scan()的次数。我不知道如何编码。这就是我现在所拥有的:

function scan(options, startKey) {
  var parms = {
    TableName: options.source,
    ProjectionExpression: "book, bid, #kee, #unt, pid, #txt",
    FilterExpression: "contains(#txt, :v_qs)",
    ExpressionAttributeNames: {
      "#unt": "unit",
      "#txt": "text",
      "#kee": "key"
    },
    ExpressionAttributeValues: {
      ":v_qs": options.queryTransformed
    }
  };

  if (startKey) {
     parms.ExclusiveStartKey = startKey;
  }

  return dynamoDb.scan(parms).promise();
}

scan(options).then(function(response) {
  if (response.LastEvaluatedKey) {
    searchResults.push(response);
    return scan(options, response.LastEvaluatedkey);
  }
  else {
    return response
  }
}).then(function(response) {
  if (response.LastEvaluatedKey) {
    searchResults.push(response);
    return scan(options, response.LastEvaluatedKey);
  }
  else {
    return response;
  }
}).then(function(response) {
  //this is crazy - bailing even if another call is needed
  searchResults.push(response);
  return generateSearchResponse();
});

这显然不是这样做的方法。此代码是AWS Lambda节点功能的一部分。在此先感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

递归函数应该是你的朋友。我自己无法测试,但这样的事情应该有效。如果您遇到任何错误,请告诉我

function scan(options, startKey) {
  var parms = {
    TableName: options.source,
    ProjectionExpression: "book, bid, #kee, #unt, pid, #txt",
    FilterExpression: "contains(#txt, :v_qs)",
    ExpressionAttributeNames: {
      "#unt": "unit",
      "#txt": "text",
      "#kee": "key"
    },
    ExpressionAttributeValues: {
      ":v_qs": options.queryTransformed
    }
  };

  if (startKey) {
     parms.ExclusiveStartKey = startKey;
  }

  return dynamoDb.scan(parms).promise();
}

function processResponse(options, response) {
    return (response && response.LastEvaluatedKey) ?
        scan(options, response.LastEvaluatedkey)
            .then(function(next_response) {
                return [response].concat(processResponse(next_response));
            })
        :
        Promise.resolve(response); // replace 'Promise' by whatever Promise library you are using (when, bluebird, ...)
}

scan(options)
    .then(processResponse.bind(null, options))
    .then(function(response_list) {
        // response_list is an array of response
        return generateSearchResponse();
    });