/*
* Reconnect if connection is broken
*/
static int reconnect(const db1_con_t* _h)
{
int ret = 0;
SQLCHAR outstr[1024];
SQLSMALLINT outstrlen;
char conn_str[MAX_CONN_STR_LEN];
LM_ERR("Attempting DB reconnect\n");
/* Disconnect */
SQLDisconnect (CON_CONNECTION(_h));
/* Reconnect */
if (!db_unixodbc_build_conn_str(CON_ID(_h), conn_str)) {
LM_ERR("failed to build connection string\n");
return ret;
}
ret = SQLDriverConnect(CON_CONNECTION(_h), (void *)1,
(SQLCHAR*)conn_str, SQL_NTS, outstr, sizeof(outstr),
&outstrlen, SQL_DRIVER_COMPLETE);
if (!SQL_SUCCEEDED(ret)) {
LM_ERR("failed to connect\n");
db_unixodbc_extract_error("SQLDriverConnect", CON_CONNECTION(_h),
SQL_HANDLE_DBC, NULL);
return ret;
}
ret = SQLAllocHandle(SQL_HANDLE_STMT, CON_CONNECTION(_h),
&CON_RESULT(_h));
if (!SQL_SUCCEEDED(ret)) {
LM_ERR("Statement allocation error %d\n", (int)(long)CON_CONNECTION(_h));
db_unixodbc_extract_error("SQLAllocStmt", CON_CONNECTION(_h), SQL_HANDLE_DBC,NULL);
return ret;
}
return ret;
}
/*
* Create a new connection structure,
* open the UNIXODBC connection and set reference count to 1
*/
struct my_con* db_unixodbc_new_connection(struct db_id* id)
{
SQLCHAR outstr[1024];
SQLSMALLINT outstrlen;
int ret;
struct my_con* ptr;
char conn_str[MAX_CONN_STR_LEN];
if (!id)
{
LM_ERR("invalid parameter value\n");
return 0;
}
ptr = (struct my_con*)pkg_malloc(sizeof(struct my_con));
if (!ptr)
{
LM_ERR("no more memory left\n");
return 0;
}
memset(ptr, 0, sizeof(struct my_con));
ptr->ref = 1;
// allocate environment handle
ret = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &(ptr->env));
if ((ret != SQL_SUCCESS) && (ret != SQL_SUCCESS_WITH_INFO))
{
LM_ERR("could not alloc a SQL handle\n");
if (ptr) pkg_free(ptr);
return 0;
}
// set the environment
ret = SQLSetEnvAttr(ptr->env, SQL_ATTR_ODBC_VERSION, (void *) SQL_OV_ODBC3, 0);
if ((ret != SQL_SUCCESS) && (ret != SQL_SUCCESS_WITH_INFO))
{
LM_ERR("could not set the environment\n");
goto err1;
}
// allocate connection handle
ret = SQLAllocHandle(SQL_HANDLE_DBC, ptr->env, &(ptr->dbc));
if ((ret != SQL_SUCCESS) && (ret != SQL_SUCCESS_WITH_INFO))
{
LM_ERR("could not alloc a connection handle %d\n", ret);
goto err1;
}
if (!db_unixodbc_build_conn_str(id, conn_str)) {
LM_ERR("failed to build connection string\n");
goto err2;
}
LM_DBG("opening connection: unixodbc://xxxx:xxxx@%s/%s\n", ZSW(id->host),
ZSW(id->database));
ret = SQLDriverConnect(ptr->dbc, NULL, (SQLCHAR*)conn_str, SQL_NTS,
outstr, sizeof(outstr), &outstrlen,
SQL_DRIVER_COMPLETE);
if (SQL_SUCCEEDED(ret))
{
LM_DBG("connection succeeded with reply <%s>\n", outstr);
if (ret == SQL_SUCCESS_WITH_INFO)
{
LM_DBG("driver reported the following diagnostics\n");
db_unixodbc_extract_error("SQLDriverConnect", ptr->dbc, SQL_HANDLE_DBC, NULL);
}
}
else
{
LM_ERR("failed to connect\n");
db_unixodbc_extract_error("SQLDriverConnect", ptr->dbc, SQL_HANDLE_DBC, NULL);
goto err2;
}
ptr->stmt_handle = NULL;
ptr->timestamp = time(0);
ptr->id = id;
return ptr;
err1:
SQLFreeHandle(SQL_HANDLE_ENV, &(ptr->env));
if (ptr) pkg_free(ptr);
return 0;
err2:
SQLFreeHandle(SQL_HANDLE_ENV, &(ptr->env));
SQLFreeHandle(SQL_HANDLE_DBC, &(ptr->dbc));
if (ptr) pkg_free(ptr);
return 0;
}