Windshaft-cartodb/lib/backends/pubsub-metrics.js

'use strict';

const { PubSub } = require('@google-cloud/pubsub');

module.exports = class PubSubMetricsBackend {
    constructor (options = {}) {
        const { project_id: projectId, credentials: keyFilename, topic } = options;
        this._pubsub = new PubSub({ projectId, keyFilename });
        this._topicName = topic;
    }

    send (event, attributes) {
        const data = Buffer.from(event);

        this._pubsub.topic(this._topicName).publish(data, attributes)
            .then(() => {
                console.log(`PubSubTracker: event '${event}' published to '${this._topicName}'`);
            })
            .catch((error) => {
                console.error(`ERROR: pubsub middleware failed to publish event '${event}': ${error.message}`);
            });
    }
};