-
Notifications
You must be signed in to change notification settings - Fork 1
/
eventhub.js
41 lines (36 loc) · 1.29 KB
/
eventhub.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
const EventHubClient = require('azure-event-hubs').Client;
const { Message } = require('@airx/proto');
const getPartitionIds = async (client) => {
const partitionIds = await client.getPartitionIds();
return partitionIds;
};
const defaultErrorHandler = async (err) => {
console.error(err);
};
const defaultMessageHandler = (message) => {
try {
const decodedMessage = Message.decode(message.body);
return decodedMessage;
} catch (e) {
console.error(e);
return {};
}
};
const createReceiver = async ({ connStr = '', messageHandler = defaultMessageHandler } = {}, errorHandler = defaultErrorHandler) => {
const client = EventHubClient.fromConnectionString(connStr);
await client.open();
const partitionIds = await getPartitionIds(client);
partitionIds.forEach(async (partitionId) => {
const receiver = await client.createReceiver('$Default', partitionId, { startAfterTime: Date.now() });
receiver.on('errorReceived', async (err) => {
if (err.transport && err.transport.name === 'AmqpProtocolError') {
console.log(`Restart #${partitionId}`);
await createReceiver({ connStr, errorHandler, messageHandler });
} else {
await errorHandler(err);
}
});
receiver.on('message', messageHandler);
});
};
module.exports = createReceiver;