130 lines
3.3 KiB
C++
130 lines
3.3 KiB
C++
/**
|
|
* cdrpgsql.cpp
|
|
* This file is part of the YATE Project http://YATE.null.ro
|
|
*
|
|
* Write the CDR to a PostgreSQL database
|
|
*/
|
|
|
|
#include <telengine.h>
|
|
|
|
#include <stdio.h>
|
|
#include <libpq-fe.h>
|
|
|
|
using namespace TelEngine;
|
|
|
|
static PGconn *conn=0;
|
|
Mutex dbmutex;
|
|
|
|
class CdrPgsqlHandler : public MessageHandler
|
|
{
|
|
public:
|
|
CdrPgsqlHandler(const char *name)
|
|
: MessageHandler(name) { }
|
|
virtual bool received(Message &msg);
|
|
private:
|
|
};
|
|
|
|
bool CdrPgsqlHandler::received(Message &msg)
|
|
{
|
|
// const char *calltime = c_safe(msg.getValue("time"));
|
|
const char *channel = c_safe(msg.getValue("channel"));
|
|
const char *called = c_safe(msg.getValue("called"));
|
|
const char *caller = c_safe(msg.getValue("caller"));
|
|
const char *billtime = c_safe(msg.getValue("billtime"));
|
|
const char *ringtime = c_safe(msg.getValue("ringtime"));
|
|
const char *duration = c_safe(msg.getValue("duration"));
|
|
const char *status = c_safe(msg.getValue("status"));
|
|
|
|
Lock lock(dbmutex);
|
|
if (!conn)
|
|
return false;
|
|
|
|
char buffer[2048];
|
|
snprintf(buffer,sizeof(buffer),"INSERT INTO cdr"
|
|
" (channel,caller,called,billtime,ringtime,duration,status)"
|
|
" VALUES ('%s','%s','%s','%s','%s','%s','%s')",
|
|
channel,caller,called,billtime,ringtime,duration,status);
|
|
|
|
PGresult *respgsql = PQexec(conn,buffer);
|
|
if (!respgsql || PQresultStatus(respgsql) != PGRES_COMMAND_OK)
|
|
Debug(DebugFail,"Failed to insert in database: %s",
|
|
PQerrorMessage(conn));
|
|
return false;
|
|
};
|
|
|
|
|
|
class StatusHandler : public MessageHandler
|
|
{
|
|
public:
|
|
StatusHandler(const char *name, unsigned prio = 1)
|
|
: MessageHandler(name,prio) { }
|
|
virtual bool received(Message &msg);
|
|
};
|
|
|
|
bool StatusHandler::received(Message &msg)
|
|
{
|
|
// msg.addParam("mod","cdrpgsql");
|
|
msg.retValue() << "CdrPgsql,conn=" << (conn != 0) <<"\n";
|
|
return false;
|
|
}
|
|
|
|
|
|
class CdrPgsqlPlugin : public Plugin
|
|
{
|
|
public:
|
|
CdrPgsqlPlugin();
|
|
~CdrPgsqlPlugin();
|
|
virtual void initialize();
|
|
private:
|
|
CdrPgsqlHandler *m_handler;
|
|
};
|
|
|
|
CdrPgsqlPlugin::CdrPgsqlPlugin()
|
|
: m_handler(0)
|
|
{
|
|
Output("Loaded module CdrFile");
|
|
}
|
|
|
|
CdrPgsqlPlugin::~CdrPgsqlPlugin()
|
|
{
|
|
if (conn) {
|
|
PQfinish(conn);
|
|
conn = 0;
|
|
}
|
|
}
|
|
|
|
void CdrPgsqlPlugin::initialize()
|
|
{
|
|
char *pgoptions=NULL,
|
|
*pgtty=NULL;
|
|
Output("Initializing module Cdr for PostgreSQL");
|
|
Configuration cfg(Engine::configFile("cdrpgsql"));
|
|
const char *pghost = c_safe(cfg.getValue("general","host","localhost"));
|
|
const char *pgport = c_safe(cfg.getValue("general","port","5432"));
|
|
const char *dbName = c_safe(cfg.getValue("general","database","yate"));
|
|
const char *dbUser = c_safe(cfg.getValue("general","user","postgres"));
|
|
const char *dbPass = c_safe(cfg.getValue("general","password"));
|
|
|
|
Lock lock(dbmutex);
|
|
if (conn)
|
|
PQfinish(conn);
|
|
conn = PQsetdbLogin(pghost,pgport,pgoptions,pgtty,dbName,dbUser,dbPass);
|
|
if (PQstatus(conn) == CONNECTION_BAD) {
|
|
Debug(DebugFail, "Connection to database '%s' failed.", dbName);
|
|
Debug(DebugFail, "%s", PQerrorMessage(conn));
|
|
PQfinish(conn);
|
|
conn = 0;
|
|
return;
|
|
}
|
|
if (!m_handler) {
|
|
Output("Installing Cdr for PostgreSQL handler");
|
|
m_handler = new CdrPgsqlHandler("cdr");
|
|
Engine::install(m_handler);
|
|
Engine::install(new StatusHandler("status"));
|
|
}
|
|
}
|
|
|
|
INIT_PLUGIN(CdrPgsqlPlugin);
|
|
|
|
/* vi: set ts=8 sw=4 sts=4 noet: */
|