mirror of
https://github.com/inspircd/inspircd.git
synced 2025-03-09 18:49:03 -04:00
125 lines
3.0 KiB
C++
125 lines
3.0 KiB
C++
/*
|
|
* InspIRCd -- Internet Relay Chat Daemon
|
|
*
|
|
* Copyright (C) 2022-2023 Sadie Powell <sadie@witchery.services>
|
|
*
|
|
* This file is part of InspIRCd. InspIRCd is free software: you can
|
|
* redistribute it and/or modify it under the terms of the GNU General Public
|
|
* License as published by the Free Software Foundation, version 2.
|
|
*
|
|
* 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 General Public License for more
|
|
* details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
|
|
#include "inspircd.h"
|
|
#include "modules/sql.h"
|
|
|
|
namespace
|
|
{
|
|
Module* thismod;
|
|
}
|
|
|
|
class SQLQuery final
|
|
: public SQL::Query
|
|
{
|
|
public:
|
|
SQLQuery(Module* mod) ATTR_NOT_NULL(2)
|
|
: SQL::Query(mod)
|
|
{
|
|
}
|
|
|
|
void OnResult(SQL::Result& res) override
|
|
{
|
|
// Nothing to do here.
|
|
}
|
|
|
|
void OnError(const SQL::Error& error) override
|
|
{
|
|
ServerInstance->SNO.WriteGlobalSno('a', "Unable to write to SQL log (query error: {}).", error.ToString());
|
|
}
|
|
};
|
|
|
|
class SQLMethod final
|
|
: public Log::Method
|
|
{
|
|
private:
|
|
std::string query;
|
|
dynamic_reference<SQL::Provider> sql;
|
|
time_t lastwarning = 0;
|
|
|
|
public:
|
|
SQLMethod(const dynamic_reference<SQL::Provider>& s, const std::string& q)
|
|
: query(q)
|
|
, sql(s)
|
|
{
|
|
}
|
|
|
|
void OnLog(time_t time, Log::Level level, const std::string& type, const std::string& message) override
|
|
{
|
|
if (!sql)
|
|
{
|
|
// Only give a warning every 5 minutes to avoid log spam.
|
|
if (ServerInstance->Time() - lastwarning > 300)
|
|
{
|
|
lastwarning = ServerInstance->Time();
|
|
ServerInstance->SNO.WriteGlobalSno('a', "Unable to write to SQL log (database {} not available).", sql->GetId());
|
|
}
|
|
return;
|
|
}
|
|
|
|
SQL::ParamMap params = {
|
|
{ "level", ConvToStr(static_cast<uint8_t>(level)) },
|
|
{ "levelstr", Log::LevelToString(level) },
|
|
{ "message", message },
|
|
{ "time", ConvToStr(time) },
|
|
{ "type", type },
|
|
};
|
|
sql->Submit(new SQLQuery(thismod), query, params);
|
|
}
|
|
};
|
|
|
|
class SQLEngine final
|
|
: public Log::Engine
|
|
{
|
|
public:
|
|
SQLEngine(Module* Creator) ATTR_NOT_NULL(2)
|
|
: Log::Engine(Creator, "sql")
|
|
{
|
|
}
|
|
|
|
Log::MethodPtr Create(const std::shared_ptr<ConfigTag>& tag) override
|
|
{
|
|
dynamic_reference<SQL::Provider> sql(creator, "SQL");
|
|
const std::string dbid = tag->getString("dbid");
|
|
if (!dbid.empty())
|
|
sql.SetProvider("SQL/" + dbid);
|
|
|
|
const std::string query = tag->getString("query", "INSERT INTO ircd_log (time, type, message) VALUES (FROM_UNIXTIME($time), '$type', '$message');", 1);
|
|
return std::make_shared<SQLMethod>(sql, query);
|
|
}
|
|
};
|
|
|
|
class ModuleLogSQL final
|
|
: public Module
|
|
{
|
|
private:
|
|
SQLEngine engine;
|
|
|
|
public:
|
|
ModuleLogSQL()
|
|
: Module(VF_VENDOR, "Provides the ability to log to a SQL database.")
|
|
, engine(this)
|
|
{
|
|
thismod = this;
|
|
}
|
|
|
|
};
|
|
|
|
MODULE_INIT(ModuleLogSQL)
|