mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
84 lines
2.6 KiB
C++
84 lines
2.6 KiB
C++
/*----------------------------------------------------------------------------*/
|
|
/* Copyright (c) FIRST 2015-2018. All Rights Reserved. */
|
|
/* Open Source Software - may be modified and shared by FRC teams. The code */
|
|
/* must be accompanied by the FIRST BSD license file in the root directory of */
|
|
/* the project. */
|
|
/*----------------------------------------------------------------------------*/
|
|
|
|
#ifndef NTCORE_LOGGERIMPL_H_
|
|
#define NTCORE_LOGGERIMPL_H_
|
|
|
|
#include "CallbackManager.h"
|
|
#include "Handle.h"
|
|
#include "ntcore_cpp.h"
|
|
|
|
namespace nt {
|
|
|
|
namespace impl {
|
|
|
|
struct LoggerListenerData
|
|
: public ListenerData<std::function<void(const LogMessage& msg)>> {
|
|
LoggerListenerData() = default;
|
|
LoggerListenerData(std::function<void(const LogMessage& msg)> callback_,
|
|
unsigned int min_level_, unsigned int max_level_)
|
|
: ListenerData(callback_), min_level(min_level_), max_level(max_level_) {}
|
|
LoggerListenerData(unsigned int poller_uid_, unsigned int min_level_,
|
|
unsigned int max_level_)
|
|
: ListenerData(poller_uid_),
|
|
min_level(min_level_),
|
|
max_level(max_level_) {}
|
|
|
|
unsigned int min_level;
|
|
unsigned int max_level;
|
|
};
|
|
|
|
class LoggerThread
|
|
: public CallbackThread<LoggerThread, LogMessage, LoggerListenerData> {
|
|
public:
|
|
explicit LoggerThread(int inst) : m_inst(inst) {}
|
|
|
|
bool Matches(const LoggerListenerData& listener, const LogMessage& data) {
|
|
return data.level >= listener.min_level && data.level <= listener.max_level;
|
|
}
|
|
|
|
void SetListener(LogMessage* data, unsigned int listener_uid) {
|
|
data->logger = Handle(m_inst, listener_uid, Handle::kLogger).handle();
|
|
}
|
|
|
|
void DoCallback(std::function<void(const LogMessage& msg)> callback,
|
|
const LogMessage& data) {
|
|
callback(data);
|
|
}
|
|
|
|
int m_inst;
|
|
};
|
|
|
|
} // namespace impl
|
|
|
|
class LoggerImpl : public CallbackManager<LoggerImpl, impl::LoggerThread> {
|
|
friend class LoggerTest;
|
|
friend class CallbackManager<LoggerImpl, impl::LoggerThread>;
|
|
|
|
public:
|
|
explicit LoggerImpl(int inst);
|
|
|
|
void Start();
|
|
|
|
unsigned int Add(std::function<void(const LogMessage& msg)> callback,
|
|
unsigned int min_level, unsigned int max_level);
|
|
unsigned int AddPolled(unsigned int poller_uid, unsigned int min_level,
|
|
unsigned int max_level);
|
|
|
|
unsigned int GetMinLevel();
|
|
|
|
void Log(unsigned int level, const char* file, unsigned int line,
|
|
const char* msg);
|
|
|
|
private:
|
|
int m_inst;
|
|
};
|
|
|
|
} // namespace nt
|
|
|
|
#endif // NTCORE_LOGGERIMPL_H_
|