You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
kamailio/modules/db_cassandra/db_cassandra.c

152 lines
4.0 KiB

/*
* $Id$
*
* CASSANDRA module interface
*
* Copyright (C) 2012 1&1 Internet AG
*
* This file is part of Kamailio, a free SIP server.
*
* Kamailio 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; either version 2 of the License, or
* (at your option) any later version
*
* Kamailio 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, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*
* History:
* --------
* 2012-01 first version (Anca Vamanu)
*/
#include <stdio.h>
#include <stdlib.h>
#include "../../sr_module.h"
#include "../../lib/srdb1/db.h"
#include "../../dprint.h"
#include "dbcassa_base.h"
#include "dbcassa_table.h"
unsigned int cassa_conn_timeout= 1000;
unsigned int cassa_send_timeout= 2000;
unsigned int cassa_recv_timeout= 4000;
unsigned int cassa_retries= 1;
unsigned int cassa_auto_reconnect = 1;
static int cassa_mod_init(void);
static void mod_destroy(void);
int db_cassa_bind_api(db_func_t *dbb);
str dbcassa_schema_path={0, 0};
MODULE_VERSION
/*
* database module interface
*/
static cmd_export_t cmds[] = {
{"db_bind_api", (cmd_function)db_cassa_bind_api, 0, 0, 0},
{0, 0, 0, 0, 0}
};
static param_export_t params[] = {
{"schema_path", PARAM_STR, &dbcassa_schema_path.s},
{"connect_timeout", PARAM_INT, &cassa_conn_timeout},
{"send_timeout", PARAM_INT, &cassa_send_timeout},
{"receive_timeout", PARAM_INT, &cassa_recv_timeout},
{"retries", PARAM_INT, &cassa_retries},
{"auto_reconnect", INT_PARAM, &cassa_auto_reconnect},
{0, 0, 0}
};
struct module_exports exports = {
"db_cassandra",
DEFAULT_DLFLAGS, /* dlopen flags */
cmds,
params, /* module parameters */
0, /* exported statistics */
0, /* exported MI functions */
0, /* exported pseudo-variables */
0, /* extra processes */
cassa_mod_init, /* module initialization function */
0, /* response function*/
mod_destroy, /* destroy function */
0 /* per-child init function */
};
static int cassa_mod_init(void)
{
if(!dbcassa_schema_path.s) {
LM_ERR("Set the schema_path parameter to the path of the directory"
" where the table schemas are found (they must be described in cassa special format)\n");
return -1;
}
dbcassa_schema_path.len = strlen(dbcassa_schema_path.s);
return dbcassa_read_table_schemas();
}
db1_con_t *db_cassa_init(const str* _url)
{
return db_do_init(_url, (void* (*)()) db_cassa_new_connection);
}
/*!
* \brief Close database when the database is no longer needed
* \param _h closed connection, as returned from db_cassa_init
* \note free all memory and resources
*/
void db_cassa_close(db1_con_t* _h)
{
db_do_close(_h, (void (*)()) db_cassa_free_connection);
}
/*
* Store name of table that will be used by
* subsequent database functions
*/
int db_cassa_use_table(db1_con_t* _h, const str* _t)
{
return db_use_table(_h, _t);
}
int db_cassa_bind_api(db_func_t *dbb)
{
if(dbb==NULL)
return -1;
memset(dbb, 0, sizeof(db_func_t));
dbb->use_table = db_cassa_use_table;
dbb->init = db_cassa_init;
dbb->close = db_cassa_close;
dbb->query = db_cassa_query;
dbb->free_result = db_cassa_free_result;
dbb->insert = db_cassa_insert;
dbb->replace = db_cassa_replace;
dbb->insert_update = db_cassa_insert;
dbb->delete = db_cassa_delete;
dbb->update = db_cassa_update;
dbb->raw_query = db_cassa_raw_query;
return 0;
}
static void mod_destroy(void)
{
dbcassa_destroy_htable();
}