aws-v3-sdk 1.0.12 → 1.0.14

package/clients/dynamo/DocumentClientAsync.js

@@ -0,0 +1,131 @@
+const { DynamoDB, DynamoDBClient } = require('@aws-sdk/client-dynamodb')
+const { DynamoDBDocument, DynamoDBDocumentClient, GetCommand, BatchWriteCommand, PutCommand, QueryCommand, ScanCommand, DeleteCommand } = require('@aws-sdk/lib-dynamodb')
+const ddb = new DynamoDBClient()
+const documentClient = DynamoDBDocumentClient.from(ddb)
+
+class DocumentClientAsync {
+  get (params, callback) {
+    return new Promise((resolve, reject) => {
+      const getCommand = new GetCommand(params)
+      documentClient.send(getCommand, (err, response) => {
+        if (err || !response) {
+          return reject(err)
+        }
+        if (response.Item) {
+          return resolve(response)
+        } else if (!response.Item) {
+          return resolve({})
+        }
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
+
+  query (params, callback) {
+    return new Promise((resolve, reject) => {
+      const getCommand = new QueryCommand(params)
+      documentClient.send(getCommand, (err, response) => {
+        if (err || !response) {
+          return resolve(err)
+        }
+        return resolve(response)
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
+
+  put (params, callback) {
+    return new Promise((resolve, reject) => {
+      let item = params.Item
+      try {
+        item = Object.fromEntries(Object.entries(item).filter(([key, value]) => value !== undefined))
+      } catch (error) {
+
+      }
+      params.Item = item
+      const putCommand = new PutCommand(params)
+      documentClient.send(putCommand, (err, response) => {
+        if (err || !response) {
+          return resolve(err)
+        }
+        return resolve(response)
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
+
+  delete (params, callback) {
+    return new Promise((resolve, reject) => {
+      const docClient = DynamoDBDocument.from(new DynamoDB())
+      docClient.delete(params, (err, response) => {
+        if (err) {
+          return resolve(err)
+        } else {
+          return resolve(response)
+        }
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
+
+  update (params, callback) {
+    return new Promise((resolve, reject) => {
+      const docClient = DynamoDBDocument.from(new DynamoDB())
+      docClient.update(params, (err, response) => {
+        if (err) {
+          return resolve(err)
+        } else {
+          return resolve(response)
+        }
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
+}
+
+module.exports = DocumentClientAsync

package/clients/dynamoDB.js

@@ -1,8 +1,10 @@
 const DocumentClient = require('./dynamo/DocumentClient')
+const DocumentClientAsync = require('./dynamo/DocumentClientAsync')
 
 class DynamoDB {
   constructor () {
     this.DocumentClient = DocumentClient
+    this.DocumentClientAsync = DocumentClientAsync
   }
 }
 

package/clients/lambda.js

@@ -19,6 +19,28 @@ class Lambda {
       callback(err)
     })
   }
+
+  invokeAsync (params, callback) {
+    return new Promise((resolve, reject) => {
+      const command = new InvokeCommand(params)
+      this.client.send(command).then(response => {
+        response.Payload = toUtf8(response.Payload)
+        resolve(response)
+      }).catch(err => {
+        resolve(err)
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
 }
 
 module.exports = Lambda

package/es-handler/es-wrapper.js

@@ -32,23 +32,35 @@ const getClient = async () => {
 }
 
 exports.sendReqToES = async (method, index, type, id, data) => {
-  const client = await getClient()
-  if (method === 'DELETE') {
-    const response = await client.delete({
-      index,
-      type,
-      id
-    })
-    return response.body
-  } else {
-    const response = await client.index({
-      id,
-      index,
-      type,
-      body: data,
-      refresh: true
-
-    })
-    return response.body
+  try {
+    const client = await getClient()
+    if (method === 'DELETE') {
+      const response = await client.delete({
+        index,
+        type,
+        id
+      })
+      return response.body
+    } else if (method === 'UPDATE') {
+      const response = await client.update({
+        id,
+        index,
+        type,
+        body: data,
+        refresh: true
+      })
+      return response.body
+    } else {
+      const response = await client.index({
+        id,
+        index,
+        type,
+        body: data,
+        refresh: true
+      })
+      return response.body
+    }
+  } catch (error) {
+    return error
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aws-v3-sdk",
-  "version": "1.0.12",
+  "version": "1.0.14",
   "description": "AWS v3 SDK Utils",
   "main": "index.js",
   "scripts": {