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 "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;

        inline void setCmdTimeout( timeout_t msec ) noexcept
        {
            cmdTimeout = msec;
        }

        inline void setSessionLog( Debug::type t ) noexcept
        {
            sessLogLevel = t;
        }
        inline void setMaxSessionCount( int num ) noexcept
        {
            sessMaxCount = num;
        }

        void run( const std::string& addr, Poco::UInt16 port, bool thread = true );
        void terminate();

        inline bool isRunning() const noexcept
        {
            return isrunning;
        }

        bool check( bool restart_if_fail = true );

        void init( const std::string& prefix, xmlNode* cnode = 0 );

        static std::string help_print( const std::string& prefix );

        std::string getShortInfo();

#ifndef DISABLE_REST_API
        Poco::JSON::Object::Ptr httpGetShortInfo();
#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
        {
            return myname;
        }

        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 timeout = { UniSetTimer::WaitUpTime };
        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_
// -------------------------------------------------------------------------