mirror of
https://github.com/postgres/postgres.git
synced 2025-11-03 09:13:20 +03:00
Version 06-30-0248
This commit is contained in:
Byron Nikolaidis
775
src/interfaces/odbc/parse.c
Normal file
775
src/interfaces/odbc/parse.c
Normal file
@@ -0,0 +1,775 @@
|
||||
|
||||
/* Module: parse.c
|
||||
*
|
||||
* Description: This module contains routines related to parsing SQL statements.
|
||||
* This can be useful for two reasons:
|
||||
*
|
||||
* 1. So the query does not actually have to be executed to return data about it
|
||||
*
|
||||
* 2. To be able to return information about precision, nullability, aliases, etc.
|
||||
* in the functions SQLDescribeCol and SQLColAttributes. Currently, Postgres
|
||||
* doesn't return any information about these things in a query.
|
||||
*
|
||||
* Classes: none
|
||||
*
|
||||
* API functions: none
|
||||
*
|
||||
* Comments: See "notice.txt" for copyright and license information.
|
||||
*
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "statement.h"
|
||||
#include "connection.h"
|
||||
#include "qresult.h"
|
||||
#include "pgtypes.h"
|
||||
|
||||
#ifdef UNIX
|
||||
#if !HAVE_STRICMP
|
||||
#define stricmp(s1,s2) strcasecmp(s1,s2)
|
||||
#define strnicmp(s1,s2,n) strncasecmp(s1,s2,n)
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#define FLD_INCR 32
|
||||
#define TAB_INCR 8
|
||||
#define COL_INCR 16
|
||||
|
||||
char *
|
||||
getNextToken(char *s, char *token, int smax, char *delim, char *quote, char *dquote, char *numeric)
|
||||
{
|
||||
int i = 0;
|
||||
int out = 0;
|
||||
char qc, in_escape = FALSE;
|
||||
|
||||
if (smax <= 1)
|
||||
return NULL;
|
||||
|
||||
smax--;
|
||||
|
||||
/* skip leading delimiters */
|
||||
while (isspace(s[i]) || s[i] == ',') {
|
||||
// mylog("skipping '%c'\n", s[i]);
|
||||
i++;
|
||||
}
|
||||
|
||||
if (s[0] == '\0') {
|
||||
token[0] = '\0';
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (quote) *quote = FALSE;
|
||||
if (dquote) *dquote = FALSE;
|
||||
if (numeric) *numeric = FALSE;
|
||||
|
||||
/* get the next token */
|
||||
while ( ! isspace(s[i]) && s[i] != ',' && s[i] != '\0' && out != smax) {
|
||||
|
||||
/* Handle quoted stuff */
|
||||
if ( out == 0 && (s[i] == '\"' || s[i] == '\'')) {
|
||||
qc = s[i];
|
||||
if (qc == '\"') {
|
||||
if (dquote) *dquote = TRUE;
|
||||
}
|
||||
if (qc == '\'') {
|
||||
if (quote) *quote = TRUE;
|
||||
}
|
||||
|
||||
i++; /* dont return the quote */
|
||||
while (s[i] != '\0' && out != smax) {
|
||||
if (s[i] == qc && ! in_escape) {
|
||||
break;
|
||||
}
|
||||
if (s[i] == '\\' && ! in_escape) {
|
||||
in_escape = TRUE;
|
||||
}
|
||||
else {
|
||||
in_escape = FALSE;
|
||||
token[out++] = s[i];
|
||||
}
|
||||
i++;
|
||||
}
|
||||
if (s[i] == qc)
|
||||
i++;
|
||||
break;
|
||||
}
|
||||
|
||||
/* Check for numeric literals */
|
||||
if ( out == 0 && isdigit(s[i])) {
|
||||
if (numeric) *numeric = TRUE;
|
||||
token[out++] = s[i++];
|
||||
while ( isalnum(s[i]) || s[i] == '.')
|
||||
token[out++] = s[i++];
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
if ( ispunct(s[i])) {
|
||||
if (out == 0) {
|
||||
token[out++] = s[i++];
|
||||
break;
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
|
||||
if (out != smax)
|
||||
token[out++] = s[i];
|
||||
|
||||
i++;
|
||||
}
|
||||
|
||||
// mylog("done -- s[%d] = '%c'\n", i, s[i]);
|
||||
|
||||
token[out] = '\0';
|
||||
|
||||
/* find the delimiter */
|
||||
while ( isspace(s[i]))
|
||||
i++;
|
||||
|
||||
/* return the most priority delimiter */
|
||||
if (s[i] == ',') {
|
||||
if (delim) *delim = s[i];
|
||||
}
|
||||
else if (s[i] == '\0') {
|
||||
if (delim) *delim = '\0';
|
||||
}
|
||||
else {
|
||||
if (delim) *delim = ' ';
|
||||
}
|
||||
|
||||
/* skip trailing blanks */
|
||||
while ( isspace(s[i])) {
|
||||
i++;
|
||||
}
|
||||
|
||||
return &s[i];
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
QR_set_num_fields(stmt->result, 14);
|
||||
QR_set_field_info(stmt->result, 0, "TABLE_QUALIFIER", PG_TYPE_TEXT, MAX_INFO_STRING);
|
||||
QR_set_field_info(stmt->result, 1, "TABLE_OWNER", PG_TYPE_TEXT, MAX_INFO_STRING);
|
||||
QR_set_field_info(stmt->result, 2, "TABLE_NAME", PG_TYPE_TEXT, MAX_INFO_STRING);
|
||||
QR_set_field_info(stmt->result, 3, "COLUMN_NAME", PG_TYPE_TEXT, MAX_INFO_STRING);
|
||||
QR_set_field_info(stmt->result, 4, "DATA_TYPE", PG_TYPE_INT2, 2);
|
||||
QR_set_field_info(stmt->result, 5, "TYPE_NAME", PG_TYPE_TEXT, MAX_INFO_STRING);
|
||||
QR_set_field_info(stmt->result, 6, "PRECISION", PG_TYPE_INT4, 4);
|
||||
QR_set_field_info(stmt->result, 7, "LENGTH", PG_TYPE_INT4, 4);
|
||||
QR_set_field_info(stmt->result, 8, "SCALE", PG_TYPE_INT2, 2);
|
||||
QR_set_field_info(stmt->result, 9, "RADIX", PG_TYPE_INT2, 2);
|
||||
QR_set_field_info(stmt->result, 10, "NULLABLE", PG_TYPE_INT2, 2);
|
||||
QR_set_field_info(stmt->result, 11, "REMARKS", PG_TYPE_TEXT, 254);
|
||||
// User defined fields
|
||||
QR_set_field_info(stmt->result, 12, "DISPLAY_SIZE", PG_TYPE_INT4, 4);
|
||||
QR_set_field_info(stmt->result, 13, "FIELD_TYPE", PG_TYPE_INT4, 4);
|
||||
*/
|
||||
|
||||
void
|
||||
getColInfo(COL_INFO *col_info, FIELD_INFO *fi, int k)
|
||||
{
|
||||
if (fi->name[0] == '\0')
|
||||
strcpy(fi->name, QR_get_value_manual(col_info->result, k, 3));
|
||||
|
||||
fi->type = atoi( QR_get_value_manual(col_info->result, k, 13));
|
||||
fi->precision = atoi( QR_get_value_manual(col_info->result, k, 6));
|
||||
fi->length = atoi( QR_get_value_manual(col_info->result, k, 7));
|
||||
fi->nullable = atoi( QR_get_value_manual(col_info->result, k, 10));
|
||||
fi->display_size = atoi( QR_get_value_manual(col_info->result, k, 12));
|
||||
}
|
||||
|
||||
char
|
||||
searchColInfo(COL_INFO *col_info, FIELD_INFO *fi)
|
||||
{
|
||||
int k;
|
||||
char *col;
|
||||
|
||||
|
||||
for (k = 0; k < QR_get_num_tuples(col_info->result); k++) {
|
||||
col = QR_get_value_manual(col_info->result, k, 3);
|
||||
if ( ! strcmp(col, fi->name)) {
|
||||
getColInfo(col_info, fi, k);
|
||||
|
||||
mylog("PARSE: searchColInfo: \n");
|
||||
return TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
|
||||
char
|
||||
parse_statement(StatementClass *stmt)
|
||||
{
|
||||
char token[256];
|
||||
char delim, quote, dquote, numeric, unquoted;
|
||||
char *ptr;
|
||||
char in_select = FALSE, in_distinct = FALSE, in_on = FALSE, in_from = FALSE, in_where = FALSE, in_table = FALSE;
|
||||
char in_field = FALSE, in_expr = FALSE, in_func = FALSE, in_dot = FALSE, in_as = FALSE;
|
||||
int j, i, k, n, blevel = 0;
|
||||
FIELD_INFO **fi;
|
||||
TABLE_INFO **ti;
|
||||
char parse;
|
||||
ConnectionClass *conn = stmt->hdbc;
|
||||
HSTMT hcol_stmt;
|
||||
StatementClass *col_stmt;
|
||||
RETCODE result;
|
||||
|
||||
|
||||
ptr = stmt->statement;
|
||||
fi = stmt->fi;
|
||||
ti = stmt->ti;
|
||||
|
||||
stmt->nfld = 0;
|
||||
stmt->ntab = 0;
|
||||
|
||||
while (ptr = getNextToken(ptr, token, sizeof(token), &delim, "e, &dquote, &numeric)) {
|
||||
|
||||
unquoted = ! ( quote || dquote );
|
||||
|
||||
mylog("unquoted=%d, quote=%d, dquote=%d, numeric=%d, delim='%c', token='%s', ptr='%s'\n", unquoted, quote, dquote, numeric, delim, token, ptr);
|
||||
|
||||
if ( unquoted && ! stricmp(token, "select")) {
|
||||
in_select = TRUE;
|
||||
|
||||
mylog("SELECT\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( unquoted && in_select && ! stricmp(token, "distinct")) {
|
||||
in_distinct = TRUE;
|
||||
|
||||
mylog("DISTINCT\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( unquoted && ! stricmp(token, "into")) {
|
||||
in_select = FALSE;
|
||||
|
||||
mylog("INTO\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( unquoted && ! stricmp(token, "from")) {
|
||||
in_select = FALSE;
|
||||
in_from = TRUE;
|
||||
|
||||
mylog("FROM\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( unquoted && (! stricmp(token, "where") ||
|
||||
! stricmp(token, "union") ||
|
||||
! stricmp(token, "order") ||
|
||||
! stricmp(token, "group") ||
|
||||
! stricmp(token, "having"))) {
|
||||
|
||||
in_select = FALSE;
|
||||
in_from = FALSE;
|
||||
in_where = TRUE;
|
||||
|
||||
mylog("WHERE...\n");
|
||||
break;
|
||||
}
|
||||
|
||||
if (in_select) {
|
||||
|
||||
if ( in_distinct) {
|
||||
mylog("in distinct\n");
|
||||
|
||||
if (unquoted && ! stricmp(token, "on")) {
|
||||
in_on = TRUE;
|
||||
mylog("got on\n");
|
||||
continue;
|
||||
}
|
||||
if (in_on) {
|
||||
in_distinct = FALSE;
|
||||
in_on = FALSE;
|
||||
continue; /* just skip the unique on field */
|
||||
}
|
||||
mylog("done distinct\n");
|
||||
in_distinct = FALSE;
|
||||
}
|
||||
|
||||
if ( in_expr || in_func) { /* just eat the expression */
|
||||
mylog("in_expr=%d or func=%d\n", in_expr, in_func);
|
||||
if (quote || dquote)
|
||||
continue;
|
||||
|
||||
if (in_expr && blevel == 0 && delim == ',') {
|
||||
mylog("**** in_expr and Got comma\n");
|
||||
in_expr = FALSE;
|
||||
in_field = FALSE;
|
||||
}
|
||||
|
||||
else if (token[0] == '(') {
|
||||
blevel++;
|
||||
mylog("blevel++ = %d\n", blevel);
|
||||
}
|
||||
else if (token[0] == ')') {
|
||||
blevel--;
|
||||
mylog("blevel-- = %d\n", blevel);
|
||||
if (delim==',') {
|
||||
in_func = FALSE;
|
||||
in_expr = FALSE;
|
||||
in_field = FALSE;
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( ! in_field) {
|
||||
|
||||
if ( ! token[0])
|
||||
continue;
|
||||
|
||||
if ( ! (stmt->nfld % FLD_INCR)) {
|
||||
mylog("reallocing at nfld=%d\n", stmt->nfld);
|
||||
fi = (FIELD_INFO **) realloc(fi, (stmt->nfld + FLD_INCR) * sizeof(FIELD_INFO *));
|
||||
if ( ! fi) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
stmt->fi = fi;
|
||||
}
|
||||
|
||||
fi[stmt->nfld] = (FIELD_INFO *) malloc( sizeof(FIELD_INFO));
|
||||
if (fi[stmt->nfld] == NULL) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/* Initialize the field info */
|
||||
memset(fi[stmt->nfld], 0, sizeof(FIELD_INFO));
|
||||
|
||||
/* double quotes are for qualifiers */
|
||||
if (dquote)
|
||||
fi[stmt->nfld]->dquote = TRUE;
|
||||
|
||||
if (quote) {
|
||||
fi[stmt->nfld++]->quote = TRUE;
|
||||
continue;
|
||||
}
|
||||
else if (numeric) {
|
||||
mylog("**** got numeric: nfld = %d\n", stmt->nfld);
|
||||
fi[stmt->nfld]->numeric = TRUE;
|
||||
}
|
||||
else if (token[0] == '(') { /* expression */
|
||||
mylog("got EXPRESSION\n");
|
||||
fi[stmt->nfld++]->expr = TRUE;
|
||||
in_expr = TRUE;
|
||||
blevel = 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
else {
|
||||
strcpy(fi[stmt->nfld]->name, token);
|
||||
fi[stmt->nfld]->dot[0] = '\0';
|
||||
}
|
||||
mylog("got field='%s', dot='%s'\n", fi[stmt->nfld]->name, fi[stmt->nfld]->dot);
|
||||
|
||||
if (delim == ',') {
|
||||
mylog("comma (1)\n");
|
||||
}
|
||||
else {
|
||||
in_field = TRUE;
|
||||
}
|
||||
stmt->nfld++;
|
||||
continue;
|
||||
}
|
||||
|
||||
/**************************/
|
||||
/* We are in a field now */
|
||||
/**************************/
|
||||
if (in_dot) {
|
||||
stmt->nfld--;
|
||||
strcpy(fi[stmt->nfld]->dot, fi[stmt->nfld]->name);
|
||||
strcpy(fi[stmt->nfld]->name, token);
|
||||
stmt->nfld++;
|
||||
in_dot = FALSE;
|
||||
|
||||
if (delim == ',') {
|
||||
mylog("in_dot: got comma\n");
|
||||
in_field = FALSE;
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (in_as) {
|
||||
stmt->nfld--;
|
||||
strcpy(fi[stmt->nfld]->alias, token);
|
||||
mylog("alias for field '%s' is '%s'\n", fi[stmt->nfld]->name, fi[stmt->nfld]->alias);
|
||||
in_as = FALSE;
|
||||
in_field = FALSE;
|
||||
|
||||
stmt->nfld++;
|
||||
|
||||
if (delim == ',') {
|
||||
mylog("comma(2)\n");
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Function */
|
||||
if (token[0] == '(') {
|
||||
in_func = TRUE;
|
||||
blevel = 1;
|
||||
fi[stmt->nfld-1]->func = TRUE;
|
||||
/* name will have the function name -- maybe useful some day */
|
||||
mylog("**** got function = '%s'\n", fi[stmt->nfld-1]->name);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (token[0] == '.') {
|
||||
in_dot = TRUE;
|
||||
mylog("got dot\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
if ( ! stricmp(token, "as")) {
|
||||
in_as = TRUE;
|
||||
mylog("got AS\n");
|
||||
continue;
|
||||
}
|
||||
|
||||
/* otherwise, its probably an expression */
|
||||
in_expr = TRUE;
|
||||
fi[stmt->nfld-1]->expr = TRUE;
|
||||
fi[stmt->nfld-1]->name[0] = '\0';
|
||||
mylog("*** setting expression\n");
|
||||
|
||||
}
|
||||
|
||||
if (in_from) {
|
||||
|
||||
if ( ! in_table) {
|
||||
if ( ! token[0])
|
||||
continue;
|
||||
|
||||
if ( ! (stmt->ntab % TAB_INCR)) {
|
||||
ti = (TABLE_INFO **) realloc(ti, (stmt->ntab + TAB_INCR) * sizeof(TABLE_INFO *));
|
||||
if ( ! ti) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
stmt->ti = ti;
|
||||
}
|
||||
ti[stmt->ntab] = (TABLE_INFO *) malloc(sizeof(TABLE_INFO));
|
||||
if (ti[stmt->ntab] == NULL) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
ti[stmt->ntab]->alias[0] = '\0';
|
||||
|
||||
strcpy(ti[stmt->ntab]->name, token);
|
||||
mylog("got table = '%s'\n", ti[stmt->ntab]->name);
|
||||
|
||||
if (delim == ',') {
|
||||
mylog("more than 1 tables\n");
|
||||
}
|
||||
else {
|
||||
in_table = TRUE;
|
||||
}
|
||||
stmt->ntab++;
|
||||
continue;
|
||||
}
|
||||
|
||||
strcpy(ti[stmt->ntab-1]->alias, token);
|
||||
mylog("alias for table '%s' is '%s'\n", ti[stmt->ntab-1]->name, ti[stmt->ntab-1]->alias);
|
||||
in_table = FALSE;
|
||||
if (delim == ',') {
|
||||
mylog("more than 1 tables\n");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/*************************************************/
|
||||
/* Resolve any possible field names with tables */
|
||||
/*************************************************/
|
||||
|
||||
parse = TRUE;
|
||||
|
||||
/* Resolve field names with tables */
|
||||
for (i = 0; i < stmt->nfld; i++) {
|
||||
|
||||
if (fi[i]->func || fi[i]->expr || fi[i]->numeric) {
|
||||
fi[i]->ti = NULL;
|
||||
fi[i]->type = -1;
|
||||
parse = FALSE;
|
||||
continue;
|
||||
}
|
||||
|
||||
else if (fi[i]->quote) { /* handle as text */
|
||||
fi[i]->ti = NULL;
|
||||
fi[i]->type = PG_TYPE_TEXT;
|
||||
fi[i]->precision = 0;
|
||||
continue;
|
||||
}
|
||||
|
||||
/* its a dot, resolve to table or alias */
|
||||
else if (fi[i]->dot[0]) {
|
||||
for (k = 0; k < stmt->ntab; k++) {
|
||||
if ( ! stricmp(ti[k]->name, fi[i]->dot)) {
|
||||
fi[i]->ti = ti[k];
|
||||
break;
|
||||
}
|
||||
else if ( ! stricmp(ti[k]->alias, fi[i]->dot)) {
|
||||
fi[i]->ti = ti[k];
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (stmt->ntab == 1)
|
||||
fi[i]->ti = ti[0];
|
||||
}
|
||||
|
||||
mylog("--------------------------------------------\n");
|
||||
mylog("nfld=%d, ntab=%d\n", stmt->nfld, stmt->ntab);
|
||||
|
||||
for (i=0; i < stmt->nfld; i++) {
|
||||
mylog("Field %d: expr=%d, func=%d, quote=%d, dquote=%d, numeric=%d, name='%s', alias='%s', dot='%s'\n", i, fi[i]->expr, fi[i]->func, fi[i]->quote, fi[i]->dquote, fi[i]->numeric, fi[i]->name, fi[i]->alias, fi[i]->dot);
|
||||
if (fi[i]->ti)
|
||||
mylog(" ----> table_name='%s', table_alias='%s'\n", fi[i]->ti->name, fi[i]->ti->alias);
|
||||
}
|
||||
|
||||
for (i=0; i < stmt->ntab; i++) {
|
||||
mylog("Table %d: name='%s', alias='%s'\n", i, ti[i]->name, ti[i]->alias);
|
||||
}
|
||||
|
||||
|
||||
/******************************************************/
|
||||
/* Now save the SQLColumns Info for the parse tables */
|
||||
/******************************************************/
|
||||
|
||||
|
||||
/* Call SQLColumns for each table and store the result */
|
||||
for (i = 0; i < stmt->ntab; i++) {
|
||||
|
||||
/* See if already got it */
|
||||
char found = FALSE;
|
||||
for (k = 0; k < conn->ntables; k++) {
|
||||
if ( ! stricmp(conn->col_info[k]->name, ti[i]->name)) {
|
||||
mylog("FOUND col_info table='%s'\n", ti[i]->name);
|
||||
found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if ( ! found) {
|
||||
|
||||
mylog("PARSE: Getting SQLColumns for table[%d]='%s'\n", i, ti[i]->name);
|
||||
|
||||
result = SQLAllocStmt( stmt->hdbc, &hcol_stmt);
|
||||
if((result != SQL_SUCCESS) && (result != SQL_SUCCESS_WITH_INFO)) {
|
||||
stmt->errormsg = "SQLAllocStmt failed in parse_statement for columns.";
|
||||
stmt->errornumber = STMT_NO_MEMORY_ERROR;
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
col_stmt = (StatementClass *) hcol_stmt;
|
||||
col_stmt->internal = TRUE;
|
||||
|
||||
result = SQLColumns(hcol_stmt, "", 0, "", 0,
|
||||
ti[i]->name, (SWORD) strlen(ti[i]->name), "", 0);
|
||||
|
||||
mylog(" Past SQLColumns\n");
|
||||
if (result == SQL_SUCCESS) {
|
||||
mylog(" Success\n");
|
||||
if ( ! (conn->ntables % COL_INCR)) {
|
||||
|
||||
mylog("PARSE: Allocing col_info at ntables=%d\n", conn->ntables);
|
||||
|
||||
conn->col_info = (COL_INFO **) realloc(conn->col_info, (conn->ntables + COL_INCR) * sizeof(COL_INFO *));
|
||||
if ( ! conn->col_info) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
}
|
||||
|
||||
mylog("PARSE: malloc at conn->col_info[%d]\n", conn->ntables);
|
||||
conn->col_info[conn->ntables] = (COL_INFO *) malloc(sizeof(COL_INFO));
|
||||
if ( ! conn->col_info[conn->ntables]) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/* Store the table name and the SQLColumns result structure */
|
||||
strcpy(conn->col_info[conn->ntables]->name, ti[i]->name);
|
||||
conn->col_info[conn->ntables]->result = col_stmt->result;
|
||||
|
||||
/* The connection will now free the result structures, so make
|
||||
sure that the statement doesn't free it
|
||||
*/
|
||||
col_stmt->result = NULL;
|
||||
|
||||
conn->ntables++;
|
||||
|
||||
SQLFreeStmt(hcol_stmt, SQL_DROP);
|
||||
mylog("Created col_info table='%s', ntables=%d\n", ti[i]->name, conn->ntables);
|
||||
}
|
||||
else {
|
||||
SQLFreeStmt(hcol_stmt, SQL_DROP);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/* Associate a table from the statement with a SQLColumn info */
|
||||
ti[i]->col_info = conn->col_info[k];
|
||||
mylog("associate col_info: i=%d, k=%d\n", i, k);
|
||||
}
|
||||
|
||||
|
||||
mylog("Done SQLColumns\n");
|
||||
|
||||
/******************************************************/
|
||||
/* Now resolve the fields to point to column info */
|
||||
/******************************************************/
|
||||
|
||||
|
||||
|
||||
for (i = 0; i < stmt->nfld;) {
|
||||
|
||||
/* Dont worry about functions or quotes */
|
||||
if (fi[i]->func || fi[i]->quote || fi[i]->numeric) {
|
||||
i++;
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Stars get expanded to all fields in the table */
|
||||
else if (fi[i]->name[0] == '*') {
|
||||
|
||||
char do_all_tables;
|
||||
int total_cols, old_size, need, cols;
|
||||
|
||||
mylog("expanding field %d\n", i);
|
||||
|
||||
total_cols = 0;
|
||||
|
||||
if (fi[i]->ti) /* The star represents only the qualified table */
|
||||
total_cols = QR_get_num_tuples(fi[i]->ti->col_info->result);
|
||||
|
||||
else { /* The star represents all tables */
|
||||
|
||||
/* Calculate the total number of columns after expansion */
|
||||
for (k = 0; k < stmt->ntab; k++) {
|
||||
total_cols += QR_get_num_tuples(ti[k]->col_info->result);
|
||||
}
|
||||
}
|
||||
total_cols--; /* makes up for the star */
|
||||
|
||||
/* Allocate some more field pointers if necessary */
|
||||
//-------------------------------------------------------------
|
||||
old_size = (stmt->nfld / FLD_INCR * FLD_INCR + FLD_INCR);
|
||||
need = total_cols - (old_size - stmt->nfld);
|
||||
|
||||
mylog("k=%d, total_cols=%d, old_size=%d, need=%d\n", k,total_cols,old_size,need);
|
||||
|
||||
if (need > 0) {
|
||||
int new_size = need / FLD_INCR * FLD_INCR + FLD_INCR;
|
||||
mylog("need more cols: new_size = %d\n", new_size);
|
||||
fi = (FIELD_INFO **) realloc(fi, (old_size + new_size) * sizeof(FIELD_INFO *));
|
||||
if ( ! fi) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
}
|
||||
|
||||
//-------------------------------------------------------------
|
||||
// copy any other fields (if there are any) up past the expansion
|
||||
for (j = stmt->nfld - 1; j > i; j--) {
|
||||
mylog("copying field %d to %d\n", j, total_cols + j);
|
||||
fi[total_cols + j] = fi[j];
|
||||
}
|
||||
mylog("done copying fields\n");
|
||||
|
||||
//-------------------------------------------------------------
|
||||
// Set the new number of fields
|
||||
stmt->nfld += total_cols;
|
||||
mylog("stmt->nfld now at %d\n", stmt->nfld);
|
||||
|
||||
|
||||
//-------------------------------------------------------------
|
||||
// copy the new field info
|
||||
|
||||
|
||||
do_all_tables = (fi[i]->ti ? FALSE : TRUE);
|
||||
|
||||
for (k = 0; k < (do_all_tables ? stmt->ntab : 1); k++) {
|
||||
|
||||
TABLE_INFO *the_ti = do_all_tables ? ti[k] : fi[i]->ti;
|
||||
|
||||
cols = QR_get_num_tuples(the_ti->col_info->result);
|
||||
|
||||
for (n = 0; n < cols; n++) {
|
||||
mylog("creating field info: n=%d\n", n);
|
||||
// skip malloc (already did it for the Star)
|
||||
if (k > 0 || n > 0) {
|
||||
mylog("allocating field info at %d\n", n + i);
|
||||
fi[n + i] = (FIELD_INFO *) malloc( sizeof(FIELD_INFO));
|
||||
if (fi[n + i] == NULL) {
|
||||
stmt->parse_status = STMT_PARSE_FATAL;
|
||||
return FALSE;
|
||||
}
|
||||
}
|
||||
/* Initialize the new space (or the * field) */
|
||||
memset(fi[n + i], 0, sizeof(FIELD_INFO));
|
||||
fi[n + i]->ti = the_ti;
|
||||
|
||||
mylog("about to copy at %d\n", n + i);
|
||||
|
||||
getColInfo(the_ti->col_info, fi[n + i], n);
|
||||
|
||||
mylog("done copying\n");
|
||||
}
|
||||
|
||||
i += cols;
|
||||
mylog("i now at %d\n", i);
|
||||
}
|
||||
|
||||
//-------------------------------------------------------------
|
||||
}
|
||||
|
||||
/* We either know which table the field was in because it was qualified
|
||||
with a table name or alias -OR- there was only 1 table.
|
||||
*/
|
||||
else if (fi[i]->ti) {
|
||||
|
||||
if ( ! searchColInfo(fi[i]->ti->col_info, fi[i]))
|
||||
parse = FALSE;
|
||||
|
||||
i++;
|
||||
}
|
||||
|
||||
/* Don't know the table -- search all tables in "from" list */
|
||||
else {
|
||||
parse = FALSE;
|
||||
for (k = 0; k < stmt->ntab; k++) {
|
||||
if ( searchColInfo(ti[k]->col_info, fi[i])) {
|
||||
fi[i]->ti = ti[k]; /* now know the table */
|
||||
parse = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
i++;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
if ( ! parse)
|
||||
stmt->parse_status = STMT_PARSE_INCOMPLETE;
|
||||
else
|
||||
stmt->parse_status = STMT_PARSE_COMPLETE;
|
||||
|
||||
|
||||
mylog("done parse_statement: parse=%d, parse_status=%d\n", parse, stmt->parse_status);
|
||||
return parse;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user