IXWebSocket/examples/satori_publisher/satori_publisher.cpp

/*
 *  satori_publisher.cpp
 *  Author: Benjamin Sergeant
 *  Copyright (c) 2018 Machine Zone, Inc. All rights reserved.
 */

#include <iostream>
#include <sstream>
#include <fstream>
#include <atomic>
#include <ixwebsocket/IXWebSocket.h>
#include "IXSatoriConnection.h"
#include "jsoncpp/json/json.h"

void msleep(int ms)
{
    std::chrono::duration<double, std::milli> duration(ms);
    std::this_thread::sleep_for(duration);
}

int main(int argc, char* argv[])
{
    std::string endpoint = argv[1];
    std::string appkey = argv[2];
    std::string channel = argv[3];
    std::string rolename = argv[4];
    std::string rolesecret = argv[5];
    std::string path = argv[6];

    std::atomic<size_t> incomingBytes(0);
    std::atomic<size_t> outgoingBytes(0);
    ix::SatoriConnection::setTrafficTrackerCallback(
        [&incomingBytes, &outgoingBytes](size_t size, bool incoming)
        {
            if (incoming)
            {
                incomingBytes += size;
            }
            else
            {
                outgoingBytes += size;
            }
        }
    );

    bool done = false;
    ix::SatoriConnection satoriConnection;
    satoriConnection.configure(appkey, endpoint, rolename, rolesecret);
    satoriConnection.connect();
    satoriConnection.setAuthenticatedCallback(
        [&satoriConnection, channel, path, &done]()
        {
            std::cout << "Authenticated" << std::endl;;

            std::string line;
            std::ifstream f(path);
            if (!f.is_open())
            {
                std::cerr << "Error while opening file: " << path << std::endl;
            }

            while (getline(f, line))
            {
                Json::Value value;
                Json::Reader reader;
                reader.parse(line, value);

                satoriConnection.publish(channel, value);
            }

            if (f.bad())
            {
                std::cerr << "Error while opening file: " << path << std::endl;
            }

            done = true;
        }
    );
    satoriConnection.setErrorCallback(
        [&done](const std::string& errMsg)
        {
            std::cerr << "Satori Error received: " << errMsg << std::endl;
            done = true;
        }
    );

    while (!done)
    {
        msleep(10);
    }

    std::cout << incomingBytes << std::endl;
    std::cout << "Incoming bytes: " << incomingBytes << std::endl;
    std::cout << "Outgoing bytes: " << outgoingBytes << std::endl;

    return 0;
}