solace-backend/dynamo.js at master · UT-Projects/solace-backend · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
require('dotenv').config()
const AWS = require('aws-sdk')

AWS.config.update({
    endpoint: process.env.AWS_ENDPOINT || "http://localhost:8000",
    region: process.env.AWS_DEFAULT_REGION || "us-east-1",
    accessKeyId: process.env.AWS_ACCESS_KEY_ID || "anything",
    secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY || "anything",
});

const docClient = new AWS.DynamoDB.DocumentClient();

const getDocument = (tableName, uuid) => {
    var params = {
        TableName: tableName,
        Key: {
            "uuid": uuid,
        },
    };

    return new Promise(function(resolve, reject) {
        docClient.get(params, function(error, data) {
            if (error) {
                reject(JSON.stringify(error, null, 2));
            } else {
                resolve(JSON.stringify(data, null, 2));
            }
        });
    })
}

const updateDocument = (tableName, uuid, key, value) => {
    var params = {
        TableName: tableName,
        Key: {
            "uuid": uuid,
        },
        UpdateExpression: "set #name = :n",
        ExpressionAttributeNames: { // use because key might be reserved word in DynamoDB
            "#name" : key,
        },
        ExpressionAttributeValues: {
            ":n" : value,
        },
        ReturnValues: "UPDATED_NEW", // return only updated values
    };

    return new Promise(function(resolve, reject) {
        docClient.update(params, function(error, data) {
            if (error) {
                reject(JSON.stringify(error, null, 2));
            } else {
                resolve(JSON.stringify(data, null, 2));
            }
        });
    })
}

const createDocument = (tableName, item) => {
    var params = {
        TableName: tableName,
        Item: item,
        ConditionExpression: "attribute_not_exists(#uuid)",
        ExpressionAttributeNames: {
            "#uuid": "uuid",
        },
    };

    return new Promise(function(resolve, reject) {
        docClient.put(params, function(error, data) {
            if (error) {
                reject(JSON.stringify(error, null, 2));
            } else {
                resolve(JSON.stringify(data, null, 2));
            }
        })
    });
}

const deleteDocument = (tableName, primaryKey) => {
    var params = {
        TableName: tableName,
        Key: primaryKey,
    };

    return new Promise(function(resolve, reject) {
        docClient.delete(params, function(error, data) {
            if (error) {
                reject(JSON.stringify(error, null, 2));
            } else {
                resolve(JSON.stringify(data, null, 2));
            }
        })
    });
}

module.exports = {
    getDocument,
    updateDocument,
    createDocument,
    deleteDocument,
    docClient
}