db_cassandra.c

/*
 * $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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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);
}


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_insert;
        dbb->insert_update    = db_cassa_insert;
        dbb->delete           = db_cassa_delete;
        dbb->update           = db_cassa_update;

        return 0;
}

static void mod_destroy(void)
{
        dbcassa_destroy_htable();
}