在调用 AWS lambda handlers` 回调时遇到 Neptune Gremlin 连接问题

Question

我将 gremlin@3.3.5 用于带有 AWS Lambda 的 Node.js 8.10 应用程序。对于单次调用，该过程一切正常。这是我的示例代码。

const gremlin = require('gremlin');
const DriverRemoteConnection = gremlin.driver.DriverRemoteConnection;
const Graph = gremlin.structure.Graph;

exports.handler = (event, context, callback) => {
    dc = new DriverRemoteConnection('wss://your-neptune-endpoint:8182/gremlin');

    const graph = new Graph();
    const g = graph.traversal().withRemote(dc);
    try {
        const result = await g.V().limit(1).count().next();
        dc.close();
        callback(null, { result: result });
    } catch (exception) {
        callback('Error');
        throw error;
    }
}

当我为单次调用运行此进程时，它似乎一切正常，但是当我尝试运行批处理操作（例如 100,000 个请求/小时）时，我在 CloudWatch 日志指标中遇到了我的连接没有成功关闭。我已经尝试了一些这样的实现，比如 callbackWaitForEventLoopEmpty，但它抓住了 lambda。当我删除回调（或类似地返回）时，这个过程也适用于批处理操作。但我确实想从这个 lambda 返回数据，其中包含传递给我的 step 函数的信息，以根据该信息触发另一个 lambda。

Answer 1

在做了一些研究之后，我发现问题在于 gremlin 包如何处理关闭连接的事件不利于无服务器架构。当触发 driver.close() 时。当驱动程序被实例化时，它会创建客户端实例，它在自身内部创建连接实例，它使用 ws 库创建 websocket 实例。现在 ws.close() 事件优雅地关闭了所有事件，它不会在我的回调被调用之前等待事件被调用并且该事件保持打开并泄漏。因此，在连接实例上显式调用 dc._client._connection.ws.terminate() 然后 dc.close() 立即关闭连接。

Answer 2

g.V().limit(1).count().next() 是异步的。

试试这个：

exports.handler = async (event) => {

  try {
      dc = new DriverRemoteConnection('wss://your-neptune-endpoint:8182/gremlin');    
      const graph = new Graph();
      const g = graph.traversal().withRemote(dc);
      const result = await g.V().limit(1).count().next();
      dc.close();
      return result;
  } catch (error) {
      throw error;
  }
}

由于您的 Lambda 运行时是 Node.js 8.10，您不需要使用 callback。