DBServer_MySQL.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 DBServer_MySQL_H_
#define DBServer_MySQL_H_
// --------------------------------------------------------------------------
#include <unordered_map>
#include <queue>
#include "UniSetTypes.h"
#include "MySQLInterface.h"
#include "DBServer.h"
// -------------------------------------------------------------------------
namespace uniset
{
    //------------------------------------------------------------------------------------------
    class DBServer_MySQL:
        public DBServer
    {
        public:
            DBServer_MySQL( uniset::ObjectId id, const std::string& prefix );
            explicit DBServer_MySQL( const std::string& prefix );
            virtual ~DBServer_MySQL();
 
            static std::shared_ptr<DBServer_MySQL> init_dbserver( int argc, const char* const* argv, const std::string& prefix = "mysql" );
 
            static void help_print( int argc, const char* const* argv );
 
            inline std::shared_ptr<LogAgregator> logAggregator()
            {
                return loga;
            }
            inline std::shared_ptr<DebugStream> log()
            {
                return dblog;
            }
 
        protected:
            typedef std::unordered_map<int, std::string> DBTableMap;
 
            virtual void initDBServer() override;
            virtual void initDB( const std::unique_ptr<MySQLInterface>& db ) {};
            virtual void initDBTableMap(DBTableMap& tblMap) {};
 
            virtual void timerInfo( const uniset::TimerMessage* tm ) override;
            virtual void sysCommand( const uniset::SystemMessage* sm ) override;
            virtual void sensorInfo( const uniset::SensorMessage* sm ) override;
            virtual void confirmInfo( const uniset::ConfirmMessage* cmsg ) override;
            virtual void onTextMessage( const uniset::TextMessage* msg ) override;
            virtual std::string getMonitInfo( const std::string& params ) override;
 
            bool writeToBase( const std::string& query );
            void createTables( MySQLInterface* db );
 
            inline std::string tblName( int key )
            {
                return tblMap[key].c_str();
            }
 
            enum Timers
            {
                PingTimer,        
                ReconnectTimer,   
                lastNumberOfTimer
            };
 
            std::unique_ptr<MySQLInterface> db;
            int PingTime = { 15000 };
            int ReconnectTime = { 30000 };
            bool connect_ok = { false };     
            bool activate = { false };
 
            typedef std::queue<std::string> QueryBuffer;
 
            QueryBuffer qbuf;
            size_t qbufSize = { 200 }; // размер буфера сообщений.
            bool lastRemove = { false };
 
            void flushBuffer();
            uniset::uniset_rwmutex mqbuf;
 
        private:
            DBTableMap tblMap;
 
    };
    // ----------------------------------------------------------------------------------
} // end of namespace uniset
//------------------------------------------------------------------------------------------
#endif