RRDServer.h

/*
 * Copyright (c) 2015 Pavel Vainerman.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, version 2.1.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
// -----------------------------------------------------------------------------
#ifndef _RRDServer_H_
#define _RRDServer_H_
// -----------------------------------------------------------------------------
#include <unordered_map>
#include <list>
#include <memory>
#include "UObject_SK.h"
#include "SMInterface.h"
#include "SharedMemory.h"
#include "extensions/Extensions.h"
// --------------------------------------------------------------------------
namespace uniset
{
// -----------------------------------------------------------------------------
class RRDServer:
    public UObject_SK
{
    public:
        RRDServer( uniset::ObjectId objId, xmlNode* cnode, uniset::ObjectId shmID, const std::shared_ptr<SharedMemory>& ic = nullptr,
                   const std::string& prefix = "rrd" );
        virtual ~RRDServer();

        static std::shared_ptr<RRDServer> init_rrdstorage( int argc, const char* const* argv,
                uniset::ObjectId shmID, const std::shared_ptr<SharedMemory>& ic = nullptr,
                const std::string& prefix = "rrd" );

        static void help_print( int argc, const char* const* argv );

        inline std::shared_ptr<LogAgregator> getLogAggregator()
        {
            return loga;
        }
        inline std::shared_ptr<DebugStream> log()
        {
            return mylog;
        }

        const size_t RRD_MAX_DSNAME_LEN = 19; 
    protected:
        RRDServer();

        virtual void askSensors( UniversalIO::UIOCommand cmd ) override;
        virtual void sensorInfo( const uniset::SensorMessage* sm ) override;
        virtual void timerInfo( const uniset::TimerMessage* tm ) override;
        virtual void sysCommand( const uniset::SystemMessage* sm ) override;

        void initRRD( xmlNode* cnode, int tmID );
        virtual void step() override;

        std::shared_ptr<SMInterface> shm;

        struct DSInfo
        {
            uniset::ObjectId sid;
            std::string dsname;
            long value;

            DSInfo( uniset::ObjectId id, const std::string& dsname, long defval ):
                sid(id), dsname(dsname), value(defval) {}
        };

        // Т.к. RRD требует чтобы данные записывались именно в том порядке в котором они были добавлены
        // при инициализации и при этом, нам нужен быстрый доступ в обработчике sensorInfo.
        // То создаём list<> для последовательного прохода по элементам в нужном порядке
        // и unordered_map<> для быстрого доступа к элементам в sensorInfo
        // При этом используем shared_ptr чтобы элементы указывали на один и тот же DSInfo
        typedef std::unordered_map<uniset::ObjectId, std::shared_ptr<DSInfo>> DSMap;
        typedef std::list<std::shared_ptr<DSInfo>> DSList;

        struct RRDInfo
        {
            std::string filename;
            long tid;
            long sec;
            DSMap dsmap;
            DSList dslist;

            RRDInfo( const std::string& fname, long tmID, long sec, const DSList& lst );
        };

        typedef std::list<RRDInfo> RRDList;

        RRDList rrdlist;

    private:

        std::string prefix;
};
// --------------------------------------------------------------------------
} // end of namespace uniset
// -----------------------------------------------------------------------------
#endif // _RRDServer_H_
// -----------------------------------------------------------------------------