]> git.saurik.com Git - wxWidgets.git/blob - src/common/dbtable.cpp
Reverted due to Darwin conflict.
[wxWidgets.git] / src / common / dbtable.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: dbtable.cpp
3 // Purpose: Implementation of the wxDbTable class.
4 // Author: Doug Card
5 // Modified by: George Tasker
6 // Bart Jourquin
7 // Mark Johnson
8 // Created: 9.96
9 // RCS-ID: $Id$
10 // Copyright: (c) 1996 Remstar International, Inc.
11 // Licence: wxWindows licence
12 ///////////////////////////////////////////////////////////////////////////////
13
14 /*
15 // SYNOPSIS START
16 // SYNOPSIS STOP
17 */
18
19 #if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
20 #pragma implementation "dbtable.h"
21 #endif
22
23 #include "wx/wxprec.h"
24
25 #ifdef __BORLANDC__
26 #pragma hdrstop
27 #endif
28
29 #ifdef DBDEBUG_CONSOLE
30 #if wxUSE_IOSTREAMH
31 #include <iostream.h>
32 #else
33 #include <iostream>
34 #endif
35 #include "wx/ioswrap.h"
36 #endif
37
38 #ifndef WX_PRECOMP
39 #include "wx/string.h"
40 #include "wx/object.h"
41 #include "wx/list.h"
42 #include "wx/utils.h"
43 #include "wx/log.h"
44 #endif
45 #include "wx/filefn.h"
46
47 #if wxUSE_ODBC
48
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <string.h>
52
53 #include "wx/dbtable.h"
54
55 #ifdef __UNIX__
56 // The HPUX preprocessor lines below were commented out on 8/20/97
57 // because macros.h currently redefines DEBUG and is unneeded.
58 // # ifdef HPUX
59 // # include <macros.h>
60 // # endif
61 # ifdef LINUX
62 # include <sys/minmax.h>
63 # endif
64 #endif
65
66 ULONG lastTableID = 0;
67
68
69 #ifdef __WXDEBUG__
70 wxList TablesInUse;
71 #endif
72
73
74 void csstrncpyt(wxChar *target, const wxChar *source, int n)
75 {
76 while ( (*target++ = *source++) != '\0' && --n )
77 ;
78
79 *target = '\0';
80 }
81
82
83
84 /********** wxDbColDef::wxDbColDef() Constructor **********/
85 wxDbColDef::wxDbColDef()
86 {
87 Initialize();
88 } // Constructor
89
90
91 bool wxDbColDef::Initialize()
92 {
93 ColName[0] = 0;
94 DbDataType = DB_DATA_TYPE_INTEGER;
95 SqlCtype = SQL_C_LONG;
96 PtrDataObj = NULL;
97 SzDataObj = 0;
98 KeyField = false;
99 Updateable = false;
100 InsertAllowed = false;
101 DerivedCol = false;
102 CbValue = 0;
103 Null = false;
104
105 return true;
106 } // wxDbColDef::Initialize()
107
108
109 /********** wxDbTable::wxDbTable() Constructor **********/
110 wxDbTable::wxDbTable(wxDb *pwxDb, const wxString &tblName, const UWORD numColumns,
111 const wxString &qryTblName, bool qryOnly, const wxString &tblPath)
112 {
113 if (!initialize(pwxDb, tblName, numColumns, qryTblName, qryOnly, tblPath))
114 cleanup();
115 } // wxDbTable::wxDbTable()
116
117
118 /***** DEPRECATED: use wxDbTable::wxDbTable() format above *****/
119 #if WXWIN_COMPATIBILITY_2_4
120 wxDbTable::wxDbTable(wxDb *pwxDb, const wxString &tblName, const UWORD numColumns,
121 const wxChar *qryTblName, bool qryOnly, const wxString &tblPath)
122 {
123 wxString tempQryTblName;
124 tempQryTblName = qryTblName;
125 if (!initialize(pwxDb, tblName, numColumns, tempQryTblName, qryOnly, tblPath))
126 cleanup();
127 } // wxDbTable::wxDbTable()
128 #endif // WXWIN_COMPATIBILITY_2_4
129
130
131 /********** wxDbTable::~wxDbTable() **********/
132 wxDbTable::~wxDbTable()
133 {
134 this->cleanup();
135 } // wxDbTable::~wxDbTable()
136
137
138 bool wxDbTable::initialize(wxDb *pwxDb, const wxString &tblName, const UWORD numColumns,
139 const wxString &qryTblName, bool qryOnly, const wxString &tblPath)
140 {
141 // Initializing member variables
142 pDb = pwxDb; // Pointer to the wxDb object
143 henv = 0;
144 hdbc = 0;
145 hstmt = 0;
146 m_hstmtGridQuery = 0;
147 hstmtDefault = 0; // Initialized below
148 hstmtCount = 0; // Initialized first time it is needed
149 hstmtInsert = 0;
150 hstmtDelete = 0;
151 hstmtUpdate = 0;
152 hstmtInternal = 0;
153 colDefs = 0;
154 tableID = 0;
155 noCols = numColumns; // Number of cols in the table
156 where.Empty(); // Where clause
157 orderBy.Empty(); // Order By clause
158 from.Empty(); // From clause
159 selectForUpdate = false; // SELECT ... FOR UPDATE; Indicates whether to include the FOR UPDATE phrase
160 queryOnly = qryOnly;
161 insertable = true;
162 tablePath.Empty();
163 tableName.Empty();
164 queryTableName.Empty();
165
166 wxASSERT(tblName.Length());
167 wxASSERT(pDb);
168
169 if (!pDb)
170 return false;
171
172 tableName = tblName; // Table Name
173 if (tblPath.Length())
174 tablePath = tblPath; // Table Path - used for dBase files
175 else
176 tablePath.Empty();
177
178 if (qryTblName.Length()) // Name of the table/view to query
179 queryTableName = qryTblName;
180 else
181 queryTableName = tblName;
182
183 pDb->incrementTableCount();
184
185 wxString s;
186 tableID = ++lastTableID;
187 s.Printf(wxT("wxDbTable constructor (%-20s) tableID:[%6lu] pDb:[%p]"), tblName.c_str(), tableID, pDb);
188
189 #ifdef __WXDEBUG__
190 wxTablesInUse *tableInUse;
191 tableInUse = new wxTablesInUse();
192 tableInUse->tableName = tblName;
193 tableInUse->tableID = tableID;
194 tableInUse->pDb = pDb;
195 TablesInUse.Append(tableInUse);
196 #endif
197
198 pDb->WriteSqlLog(s);
199
200 // Grab the HENV and HDBC from the wxDb object
201 henv = pDb->GetHENV();
202 hdbc = pDb->GetHDBC();
203
204 // Allocate space for column definitions
205 if (noCols)
206 colDefs = new wxDbColDef[noCols]; // Points to the first column definition
207
208 // Allocate statement handles for the table
209 if (!queryOnly)
210 {
211 // Allocate a separate statement handle for performing inserts
212 if (SQLAllocStmt(hdbc, &hstmtInsert) != SQL_SUCCESS)
213 pDb->DispAllErrors(henv, hdbc);
214 // Allocate a separate statement handle for performing deletes
215 if (SQLAllocStmt(hdbc, &hstmtDelete) != SQL_SUCCESS)
216 pDb->DispAllErrors(henv, hdbc);
217 // Allocate a separate statement handle for performing updates
218 if (SQLAllocStmt(hdbc, &hstmtUpdate) != SQL_SUCCESS)
219 pDb->DispAllErrors(henv, hdbc);
220 }
221 // Allocate a separate statement handle for internal use
222 if (SQLAllocStmt(hdbc, &hstmtInternal) != SQL_SUCCESS)
223 pDb->DispAllErrors(henv, hdbc);
224
225 // Set the cursor type for the statement handles
226 cursorType = SQL_CURSOR_STATIC;
227
228 if (SQLSetStmtOption(hstmtInternal, SQL_CURSOR_TYPE, cursorType) != SQL_SUCCESS)
229 {
230 // Check to see if cursor type is supported
231 pDb->GetNextError(henv, hdbc, hstmtInternal);
232 if (! wxStrcmp(pDb->sqlState, wxT("01S02"))) // Option Value Changed
233 {
234 // Datasource does not support static cursors. Driver
235 // will substitute a cursor type. Call SQLGetStmtOption()
236 // to determine which cursor type was selected.
237 if (SQLGetStmtOption(hstmtInternal, SQL_CURSOR_TYPE, &cursorType) != SQL_SUCCESS)
238 pDb->DispAllErrors(henv, hdbc, hstmtInternal);
239 #ifdef DBDEBUG_CONSOLE
240 cout << wxT("Static cursor changed to: ");
241 switch(cursorType)
242 {
243 case SQL_CURSOR_FORWARD_ONLY:
244 cout << wxT("Forward Only");
245 break;
246 case SQL_CURSOR_STATIC:
247 cout << wxT("Static");
248 break;
249 case SQL_CURSOR_KEYSET_DRIVEN:
250 cout << wxT("Keyset Driven");
251 break;
252 case SQL_CURSOR_DYNAMIC:
253 cout << wxT("Dynamic");
254 break;
255 }
256 cout << endl << endl;
257 #endif
258 // BJO20000425
259 if (pDb->FwdOnlyCursors() && cursorType != SQL_CURSOR_FORWARD_ONLY)
260 {
261 // Force the use of a forward only cursor...
262 cursorType = SQL_CURSOR_FORWARD_ONLY;
263 if (SQLSetStmtOption(hstmtInternal, SQL_CURSOR_TYPE, cursorType) != SQL_SUCCESS)
264 {
265 // Should never happen
266 pDb->GetNextError(henv, hdbc, hstmtInternal);
267 return false;
268 }
269 }
270 }
271 else
272 {
273 pDb->DispNextError();
274 pDb->DispAllErrors(henv, hdbc, hstmtInternal);
275 }
276 }
277 #ifdef DBDEBUG_CONSOLE
278 else
279 cout << wxT("Cursor Type set to STATIC") << endl << endl;
280 #endif
281
282 if (!queryOnly)
283 {
284 // Set the cursor type for the INSERT statement handle
285 if (SQLSetStmtOption(hstmtInsert, SQL_CURSOR_TYPE, SQL_CURSOR_FORWARD_ONLY) != SQL_SUCCESS)
286 pDb->DispAllErrors(henv, hdbc, hstmtInsert);
287 // Set the cursor type for the DELETE statement handle
288 if (SQLSetStmtOption(hstmtDelete, SQL_CURSOR_TYPE, SQL_CURSOR_FORWARD_ONLY) != SQL_SUCCESS)
289 pDb->DispAllErrors(henv, hdbc, hstmtDelete);
290 // Set the cursor type for the UPDATE statement handle
291 if (SQLSetStmtOption(hstmtUpdate, SQL_CURSOR_TYPE, SQL_CURSOR_FORWARD_ONLY) != SQL_SUCCESS)
292 pDb->DispAllErrors(henv, hdbc, hstmtUpdate);
293 }
294
295 // Make the default cursor the active cursor
296 hstmtDefault = GetNewCursor(false,false);
297 wxASSERT(hstmtDefault);
298 hstmt = *hstmtDefault;
299
300 return true;
301
302 } // wxDbTable::initialize()
303
304
305 void wxDbTable::cleanup()
306 {
307 wxString s;
308 if (pDb)
309 {
310 s.Printf(wxT("wxDbTable destructor (%-20s) tableID:[%6lu] pDb:[%p]"), tableName.c_str(), tableID, pDb);
311 pDb->WriteSqlLog(s);
312 }
313
314 #ifdef __WXDEBUG__
315 if (tableID)
316 {
317 bool found = false;
318
319 wxList::compatibility_iterator pNode;
320 pNode = TablesInUse.GetFirst();
321 while (pNode && !found)
322 {
323 if (((wxTablesInUse *)pNode->GetData())->tableID == tableID)
324 {
325 found = true;
326 delete (wxTablesInUse *)pNode->GetData();
327 TablesInUse.Erase(pNode);
328 }
329 else
330 pNode = pNode->GetNext();
331 }
332 if (!found)
333 {
334 wxString msg;
335 msg.Printf(wxT("Unable to find the tableID in the linked\nlist of tables in use.\n\n%s"),s.c_str());
336 wxLogDebug (msg,wxT("NOTICE..."));
337 }
338 }
339 #endif
340
341 // Decrement the wxDb table count
342 if (pDb)
343 pDb->decrementTableCount();
344
345 // Delete memory allocated for column definitions
346 if (colDefs)
347 delete [] colDefs;
348
349 // Free statement handles
350 if (!queryOnly)
351 {
352 if (hstmtInsert)
353 {
354 /*
355 ODBC 3.0 says to use this form
356 if (SQLFreeHandle(*hstmtDel, SQL_DROP) != SQL_SUCCESS)
357 */
358 if (SQLFreeStmt(hstmtInsert, SQL_DROP) != SQL_SUCCESS)
359 pDb->DispAllErrors(henv, hdbc);
360 }
361
362 if (hstmtDelete)
363 {
364 /*
365 ODBC 3.0 says to use this form
366 if (SQLFreeHandle(*hstmtDel, SQL_DROP) != SQL_SUCCESS)
367 */
368 if (SQLFreeStmt(hstmtDelete, SQL_DROP) != SQL_SUCCESS)
369 pDb->DispAllErrors(henv, hdbc);
370 }
371
372 if (hstmtUpdate)
373 {
374 /*
375 ODBC 3.0 says to use this form
376 if (SQLFreeHandle(*hstmtDel, SQL_DROP) != SQL_SUCCESS)
377 */
378 if (SQLFreeStmt(hstmtUpdate, SQL_DROP) != SQL_SUCCESS)
379 pDb->DispAllErrors(henv, hdbc);
380 }
381 }
382
383 if (hstmtInternal)
384 {
385 if (SQLFreeStmt(hstmtInternal, SQL_DROP) != SQL_SUCCESS)
386 pDb->DispAllErrors(henv, hdbc);
387 }
388
389 // Delete dynamically allocated cursors
390 if (hstmtDefault)
391 DeleteCursor(hstmtDefault);
392
393 if (hstmtCount)
394 DeleteCursor(hstmtCount);
395
396 if (m_hstmtGridQuery)
397 DeleteCursor(m_hstmtGridQuery);
398
399 } // wxDbTable::cleanup()
400
401
402 /***************************** PRIVATE FUNCTIONS *****************************/
403
404
405 void wxDbTable::setCbValueForColumn(int columnIndex)
406 {
407 switch(colDefs[columnIndex].DbDataType)
408 {
409 case DB_DATA_TYPE_VARCHAR:
410 if (colDefs[columnIndex].Null)
411 colDefs[columnIndex].CbValue = SQL_NULL_DATA;
412 else
413 colDefs[columnIndex].CbValue = SQL_NTS;
414 break;
415 case DB_DATA_TYPE_INTEGER:
416 if (colDefs[columnIndex].Null)
417 colDefs[columnIndex].CbValue = SQL_NULL_DATA;
418 else
419 colDefs[columnIndex].CbValue = 0;
420 break;
421 case DB_DATA_TYPE_FLOAT:
422 if (colDefs[columnIndex].Null)
423 colDefs[columnIndex].CbValue = SQL_NULL_DATA;
424 else
425 colDefs[columnIndex].CbValue = 0;
426 break;
427 case DB_DATA_TYPE_DATE:
428 if (colDefs[columnIndex].Null)
429 colDefs[columnIndex].CbValue = SQL_NULL_DATA;
430 else
431 colDefs[columnIndex].CbValue = 0;
432 break;
433 case DB_DATA_TYPE_BLOB:
434 if (colDefs[columnIndex].Null)
435 colDefs[columnIndex].CbValue = SQL_NULL_DATA;
436 else
437 if (colDefs[columnIndex].SqlCtype == SQL_C_CHAR)
438 colDefs[columnIndex].CbValue = SQL_NTS;
439 else
440 colDefs[columnIndex].CbValue = SQL_LEN_DATA_AT_EXEC(colDefs[columnIndex].SzDataObj);
441 break;
442 }
443 }
444
445 /********** wxDbTable::bindParams() **********/
446 bool wxDbTable::bindParams(bool forUpdate)
447 {
448 wxASSERT(!queryOnly);
449 if (queryOnly)
450 return false;
451
452 SWORD fSqlType = 0;
453 SDWORD precision = 0;
454 SWORD scale = 0;
455
456 // Bind each column of the table that should be bound
457 // to a parameter marker
458 int i;
459 UWORD colNo;
460
461 for (i=0, colNo=1; i < noCols; i++)
462 {
463 if (forUpdate)
464 {
465 if (!colDefs[i].Updateable)
466 continue;
467 }
468 else
469 {
470 if (!colDefs[i].InsertAllowed)
471 continue;
472 }
473
474 switch(colDefs[i].DbDataType)
475 {
476 case DB_DATA_TYPE_VARCHAR:
477 fSqlType = pDb->GetTypeInfVarchar().FsqlType;
478 precision = colDefs[i].SzDataObj;
479 scale = 0;
480 break;
481 case DB_DATA_TYPE_INTEGER:
482 fSqlType = pDb->GetTypeInfInteger().FsqlType;
483 precision = pDb->GetTypeInfInteger().Precision;
484 scale = 0;
485 break;
486 case DB_DATA_TYPE_FLOAT:
487 fSqlType = pDb->GetTypeInfFloat().FsqlType;
488 precision = pDb->GetTypeInfFloat().Precision;
489 scale = pDb->GetTypeInfFloat().MaximumScale;
490 // SQL Sybase Anywhere v5.5 returned a negative number for the
491 // MaxScale. This caused ODBC to kick out an error on ibscale.
492 // I check for this here and set the scale = precision.
493 //if (scale < 0)
494 // scale = (short) precision;
495 break;
496 case DB_DATA_TYPE_DATE:
497 fSqlType = pDb->GetTypeInfDate().FsqlType;
498 precision = pDb->GetTypeInfDate().Precision;
499 scale = 0;
500 break;
501 case DB_DATA_TYPE_BLOB:
502 fSqlType = pDb->GetTypeInfBlob().FsqlType;
503 precision = colDefs[i].SzDataObj;
504 scale = 0;
505 break;
506 }
507
508 setCbValueForColumn(i);
509
510 if (forUpdate)
511 {
512 if (SQLBindParameter(hstmtUpdate, colNo++, SQL_PARAM_INPUT, colDefs[i].SqlCtype,
513 fSqlType, precision, scale, (UCHAR*) colDefs[i].PtrDataObj,
514 precision+1, &colDefs[i].CbValue) != SQL_SUCCESS)
515 {
516 return(pDb->DispAllErrors(henv, hdbc, hstmtUpdate));
517 }
518 }
519 else
520 {
521 if (SQLBindParameter(hstmtInsert, colNo++, SQL_PARAM_INPUT, colDefs[i].SqlCtype,
522 fSqlType, precision, scale, (UCHAR*) colDefs[i].PtrDataObj,
523 precision+1, &colDefs[i].CbValue) != SQL_SUCCESS)
524 {
525 return(pDb->DispAllErrors(henv, hdbc, hstmtInsert));
526 }
527 }
528 }
529
530 // Completed successfully
531 return true;
532
533 } // wxDbTable::bindParams()
534
535
536 /********** wxDbTable::bindInsertParams() **********/
537 bool wxDbTable::bindInsertParams(void)
538 {
539 return bindParams(false);
540 } // wxDbTable::bindInsertParams()
541
542
543 /********** wxDbTable::bindUpdateParams() **********/
544 bool wxDbTable::bindUpdateParams(void)
545 {
546 return bindParams(true);
547 } // wxDbTable::bindUpdateParams()
548
549
550 /********** wxDbTable::bindCols() **********/
551 bool wxDbTable::bindCols(HSTMT cursor)
552 {
553 static SDWORD cb;
554
555 // Bind each column of the table to a memory address for fetching data
556 UWORD i;
557 for (i = 0; i < noCols; i++)
558 {
559 cb = colDefs[i].CbValue;
560 if (SQLBindCol(cursor, (UWORD)(i+1), colDefs[i].SqlCtype, (UCHAR*) colDefs[i].PtrDataObj,
561 colDefs[i].SzDataObj, &cb ) != SQL_SUCCESS)
562 return (pDb->DispAllErrors(henv, hdbc, cursor));
563 }
564
565 // Completed successfully
566 return true;
567
568 } // wxDbTable::bindCols()
569
570
571 /********** wxDbTable::getRec() **********/
572 bool wxDbTable::getRec(UWORD fetchType)
573 {
574 RETCODE retcode;
575
576 if (!pDb->FwdOnlyCursors())
577 {
578 // Fetch the NEXT, PREV, FIRST or LAST record, depending on fetchType
579 UDWORD cRowsFetched;
580 UWORD rowStatus;
581
582 retcode = SQLExtendedFetch(hstmt, fetchType, 0, &cRowsFetched, &rowStatus);
583 if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)
584 {
585 if (retcode == SQL_NO_DATA_FOUND)
586 return false;
587 else
588 return(pDb->DispAllErrors(henv, hdbc, hstmt));
589 }
590 else
591 {
592 // Set the Null member variable to indicate the Null state
593 // of each column just read in.
594 int i;
595 for (i = 0; i < noCols; i++)
596 colDefs[i].Null = (colDefs[i].CbValue == SQL_NULL_DATA);
597 }
598 }
599 else
600 {
601 // Fetch the next record from the record set
602 retcode = SQLFetch(hstmt);
603 if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)
604 {
605 if (retcode == SQL_NO_DATA_FOUND)
606 return false;
607 else
608 return(pDb->DispAllErrors(henv, hdbc, hstmt));
609 }
610 else
611 {
612 // Set the Null member variable to indicate the Null state
613 // of each column just read in.
614 int i;
615 for (i = 0; i < noCols; i++)
616 colDefs[i].Null = (colDefs[i].CbValue == SQL_NULL_DATA);
617 }
618 }
619
620 // Completed successfully
621 return true;
622
623 } // wxDbTable::getRec()
624
625
626 /********** wxDbTable::execDelete() **********/
627 bool wxDbTable::execDelete(const wxString &pSqlStmt)
628 {
629 RETCODE retcode;
630
631 // Execute the DELETE statement
632 retcode = SQLExecDirect(hstmtDelete, (SQLTCHAR FAR *) pSqlStmt.c_str(), SQL_NTS);
633
634 if (retcode == SQL_SUCCESS ||
635 retcode == SQL_NO_DATA_FOUND ||
636 retcode == SQL_SUCCESS_WITH_INFO)
637 {
638 // Record deleted successfully
639 return true;
640 }
641
642 // Problem deleting record
643 return(pDb->DispAllErrors(henv, hdbc, hstmtDelete));
644
645 } // wxDbTable::execDelete()
646
647
648 /********** wxDbTable::execUpdate() **********/
649 bool wxDbTable::execUpdate(const wxString &pSqlStmt)
650 {
651 RETCODE retcode;
652
653 // Execute the UPDATE statement
654 retcode = SQLExecDirect(hstmtUpdate, (SQLTCHAR FAR *) pSqlStmt.c_str(), SQL_NTS);
655
656 if (retcode == SQL_SUCCESS ||
657 retcode == SQL_NO_DATA_FOUND ||
658 retcode == SQL_SUCCESS_WITH_INFO)
659 {
660 // Record updated successfully
661 return true;
662 }
663 else if (retcode == SQL_NEED_DATA)
664 {
665 PTR pParmID;
666 retcode = SQLParamData(hstmtUpdate, &pParmID);
667 while (retcode == SQL_NEED_DATA)
668 {
669 // Find the parameter
670 int i;
671 for (i=0; i < noCols; i++)
672 {
673 if (colDefs[i].PtrDataObj == pParmID)
674 {
675 // We found it. Store the parameter.
676 retcode = SQLPutData(hstmtUpdate, pParmID, colDefs[i].SzDataObj);
677 if (retcode != SQL_SUCCESS)
678 {
679 pDb->DispNextError();
680 return pDb->DispAllErrors(henv, hdbc, hstmtUpdate);
681 }
682 break;
683 }
684 }
685 retcode = SQLParamData(hstmtUpdate, &pParmID);
686 }
687 if (retcode == SQL_SUCCESS ||
688 retcode == SQL_NO_DATA_FOUND ||
689 retcode == SQL_SUCCESS_WITH_INFO)
690 {
691 // Record updated successfully
692 return true;
693 }
694 }
695
696 // Problem updating record
697 return(pDb->DispAllErrors(henv, hdbc, hstmtUpdate));
698
699 } // wxDbTable::execUpdate()
700
701
702 /********** wxDbTable::query() **********/
703 bool wxDbTable::query(int queryType, bool forUpdate, bool distinct, const wxString &pSqlStmt)
704 {
705 wxString sqlStmt;
706
707 if (forUpdate)
708 // The user may wish to select for update, but the DBMS may not be capable
709 selectForUpdate = CanSelectForUpdate();
710 else
711 selectForUpdate = false;
712
713 // Set the SQL SELECT string
714 if (queryType != DB_SELECT_STATEMENT) // A select statement was not passed in,
715 { // so generate a select statement.
716 BuildSelectStmt(sqlStmt, queryType, distinct);
717 pDb->WriteSqlLog(sqlStmt);
718 }
719
720 // Make sure the cursor is closed first
721 if (!CloseCursor(hstmt))
722 return false;
723
724 // Execute the SQL SELECT statement
725 int retcode;
726 retcode = SQLExecDirect(hstmt, (SQLTCHAR FAR *) (queryType == DB_SELECT_STATEMENT ? pSqlStmt.c_str() : sqlStmt.c_str()), SQL_NTS);
727 if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)
728 return(pDb->DispAllErrors(henv, hdbc, hstmt));
729
730 // Completed successfully
731 return true;
732
733 } // wxDbTable::query()
734
735
736 /***************************** PUBLIC FUNCTIONS *****************************/
737
738
739 /********** wxDbTable::Open() **********/
740 bool wxDbTable::Open(bool checkPrivileges, bool checkTableExists)
741 {
742 if (!pDb)
743 return false;
744
745 int i;
746 wxString sqlStmt;
747 wxString s;
748 // int NumKeyCols=0;
749
750 // Calculate the maximum size of the concatenated
751 // keys for use with wxDbGrid
752 m_keysize = 0;
753 for (i=0; i < noCols; i++)
754 {
755 if (colDefs[i].KeyField)
756 {
757 // NumKeyCols++;
758 m_keysize += colDefs[i].SzDataObj;
759 }
760 }
761
762 s.Empty();
763 // Verify that the table exists in the database
764 if (checkTableExists && !pDb->TableExists(tableName, pDb->GetUsername(), tablePath))
765 {
766 s = wxT("Table/view does not exist in the database");
767 if ( *(pDb->dbInf.accessibleTables) == wxT('Y'))
768 s += wxT(", or you have no permissions.\n");
769 else
770 s += wxT(".\n");
771 }
772 else if (checkPrivileges)
773 {
774 // Verify the user has rights to access the table.
775 // Shortcut boolean evaluation to optimize out call to
776 // TablePrivileges
777 //
778 // Unfortunately this optimization doesn't seem to be
779 // reliable!
780 if (// *(pDb->dbInf.accessibleTables) == 'N' &&
781 !pDb->TablePrivileges(tableName,wxT("SELECT"), pDb->GetUsername(), pDb->GetUsername(), tablePath))
782 s = wxT("Current logged in user does not have sufficient privileges to access this table.\n");
783 }
784
785 if (!s.IsEmpty())
786 {
787 wxString p;
788
789 if (!tablePath.IsEmpty())
790 p.Printf(wxT("Error opening '%s/%s'.\n"),tablePath.c_str(),tableName.c_str());
791 else
792 p.Printf(wxT("Error opening '%s'.\n"), tableName.c_str());
793
794 p += s;
795 pDb->LogError(p.GetData());
796
797 return false;
798 }
799
800 // Bind the member variables for field exchange between
801 // the wxDbTable object and the ODBC record.
802 if (!queryOnly)
803 {
804 if (!bindInsertParams()) // Inserts
805 return false;
806
807 if (!bindUpdateParams()) // Updates
808 return false;
809 }
810
811 if (!bindCols(*hstmtDefault)) // Selects
812 return false;
813
814 if (!bindCols(hstmtInternal)) // Internal use only
815 return false;
816
817 /*
818 * Do NOT bind the hstmtCount cursor!!!
819 */
820
821 // Build an insert statement using parameter markers
822 if (!queryOnly && noCols > 0)
823 {
824 bool needComma = false;
825 sqlStmt.Printf(wxT("INSERT INTO %s ("),
826 pDb->SQLTableName(tableName.c_str()).c_str());
827 for (i = 0; i < noCols; i++)
828 {
829 if (! colDefs[i].InsertAllowed)
830 continue;
831 if (needComma)
832 sqlStmt += wxT(",");
833 sqlStmt += pDb->SQLColumnName(colDefs[i].ColName);
834 // sqlStmt += colDefs[i].ColName;
835 needComma = true;
836 }
837 needComma = false;
838 sqlStmt += wxT(") VALUES (");
839
840 int insertableCount = 0;
841
842 for (i = 0; i < noCols; i++)
843 {
844 if (! colDefs[i].InsertAllowed)
845 continue;
846 if (needComma)
847 sqlStmt += wxT(",");
848 sqlStmt += wxT("?");
849 needComma = true;
850 insertableCount++;
851 }
852 sqlStmt += wxT(")");
853
854 // Prepare the insert statement for execution
855 if (insertableCount)
856 {
857 if (SQLPrepare(hstmtInsert, (SQLTCHAR FAR *) sqlStmt.c_str(), SQL_NTS) != SQL_SUCCESS)
858 return(pDb->DispAllErrors(henv, hdbc, hstmtInsert));
859 }
860 else
861 insertable = false;
862 }
863
864 // Completed successfully
865 return true;
866
867 } // wxDbTable::Open()
868
869
870 /********** wxDbTable::Query() **********/
871 bool wxDbTable::Query(bool forUpdate, bool distinct)
872 {
873
874 return(query(DB_SELECT_WHERE, forUpdate, distinct));
875
876 } // wxDbTable::Query()
877
878
879 /********** wxDbTable::QueryBySqlStmt() **********/
880 bool wxDbTable::QueryBySqlStmt(const wxString &pSqlStmt)
881 {
882 pDb->WriteSqlLog(pSqlStmt);
883
884 return(query(DB_SELECT_STATEMENT, false, false, pSqlStmt));
885
886 } // wxDbTable::QueryBySqlStmt()
887
888
889 /********** wxDbTable::QueryMatching() **********/
890 bool wxDbTable::QueryMatching(bool forUpdate, bool distinct)
891 {
892
893 return(query(DB_SELECT_MATCHING, forUpdate, distinct));
894
895 } // wxDbTable::QueryMatching()
896
897
898 /********** wxDbTable::QueryOnKeyFields() **********/
899 bool wxDbTable::QueryOnKeyFields(bool forUpdate, bool distinct)
900 {
901
902 return(query(DB_SELECT_KEYFIELDS, forUpdate, distinct));
903
904 } // wxDbTable::QueryOnKeyFields()
905
906
907 /********** wxDbTable::GetPrev() **********/
908 bool wxDbTable::GetPrev(void)
909 {
910 if (pDb->FwdOnlyCursors())
911 {
912 wxFAIL_MSG(wxT("GetPrev()::Backward scrolling cursors are not enabled for this instance of wxDbTable"));
913 return false;
914 }
915 else
916 return(getRec(SQL_FETCH_PRIOR));
917
918 } // wxDbTable::GetPrev()
919
920
921 /********** wxDbTable::operator-- **********/
922 bool wxDbTable::operator--(int)
923 {
924 if (pDb->FwdOnlyCursors())
925 {
926 wxFAIL_MSG(wxT("operator--:Backward scrolling cursors are not enabled for this instance of wxDbTable"));
927 return false;
928 }
929 else
930 return(getRec(SQL_FETCH_PRIOR));
931
932 } // wxDbTable::operator--
933
934
935 /********** wxDbTable::GetFirst() **********/
936 bool wxDbTable::GetFirst(void)
937 {
938 if (pDb->FwdOnlyCursors())
939 {
940 wxFAIL_MSG(wxT("GetFirst():Backward scrolling cursors are not enabled for this instance of wxDbTable"));
941 return false;
942 }
943 else
944 return(getRec(SQL_FETCH_FIRST));
945
946 } // wxDbTable::GetFirst()
947
948
949 /********** wxDbTable::GetLast() **********/
950 bool wxDbTable::GetLast(void)
951 {
952 if (pDb->FwdOnlyCursors())
953 {
954 wxFAIL_MSG(wxT("GetLast()::Backward scrolling cursors are not enabled for this instance of wxDbTable"));
955 return false;
956 }
957 else
958 return(getRec(SQL_FETCH_LAST));
959
960 } // wxDbTable::GetLast()
961
962
963 /********** wxDbTable::BuildDeleteStmt() **********/
964 void wxDbTable::BuildDeleteStmt(wxString &pSqlStmt, int typeOfDel, const wxString &pWhereClause)
965 {
966 wxASSERT(!queryOnly);
967 if (queryOnly)
968 return;
969
970 wxString whereClause;
971
972 whereClause.Empty();
973
974 // Handle the case of DeleteWhere() and the where clause is blank. It should
975 // delete all records from the database in this case.
976 if (typeOfDel == DB_DEL_WHERE && (pWhereClause.Length() == 0))
977 {
978 pSqlStmt.Printf(wxT("DELETE FROM %s"),
979 pDb->SQLTableName(tableName.c_str()).c_str());
980 return;
981 }
982
983 pSqlStmt.Printf(wxT("DELETE FROM %s WHERE "),
984 pDb->SQLTableName(tableName.c_str()).c_str());
985
986 // Append the WHERE clause to the SQL DELETE statement
987 switch(typeOfDel)
988 {
989 case DB_DEL_KEYFIELDS:
990 // If the datasource supports the ROWID column, build
991 // the where on ROWID for efficiency purposes.
992 // e.g. DELETE FROM PARTS WHERE ROWID = '111.222.333'
993 if (CanUpdByROWID())
994 {
995 SDWORD cb;
996 wxChar rowid[wxDB_ROWID_LEN+1];
997
998 // Get the ROWID value. If not successful retreiving the ROWID,
999 // simply fall down through the code and build the WHERE clause
1000 // based on the key fields.
1001 if (SQLGetData(hstmt, (UWORD)(noCols+1), SQL_C_CHAR, (UCHAR*) rowid, wxDB_ROWID_LEN, &cb) == SQL_SUCCESS)
1002 {
1003 pSqlStmt += wxT("ROWID = '");
1004 pSqlStmt += rowid;
1005 pSqlStmt += wxT("'");
1006 break;
1007 }
1008 }
1009 // Unable to delete by ROWID, so build a WHERE
1010 // clause based on the keyfields.
1011 BuildWhereClause(whereClause, DB_WHERE_KEYFIELDS);
1012 pSqlStmt += whereClause;
1013 break;
1014 case DB_DEL_WHERE:
1015 pSqlStmt += pWhereClause;
1016 break;
1017 case DB_DEL_MATCHING:
1018 BuildWhereClause(whereClause, DB_WHERE_MATCHING);
1019 pSqlStmt += whereClause;
1020 break;
1021 }
1022
1023 } // BuildDeleteStmt()
1024
1025
1026 /***** DEPRECATED: use wxDbTable::BuildDeleteStmt(wxString &....) form *****/
1027 void wxDbTable::BuildDeleteStmt(wxChar *pSqlStmt, int typeOfDel, const wxString &pWhereClause)
1028 {
1029 wxString tempSqlStmt;
1030 BuildDeleteStmt(tempSqlStmt, typeOfDel, pWhereClause);
1031 wxStrcpy(pSqlStmt, tempSqlStmt);
1032 } // wxDbTable::BuildDeleteStmt()
1033
1034
1035 /********** wxDbTable::BuildSelectStmt() **********/
1036 void wxDbTable::BuildSelectStmt(wxString &pSqlStmt, int typeOfSelect, bool distinct)
1037 {
1038 wxString whereClause;
1039 whereClause.Empty();
1040
1041 // Build a select statement to query the database
1042 pSqlStmt = wxT("SELECT ");
1043
1044 // SELECT DISTINCT values only?
1045 if (distinct)
1046 pSqlStmt += wxT("DISTINCT ");
1047
1048 // Was a FROM clause specified to join tables to the base table?
1049 // Available for ::Query() only!!!
1050 bool appendFromClause = false;
1051 #if wxODBC_BACKWARD_COMPATABILITY
1052 if (typeOfSelect == DB_SELECT_WHERE && from && wxStrlen(from))
1053 appendFromClause = true;
1054 #else
1055 if (typeOfSelect == DB_SELECT_WHERE && from.Length())
1056 appendFromClause = true;
1057 #endif
1058
1059 // Add the column list
1060 int i;
1061 wxString tStr;
1062 for (i = 0; i < noCols; i++)
1063 {
1064 tStr = colDefs[i].ColName;
1065 // If joining tables, the base table column names must be qualified to avoid ambiguity
1066 if ((appendFromClause || pDb->Dbms() == dbmsACCESS) && tStr.Find(wxT('.')) == wxNOT_FOUND)
1067 {
1068 pSqlStmt += pDb->SQLTableName(queryTableName.c_str());
1069 pSqlStmt += wxT(".");
1070 }
1071 pSqlStmt += pDb->SQLColumnName(colDefs[i].ColName);
1072 if (i + 1 < noCols)
1073 pSqlStmt += wxT(",");
1074 }
1075
1076 // If the datasource supports ROWID, get this column as well. Exception: Don't retrieve
1077 // the ROWID if querying distinct records. The rowid will always be unique.
1078 if (!distinct && CanUpdByROWID())
1079 {
1080 // If joining tables, the base table column names must be qualified to avoid ambiguity
1081 if (appendFromClause || pDb->Dbms() == dbmsACCESS)
1082 {
1083 pSqlStmt += wxT(",");
1084 pSqlStmt += pDb->SQLTableName(queryTableName);
1085 // pSqlStmt += queryTableName;
1086 pSqlStmt += wxT(".ROWID");
1087 }
1088 else
1089 pSqlStmt += wxT(",ROWID");
1090 }
1091
1092 // Append the FROM tablename portion
1093 pSqlStmt += wxT(" FROM ");
1094 pSqlStmt += pDb->SQLTableName(queryTableName);
1095 // pSqlStmt += queryTableName;
1096
1097 // Sybase uses the HOLDLOCK keyword to lock a record during query.
1098 // The HOLDLOCK keyword follows the table name in the from clause.
1099 // Each table in the from clause must specify HOLDLOCK or
1100 // NOHOLDLOCK (the default). Note: The "FOR UPDATE" clause
1101 // is parsed but ignored in SYBASE Transact-SQL.
1102 if (selectForUpdate && (pDb->Dbms() == dbmsSYBASE_ASA || pDb->Dbms() == dbmsSYBASE_ASE))
1103 pSqlStmt += wxT(" HOLDLOCK");
1104
1105 if (appendFromClause)
1106 pSqlStmt += from;
1107
1108 // Append the WHERE clause. Either append the where clause for the class
1109 // or build a where clause. The typeOfSelect determines this.
1110 switch(typeOfSelect)
1111 {
1112 case DB_SELECT_WHERE:
1113 #if wxODBC_BACKWARD_COMPATABILITY
1114 if (where && wxStrlen(where)) // May not want a where clause!!!
1115 #else
1116 if (where.Length()) // May not want a where clause!!!
1117 #endif
1118 {
1119 pSqlStmt += wxT(" WHERE ");
1120 pSqlStmt += where;
1121 }
1122 break;
1123 case DB_SELECT_KEYFIELDS:
1124 BuildWhereClause(whereClause, DB_WHERE_KEYFIELDS);
1125 if (whereClause.Length())
1126 {
1127 pSqlStmt += wxT(" WHERE ");
1128 pSqlStmt += whereClause;
1129 }
1130 break;
1131 case DB_SELECT_MATCHING:
1132 BuildWhereClause(whereClause, DB_WHERE_MATCHING);
1133 if (whereClause.Length())
1134 {
1135 pSqlStmt += wxT(" WHERE ");
1136 pSqlStmt += whereClause;
1137 }
1138 break;
1139 }
1140
1141 // Append the ORDER BY clause
1142 #if wxODBC_BACKWARD_COMPATABILITY
1143 if (orderBy && wxStrlen(orderBy))
1144 #else
1145 if (orderBy.Length())
1146 #endif
1147 {
1148 pSqlStmt += wxT(" ORDER BY ");
1149 pSqlStmt += orderBy;
1150 }
1151
1152 // SELECT FOR UPDATE if told to do so and the datasource is capable. Sybase
1153 // parses the FOR UPDATE clause but ignores it. See the comment above on the
1154 // HOLDLOCK for Sybase.
1155 if (selectForUpdate && CanSelectForUpdate())
1156 pSqlStmt += wxT(" FOR UPDATE");
1157
1158 } // wxDbTable::BuildSelectStmt()
1159
1160
1161 /***** DEPRECATED: use wxDbTable::BuildSelectStmt(wxString &....) form *****/
1162 void wxDbTable::BuildSelectStmt(wxChar *pSqlStmt, int typeOfSelect, bool distinct)
1163 {
1164 wxString tempSqlStmt;
1165 BuildSelectStmt(tempSqlStmt, typeOfSelect, distinct);
1166 wxStrcpy(pSqlStmt, tempSqlStmt);
1167 } // wxDbTable::BuildSelectStmt()
1168
1169
1170 /********** wxDbTable::BuildUpdateStmt() **********/
1171 void wxDbTable::BuildUpdateStmt(wxString &pSqlStmt, int typeOfUpd, const wxString &pWhereClause)
1172 {
1173 wxASSERT(!queryOnly);
1174 if (queryOnly)
1175 return;
1176
1177 wxString whereClause;
1178 whereClause.Empty();
1179
1180 bool firstColumn = true;
1181
1182 pSqlStmt.Printf(wxT("UPDATE %s SET "),
1183 pDb->SQLTableName(tableName.c_str()).c_str());
1184
1185 // Append a list of columns to be updated
1186 int i;
1187 for (i = 0; i < noCols; i++)
1188 {
1189 // Only append Updateable columns
1190 if (colDefs[i].Updateable)
1191 {
1192 if (!firstColumn)
1193 pSqlStmt += wxT(",");
1194 else
1195 firstColumn = false;
1196
1197 pSqlStmt += pDb->SQLColumnName(colDefs[i].ColName);
1198 // pSqlStmt += colDefs[i].ColName;
1199 pSqlStmt += wxT(" = ?");
1200 }
1201 }
1202
1203 // Append the WHERE clause to the SQL UPDATE statement
1204 pSqlStmt += wxT(" WHERE ");
1205 switch(typeOfUpd)
1206 {
1207 case DB_UPD_KEYFIELDS:
1208 // If the datasource supports the ROWID column, build
1209 // the where on ROWID for efficiency purposes.
1210 // e.g. UPDATE PARTS SET Col1 = ?, Col2 = ? WHERE ROWID = '111.222.333'
1211 if (CanUpdByROWID())
1212 {
1213 SDWORD cb;
1214 wxChar rowid[wxDB_ROWID_LEN+1];
1215
1216 // Get the ROWID value. If not successful retreiving the ROWID,
1217 // simply fall down through the code and build the WHERE clause
1218 // based on the key fields.
1219 if (SQLGetData(hstmt, (UWORD)(noCols+1), SQL_C_CHAR, (UCHAR*) rowid, wxDB_ROWID_LEN, &cb) == SQL_SUCCESS)
1220 {
1221 pSqlStmt += wxT("ROWID = '");
1222 pSqlStmt += rowid;
1223 pSqlStmt += wxT("'");
1224 break;
1225 }
1226 }
1227 // Unable to delete by ROWID, so build a WHERE
1228 // clause based on the keyfields.
1229 BuildWhereClause(whereClause, DB_WHERE_KEYFIELDS);
1230 pSqlStmt += whereClause;
1231 break;
1232 case DB_UPD_WHERE:
1233 pSqlStmt += pWhereClause;
1234 break;
1235 }
1236 } // BuildUpdateStmt()
1237
1238
1239 /***** DEPRECATED: use wxDbTable::BuildUpdateStmt(wxString &....) form *****/
1240 void wxDbTable::BuildUpdateStmt(wxChar *pSqlStmt, int typeOfUpd, const wxString &pWhereClause)
1241 {
1242 wxString tempSqlStmt;
1243 BuildUpdateStmt(tempSqlStmt, typeOfUpd, pWhereClause);
1244 wxStrcpy(pSqlStmt, tempSqlStmt);
1245 } // BuildUpdateStmt()
1246
1247
1248 /********** wxDbTable::BuildWhereClause() **********/
1249 void wxDbTable::BuildWhereClause(wxString &pWhereClause, int typeOfWhere,
1250 const wxString &qualTableName, bool useLikeComparison)
1251 /*
1252 * Note: BuildWhereClause() currently ignores timestamp columns.
1253 * They are not included as part of the where clause.
1254 */
1255 {
1256 bool moreThanOneColumn = false;
1257 wxString colValue;
1258
1259 // Loop through the columns building a where clause as you go
1260 int colNo;
1261 for (colNo = 0; colNo < noCols; colNo++)
1262 {
1263 // Determine if this column should be included in the WHERE clause
1264 if ((typeOfWhere == DB_WHERE_KEYFIELDS && colDefs[colNo].KeyField) ||
1265 (typeOfWhere == DB_WHERE_MATCHING && (!IsColNull((UWORD)colNo))))
1266 {
1267 // Skip over timestamp columns
1268 if (colDefs[colNo].SqlCtype == SQL_C_TIMESTAMP)
1269 continue;
1270 // If there is more than 1 column, join them with the keyword "AND"
1271 if (moreThanOneColumn)
1272 pWhereClause += wxT(" AND ");
1273 else
1274 moreThanOneColumn = true;
1275
1276 // Concatenate where phrase for the column
1277 wxString tStr = colDefs[colNo].ColName;
1278
1279 if (qualTableName.Length() && tStr.Find(wxT('.')) == wxNOT_FOUND)
1280 {
1281 pWhereClause += pDb->SQLTableName(qualTableName);
1282 pWhereClause += wxT(".");
1283 }
1284 pWhereClause += pDb->SQLColumnName(colDefs[colNo].ColName);
1285
1286 if (useLikeComparison && (colDefs[colNo].SqlCtype == SQL_C_CHAR))
1287 pWhereClause += wxT(" LIKE ");
1288 else
1289 pWhereClause += wxT(" = ");
1290
1291 switch(colDefs[colNo].SqlCtype)
1292 {
1293 case SQL_C_CHAR:
1294 colValue.Printf(wxT("'%s'"), (UCHAR FAR *) colDefs[colNo].PtrDataObj);
1295 break;
1296 case SQL_C_SHORT:
1297 case SQL_C_SSHORT:
1298 colValue.Printf(wxT("%hi"), *((SWORD *) colDefs[colNo].PtrDataObj));
1299 break;
1300 case SQL_C_USHORT:
1301 colValue.Printf(wxT("%hu"), *((UWORD *) colDefs[colNo].PtrDataObj));
1302 break;
1303 case SQL_C_LONG:
1304 case SQL_C_SLONG:
1305 colValue.Printf(wxT("%li"), *((SDWORD *) colDefs[colNo].PtrDataObj));
1306 break;
1307 case SQL_C_ULONG:
1308 colValue.Printf(wxT("%lu"), *((UDWORD *) colDefs[colNo].PtrDataObj));
1309 break;
1310 case SQL_C_FLOAT:
1311 colValue.Printf(wxT("%.6f"), *((SFLOAT *) colDefs[colNo].PtrDataObj));
1312 break;
1313 case SQL_C_DOUBLE:
1314 colValue.Printf(wxT("%.6f"), *((SDOUBLE *) colDefs[colNo].PtrDataObj));
1315 break;
1316 default:
1317 {
1318 wxString strMsg;
1319 strMsg.Printf(wxT("wxDbTable::bindParams(): Unknown column type for colDefs %d colName %s"),
1320 colNo,colDefs[colNo].ColName);
1321 wxFAIL_MSG(strMsg.c_str());
1322 }
1323 break;
1324 }
1325 pWhereClause += colValue;
1326 }
1327 }
1328 } // wxDbTable::BuildWhereClause()
1329
1330
1331 /***** DEPRECATED: use wxDbTable::BuildWhereClause(wxString &....) form *****/
1332 void wxDbTable::BuildWhereClause(wxChar *pWhereClause, int typeOfWhere,
1333 const wxString &qualTableName, bool useLikeComparison)
1334 {
1335 wxString tempSqlStmt;
1336 BuildWhereClause(tempSqlStmt, typeOfWhere, qualTableName, useLikeComparison);
1337 wxStrcpy(pWhereClause, tempSqlStmt);
1338 } // wxDbTable::BuildWhereClause()
1339
1340
1341 /********** wxDbTable::GetRowNum() **********/
1342 UWORD wxDbTable::GetRowNum(void)
1343 {
1344 UDWORD rowNum;
1345
1346 if (SQLGetStmtOption(hstmt, SQL_ROW_NUMBER, (UCHAR*) &rowNum) != SQL_SUCCESS)
1347 {
1348 pDb->DispAllErrors(henv, hdbc, hstmt);
1349 return(0);
1350 }
1351
1352 // Completed successfully
1353 return((UWORD) rowNum);
1354
1355 } // wxDbTable::GetRowNum()
1356
1357
1358 /********** wxDbTable::CloseCursor() **********/
1359 bool wxDbTable::CloseCursor(HSTMT cursor)
1360 {
1361 if (SQLFreeStmt(cursor, SQL_CLOSE) != SQL_SUCCESS)
1362 return(pDb->DispAllErrors(henv, hdbc, cursor));
1363
1364 // Completed successfully
1365 return true;
1366
1367 } // wxDbTable::CloseCursor()
1368
1369
1370 /********** wxDbTable::CreateTable() **********/
1371 bool wxDbTable::CreateTable(bool attemptDrop)
1372 {
1373 if (!pDb)
1374 return false;
1375
1376 int i, j;
1377 wxString sqlStmt;
1378
1379 #ifdef DBDEBUG_CONSOLE
1380 cout << wxT("Creating Table ") << tableName << wxT("...") << endl;
1381 #endif
1382
1383 // Drop table first
1384 if (attemptDrop && !DropTable())
1385 return false;
1386
1387 // Create the table
1388 #ifdef DBDEBUG_CONSOLE
1389 for (i = 0; i < noCols; i++)
1390 {
1391 // Exclude derived columns since they are NOT part of the base table
1392 if (colDefs[i].DerivedCol)
1393 continue;
1394 cout << i + 1 << wxT(": ") << colDefs[i].ColName << wxT("; ");
1395 switch(colDefs[i].DbDataType)
1396 {
1397 case DB_DATA_TYPE_VARCHAR:
1398 cout << pDb->GetTypeInfVarchar().TypeName << wxT("(") << colDefs[i].SzDataObj << wxT(")");
1399 break;
1400 case DB_DATA_TYPE_INTEGER:
1401 cout << pDb->GetTypeInfInteger().TypeName;
1402 break;
1403 case DB_DATA_TYPE_FLOAT:
1404 cout << pDb->GetTypeInfFloat().TypeName;
1405 break;
1406 case DB_DATA_TYPE_DATE:
1407 cout << pDb->GetTypeInfDate().TypeName;
1408 break;
1409 case DB_DATA_TYPE_BLOB:
1410 cout << pDb->GetTypeInfBlob().TypeName;
1411 break;
1412 }
1413 cout << endl;
1414 }
1415 #endif
1416
1417 // Build a CREATE TABLE string from the colDefs structure.
1418 bool needComma = false;
1419
1420 sqlStmt.Printf(wxT("CREATE TABLE %s ("),
1421 pDb->SQLTableName(tableName.c_str()).c_str());
1422
1423 for (i = 0; i < noCols; i++)
1424 {
1425 // Exclude derived columns since they are NOT part of the base table
1426 if (colDefs[i].DerivedCol)
1427 continue;
1428 // Comma Delimiter
1429 if (needComma)
1430 sqlStmt += wxT(",");
1431 // Column Name
1432 sqlStmt += pDb->SQLColumnName(colDefs[i].ColName);
1433 // sqlStmt += colDefs[i].ColName;
1434 sqlStmt += wxT(" ");
1435 // Column Type
1436 switch(colDefs[i].DbDataType)
1437 {
1438 case DB_DATA_TYPE_VARCHAR:
1439 sqlStmt += pDb->GetTypeInfVarchar().TypeName;
1440 break;
1441 case DB_DATA_TYPE_INTEGER:
1442 sqlStmt += pDb->GetTypeInfInteger().TypeName;
1443 break;
1444 case DB_DATA_TYPE_FLOAT:
1445 sqlStmt += pDb->GetTypeInfFloat().TypeName;
1446 break;
1447 case DB_DATA_TYPE_DATE:
1448 sqlStmt += pDb->GetTypeInfDate().TypeName;
1449 break;
1450 case DB_DATA_TYPE_BLOB:
1451 sqlStmt += pDb->GetTypeInfBlob().TypeName;
1452 break;
1453 }
1454 // For varchars, append the size of the string
1455 if (colDefs[i].DbDataType == DB_DATA_TYPE_VARCHAR &&
1456 (pDb->Dbms() != dbmsMY_SQL || pDb->GetTypeInfVarchar().TypeName != _T("text")))// ||
1457 // colDefs[i].DbDataType == DB_DATA_TYPE_BLOB)
1458 {
1459 wxString s;
1460 s.Printf(wxT("(%d)"), colDefs[i].SzDataObj);
1461 sqlStmt += s;
1462 }
1463
1464 if (pDb->Dbms() == dbmsDB2 ||
1465 pDb->Dbms() == dbmsMY_SQL ||
1466 pDb->Dbms() == dbmsSYBASE_ASE ||
1467 pDb->Dbms() == dbmsINTERBASE ||
1468 pDb->Dbms() == dbmsMS_SQL_SERVER)
1469 {
1470 if (colDefs[i].KeyField)
1471 {
1472 sqlStmt += wxT(" NOT NULL");
1473 }
1474 }
1475
1476 needComma = true;
1477 }
1478 // If there is a primary key defined, include it in the create statement
1479 for (i = j = 0; i < noCols; i++)
1480 {
1481 if (colDefs[i].KeyField)
1482 {
1483 j++;
1484 break;
1485 }
1486 }
1487 if ( j && (pDb->Dbms() != dbmsDBASE)
1488 && (pDb->Dbms() != dbmsXBASE_SEQUITER) ) // Found a keyfield
1489 {
1490 switch (pDb->Dbms())
1491 {
1492 case dbmsACCESS:
1493 case dbmsINFORMIX:
1494 case dbmsSYBASE_ASA:
1495 case dbmsSYBASE_ASE:
1496 case dbmsMY_SQL:
1497 {
1498 // MySQL goes out on this one. We also declare the relevant key NON NULL above
1499 sqlStmt += wxT(",PRIMARY KEY (");
1500 break;
1501 }
1502 default:
1503 {
1504 sqlStmt += wxT(",CONSTRAINT ");
1505 // DB2 is limited to 18 characters for index names
1506 if (pDb->Dbms() == dbmsDB2)
1507 {
1508 wxASSERT_MSG((tableName && wxStrlen(tableName) <= 13), wxT("DB2 table/index names must be no longer than 13 characters in length.\n\nTruncating table name to 13 characters."));
1509 sqlStmt += pDb->SQLTableName(tableName.substr(0, 13).c_str());
1510 // sqlStmt += tableName.substr(0, 13);
1511 }
1512 else
1513 sqlStmt += pDb->SQLTableName(tableName.c_str());
1514 // sqlStmt += tableName;
1515
1516 sqlStmt += wxT("_PIDX PRIMARY KEY (");
1517 break;
1518 }
1519 }
1520
1521 // List column name(s) of column(s) comprising the primary key
1522 for (i = j = 0; i < noCols; i++)
1523 {
1524 if (colDefs[i].KeyField)
1525 {
1526 if (j++) // Multi part key, comma separate names
1527 sqlStmt += wxT(",");
1528 sqlStmt += pDb->SQLColumnName(colDefs[i].ColName);
1529
1530 if (pDb->Dbms() == dbmsMY_SQL &&
1531 colDefs[i].DbDataType == DB_DATA_TYPE_VARCHAR)
1532 {
1533 wxString s;
1534 s.Printf(wxT("(%d)"), colDefs[i].SzDataObj);
1535 sqlStmt += s;
1536 }
1537 }
1538 }
1539 sqlStmt += wxT(")");
1540
1541 if (pDb->Dbms() == dbmsINFORMIX ||
1542 pDb->Dbms() == dbmsSYBASE_ASA ||
1543 pDb->Dbms() == dbmsSYBASE_ASE)
1544 {
1545 sqlStmt += wxT(" CONSTRAINT ");
1546 sqlStmt += pDb->SQLTableName(tableName);
1547 // sqlStmt += tableName;
1548 sqlStmt += wxT("_PIDX");
1549 }
1550 }
1551 // Append the closing parentheses for the create table statement
1552 sqlStmt += wxT(")");
1553
1554 pDb->WriteSqlLog(sqlStmt);
1555
1556 #ifdef DBDEBUG_CONSOLE
1557 cout << endl << sqlStmt.c_str() << endl;
1558 #endif
1559
1560 // Execute the CREATE TABLE statement
1561 RETCODE retcode = SQLExecDirect(hstmt, (SQLTCHAR FAR *) sqlStmt.c_str(), SQL_NTS);
1562 if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)
1563 {
1564 pDb->DispAllErrors(henv, hdbc, hstmt);
1565 pDb->RollbackTrans();
1566 CloseCursor(hstmt);
1567 return false;
1568 }
1569
1570 // Commit the transaction and close the cursor
1571 if (!pDb->CommitTrans())
1572 return false;
1573 if (!CloseCursor(hstmt))
1574 return false;
1575
1576 // Database table created successfully
1577 return true;
1578
1579 } // wxDbTable::CreateTable()
1580
1581
1582 /********** wxDbTable::DropTable() **********/
1583 bool wxDbTable::DropTable()
1584 {
1585 // NOTE: This function returns true if the Table does not exist, but
1586 // only for identified databases. Code will need to be added
1587 // below for any other databases when those databases are defined
1588 // to handle this situation consistently
1589
1590 wxString sqlStmt;
1591
1592 sqlStmt.Printf(wxT("DROP TABLE %s"),
1593 pDb->SQLTableName(tableName.c_str()).c_str());
1594
1595 pDb->WriteSqlLog(sqlStmt);
1596
1597 #ifdef DBDEBUG_CONSOLE
1598 cout << endl << sqlStmt.c_str() << endl;
1599 #endif
1600
1601 RETCODE retcode = SQLExecDirect(hstmt, (SQLTCHAR FAR *) sqlStmt.c_str(), SQL_NTS);
1602 if (retcode != SQL_SUCCESS)
1603 {
1604 // Check for "Base table not found" error and ignore
1605 pDb->GetNextError(henv, hdbc, hstmt);
1606 if (wxStrcmp(pDb->sqlState, wxT("S0002")) /*&&
1607 wxStrcmp(pDb->sqlState, wxT("S1000"))*/) // "Base table not found"
1608 {
1609 // Check for product specific error codes
1610 if (!((pDb->Dbms() == dbmsSYBASE_ASA && !wxStrcmp(pDb->sqlState,wxT("42000"))) || // 5.x (and lower?)
1611 (pDb->Dbms() == dbmsSYBASE_ASE && !wxStrcmp(pDb->sqlState,wxT("37000"))) ||
1612 (pDb->Dbms() == dbmsPERVASIVE_SQL && !wxStrcmp(pDb->sqlState,wxT("S1000"))) || // Returns an S1000 then an S0002
1613 (pDb->Dbms() == dbmsPOSTGRES && !wxStrcmp(pDb->sqlState,wxT("08S01")))))
1614 {
1615 pDb->DispNextError();
1616 pDb->DispAllErrors(henv, hdbc, hstmt);
1617 pDb->RollbackTrans();
1618 // CloseCursor(hstmt);
1619 return false;
1620 }
1621 }
1622 }
1623
1624 // Commit the transaction and close the cursor
1625 if (! pDb->CommitTrans())
1626 return false;
1627 if (! CloseCursor(hstmt))
1628 return false;
1629
1630 return true;
1631 } // wxDbTable::DropTable()
1632
1633
1634 /********** wxDbTable::CreateIndex() **********/
1635 bool wxDbTable::CreateIndex(const wxString &idxName, bool unique, UWORD noIdxCols,
1636 wxDbIdxDef *pIdxDefs, bool attemptDrop)
1637 {
1638 wxString sqlStmt;
1639
1640 // Drop the index first
1641 if (attemptDrop && !DropIndex(idxName))
1642 return false;
1643
1644 // MySQL (and possibly Sybase ASE?? - gt) require that any columns which are used as portions
1645 // of an index have the columns defined as "NOT NULL". During initial table creation though,
1646 // it may not be known which columns are necessarily going to be part of an index (e.g. the
1647 // table was created, then months later you determine that an additional index while
1648 // give better performance, so you want to add an index).
1649 //
1650 // The following block of code will modify the column definition to make the column be
1651 // defined with the "NOT NULL" qualifier.
1652 if (pDb->Dbms() == dbmsMY_SQL)
1653 {
1654 wxString sqlStmt;
1655 int i;
1656 bool ok = true;
1657 for (i = 0; i < noIdxCols && ok; i++)
1658 {
1659 int j = 0;
1660 bool found = false;
1661 // Find the column definition that has the ColName that matches the
1662 // index column name. We need to do this to get the DB_DATA_TYPE of
1663 // the index column, as MySQL's syntax for the ALTER column requires
1664 // this information
1665 while (!found && (j < this->noCols))
1666 {
1667 if (wxStrcmp(colDefs[j].ColName,pIdxDefs[i].ColName) == 0)
1668 found = true;
1669 if (!found)
1670 j++;
1671 }
1672
1673 if (found)
1674 {
1675 ok = pDb->ModifyColumn(tableName, pIdxDefs[i].ColName,
1676 colDefs[j].DbDataType, colDefs[j].SzDataObj,
1677 wxT("NOT NULL"));
1678
1679 if (!ok)
1680 {
1681 #if 0
1682 // retcode is not used
1683 wxODBC_ERRORS retcode;
1684 // Oracle returns a DB_ERR_GENERAL_ERROR if the column is already
1685 // defined to be NOT NULL, but reportedly MySQL doesn't mind.
1686 // This line is just here for debug checking of the value
1687 retcode = (wxODBC_ERRORS)pDb->DB_STATUS;
1688 #endif
1689 }
1690 }
1691 else
1692 ok = false;
1693 }
1694 if (ok)
1695 pDb->CommitTrans();
1696 else
1697 {
1698 pDb->RollbackTrans();
1699 return false;
1700 }
1701 }
1702
1703 // Build a CREATE INDEX statement
1704 sqlStmt = wxT("CREATE ");
1705 if (unique)
1706 sqlStmt += wxT("UNIQUE ");
1707
1708 sqlStmt += wxT("INDEX ");
1709 sqlStmt += pDb->SQLTableName(idxName);
1710 sqlStmt += wxT(" ON ");
1711
1712 sqlStmt += pDb->SQLTableName(tableName);
1713 // sqlStmt += tableName;
1714 sqlStmt += wxT(" (");
1715
1716 // Append list of columns making up index
1717 int i;
1718 for (i = 0; i < noIdxCols; i++)
1719 {
1720 sqlStmt += pDb->SQLColumnName(pIdxDefs[i].ColName);
1721 // sqlStmt += pIdxDefs[i].ColName;
1722
1723 // MySQL requires a key length on VARCHAR keys
1724 if ( pDb->Dbms() == dbmsMY_SQL )
1725 {
1726 // Find the details on this column
1727 int j;
1728 for ( j = 0; j < noCols; ++j )
1729 {
1730 if ( wxStrcmp( pIdxDefs[i].ColName, colDefs[j].ColName ) == 0 )
1731 {
1732 break;
1733 }
1734 }
1735 if ( colDefs[j].DbDataType == DB_DATA_TYPE_VARCHAR)
1736 {
1737 wxString s;
1738 s.Printf(wxT("(%d)"), colDefs[i].SzDataObj);
1739 sqlStmt += s;
1740 }
1741 }
1742
1743 // Postgres and SQL Server 7 do not support the ASC/DESC keywords for index columns
1744 if (!((pDb->Dbms() == dbmsMS_SQL_SERVER) && (wxStrncmp(pDb->dbInf.dbmsVer,_T("07"),2)==0)) &&
1745 !(pDb->Dbms() == dbmsPOSTGRES))
1746 {
1747 if (pIdxDefs[i].Ascending)
1748 sqlStmt += wxT(" ASC");
1749 else
1750 sqlStmt += wxT(" DESC");
1751 }
1752 else
1753 wxASSERT_MSG(pIdxDefs[i].Ascending, _T("Datasource does not support DESCending index columns"));
1754
1755 if ((i + 1) < noIdxCols)
1756 sqlStmt += wxT(",");
1757 }
1758
1759 // Append closing parentheses
1760 sqlStmt += wxT(")");
1761
1762 pDb->WriteSqlLog(sqlStmt);
1763
1764 #ifdef DBDEBUG_CONSOLE
1765 cout << endl << sqlStmt.c_str() << endl << endl;
1766 #endif
1767
1768 // Execute the CREATE INDEX statement
1769 if (SQLExecDirect(hstmt, (SQLTCHAR FAR *) sqlStmt.c_str(), SQL_NTS) != SQL_SUCCESS)
1770 {
1771 pDb->DispAllErrors(henv, hdbc, hstmt);
1772 pDb->RollbackTrans();
1773 CloseCursor(hstmt);
1774 return false;
1775 }
1776
1777 // Commit the transaction and close the cursor
1778 if (! pDb->CommitTrans())
1779 return false;
1780 if (! CloseCursor(hstmt))
1781 return false;
1782
1783 // Index Created Successfully
1784 return true;
1785
1786 } // wxDbTable::CreateIndex()
1787
1788
1789 /********** wxDbTable::DropIndex() **********/
1790 bool wxDbTable::DropIndex(const wxString &idxName)
1791 {
1792 // NOTE: This function returns true if the Index does not exist, but
1793 // only for identified databases. Code will need to be added
1794 // below for any other databases when those databases are defined
1795 // to handle this situation consistently
1796
1797 wxString sqlStmt;
1798
1799 if (pDb->Dbms() == dbmsACCESS || pDb->Dbms() == dbmsMY_SQL ||
1800 pDb->Dbms() == dbmsDBASE /*|| Paradox needs this syntax too when we add support*/)
1801 sqlStmt.Printf(wxT("DROP INDEX %s ON %s"),
1802 pDb->SQLTableName(idxName.c_str()).c_str(),
1803 pDb->SQLTableName(tableName.c_str()).c_str());
1804 else if ((pDb->Dbms() == dbmsMS_SQL_SERVER) ||
1805 (pDb->Dbms() == dbmsSYBASE_ASE) ||
1806 (pDb->Dbms() == dbmsXBASE_SEQUITER))
1807 sqlStmt.Printf(wxT("DROP INDEX %s.%s"),
1808 pDb->SQLTableName(tableName.c_str()).c_str(),
1809 pDb->SQLTableName(idxName.c_str()).c_str());
1810 else
1811 sqlStmt.Printf(wxT("DROP INDEX %s"),
1812 pDb->SQLTableName(idxName.c_str()).c_str());
1813
1814 pDb->WriteSqlLog(sqlStmt);
1815
1816 #ifdef DBDEBUG_CONSOLE
1817 cout << endl << sqlStmt.c_str() << endl;
1818 #endif
1819
1820 if (SQLExecDirect(hstmt, (SQLTCHAR FAR *) sqlStmt.c_str(), SQL_NTS) != SQL_SUCCESS)
1821 {
1822 // Check for "Index not found" error and ignore
1823 pDb->GetNextError(henv, hdbc, hstmt);
1824 if (wxStrcmp(pDb->sqlState,wxT("S0012"))) // "Index not found"
1825 {
1826 // Check for product specific error codes
1827 if (!((pDb->Dbms() == dbmsSYBASE_ASA && !wxStrcmp(pDb->sqlState,wxT("42000"))) || // v5.x (and lower?)
1828 (pDb->Dbms() == dbmsSYBASE_ASE && !wxStrcmp(pDb->sqlState,wxT("37000"))) ||
1829 (pDb->Dbms() == dbmsMS_SQL_SERVER && !wxStrcmp(pDb->sqlState,wxT("S1000"))) ||
1830 (pDb->Dbms() == dbmsINTERBASE && !wxStrcmp(pDb->sqlState,wxT("S1000"))) ||
1831 (pDb->Dbms() == dbmsSYBASE_ASE && !wxStrcmp(pDb->sqlState,wxT("S0002"))) || // Base table not found
1832 (pDb->Dbms() == dbmsMY_SQL && !wxStrcmp(pDb->sqlState,wxT("42S12"))) || // tested by Christopher Ludwik Marino-Cebulski using v3.23.21beta
1833 (pDb->Dbms() == dbmsPOSTGRES && !wxStrcmp(pDb->sqlState,wxT("08S01")))
1834 ))
1835 {
1836 pDb->DispNextError();
1837 pDb->DispAllErrors(henv, hdbc, hstmt);
1838 pDb->RollbackTrans();
1839 CloseCursor(hstmt);
1840 return false;
1841 }
1842 }
1843 }
1844
1845 // Commit the transaction and close the cursor
1846 if (! pDb->CommitTrans())
1847 return false;
1848 if (! CloseCursor(hstmt))
1849 return false;
1850
1851 return true;
1852 } // wxDbTable::DropIndex()
1853
1854
1855 /********** wxDbTable::SetOrderByColNums() **********/
1856 bool wxDbTable::SetOrderByColNums(UWORD first, ... )
1857 {
1858 int colNo = first; // using 'int' to be able to look for wxDB_NO_MORE_COLUN_NUMBERS
1859 va_list argptr;
1860
1861 bool abort = false;
1862 wxString tempStr;
1863
1864 va_start(argptr, first); /* Initialize variable arguments. */
1865 while (!abort && (colNo != wxDB_NO_MORE_COLUMN_NUMBERS))
1866 {
1867 // Make sure the passed in column number
1868 // is within the valid range of columns
1869 //
1870 // Valid columns are 0 thru noCols-1
1871 if (colNo >= noCols || colNo < 0)
1872 {
1873 abort = true;
1874 continue;
1875 }
1876
1877 if (colNo != first)
1878 tempStr += wxT(",");
1879
1880 tempStr += colDefs[colNo].ColName;
1881 colNo = va_arg (argptr, int);
1882 }
1883 va_end (argptr); /* Reset variable arguments. */
1884
1885 SetOrderByClause(tempStr);
1886
1887 return (!abort);
1888 } // wxDbTable::SetOrderByColNums()
1889
1890
1891 /********** wxDbTable::Insert() **********/
1892 int wxDbTable::Insert(void)
1893 {
1894 wxASSERT(!queryOnly);
1895 if (queryOnly || !insertable)
1896 return(DB_FAILURE);
1897
1898 bindInsertParams();
1899
1900 // Insert the record by executing the already prepared insert statement
1901 RETCODE retcode;
1902 retcode=SQLExecute(hstmtInsert);
1903 if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO &&
1904 retcode != SQL_NEED_DATA)
1905 {
1906 // Check to see if integrity constraint was violated
1907 pDb->GetNextError(henv, hdbc, hstmtInsert);
1908 if (! wxStrcmp(pDb->sqlState, wxT("23000"))) // Integrity constraint violated
1909 return(DB_ERR_INTEGRITY_CONSTRAINT_VIOL);
1910 else
1911 {
1912 pDb->DispNextError();
1913 pDb->DispAllErrors(henv, hdbc, hstmtInsert);
1914 return(DB_FAILURE);
1915 }
1916 }
1917 if (retcode == SQL_NEED_DATA)
1918 {
1919 PTR pParmID;
1920 retcode = SQLParamData(hstmtInsert, &pParmID);
1921 while (retcode == SQL_NEED_DATA)
1922 {
1923 // Find the parameter
1924 int i;
1925 for (i=0; i < noCols; i++)
1926 {
1927 if (colDefs[i].PtrDataObj == pParmID)
1928 {
1929 // We found it. Store the parameter.
1930 retcode = SQLPutData(hstmtInsert, pParmID, colDefs[i].SzDataObj);
1931 if (retcode != SQL_SUCCESS)
1932 {
1933 pDb->DispNextError();
1934 pDb->DispAllErrors(henv, hdbc, hstmtInsert);
1935 return(DB_FAILURE);
1936 }
1937 break;
1938 }
1939 }
1940 retcode = SQLParamData(hstmtInsert, &pParmID);
1941 if (retcode != SQL_SUCCESS &&
1942 retcode != SQL_SUCCESS_WITH_INFO)
1943 {
1944 // record was not inserted
1945 pDb->DispNextError();
1946 pDb->DispAllErrors(henv, hdbc, hstmtInsert);
1947 return(DB_FAILURE);
1948 }
1949 }
1950 }
1951
1952 // Record inserted into the datasource successfully
1953 return(DB_SUCCESS);
1954
1955 } // wxDbTable::Insert()
1956
1957
1958 /********** wxDbTable::Update() **********/
1959 bool wxDbTable::Update(void)
1960 {
1961 wxASSERT(!queryOnly);
1962 if (queryOnly)
1963 return false;
1964
1965 wxString sqlStmt;
1966
1967 // Build the SQL UPDATE statement
1968 BuildUpdateStmt(sqlStmt, DB_UPD_KEYFIELDS);
1969
1970 pDb->WriteSqlLog(sqlStmt);
1971
1972 #ifdef DBDEBUG_CONSOLE
1973 cout << endl << sqlStmt.c_str() << endl << endl;
1974 #endif
1975
1976 // Execute the SQL UPDATE statement
1977 return(execUpdate(sqlStmt));
1978
1979 } // wxDbTable::Update()
1980
1981
1982 /********** wxDbTable::Update(pSqlStmt) **********/
1983 bool wxDbTable::Update(const wxString &pSqlStmt)
1984 {
1985 wxASSERT(!queryOnly);
1986 if (queryOnly)
1987 return false;
1988
1989 pDb->WriteSqlLog(pSqlStmt);
1990
1991 return(execUpdate(pSqlStmt));
1992
1993 } // wxDbTable::Update(pSqlStmt)
1994
1995
1996 /********** wxDbTable::UpdateWhere() **********/
1997 bool wxDbTable::UpdateWhere(const wxString &pWhereClause)
1998 {
1999 wxASSERT(!queryOnly);
2000 if (queryOnly)
2001 return false;
2002
2003 wxString sqlStmt;
2004
2005 // Build the SQL UPDATE statement
2006 BuildUpdateStmt(sqlStmt, DB_UPD_WHERE, pWhereClause);
2007
2008 pDb->WriteSqlLog(sqlStmt);
2009
2010 #ifdef DBDEBUG_CONSOLE
2011 cout << endl << sqlStmt.c_str() << endl << endl;
2012 #endif
2013
2014 // Execute the SQL UPDATE statement
2015 return(execUpdate(sqlStmt));
2016
2017 } // wxDbTable::UpdateWhere()
2018
2019
2020 /********** wxDbTable::Delete() **********/
2021 bool wxDbTable::Delete(void)
2022 {
2023 wxASSERT(!queryOnly);
2024 if (queryOnly)
2025 return false;
2026
2027 wxString sqlStmt;
2028 sqlStmt.Empty();
2029
2030 // Build the SQL DELETE statement
2031 BuildDeleteStmt(sqlStmt, DB_DEL_KEYFIELDS);
2032
2033 pDb->WriteSqlLog(sqlStmt);
2034
2035 // Execute the SQL DELETE statement
2036 return(execDelete(sqlStmt));
2037
2038 } // wxDbTable::Delete()
2039
2040
2041 /********** wxDbTable::DeleteWhere() **********/
2042 bool wxDbTable::DeleteWhere(const wxString &pWhereClause)
2043 {
2044 wxASSERT(!queryOnly);
2045 if (queryOnly)
2046 return false;
2047
2048 wxString sqlStmt;
2049 sqlStmt.Empty();
2050
2051 // Build the SQL DELETE statement
2052 BuildDeleteStmt(sqlStmt, DB_DEL_WHERE, pWhereClause);
2053
2054 pDb->WriteSqlLog(sqlStmt);
2055
2056 // Execute the SQL DELETE statement
2057 return(execDelete(sqlStmt));
2058
2059 } // wxDbTable::DeleteWhere()
2060
2061
2062 /********** wxDbTable::DeleteMatching() **********/
2063 bool wxDbTable::DeleteMatching(void)
2064 {
2065 wxASSERT(!queryOnly);
2066 if (queryOnly)
2067 return false;
2068
2069 wxString sqlStmt;
2070 sqlStmt.Empty();
2071
2072 // Build the SQL DELETE statement
2073 BuildDeleteStmt(sqlStmt, DB_DEL_MATCHING);
2074
2075 pDb->WriteSqlLog(sqlStmt);
2076
2077 // Execute the SQL DELETE statement
2078 return(execDelete(sqlStmt));
2079
2080 } // wxDbTable::DeleteMatching()
2081
2082
2083 /********** wxDbTable::IsColNull() **********/
2084 bool wxDbTable::IsColNull(UWORD colNo) const
2085 {
2086 /*
2087 This logic is just not right. It would indicate true
2088 if a numeric field were set to a value of 0.
2089
2090 switch(colDefs[colNo].SqlCtype)
2091 {
2092 case SQL_C_CHAR:
2093 return(((UCHAR FAR *) colDefs[colNo].PtrDataObj)[0] == 0);
2094 case SQL_C_SSHORT:
2095 return(( *((SWORD *) colDefs[colNo].PtrDataObj)) == 0);
2096 case SQL_C_USHORT:
2097 return(( *((UWORD*) colDefs[colNo].PtrDataObj)) == 0);
2098 case SQL_C_SLONG:
2099 return(( *((SDWORD *) colDefs[colNo].PtrDataObj)) == 0);
2100 case SQL_C_ULONG:
2101 return(( *((UDWORD *) colDefs[colNo].PtrDataObj)) == 0);
2102 case SQL_C_FLOAT:
2103 return(( *((SFLOAT *) colDefs[colNo].PtrDataObj)) == 0);
2104 case SQL_C_DOUBLE:
2105 return((*((SDOUBLE *) colDefs[colNo].PtrDataObj)) == 0);
2106 case SQL_C_TIMESTAMP:
2107 TIMESTAMP_STRUCT *pDt;
2108 pDt = (TIMESTAMP_STRUCT *) colDefs[colNo].PtrDataObj;
2109 if (pDt->year == 0 && pDt->month == 0 && pDt->day == 0)
2110 return true;
2111 else
2112 return false;
2113 default:
2114 return true;
2115 }
2116 */
2117 return (colDefs[colNo].Null);
2118 } // wxDbTable::IsColNull()
2119
2120
2121 /********** wxDbTable::CanSelectForUpdate() **********/
2122 bool wxDbTable::CanSelectForUpdate(void)
2123 {
2124 if (queryOnly)
2125 return false;
2126
2127 if (pDb->Dbms() == dbmsMY_SQL)
2128 return false;
2129
2130 if ((pDb->Dbms() == dbmsORACLE) ||
2131 (pDb->dbInf.posStmts & SQL_PS_SELECT_FOR_UPDATE))
2132 return true;
2133 else
2134 return false;
2135
2136 } // wxDbTable::CanSelectForUpdate()
2137
2138
2139 /********** wxDbTable::CanUpdByROWID() **********/
2140 bool wxDbTable::CanUpdByROWID(void)
2141 {
2142 /*
2143 * NOTE: Returning false for now until this can be debugged,
2144 * as the ROWID is not getting updated correctly
2145 */
2146 return false;
2147 /*
2148 if (pDb->Dbms() == dbmsORACLE)
2149 return true;
2150 else
2151 return false;
2152 */
2153 } // wxDbTable::CanUpdByROWID()
2154
2155
2156 /********** wxDbTable::IsCursorClosedOnCommit() **********/
2157 bool wxDbTable::IsCursorClosedOnCommit(void)
2158 {
2159 if (pDb->dbInf.cursorCommitBehavior == SQL_CB_PRESERVE)
2160 return false;
2161 else
2162 return true;
2163
2164 } // wxDbTable::IsCursorClosedOnCommit()
2165
2166
2167
2168 /********** wxDbTable::ClearMemberVar() **********/
2169 void wxDbTable::ClearMemberVar(UWORD colNo, bool setToNull)
2170 {
2171 wxASSERT(colNo < noCols);
2172
2173 switch(colDefs[colNo].SqlCtype)
2174 {
2175 case SQL_C_CHAR:
2176 ((UCHAR FAR *) colDefs[colNo].PtrDataObj)[0] = 0;
2177 break;
2178 case SQL_C_SSHORT:
2179 *((SWORD *) colDefs[colNo].PtrDataObj) = 0;
2180 break;
2181 case SQL_C_USHORT:
2182 *((UWORD*) colDefs[colNo].PtrDataObj) = 0;
2183 break;
2184 case SQL_C_LONG:
2185 case SQL_C_SLONG:
2186 *((SDWORD *) colDefs[colNo].PtrDataObj) = 0;
2187 break;
2188 case SQL_C_ULONG:
2189 *((UDWORD *) colDefs[colNo].PtrDataObj) = 0;
2190 break;
2191 case SQL_C_FLOAT:
2192 *((SFLOAT *) colDefs[colNo].PtrDataObj) = 0.0f;
2193 break;
2194 case SQL_C_DOUBLE:
2195 *((SDOUBLE *) colDefs[colNo].PtrDataObj) = 0.0f;
2196 break;
2197 case SQL_C_TIMESTAMP:
2198 TIMESTAMP_STRUCT *pDt;
2199 pDt = (TIMESTAMP_STRUCT *) colDefs[colNo].PtrDataObj;
2200 pDt->year = 0;
2201 pDt->month = 0;
2202 pDt->day = 0;
2203 pDt->hour = 0;
2204 pDt->minute = 0;
2205 pDt->second = 0;
2206 pDt->fraction = 0;
2207 break;
2208 }
2209
2210 if (setToNull)
2211 SetColNull(colNo);
2212 } // wxDbTable::ClearMemberVar()
2213
2214
2215 /********** wxDbTable::ClearMemberVars() **********/
2216 void wxDbTable::ClearMemberVars(bool setToNull)
2217 {
2218 int i;
2219
2220 // Loop through the columns setting each member variable to zero
2221 for (i=0; i < noCols; i++)
2222 ClearMemberVar((UWORD)i,setToNull);
2223
2224 } // wxDbTable::ClearMemberVars()
2225
2226
2227 /********** wxDbTable::SetQueryTimeout() **********/
2228 bool wxDbTable::SetQueryTimeout(UDWORD nSeconds)
2229 {
2230 if (SQLSetStmtOption(hstmtInsert, SQL_QUERY_TIMEOUT, nSeconds) != SQL_SUCCESS)
2231 return(pDb->DispAllErrors(henv, hdbc, hstmtInsert));
2232 if (SQLSetStmtOption(hstmtUpdate, SQL_QUERY_TIMEOUT, nSeconds) != SQL_SUCCESS)
2233 return(pDb->DispAllErrors(henv, hdbc, hstmtUpdate));
2234 if (SQLSetStmtOption(hstmtDelete, SQL_QUERY_TIMEOUT, nSeconds) != SQL_SUCCESS)
2235 return(pDb->DispAllErrors(henv, hdbc, hstmtDelete));
2236 if (SQLSetStmtOption(hstmtInternal, SQL_QUERY_TIMEOUT, nSeconds) != SQL_SUCCESS)
2237 return(pDb->DispAllErrors(henv, hdbc, hstmtInternal));
2238
2239 // Completed Successfully
2240 return true;
2241
2242 } // wxDbTable::SetQueryTimeout()
2243
2244
2245 /********** wxDbTable::SetColDefs() **********/
2246 void wxDbTable::SetColDefs(UWORD index, const wxString &fieldName, int dataType, void *pData,
2247 SWORD cType, int size, bool keyField, bool upd,
2248 bool insAllow, bool derivedCol)
2249 {
2250 wxASSERT_MSG( index < noCols,
2251 _T("Specified column index exceeds the maximum number of columns for this table.") );
2252
2253 if (!colDefs) // May happen if the database connection fails
2254 return;
2255
2256 if (fieldName.Length() > (unsigned int) DB_MAX_COLUMN_NAME_LEN)
2257 {
2258 wxStrncpy(colDefs[index].ColName, fieldName, DB_MAX_COLUMN_NAME_LEN);
2259 colDefs[index].ColName[DB_MAX_COLUMN_NAME_LEN] = 0;
2260
2261 #ifdef __WXDEBUG__
2262 wxString tmpMsg;
2263 tmpMsg.Printf(_T("Column name '%s' is too long. Truncated to '%s'."),
2264 fieldName.c_str(),colDefs[index].ColName);
2265 wxFAIL_MSG(tmpMsg);
2266 #endif // __WXDEBUG__
2267 }
2268 else
2269 wxStrcpy(colDefs[index].ColName, fieldName);
2270
2271 colDefs[index].DbDataType = dataType;
2272 colDefs[index].PtrDataObj = pData;
2273 colDefs[index].SqlCtype = cType;
2274 colDefs[index].SzDataObj = size;
2275 colDefs[index].KeyField = keyField;
2276 colDefs[index].DerivedCol = derivedCol;
2277 // Derived columns by definition would NOT be "Insertable" or "Updateable"
2278 if (derivedCol)
2279 {
2280 colDefs[index].Updateable = false;
2281 colDefs[index].InsertAllowed = false;
2282 }
2283 else
2284 {
2285 colDefs[index].Updateable = upd;
2286 colDefs[index].InsertAllowed = insAllow;
2287 }
2288
2289 colDefs[index].Null = false;
2290
2291 } // wxDbTable::SetColDefs()
2292
2293
2294 /********** wxDbTable::SetColDefs() **********/
2295 wxDbColDataPtr* wxDbTable::SetColDefs(wxDbColInf *pColInfs, UWORD numCols)
2296 {
2297 wxASSERT(pColInfs);
2298 wxDbColDataPtr *pColDataPtrs = NULL;
2299
2300 if (pColInfs)
2301 {
2302 UWORD index;
2303
2304 pColDataPtrs = new wxDbColDataPtr[numCols+1];
2305
2306 for (index = 0; index < numCols; index++)
2307 {
2308 // Process the fields
2309 switch (pColInfs[index].dbDataType)
2310 {
2311 case DB_DATA_TYPE_VARCHAR:
2312 pColDataPtrs[index].PtrDataObj = new wxChar[pColInfs[index].bufferLength+1];
2313 pColDataPtrs[index].SzDataObj = pColInfs[index].columnSize;
2314 pColDataPtrs[index].SqlCtype = SQL_C_CHAR;
2315 break;
2316 case DB_DATA_TYPE_INTEGER:
2317 // Can be long or short
2318 if (pColInfs[index].bufferLength == sizeof(long))
2319 {
2320 pColDataPtrs[index].PtrDataObj = new long;
2321 pColDataPtrs[index].SzDataObj = sizeof(long);
2322 pColDataPtrs[index].SqlCtype = SQL_C_SLONG;
2323 }
2324 else
2325 {
2326 pColDataPtrs[index].PtrDataObj = new short;
2327 pColDataPtrs[index].SzDataObj = sizeof(short);
2328 pColDataPtrs[index].SqlCtype = SQL_C_SSHORT;
2329 }
2330 break;
2331 case DB_DATA_TYPE_FLOAT:
2332 // Can be float or double
2333 if (pColInfs[index].bufferLength == sizeof(float))
2334 {
2335 pColDataPtrs[index].PtrDataObj = new float;
2336 pColDataPtrs[index].SzDataObj = sizeof(float);
2337 pColDataPtrs[index].SqlCtype = SQL_C_FLOAT;
2338 }
2339 else
2340 {
2341 pColDataPtrs[index].PtrDataObj = new double;
2342 pColDataPtrs[index].SzDataObj = sizeof(double);
2343 pColDataPtrs[index].SqlCtype = SQL_C_DOUBLE;
2344 }
2345 break;
2346 case DB_DATA_TYPE_DATE:
2347 pColDataPtrs[index].PtrDataObj = new TIMESTAMP_STRUCT;
2348 pColDataPtrs[index].SzDataObj = sizeof(TIMESTAMP_STRUCT);
2349 pColDataPtrs[index].SqlCtype = SQL_C_TIMESTAMP;
2350 break;
2351 case DB_DATA_TYPE_BLOB:
2352 wxFAIL_MSG(wxT("This form of ::SetColDefs() cannot be used with BLOB columns"));
2353 pColDataPtrs[index].PtrDataObj = /*BLOB ADDITION NEEDED*/NULL;
2354 pColDataPtrs[index].SzDataObj = /*BLOB ADDITION NEEDED*/sizeof(void *);
2355 pColDataPtrs[index].SqlCtype = SQL_VARBINARY;
2356 break;
2357 }
2358 if (pColDataPtrs[index].PtrDataObj != NULL)
2359 SetColDefs (index,pColInfs[index].colName,pColInfs[index].dbDataType, pColDataPtrs[index].PtrDataObj, pColDataPtrs[index].SqlCtype, pColDataPtrs[index].SzDataObj);
2360 else
2361 {
2362 // Unable to build all the column definitions, as either one of
2363 // the calls to "new" failed above, or there was a BLOB field
2364 // to have a column definition for. If BLOBs are to be used,
2365 // the other form of ::SetColDefs() must be used, as it is impossible
2366 // to know the maximum size to create the PtrDataObj to be.
2367 delete [] pColDataPtrs;
2368 return NULL;
2369 }
2370 }
2371 }
2372
2373 return (pColDataPtrs);
2374
2375 } // wxDbTable::SetColDefs()
2376
2377
2378 /********** wxDbTable::SetCursor() **********/
2379 void wxDbTable::SetCursor(HSTMT *hstmtActivate)
2380 {
2381 if (hstmtActivate == wxDB_DEFAULT_CURSOR)
2382 hstmt = *hstmtDefault;
2383 else
2384 hstmt = *hstmtActivate;
2385
2386 } // wxDbTable::SetCursor()
2387
2388
2389 /********** wxDbTable::Count(const wxString &) **********/
2390 ULONG wxDbTable::Count(const wxString &args)
2391 {
2392 ULONG count;
2393 wxString sqlStmt;
2394 SDWORD cb;
2395
2396 // Build a "SELECT COUNT(*) FROM queryTableName [WHERE whereClause]" SQL Statement
2397 sqlStmt = wxT("SELECT COUNT(");
2398 sqlStmt += args;
2399 sqlStmt += wxT(") FROM ");
2400 sqlStmt += pDb->SQLTableName(queryTableName);
2401 // sqlStmt += queryTableName;
2402 #if wxODBC_BACKWARD_COMPATABILITY
2403 if (from && wxStrlen(from))
2404 #else
2405 if (from.Length())
2406 #endif
2407 sqlStmt += from;
2408
2409 // Add the where clause if one is provided
2410 #if wxODBC_BACKWARD_COMPATABILITY
2411 if (where && wxStrlen(where))
2412 #else
2413 if (where.Length())
2414 #endif
2415 {
2416 sqlStmt += wxT(" WHERE ");
2417 sqlStmt += where;
2418 }
2419
2420 pDb->WriteSqlLog(sqlStmt);
2421
2422 // Initialize the Count cursor if it's not already initialized
2423 if (!hstmtCount)
2424 {
2425 hstmtCount = GetNewCursor(false,false);
2426 wxASSERT(hstmtCount);
2427 if (!hstmtCount)
2428 return(0);
2429 }
2430
2431 // Execute the SQL statement
2432 if (SQLExecDirect(*hstmtCount, (SQLTCHAR FAR *) sqlStmt.c_str(), SQL_NTS) != SQL_SUCCESS)
2433 {
2434 pDb->DispAllErrors(henv, hdbc, *hstmtCount);
2435 return(0);
2436 }
2437
2438 // Fetch the record
2439 if (SQLFetch(*hstmtCount) != SQL_SUCCESS)
2440 {
2441 pDb->DispAllErrors(henv, hdbc, *hstmtCount);
2442 return(0);
2443 }
2444
2445 // Obtain the result
2446 if (SQLGetData(*hstmtCount, (UWORD)1, SQL_C_ULONG, &count, sizeof(count), &cb) != SQL_SUCCESS)
2447 {
2448 pDb->DispAllErrors(henv, hdbc, *hstmtCount);
2449 return(0);
2450 }
2451
2452 // Free the cursor
2453 if (SQLFreeStmt(*hstmtCount, SQL_CLOSE) != SQL_SUCCESS)
2454 pDb->DispAllErrors(henv, hdbc, *hstmtCount);
2455
2456 // Return the record count
2457 return(count);
2458
2459 } // wxDbTable::Count()
2460
2461
2462 /********** wxDbTable::Refresh() **********/
2463 bool wxDbTable::Refresh(void)
2464 {
2465 bool result = true;
2466
2467 // Switch to the internal cursor so any active cursors are not corrupted
2468 HSTMT currCursor = GetCursor();
2469 hstmt = hstmtInternal;
2470 #if wxODBC_BACKWARD_COMPATABILITY
2471 // Save the where and order by clauses
2472 wxChar *saveWhere = where;
2473 wxChar *saveOrderBy = orderBy;
2474 #else
2475 wxString saveWhere = where;
2476 wxString saveOrderBy = orderBy;
2477 #endif
2478 // Build a where clause to refetch the record with. Try and use the
2479 // ROWID if it's available, ow use the key fields.
2480 wxString whereClause;
2481 whereClause.Empty();
2482
2483 if (CanUpdByROWID())
2484 {
2485 SDWORD cb;
2486 wxChar rowid[wxDB_ROWID_LEN+1];
2487
2488 // Get the ROWID value. If not successful retreiving the ROWID,
2489 // simply fall down through the code and build the WHERE clause
2490 // based on the key fields.
2491 if (SQLGetData(hstmt, (UWORD)(noCols+1), SQL_C_CHAR, (UCHAR*) rowid, wxDB_ROWID_LEN, &cb) == SQL_SUCCESS)
2492 {
2493 whereClause += pDb->SQLTableName(queryTableName);
2494 // whereClause += queryTableName;
2495 whereClause += wxT(".ROWID = '");
2496 whereClause += rowid;
2497 whereClause += wxT("'");
2498 }
2499 }
2500
2501 // If unable to use the ROWID, build a where clause from the keyfields
2502 if (wxStrlen(whereClause) == 0)
2503 BuildWhereClause(whereClause, DB_WHERE_KEYFIELDS, queryTableName);
2504
2505 // Requery the record
2506 where = whereClause;
2507 orderBy.Empty();
2508 if (!Query())
2509 result = false;
2510
2511 if (result && !GetNext())
2512 result = false;
2513
2514 // Switch back to original cursor
2515 SetCursor(&currCursor);
2516
2517 // Free the internal cursor
2518 if (SQLFreeStmt(hstmtInternal, SQL_CLOSE) != SQL_SUCCESS)
2519 pDb->DispAllErrors(henv, hdbc, hstmtInternal);
2520
2521 // Restore the original where and order by clauses
2522 where = saveWhere;
2523 orderBy = saveOrderBy;
2524
2525 return(result);
2526
2527 } // wxDbTable::Refresh()
2528
2529
2530 /********** wxDbTable::SetColNull() **********/
2531 bool wxDbTable::SetColNull(UWORD colNo, bool set)
2532 {
2533 if (colNo < noCols)
2534 {
2535 colDefs[colNo].Null = set;
2536 if (set) // Blank out the values in the member variable
2537 ClearMemberVar(colNo, false); // Must call with false here, or infinite recursion will happen
2538
2539 setCbValueForColumn(colNo);
2540
2541 return true;
2542 }
2543 else
2544 return false;
2545
2546 } // wxDbTable::SetColNull()
2547
2548
2549 /********** wxDbTable::SetColNull() **********/
2550 bool wxDbTable::SetColNull(const wxString &colName, bool set)
2551 {
2552 int colNo;
2553 for (colNo = 0; colNo < noCols; colNo++)
2554 {
2555 if (!wxStricmp(colName, colDefs[colNo].ColName))
2556 break;
2557 }
2558
2559 if (colNo < noCols)
2560 {
2561 colDefs[colNo].Null = set;
2562 if (set) // Blank out the values in the member variable
2563 ClearMemberVar((UWORD)colNo,false); // Must call with false here, or infinite recursion will happen
2564
2565 setCbValueForColumn(colNo);
2566
2567 return true;
2568 }
2569 else
2570 return false;
2571
2572 } // wxDbTable::SetColNull()
2573
2574
2575 /********** wxDbTable::GetNewCursor() **********/
2576 HSTMT *wxDbTable::GetNewCursor(bool setCursor, bool bindColumns)
2577 {
2578 HSTMT *newHSTMT = new HSTMT;
2579 wxASSERT(newHSTMT);
2580 if (!newHSTMT)
2581 return(0);
2582
2583 if (SQLAllocStmt(hdbc, newHSTMT) != SQL_SUCCESS)
2584 {
2585 pDb->DispAllErrors(henv, hdbc);
2586 delete newHSTMT;
2587 return(0);
2588 }
2589
2590 if (SQLSetStmtOption(*newHSTMT, SQL_CURSOR_TYPE, cursorType) != SQL_SUCCESS)
2591 {
2592 pDb->DispAllErrors(henv, hdbc, *newHSTMT);
2593 delete newHSTMT;
2594 return(0);
2595 }
2596
2597 if (bindColumns)
2598 {
2599 if (!bindCols(*newHSTMT))
2600 {
2601 delete newHSTMT;
2602 return(0);
2603 }
2604 }
2605
2606 if (setCursor)
2607 SetCursor(newHSTMT);
2608
2609 return(newHSTMT);
2610
2611 } // wxDbTable::GetNewCursor()
2612
2613
2614 /********** wxDbTable::DeleteCursor() **********/
2615 bool wxDbTable::DeleteCursor(HSTMT *hstmtDel)
2616 {
2617 bool result = true;
2618
2619 if (!hstmtDel) // Cursor already deleted
2620 return(result);
2621
2622 /*
2623 ODBC 3.0 says to use this form
2624 if (SQLFreeHandle(*hstmtDel, SQL_DROP) != SQL_SUCCESS)
2625
2626 */
2627 if (SQLFreeStmt(*hstmtDel, SQL_DROP) != SQL_SUCCESS)
2628 {
2629 pDb->DispAllErrors(henv, hdbc);
2630 result = false;
2631 }
2632
2633 delete hstmtDel;
2634
2635 return(result);
2636
2637 } // wxDbTable::DeleteCursor()
2638
2639 //////////////////////////////////////////////////////////////
2640 // wxDbGrid support functions
2641 //////////////////////////////////////////////////////////////
2642
2643 void wxDbTable::SetRowMode(const rowmode_t rowmode)
2644 {
2645 if (!m_hstmtGridQuery)
2646 {
2647 m_hstmtGridQuery = GetNewCursor(false,false);
2648 if (!bindCols(*m_hstmtGridQuery))
2649 return;
2650 }
2651
2652 m_rowmode = rowmode;
2653 switch (m_rowmode)
2654 {
2655 case WX_ROW_MODE_QUERY:
2656 SetCursor(m_hstmtGridQuery);
2657 break;
2658 case WX_ROW_MODE_INDIVIDUAL:
2659 SetCursor(hstmtDefault);
2660 break;
2661 default:
2662 wxASSERT(0);
2663 }
2664 } // wxDbTable::SetRowMode()
2665
2666
2667 wxVariant wxDbTable::GetCol(const int colNo) const
2668 {
2669 wxVariant val;
2670 if ((colNo < noCols) && (!IsColNull((UWORD)colNo)))
2671 {
2672 switch (colDefs[colNo].SqlCtype)
2673 {
2674 case SQL_CHAR:
2675 case SQL_VARCHAR:
2676 val = (wxChar *)(colDefs[colNo].PtrDataObj);
2677 break;
2678 case SQL_C_LONG:
2679 case SQL_C_SLONG:
2680 val = *(long *)(colDefs[colNo].PtrDataObj);
2681 break;
2682 case SQL_C_SHORT:
2683 case SQL_C_SSHORT:
2684 val = (long int )(*(short *)(colDefs[colNo].PtrDataObj));
2685 break;
2686 case SQL_C_ULONG:
2687 val = (long)(*(unsigned long *)(colDefs[colNo].PtrDataObj));
2688 break;
2689 case SQL_C_TINYINT:
2690 val = (long)(*(wxChar *)(colDefs[colNo].PtrDataObj));
2691 break;
2692 case SQL_C_UTINYINT:
2693 val = (long)(*(wxChar *)(colDefs[colNo].PtrDataObj));
2694 break;
2695 case SQL_C_USHORT:
2696 val = (long)(*(UWORD *)(colDefs[colNo].PtrDataObj));
2697 break;
2698 case SQL_C_DATE:
2699 val = (DATE_STRUCT *)(colDefs[colNo].PtrDataObj);
2700 break;
2701 case SQL_C_TIME:
2702 val = (TIME_STRUCT *)(colDefs[colNo].PtrDataObj);
2703 break;
2704 case SQL_C_TIMESTAMP:
2705 val = (TIMESTAMP_STRUCT *)(colDefs[colNo].PtrDataObj);
2706 break;
2707 case SQL_C_DOUBLE:
2708 val = *(double *)(colDefs[colNo].PtrDataObj);
2709 break;
2710 default:
2711 assert(0);
2712 }
2713 }
2714 return val;
2715 } // wxDbTable::GetCol()
2716
2717
2718 void wxDbTable::SetCol(const int colNo, const wxVariant val)
2719 {
2720 //FIXME: Add proper wxDateTime support to wxVariant..
2721 wxDateTime dateval;
2722
2723 SetColNull((UWORD)colNo, val.IsNull());
2724
2725 if (!val.IsNull())
2726 {
2727 if ((colDefs[colNo].SqlCtype == SQL_C_DATE)
2728 || (colDefs[colNo].SqlCtype == SQL_C_TIME)
2729 || (colDefs[colNo].SqlCtype == SQL_C_TIMESTAMP))
2730 {
2731 //Returns null if invalid!
2732 if (!dateval.ParseDate(val.GetString()))
2733 SetColNull((UWORD)colNo, true);
2734 }
2735
2736 switch (colDefs[colNo].SqlCtype)
2737 {
2738 case SQL_CHAR:
2739 case SQL_VARCHAR:
2740 csstrncpyt((wxChar *)(colDefs[colNo].PtrDataObj),
2741 val.GetString().c_str(),
2742 colDefs[colNo].SzDataObj-1);
2743 break;
2744 case SQL_C_LONG:
2745 case SQL_C_SLONG:
2746 *(long *)(colDefs[colNo].PtrDataObj) = val;
2747 break;
2748 case SQL_C_SHORT:
2749 case SQL_C_SSHORT:
2750 *(short *)(colDefs[colNo].PtrDataObj) = (short)val.GetLong();
2751 break;
2752 case SQL_C_ULONG:
2753 *(unsigned long *)(colDefs[colNo].PtrDataObj) = val.GetLong();
2754 break;
2755 case SQL_C_TINYINT:
2756 *(wxChar *)(colDefs[colNo].PtrDataObj) = val.GetChar();
2757 break;
2758 case SQL_C_UTINYINT:
2759 *(wxChar *)(colDefs[colNo].PtrDataObj) = val.GetChar();
2760 break;
2761 case SQL_C_USHORT:
2762 *(unsigned short *)(colDefs[colNo].PtrDataObj) = (unsigned short)val.GetLong();
2763 break;
2764 //FIXME: Add proper wxDateTime support to wxVariant..
2765 case SQL_C_DATE:
2766 {
2767 DATE_STRUCT *dataptr =
2768 (DATE_STRUCT *)colDefs[colNo].PtrDataObj;
2769
2770 dataptr->year = (SWORD)dateval.GetYear();
2771 dataptr->month = (UWORD)(dateval.GetMonth()+1);
2772 dataptr->day = (UWORD)dateval.GetDay();
2773 }
2774 break;
2775 case SQL_C_TIME:
2776 {
2777 TIME_STRUCT *dataptr =
2778 (TIME_STRUCT *)colDefs[colNo].PtrDataObj;
2779
2780 dataptr->hour = dateval.GetHour();
2781 dataptr->minute = dateval.GetMinute();
2782 dataptr->second = dateval.GetSecond();
2783 }
2784 break;
2785 case SQL_C_TIMESTAMP:
2786 {
2787 TIMESTAMP_STRUCT *dataptr =
2788 (TIMESTAMP_STRUCT *)colDefs[colNo].PtrDataObj;
2789 dataptr->year = (SWORD)dateval.GetYear();
2790 dataptr->month = (UWORD)(dateval.GetMonth()+1);
2791 dataptr->day = (UWORD)dateval.GetDay();
2792
2793 dataptr->hour = dateval.GetHour();
2794 dataptr->minute = dateval.GetMinute();
2795 dataptr->second = dateval.GetSecond();
2796 }
2797 break;
2798 case SQL_C_DOUBLE:
2799 *(double *)(colDefs[colNo].PtrDataObj) = val;
2800 break;
2801 default:
2802 assert(0);
2803 } // switch
2804 } // if (!val.IsNull())
2805 } // wxDbTable::SetCol()
2806
2807
2808 GenericKey wxDbTable::GetKey()
2809 {
2810 void *blk;
2811 wxChar *blkptr;
2812
2813 blk = malloc(m_keysize);
2814 blkptr = (wxChar *) blk;
2815
2816 int i;
2817 for (i=0; i < noCols; i++)
2818 {
2819 if (colDefs[i].KeyField)
2820 {
2821 memcpy(blkptr,colDefs[i].PtrDataObj, colDefs[i].SzDataObj);
2822 blkptr += colDefs[i].SzDataObj;
2823 }
2824 }
2825
2826 GenericKey k = GenericKey(blk, m_keysize);
2827 free(blk);
2828
2829 return k;
2830 } // wxDbTable::GetKey()
2831
2832
2833 void wxDbTable::SetKey(const GenericKey& k)
2834 {
2835 void *blk;
2836 wxChar *blkptr;
2837
2838 blk = k.GetBlk();
2839 blkptr = (wxChar *)blk;
2840
2841 int i;
2842 for (i=0; i < noCols; i++)
2843 {
2844 if (colDefs[i].KeyField)
2845 {
2846 SetColNull((UWORD)i, false);
2847 memcpy(colDefs[i].PtrDataObj, blkptr, colDefs[i].SzDataObj);
2848 blkptr += colDefs[i].SzDataObj;
2849 }
2850 }
2851 } // wxDbTable::SetKey()
2852
2853
2854 #endif // wxUSE_ODBC
2855