LogAgregator.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 LogAgregator_H_
#define LogAgregator_H_
// -------------------------------------------------------------------------
#include <string>
#include <memory>
#include <regex>
#include <list>
#include <vector>
#include <unordered_map>
#include "DebugStream.h"
#include "LogServerTypes.h"
// -------------------------------------------------------------------------
namespace uniset
{
    // -------------------------------------------------------------------------
    /* Т.к. в других агрегаторах может тоже встречаться такие же логи, приходится отдельно вести
     * учёт подключений (conmap) и подключаться к потокам напрямую, а не к агрегатору
     * иначе будет происходить дублирование информации на экране (логи смешиваются от разных агрегаторов)
    */
    class LogAgregator:
        public DebugStream
    {
        public:
 
            static const std::string sep; /*< разделитель для имён подчинённых агрегаторов ('/') */
 
            explicit LogAgregator( const std::string& name, Debug::type t );
            explicit LogAgregator( const std::string& name = "" );
 
            virtual ~LogAgregator();
 
            virtual void logFile( const std::string& f, bool truncate = false ) override;
 
            void add( std::shared_ptr<LogAgregator> log, const std::string& lname = "" );
            void add( std::shared_ptr<DebugStream> log, const std::string& lname = "" );
 
            std::shared_ptr<DebugStream> create( const std::string& logname );
 
            // Управление "подчинёнными" логами
            void addLevel( const std::string& logname, Debug::type t );
            void delLevel( const std::string& logname, Debug::type t );
            void level( const std::string& logname, Debug::type t );
            void offLogFile( const std::string& logname );
            void onLogFile( const std::string& logname );
 
            // найти лог.. (по полному составному имени)
            std::shared_ptr<DebugStream> getLog( const std::string& logname );
            bool logExist( std::shared_ptr<DebugStream>& l ) const;
            std::shared_ptr<DebugStream> findByLogName( const std::string& logname ) const;
            // -------------------------------------------------------------------------
 
            struct iLog
            {
                iLog( const std::shared_ptr<DebugStream>& l, const std::string& nm ): log(l), name(nm) {}
                std::shared_ptr<DebugStream> log;
                std::string name;
 
                // для сортировки "по алфавиту"
                inline bool operator < ( const iLog& r ) const
                {
                    return name < r.name;
                }
            };
 
            std::list<iLog> getLogList() const;
            std::list<iLog> getLogList( const std::string& regexp_str ) const;
 
            friend std::ostream& operator<<(std::ostream& os, LogAgregator& la );
            friend std::ostream& operator<<(std::ostream& os, std::shared_ptr<LogAgregator> la );
 
            static std::vector<std::string> splitFirst( const std::string& lname, const std::string s = "/" );
 
            std::ostream& printLogList( std::ostream& os, const std::string& regexp_str = "" ) const;
            static std::ostream& printLogList( std::ostream& os, std::list<iLog>& lst );
 
        protected:
            void logOnEvent( const std::string& s );
            void addLog( std::shared_ptr<DebugStream> l, const std::string& lname, bool connect );
            void addLogAgregator( std::shared_ptr<LogAgregator> la, const std::string& lname );
 
            // поиск лога по составному логу.."agregator/agregator2/.../logname"
            std::shared_ptr<DebugStream> findLog( const std::string& lname ) const;
 
            // вывод в виде "дерева"
            std::ostream& printTree(std::ostream& os, const std::string& g_tab = "") const;
 
            // получить список с именами (длинными) и с указателями на логи
            std::list<iLog> makeLogNameList( const std::string& prefix ) const;
 
            // получить список по именам удовлетворяющим регулярному выражению (рекурсивная функция)
            void getListByLogNameWithRule( std::list<iLog>& lst, const std::regex& rule, const std::string& prefix ) const;
 
        private:
            typedef std::unordered_map<std::string, std::shared_ptr<DebugStream>> LogMap;
            LogMap lmap;
 
            typedef std::unordered_map<std::shared_ptr<DebugStream>, sigc::connection> ConnectionMap;
            ConnectionMap conmap;
    };
    // -------------------------------------------------------------------------
} // end of uniset namespace
// -------------------------------------------------------------------------
#endif // LogAgregator_H_
// -------------------------------------------------------------------------