/
index.js
92 lines (80 loc) · 2.8 KB
/
index.js
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
import 'source-map-support/register';
import AWSXray from 'aws-xray-sdk';
const AWS = AWSXray.captureAWS(require('aws-sdk'));
import { SubscriptionManager, PubSub } from 'graphql-aws-iot-server';
import schema from './root.schema';
let db;
let manager;
let pubsub;
export const handler = (event, context, callback) => {
console.log('Todo api handler running');
console.log(JSON.stringify(event));
if (!db) {
db = new AWS.DynamoDB.DocumentClient();
}
if (!pubsub) {
// class that invokes publisher lambda to trigger a subscription publish has only one method publish
pubsub = new PubSub(process.env.SubscriptionPublisherFunctionName);
}
if (!manager) {
const subscriptionManagerOptions = {
appPrefix: process.env.AppPrefix,
iotEndpoint: process.env.IotEndpoint,
schema,
addSubscriptionFunction: (subscription) => {
const putParams = {
TableName: process.env.SubscriptionsTableName,
Item: subscription
};
return db.put(putParams).promise();
},
removeSubscriptionFunction: ({ clientId, subscriptionName }) => {
const params = {
TableName: process.env.SubscriptionsTableName,
Key: {
clientId,
subscriptionName: subscriptionName
}
};
return db.delete(params).promise();
}
};
manager = new SubscriptionManager(subscriptionManagerOptions);
}
const { data, clientId } = event;
const parsedMessage = JSON.parse(data);
const graphqlContext = {
documentClient: db,
TableName: process.env.TodosTableName,
pubsub
};
manager.onMessage(parsedMessage, clientId, graphqlContext)
.then(_ => {
callback();
})
.catch(err => {
console.log('Todo Api Handler Error');
console.log(JSON.stringify(err));
callback();
});
};
// Used to generate schema for apollo code gen / automatic type generation
// TODO take this out of handler as it is not used post deployment
export const schemaGenHandler = ({ query, variables }, context, callback) => {
const runQuery = require('graphql-server-core').runQuery;
const queryOptions = {
schema,
query,
variables,
// context: graphQLcontext,
debug: false // unnecessarily shows stacktrace when I throw a custom error message (like accountId not found)
};
runQuery(queryOptions)
.then(res => {
callback(null, res);
})
.catch(err => {
console.log('error is ' + err);
callback(err);
});
};