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.h"
#include "USingleProcess.h"
// --------------------------------------------------------------------------
namespace uniset
{
    // -----------------------------------------------------------------------------
    // -----------------------------------------------------------------------------
    class RRDServer:
        private USingleProcess,
        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 );
 
            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.
            // То создаём vector<> для последовательного прохода по элементам в нужном порядке
            // и unordered_map<> для быстрого доступа к элементам в sensorInfo
            // При этом используем shared_ptr чтобы элементы указывали на один и тот же DSInfo
            typedef std::unordered_map<uniset::ObjectId, std::shared_ptr<DSInfo>> DSMap;
            typedef std::vector<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::vector<RRDInfo> RRDList;
 
            RRDList rrdlist;
 
        private:
 
            std::string prefix;
    };
    // --------------------------------------------------------------------------
} // end of namespace uniset
// -----------------------------------------------------------------------------
#endif // _RRDServer_H_
// -----------------------------------------------------------------------------