LogServer.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 LogServer_H_
#define LogServer_H_
// -------------------------------------------------------------------------
#include <vector>
#include <string>
#include <memory>
#include <unordered_map>
#include <ev++.h>
#include <Poco/JSON/Object.h>
#include "Mutex.h"
#include "UniXML.h"
#include "DebugStream.h"
#include "ThreadCreator.h"
#include "UTCPSocket.h"
#include "CommonEventLoop.h"
#include "LogServerTypes.h"
 
#ifndef DISABLE_REST_API
#include <Poco/JSON/Object.h>
#endif
// -------------------------------------------------------------------------
namespace uniset
{
    // -------------------------------------------------------------------------
    class LogSession;
    class LogAgregator;
    class NullLogSession;
    // -------------------------------------------------------------------------
    // -------------------------------------------------------------------------
    class LogServer:
        protected EvWatcher
    {
        public:
 
            LogServer( std::shared_ptr<DebugStream> log );
            LogServer( std::shared_ptr<LogAgregator> log );
            virtual ~LogServer() noexcept;
 
            void setCmdTimeout( timeout_t msec ) noexcept;
            void setSessionLog( Debug::type t ) noexcept;
            void setMaxSessionCount( size_t num ) noexcept;
 
            // port <= 0: select an available port automatically
            bool async_run( const std::string& addr, int port );
            bool run( const std::string& addr, int port );
 
            void terminate();
 
            bool isRunning() const noexcept;
 
            bool check( bool restart_if_fail = true );
 
            void init( const std::string& prefix, xmlNode* cnode = nullptr, int argc = 0, const char* const argv[] = nullptr );
 
            static std::string help_print( const std::string& prefix );
 
            std::string getShortInfo();
 
#ifndef DISABLE_REST_API
            Poco::JSON::Object::Ptr httpGetShortInfo();
            static Poco::JSON::Object::Ptr httpLogServerInfo( const std::shared_ptr<LogServer>& logserv,
                                                              const std::string& host,
                                                              int port );
#endif
 
        protected:
            LogServer();
 
            virtual void evprepare( const ev::loop_ref& loop ) override;
            virtual void evfinish( const ev::loop_ref& loop ) override;
            virtual std::string wname() const noexcept override;
 
            void ioAccept( ev::io& watcher, int revents );
            void sessionFinished( LogSession* s );
            void saveDefaultLogLevels( const std::string& logname );
            void restoreDefaultLogLevels( const std::string& logname );
            std::string onCommand( LogSession* s, LogServerTypes::Command cmd, const std::string& logname );
 
        private:
 
            timeout_t cmdTimeout = { 2000 };
            Debug::type sessLogLevel = { Debug::NONE };
            size_t sessMaxCount = { 10 };
 
            typedef std::vector< std::shared_ptr<LogSession> > SessionList;
            SessionList slist;
            uniset::uniset_rwmutex mutSList;
 
            DebugStream mylog;
            ev::io io;
 
            // делаем loop общим.. одним на всех!
            static CommonEventLoop loop;
 
            std::shared_ptr<UTCPSocket> sock;
            std::shared_ptr<DebugStream> elog; // eventlog..
 
            // map с уровнями логов по умолчанию (инициализируются при создании первой сессии),
            // (они необходимы для восстановления настроек после завершения всех (!) сессий)
            // т.к. shared_ptr-ов может быть много, то в качестве ключа используем указатель на "реальный объект"(внутри shared_ptr)
            // но только для этого(!), пользоваться этим указателем ни в коем случае нельзя (и нужно проверять shared_ptr на существование)
            std::unordered_map< DebugStream*, Debug::type > defaultLogLevels;
 
            std::string myname = { "LogServer" };
            std::string addr = { "" };
            Poco::UInt16 port = { 0 };
 
            std::atomic_bool isrunning = { false };
    };
    // -------------------------------------------------------------------------
} // end of uniset namespace
// -------------------------------------------------------------------------
#endif // LogServer_H_
// -------------------------------------------------------------------------