dbt_table_p dbt_load_file(const str *tbn, const str *dbn)
{
FILE *fin=NULL;
char path[512], buf[4096];
int c, crow, ccol, bp, sign, max_auto;
dbt_val_t dtval;
dbt_table_p dtp = NULL;
dbt_column_p colp, colp0 = NULL;
dbt_row_p rowp, rowp0 = NULL;
enum {DBT_FLINE_ST, DBT_NLINE_ST, DBT_DATA_ST} state;
LM_DBG("request for table [%.*s]\n", tbn->len, tbn->s);
if(!tbn || !tbn->s || tbn->len<=0 || tbn->len>=255)
return NULL;
path[0] = 0;
if(dbn && dbn->s && dbn->len>0)
{
LM_DBG("db is [%.*s]\n", dbn->len, dbn->s);
if(dbn->len+tbn->len<511)
{
strncpy(path, dbn->s, dbn->len);
path[dbn->len] = '/';
strncpy(path+dbn->len+1, tbn->s, tbn->len);
path[dbn->len+tbn->len+1] = 0;
}
}
if(path[0] == 0)
{
strncpy(path, tbn->s, tbn->len);
path[tbn->len] = 0;
}
LM_DBG("loading file [%s]\n", path);
fin = fopen(path, "rt");
if(!fin)
return NULL;
dtp = dbt_table_new(tbn, dbn, path);
if(!dtp)
goto done;
state = DBT_FLINE_ST;
crow = ccol = -1;
colp = colp0 = NULL;
rowp = rowp0 = NULL;
c = fgetc(fin);
max_auto = 0;
while(c!=EOF)
{
switch(state)
{
case DBT_FLINE_ST:
//LM_DBG("state FLINE!\n");
bp = 0;
while(c==DBT_DELIM_C)
c = fgetc(fin);
if(c==DBT_DELIM_R && !colp0)
goto clean;
if(c==DBT_DELIM_R)
{
if(dtp->nrcols <= 0)
goto clean;
dtp->colv = (dbt_column_p*)
shm_malloc(dtp->nrcols*sizeof(dbt_column_p));
if(!dtp->colv)
goto clean;
colp0 = dtp->cols;
for(ccol=0; ccol<dtp->nrcols && colp0; ccol++)
{
dtp->colv[ccol] = colp0;
colp0 = colp0->next;
}
state = DBT_NLINE_ST;
break;
}
while(c!=DBT_DELIM_C && c!='(' && c!=DBT_DELIM_R)
{
if(c==EOF)
goto clean;
buf[bp++] = c;
c = fgetc(fin);
}
colp = dbt_column_new(buf, bp);
if(!colp)
goto clean;
//LM_DBG("new col [%.*s]\n", bp, buf);
while(c==DBT_DELIM_C)
c = fgetc(fin);
if(c!='(')
goto clean;
c = fgetc(fin);
while(c==DBT_DELIM_C)
c = fgetc(fin);
switch(c)
{
case 's':
case 'S':
colp->type = DB_STR;
LM_DBG("column[%d] is STR!\n", ccol+1);
break;
case 'i':
case 'I':
colp->type = DB_INT;
LM_DBG("column[%d] is INT!\n", ccol+1);
break;
case 'l':
case 'L':
colp->type = DB_BIGINT;
LM_DBG("column[%d] is BIGINT!\n", ccol+1);
break;
case 'd':
case 'D':
colp->type = DB_DOUBLE;
LM_DBG("column[%d] is DOUBLE!\n", ccol+1);
break;
case 'b':
case 'B':
colp->type = DB_BLOB;
LM_DBG("column[%d] is BLOB!\n", ccol+1);
break;
case 't':
case 'T':
colp->type = DB_DATETIME;
LM_DBG("column[%d] is TIME!\n", ccol+1);
break;
default:
LM_DBG("wrong column type!\n");
goto clean;
}
while(c!='\n' && c!=EOF && c!=')' && c!= ',')
{
if(colp->type == DB_STR && (c=='i'|| c=='I'))
{
colp->type = DB_STRING;
LM_DBG("column[%d] is actually STRING!\n", ccol+1);
}
c = fgetc(fin);
}
if(c==',')
{
//LM_DBG("c=%c!\n", c);
c = fgetc(fin);
while(c==DBT_DELIM_C)
c = fgetc(fin);
if(c=='N' || c=='n')
{
//LM_DBG("NULL flag set!\n");
colp->flag |= DBT_FLAG_NULL;
}
else if(colp->type==DB_INT && dtp->auto_col<0
&& (c=='A' || c=='a'))
{
//LM_DBG("AUTO flag set!\n");
colp->flag |= DBT_FLAG_AUTO;
dtp->auto_col = ccol+1;
}
else
goto clean;
while(c!=')' && c!=DBT_DELIM_R && c!=EOF)
c = fgetc(fin);
}
if(c == ')')
{
//LM_DBG("c=%c!\n", c);
if(colp0)
{
colp->prev = colp0;
colp0->next = colp;
}
else
dtp->cols = colp;
colp0 = colp;
dtp->nrcols++;
c = fgetc(fin);
}
else
goto clean;
ccol++;
break;
case DBT_NLINE_ST:
//LM_DBG("state NLINE!\n");
while(c==DBT_DELIM_R)
c = fgetc(fin);
if(rowp)
{
if(dbt_table_check_row(dtp, rowp))
goto clean;
if(!rowp0)
dtp->rows = rowp;
else
{
rowp0->next = rowp;
rowp->prev = rowp0;
}
rowp0 = rowp;
dtp->nrrows++;
}
if(c==EOF)
break;
crow++;
ccol = 0;
rowp = dbt_row_new(dtp->nrcols);
if(!rowp)
goto clean;
state = DBT_DATA_ST;
break;
case DBT_DATA_ST:
//LM_DBG("state DATA!\n");
//while(c==DBT_DELIM)
// c = fgetc(fin);
if(ccol == dtp->nrcols && (c==DBT_DELIM_R || c==EOF))
{
state = DBT_NLINE_ST;
break;
}
if(ccol>= dtp->nrcols)
goto clean;
switch(dtp->colv[ccol]->type)
{
case DB_INT:
case DB_BIGINT:
case DB_DATETIME:
//LM_DBG("INT value!\n");
dtval.val.int_val = 0;
dtval.type = dtp->colv[ccol]->type;
if(c==DBT_DELIM ||
(ccol==dtp->nrcols-1
&& (c==DBT_DELIM_R || c==EOF)))
dtval.nul = 1;
else
{
dtval.nul = 0;
sign = 1;
if(c=='-')
{
sign = -1;
c = fgetc(fin);
}
if(c<'0' || c>'9')
goto clean;
while(c>='0' && c<='9')
{
dtval.val.int_val=dtval.val.int_val*10+c-'0';
c = fgetc(fin);
}
dtval.val.int_val *= sign;
//LM_DBG("data[%d,%d]=%d\n", crow,
// ccol, dtval.val.int_val);
}
if(c!=DBT_DELIM && c!=DBT_DELIM_R && c!=EOF)
goto clean;
if(dbt_row_set_val(rowp,&dtval,dtp->colv[ccol]->type,
ccol))
goto clean;
if(ccol == dtp->auto_col)
max_auto = (max_auto<dtval.val.int_val)?
dtval.val.int_val:max_auto;
break;
case DB_DOUBLE:
//LM_DBG("DOUBLE value!\n");
dtval.val.double_val = 0.0;
dtval.type = DB_DOUBLE;
if(c==DBT_DELIM ||
(ccol==dtp->nrcols-1
&& (c==DBT_DELIM_R || c==EOF)))
dtval.nul = 1;
else
{
dtval.nul = 0;
sign = 1;
if(c=='-')
{
sign = -1;
c = fgetc(fin);
}
if(c<'0' || c>'9')
goto clean;
while(c>='0' && c<='9')
{
dtval.val.double_val = dtval.val.double_val*10
+ c - '0';
c = fgetc(fin);
}
if(c=='.')
{
c = fgetc(fin);
bp = 1;
while(c>='0' && c<='9')
{
bp *= 10;
dtval.val.double_val+=((double)(c-'0'))/bp;
c = fgetc(fin);
}
}
dtval.val.double_val *= sign;
//LM_DBG("data[%d,%d]=%10.2f\n",
// crow, ccol, dtval.val.double_val);
}
if(c!=DBT_DELIM && c!=DBT_DELIM_R && c!=EOF)
goto clean;
if(dbt_row_set_val(rowp,&dtval,DB_DOUBLE,ccol))
goto clean;
break;
case DB_STR:
case DB_STRING:
case DB_BLOB:
//LM_DBG("STR value!\n");
dtval.val.str_val.s = NULL;
dtval.val.str_val.len = 0;
dtval.type = dtp->colv[ccol]->type;
bp = 0;
if(c==DBT_DELIM ||
(ccol == dtp->nrcols-1
&& (c == DBT_DELIM_R || c==EOF)))
dtval.nul = 1;
else
{
dtval.nul = 0;
while(c!=DBT_DELIM && c!=DBT_DELIM_R && c!=EOF)
{
if(c=='\\')
{
c = fgetc(fin);
switch(c)
{
case 'n':
c = '\n';
break;
case 'r':
c = '\r';
break;
case 't':
c = '\t';
break;
case '\\':
c = '\\';
break;
case DBT_DELIM:
c = DBT_DELIM;
break;
case '0':
c = 0;
break;
default:
goto clean;
}
}
buf[bp++] = c;
c = fgetc(fin);
}
dtval.val.str_val.s = buf;
dtval.val.str_val.len = bp;
//LM_DBG("data[%d,%d]=%.*s\n",
/// crow, ccol, bp, buf);
}
if(c!=DBT_DELIM && c!=DBT_DELIM_R && c!=EOF)
goto clean;
if(dbt_row_set_val(rowp,&dtval,dtp->colv[ccol]->type,
ccol))
goto clean;
break;
default:
goto clean;
}
if(c==DBT_DELIM)
c = fgetc(fin);
ccol++;
break; // state DBT_DATA_ST
}
}
if(max_auto)
dtp->auto_val = max_auto;
done:
if(fin)
fclose(fin);
return dtp;
clean:
/// ????? FILL IT IN - incomplete row/column
// memory leak?!?! with last incomplete row
LM_DBG("error at row=%d col=%d c=%c\n", crow+1, ccol+1, c);
if(dtp)
dbt_table_free(dtp);
return NULL;
}
int dbt_query(db1_con_t* _h, db_key_t* _k, db_op_t* _op, db_val_t* _v,
db_key_t* _c, int _n, int _nc, db_key_t _o, db1_res_t** _r)
{
dbt_table_p _tbc = NULL;
dbt_table_p _tbc_temp = NULL;
dbt_row_p _drp = NULL;
dbt_row_p *_res = NULL;
// dbt_result_p _dres = NULL;
int result = 0;
int counter = 0;
int i=0;
int *lkey=NULL, *lres=NULL;
db_key_t *_o_k=NULL; /* columns in order-by */
char *_o_op=NULL; /* operators for oder-by */
int _o_n; /* no of elements in order-by */
int *_o_l=NULL; /* column selection for order-by */
// int _o_nc; /* no of elements in _o_l but not lres */
if(_r)
*_r = NULL;
if ((!_h) || !CON_TABLE(_h))
{
LM_ERR("invalid parameters\n");
return -1;
}
if (_o)
{
if (dbt_parse_orderbyclause(&_o_k, &_o_op, &_o_n, _o) < 0)
return -1;
}
_tbc_temp = dbt_db_get_temp_table(DBT_CON_CONNECTION(_h));
if(!_tbc_temp)
{
LM_ERR("unable to allocate temp table\n");
return -1;
}
/* lock database */
_tbc = dbt_db_get_table(DBT_CON_CONNECTION(_h), CON_TABLE(_h));
if(!_tbc)
{
LM_ERR("table %.*s does not exist!\n", CON_TABLE(_h)->len, CON_TABLE(_h)->s);
dbt_db_del_table(DBT_CON_CONNECTION(_h), &_tbc_temp->name, 0);
return -1;
}
if(!_tbc || _tbc->nrcols < _nc)
{
LM_ERR("table %s not loaded! (too few columns)\n", CON_TABLE(_h)->s);
goto error;
}
if(_k)
{
lkey = dbt_get_refs(_tbc, _k, _n);
if(!lkey)
goto error;
}
if(_c)
{
lres = dbt_get_refs(_tbc, _c, _nc);
if(!lres)
goto error;
}
if(_o_k)
{
_o_l = dbt_get_refs(_tbc, _o_k, _o_n);
if (!_o_l)
goto error;
/* enlarge select-columns lres by all order-by columns, _o_nc is how many */
// if (dbt_mangle_columnselection(&lres, &_nc, &_o_nc, _o_l, _o_n) < 0)
// goto error;
}
/*
LM_DBG("new res with %d cols\n", _nc);
_dres = dbt_result_new(_tbc, lres, _nc);
if(!_dres)
goto error;
*/
dbt_column_p pPrevCol = NULL;
_tbc_temp->colv = (dbt_column_p*) shm_malloc(_nc*sizeof(dbt_column_p));
for(i=0; i < _nc; i++) {
dbt_column_p pCol = dbt_column_new(_tbc->colv[ lres[i] ]->name.s, _tbc->colv[ lres[i] ]->name.len);
pCol->type = _tbc->colv[ lres[i] ]->type;
pCol->flag = _tbc->colv[ lres[i] ]->flag;
if(pPrevCol)
{
pCol->prev = pPrevCol;
pPrevCol->next = pCol;
}
else
_tbc_temp->cols = pCol;
_tbc_temp->colv[i] = pCol;
pPrevCol = pCol;
_tbc_temp->nrcols++;
}
_res = (dbt_row_p*) pkg_malloc(_db_text_max_result_rows * sizeof(dbt_row_p));
if(!_res) {
LM_ERR("no more space to allocate for query rows\n");
goto error;
}
_drp = _tbc->rows;
while(_drp && counter < _db_text_max_result_rows)
{
if(dbt_row_match(_tbc, _drp, lkey, _op, _v, _n))
{
_res[counter] = _drp;
// if(dbt_result_extract_fields(_tbc, _drp, lres, _dres))
// {
// LM_ERR("failed to extract result fields!\n");
// goto clean;
// }
counter++;
}
_drp = _drp->next;
}
if (_o_l)
{
if (counter > 1)
{
if (dbt_sort_result_temp(_res, counter, _o_l, _o_op, _o_n) < 0)
goto error;
}
/* last but not least, remove surplus columns */
// if (_o_nc)
// dbt_project_result(_dres, _o_nc);
}
// copy results to temp table
_tbc_temp->rows = dbt_result_extract_results(_tbc, _res, counter, lres, _nc);
_tbc_temp->nrrows = (_tbc_temp->rows == NULL ? 0 : counter);
dbt_table_update_flags(_tbc, DBT_TBFL_ZERO, DBT_FL_IGN, 1);
/* unlock database */
dbt_release_table(DBT_CON_CONNECTION(_h), CON_TABLE(_h));
// DBT_CON_TEMP_TABLE(_h) = _tbc_temp;
last_temp_table = _tbc_temp;
// dbt_release_table(DBT_CON_CONNECTION(_h), &_tbc_temp->name);
// dbt_result_print(_tbc_temp);
if(lkey)
pkg_free(lkey);
if(lres)
pkg_free(lres);
if(_o_k)
pkg_free(_o_k);
if(_o_op)
pkg_free(_o_op);
if(_o_l)
pkg_free(_o_l);
if(_res)
pkg_free(_res);
if(_r) {
result = dbt_get_result(_r, _tbc_temp);
// dbt_db_del_table(DBT_CON_CONNECTION(_h), &_tbc_temp->name, 1);
if(result != 0)
dbt_result_free(_h, _tbc_temp);
}
return result;
error:
/* unlock database */
dbt_release_table(DBT_CON_CONNECTION(_h), CON_TABLE(_h));
/* delete temp table */
dbt_db_del_table(DBT_CON_CONNECTION(_h), &_tbc_temp->name, 1);
if(_res)
pkg_free(_res);
if(lkey)
pkg_free(lkey);
if(lres)
pkg_free(lres);
if(_o_k)
pkg_free(_o_k);
if(_o_op)
pkg_free(_o_op);
if(_o_l)
pkg_free(_o_l);
LM_ERR("failed to query the table!\n");
return -1;
/*
clean:
dbt_release_table(DBT_CON_CONNECTION(_h), CON_TABLE(_h));
if(lkey)
pkg_free(lkey);
if(lres)
pkg_free(lres);
if(_o_k)
pkg_free(_o_k);
if(_o_op)
pkg_free(_o_op);
if(_o_l)
pkg_free(_o_l);
return -1;
*/
}
