+ colNo++;
+ }
+ }
+ }
+ if (retcode != SQL_NO_DATA_FOUND)
+ { // Error occured, abort
+ DispAllErrors(henv, hdbc, hstmt);
+ if (colInf)
+ delete [] colInf;
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ if (numCols)
+ *numCols = 0;
+ return(0);
+ }
+ }
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ // Store Primary and Foreign Keys
+ GetKeyFields(tableName,colInf,noCols);
+
+ ///////////////////////////////////////////////////////////////////////////
+ // Now sort the the columns in order to make them appear in the right order
+ ///////////////////////////////////////////////////////////////////////////
+
+ // Build a generic SELECT statement which returns 0 rows
+ wxString Stmt;
+
+ Stmt.Printf(wxT("select * from \"%s\" where 0=1"), tableName);
+
+ // Execute query
+ if (SQLExecDirect(hstmt, (UCHAR FAR *) Stmt.c_str(), SQL_NTS) != SQL_SUCCESS)
+ {
+ DispAllErrors(henv, hdbc, hstmt);
+ return NULL;
+ }
+
+ // Get the number of result columns
+ if (SQLNumResultCols (hstmt, &noCols) != SQL_SUCCESS)
+ {
+ DispAllErrors(henv, hdbc, hstmt);
+ return NULL;
+ }
+
+ if (noCols == 0) // Probably a bogus table name
+ return NULL;
+
+ // Get the name
+ int i;
+ short colNum;
+ UCHAR name[100];
+ SWORD Sword;
+ SDWORD Sdword;
+ for (colNum = 0; colNum < noCols; colNum++)
+ {
+ if (SQLColAttributes(hstmt,colNum+1, SQL_COLUMN_NAME,
+ name, sizeof(name),
+ &Sword, &Sdword) != SQL_SUCCESS)
+ {
+ DispAllErrors(henv, hdbc, hstmt);
+ return NULL;
+ }
+
+ wxString Name1 = name;
+ Name1 = Name1.Upper();
+
+ // Where is this name in the array ?
+ for (i = colNum ; i < noCols ; i++)
+ {
+ wxString Name2 = colInf[i].colName;
+ Name2 = Name2.Upper();
+ if (Name2 == Name1)
+ {
+ if (colNum != i) // swap to sort
+ {
+ wxDbColInf tmpColInf = colInf[colNum];
+ colInf[colNum] = colInf[i];
+ colInf[i] = tmpColInf;
+ }
+ break;
+ }
+ }
+ }
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ ///////////////////////////////////////////////////////////////////////////
+ // End sorting
+ ///////////////////////////////////////////////////////////////////////////
+
+ if (numCols)
+ *numCols = noCols;
+ return colInf;
+
+} // wxDb::GetColumns()
+
+
+#endif // #else OLD_GETCOLUMNS
+
+
+/********** wxDb::GetColumnCount() **********/
+int wxDb::GetColumnCount(const wxString &tableName, const wxChar *userID)
+/*
+ * Returns a count of how many columns are in a table.
+ * If an error occurs in computing the number of columns
+ * this function will return a -1 for the count
+ *
+ * userID is evaluated in the following manner:
+ * userID == NULL ... UserID is ignored
+ * userID == "" ... UserID set equal to 'this->uid'
+ * userID != "" ... UserID set equal to 'userID'
+ *
+ * NOTE: ALL column bindings associated with this wxDb instance are unbound
+ * by this function. This function should use its own wxDb instance
+ * to avoid undesired unbinding of columns.
+ */
+{
+ UWORD noCols = 0;
+
+ RETCODE retcode;
+
+ wxString TableName;
+
+ wxString UserID;
+ convertUserID(userID,UserID);
+
+ TableName = tableName;
+ // Oracle and Interbase table names are uppercase only, so force
+ // the name to uppercase just in case programmer forgot to do this
+ if ((Dbms() == dbmsORACLE) ||
+ (Dbms() == dbmsINTERBASE))
+ TableName = TableName.Upper();
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+
+ // MySQL, SQLServer, and Access cannot accept a user name when looking up column names, so we
+ // use the call below that leaves out the user name
+ if (!UserID.IsEmpty() &&
+ Dbms() != dbmsMY_SQL &&
+ Dbms() != dbmsACCESS &&
+ Dbms() != dbmsMS_SQL_SERVER)
+ {
+ retcode = SQLColumns(hstmt,
+ NULL, 0, // All qualifiers
+ (UCHAR *) UserID.c_str(), SQL_NTS, // Owner
+ (UCHAR *) TableName.c_str(), SQL_NTS,
+ NULL, 0); // All columns
+ }
+ else
+ {
+ retcode = SQLColumns(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // Owner
+ (UCHAR *) TableName.c_str(), SQL_NTS,
+ NULL, 0); // All columns
+ }
+ if (retcode != SQL_SUCCESS)
+ { // Error occured, abort
+ DispAllErrors(henv, hdbc, hstmt);
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return(-1);
+ }
+
+ // Count the columns
+ while ((retcode = SQLFetch(hstmt)) == SQL_SUCCESS)
+ noCols++;
+
+ if (retcode != SQL_NO_DATA_FOUND)
+ { // Error occured, abort
+ DispAllErrors(henv, hdbc, hstmt);
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return(-1);
+ }
+
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return noCols;
+
+} // wxDb::GetColumnCount()
+
+
+/********** wxDb::GetCatalog() *******/
+wxDbInf *wxDb::GetCatalog(const wxChar *userID)
+/*
+ * ---------------------------------------------------------------------
+ * -- 19991203 : mj10777 : Create ------
+ * -- : Creates a wxDbInf with Tables / Cols Array ------
+ * -- : uses SQLTables and fills pTableInf; ------
+ * -- : pColInf is set to NULL and numCols to 0; ------
+ * -- : returns pDbInf (wxDbInf) ------
+ * -- - if unsuccesfull (pDbInf == NULL) ------
+ * -- : pColInf can be filled with GetColumns(..); ------
+ * -- : numCols can be filled with GetColumnCount(..); ------
+ * ---------------------------------------------------------------------
+ *
+ * userID is evaluated in the following manner:
+ * userID == NULL ... UserID is ignored
+ * userID == "" ... UserID set equal to 'this->uid'
+ * userID != "" ... UserID set equal to 'userID'
+ *
+ * NOTE: ALL column bindings associated with this wxDb instance are unbound
+ * by this function. This function should use its own wxDb instance
+ * to avoid undesired unbinding of columns.
+ */
+{
+ wxDbInf *pDbInf = NULL; // Array of catalog entries
+ int noTab = 0; // Counter while filling table entries
+ int pass;
+ RETCODE retcode;
+ SDWORD cb;
+ wxString tblNameSave;
+
+ wxString UserID;
+ convertUserID(userID,UserID);
+
+ //-------------------------------------------------------------
+ pDbInf = new wxDbInf; // Create the Database Array
+ //-------------------------------------------------------------
+ // Table Information
+ // Pass 1 - Determine how many Tables there are.
+ // Pass 2 - Create the Table array and fill it
+ // - Create the Cols array = NULL
+ //-------------------------------------------------------------
+
+ for (pass = 1; pass <= 2; pass++)
+ {
+ SQLFreeStmt(hstmt, SQL_CLOSE); // Close if Open
+ tblNameSave.Empty();
+
+ if (!UserID.IsEmpty() &&
+ Dbms() != dbmsMY_SQL &&
+ Dbms() != dbmsACCESS &&
+ Dbms() != dbmsMS_SQL_SERVER)
+ {
+ retcode = SQLTables(hstmt,
+ NULL, 0, // All qualifiers
+ (UCHAR *) UserID.c_str(), SQL_NTS, // User specified
+ NULL, 0, // All tables
+ NULL, 0); // All columns
+ }
+ else
+ {
+ retcode = SQLTables(hstmt,
+ NULL, 0, // All qualifiers
+ NULL, 0, // User specified
+ NULL, 0, // All tables
+ NULL, 0); // All columns
+ }
+
+ if (retcode != SQL_SUCCESS)
+ {
+ DispAllErrors(henv, hdbc, hstmt);
+ pDbInf = NULL;
+ SQLFreeStmt(hstmt, SQL_CLOSE);
+ return pDbInf;
+ }
+
+ while ((retcode = SQLFetch(hstmt)) == SQL_SUCCESS) // Table Information
+ {
+ if (pass == 1) // First pass, just count the Tables
+ {
+ if (pDbInf->numTables == 0)
+ {
+ GetData( 1, SQL_C_CHAR, (UCHAR*) pDbInf->catalog, 128+1, &cb);
+ GetData( 2, SQL_C_CHAR, (UCHAR*) pDbInf->schema, 128+1, &cb);
+ }
+ pDbInf->numTables++; // Counter for Tables
+ } // if (pass == 1)
+ if (pass == 2) // Create and fill the Table entries
+ {
+ if (pDbInf->pTableInf == NULL) // Has the Table Array been created
+ { // no, then create the Array
+ pDbInf->pTableInf = new wxDbTableInf[pDbInf->numTables];
projects / wxWidgets.git / blobdiff