|| || || || || || || ||

---

ServerConnector.h

Go to the documentation of this file.

#ifndef Impala_Application_Videolympics_ServerConnector_h
#define Impala_Application_Videolympics_ServerConnector_h

#include <iostream>
#include <queue>
#include "Link/PracticalSocket.h"
#include "Link/PracticalSocket.cpp"

#include "Util/StringParser.h"

namespace Impala {
namespace Application {
namespace Videolympics {

class ShotPacket
{
public:
    ShotPacket()
    {
    }

    int teamID;
    std::string shotName;
    double timestamp;
    int sequenceNo;
    int shotNo;
    int relevance;
    int selectedByOthers;
};

class GlobalStats
{
public:
    GlobalStats()
    {
        unjudged = positive = negative = 0;
        timeSinceStart = 0.0;
        contestActive = 0;
        totalRelevantShots = 0;
        connected = false;
    }

    int unjudged;
    int positive;
    int negative;
    double timeSinceStart;
    int contestActive; // 0 = not yet started, 1 = running, 2 = ended 
    int totalRelevantShots;
    bool connected;
};

class StatsPacket
{
public:
    StatsPacket()
    {
    }

    int teamID;
    int unjudged;
    int positive;
    int negative;
};


class ServerConnector
{
public:
    ServerConnector(std::string server, int port, std::string password)
    {
        mVidolServer =    server;
        mVidolPort =      port;
        mSecretPassword = password + "\n";
        mConnected = false;

        mGlobalStats = new GlobalStats();
    }

    void Connect()
    {
        if (mConnected)
            return;
        ILOG_DEBUG("establishing connection... ");
        try {
            mSocket = new TCPSocket(mVidolServer, mVidolPort);
        } catch (SocketException e) {
            ILOG_DEBUG("Could not connect to server.");
            mConnected = false;
            mStats.clear();

            mGlobalStats->connected = mConnected;
            return;
        }
        ILOG_DEBUG("connection established. Sending password...");
        mSocket->send(mSecretPassword.c_str(), strlen(mSecretPassword.c_str()));
        mConnected = true;
        mGlobalStats->connected = mConnected;
    }

    ShotPacket *GetPacket()
    {
        if (mShotQueue.empty())
            return 0;
        ShotPacket *p = (ShotPacket*)mShotQueue.front();
        mShotQueue.pop();
        return p;
    }

    StatsPacket *GetStats(int team)
    {
        if(mStats.find(team) == mStats.end())
        {
            return 0;
        }
        return mStats[team];
    }

    GlobalStats *GetGlobalStats()
    {
        return mGlobalStats;
    }

   bool Update()
    {
        Connect();
        if (!mConnected)
            return false;

        ILOG_DEBUG("Requesting POLL update...");
        try {
            mSocket->send("POLL", 4);
        } catch (SocketException e) {
            ILOG_ERROR("Server connection failed during SEND.");
            mConnected = false;
            mGlobalStats->connected = false;
            mStats.clear();
            return false;
        }
//        ILOG_DEBUG("receiving length...");
        int aantal;
        try {
            mSocket->recv(&aantal, 4);
        } catch (SocketException e) {
            ILOG_ERROR("Server connection failed during RECV.");
            mConnected = false;
            mGlobalStats->connected = false;
            mStats.clear();
            return false;
        }
        ILOG_DEBUG(aantal << " shots in queue");
        char buffer[20000];
        if (aantal > 0)
        {
            int recv = 0;
            try {
                recv = mSocket->recv(buffer, 20000);
            } catch (SocketException e) {
                ILOG_ERROR("Server connection failed during RECV.");
                mConnected = false;
                mGlobalStats->connected = false;
                mStats.clear();
                return false;
            }
            if (recv <= 0) 
            {
                ILOG_ERROR("There should be a queue, however nothing was read.");
                return false;
            }
            buffer[recv] = 0;
            ParsePoll(buffer);
        }

        ILOG_DEBUG("Requesting POLL update...");
        try {
            mSocket->send("STAT", 4);
        } catch (SocketException e) {
            ILOG_ERROR("Server connection failed.");
            mConnected = false;
            mGlobalStats->connected = false;
            mStats.clear();
            return false;
        }
        try {
            mSocket->recv(&aantal, 4);
        } catch (SocketException e) {
            ILOG_ERROR("Server connection failed during RECV.");
            mConnected = false;
            mGlobalStats->connected = false;
            mStats.clear();
            return false;
        }
        ILOG_DEBUG(aantal << " statistics received.");
        if (aantal > 0)
        {
            int recv;
            try {
                recv = mSocket->recv(buffer, 20000);
            } catch (SocketException e) {
                ILOG_ERROR("Server connection failed during RECV.");
                mConnected = false;
                mGlobalStats->connected = false;
                mStats.clear();
                return false;
            }
            if (recv <= 0)
            {
                ILOG_ERROR("There should be statistics, however nothing was read.");
                return false;
            }
            buffer[recv] = 0;
            ParseStat(buffer, aantal);
        }

        return true;
    }

    void
    ParsePoll(std::string input)
    {
        Util::StringParser p = Util::StringParser(input);
        while (1)
        {
            std::string oneline = p.GetString('\n', false);
            if (oneline.size() == 0)
                break;
            Util::StringParser line = Util::StringParser(oneline);
            ShotPacket *entry = new ShotPacket();
            entry->timestamp = line.GetDouble(',');
            entry->sequenceNo = line.GetInt(',');
            entry->teamID = line.GetInt(',');
            entry->shotName = line.GetString(',', false);
            entry->shotNo = line.GetInt(',');
            entry->relevance = line.GetInt(',');
            entry->selectedByOthers = line.GetInt(',');
            mShotQueue.push(entry);

            if (mShotQueue.size() > 100)
                mShotQueue.pop();
        }
    }
    

    void
    ParseStat(std::string input, int aantal)
    {
        Util::StringParser p = Util::StringParser(input);
        while (aantal--)
        {
            std::string oneline = p.GetString('\n', false);
            if (oneline.size() == 0)
                break;
            Util::StringParser line = Util::StringParser(oneline);
            int teamnr = line.GetInt(',');
            if (teamnr == -1)
            { // global stats packet:
                mGlobalStats->unjudged = line.GetInt(',');
                mGlobalStats->positive = line.GetInt(',');
                mGlobalStats->negative = line.GetInt(',');
                mGlobalStats->timeSinceStart = line.GetDouble(',');
                mGlobalStats->contestActive = line.GetInt(',');
                mGlobalStats->totalRelevantShots = line.GetInt(',');

                ILOG_DEBUG("Global: " << mGlobalStats->unjudged << " unjudged, " << mGlobalStats->positive << " positive, " << mGlobalStats->negative << ". contest runs for " << mGlobalStats->timeSinceStart);

                if (!mGlobalStats->contestActive) {
                    mStats.clear();
                }
            }
            else // it's a team:
            {
                StatsPacket *t = GetStats(teamnr);
                if (!t)
                {
                    t = new StatsPacket();
                    mStats[teamnr] = t;
                }
                t->unjudged = line.GetInt(',');
                t->positive = line.GetInt(',');
                t->negative = line.GetInt(',');
                ILOG_DEBUG(" TEAM: " << teamnr << " - pos=" << t->positive << " - neg=" << t->negative);
            }
        }
    }

private:
    std::string mVidolServer;
    std::string mSecretPassword;
    int         mVidolPort;
    TCPSocket   *mSocket;
    bool        mConnected;
    queue <ShotPacket*> mShotQueue;

    GlobalStats *mGlobalStats;
    map <int, StatsPacket*> mStats;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(ServerConnector, Application.Videolympics);


}
}
}

#endif

---