Add more methods

Author: Sebastian Gellweiler

lib/dynamodb.js

@@ -1,8 +1,5 @@
 const _ = require('lodash');
-
-const { DynamoDBDocument } = require("@aws-sdk/lib-dynamodb");
-const { DynamoDB } = require("@aws-sdk/client-dynamodb");
-
+const AWS = require("aws-sdk");
 const RSVP = require('rsvp');
 
 function DynamodbAdapter(logger, opts) {
@@ -14,28 +11,51 @@ function DynamodbAdapter(logger, opts) {
     maxRetries: 'maxRetries' in opts ? opts.maxRetries : 4
   };
 
-  this.settings = {
+  this.dynamodbsettings = {
     logger: logger,
     table: opts.table
   };
 
-  this.docClient = DynamoDBDocument.from(new DynamoDB(docClientOptions))
-
-  _.bindAll(this, 'get', 'set', 'del');
+  this.docClient = new AWS.DynamoDB.DocumentClient(docClientOptions)
+
+  _.bindAll(
+      this,
+      "get",
+      "set",
+      "del",
+      "isMemoryStore",
+      "getAllClientInfos",
+      "saveInstallation",
+  );
 }
 
 var proto = DynamodbAdapter.prototype;
 
+proto.isMemoryStore = function () {
+  return false;
+};
+
+proto.getAllClientInfos = async function () {
+  const self = this;
+  return await self.get({ key: "clientInfo" });
+};
+
+proto.saveInstallation = async function (val, clientKey) {
+  await this.connectionPromise;
+  const clientSetting = await this.set("clientInfo", val, clientKey);
+  return clientSetting;
+};
+
 // return a promise to a single object identified by 'key' in the data belonging to tenant 'clientKey'
 proto.get = async function (key, clientKey) {
   try {
     var data = await this.docClient.get({
-      "TableName": this.settings.table,
+      "TableName": this.dynamodbsettings.table,
       "Key": {
         "key": key,
         "clientKey": clientKey
       }
-    });
+    }).promise();
 
     if ('Item' in data) {
       data = data.Item;
@@ -54,13 +74,13 @@ proto.get = async function (key, clientKey) {
 proto.set = async function (key, value, clientKey) {
   try {
     await this.docClient.put({
-      "TableName": this.settings.table,
+      "TableName": this.dynamodbsettings.table,
       "Item": {
         "key": key,
         "clientKey": clientKey,
         "val": value
       }
-    });
+    }).promise();
     return value;
   } catch (err) {
     throw new Error("atlassian-connect-express-dynamodb set failed: " + err.message);
@@ -72,24 +92,24 @@ proto._del_multiple = function (keys, clientKey) {
   for (var i = 0; i < keys.length; i++) {
     var key = keys[i];
     promisses.push(this.docClient.delete({
-      "TableName": this.settings.table,
+      "TableName": this.dynamodbsettings.table,
       "Key": {
         "clientKey": clientKey,
         "key": key
       }
-    }));
+    }).promise());
   }
   return Promise.all(promisses);
 };
 
 proto._query_by_client_key = function (clientKey) {
   return this.docClient.query({
-    TableName: this.settings.table,
+    TableName: this.dynamodbsettings.table,
     KeyConditionExpression: "clientKey = :clientKey",
     ExpressionAttributeValues: {
       ":clientKey": clientKey
     }
-  });
+  }).promise();
 };
 
 proto.del = async function (key, clientKey) {
@@ -117,4 +137,4 @@ module.exports = function (logger, opts) {
     return DynamodbAdapter;
   }
   return new DynamodbAdapter(logger, opts);
-};
+};