Modified:
MYODBCC/include/MYODBCC.h
MYODBCShell/MYODBCShell.cpp
MYODBCShell/MYODBCShell.h
MYODBCShell/main.cpp
Log:
Modified: MYODBCC/include/MYODBCC.h
===================================================================
--- MYODBCC/include/MYODBCC.h 2006-04-18 17:39:51 UTC (rev 83)
+++ MYODBCC/include/MYODBCC.h 2006-04-18 17:43:35 UTC (rev 84)
@@ -80,6 +80,14 @@
#include <QString>
+#ifndef max
+#define max( a, b ) (((a) > (b)) ? (a) : (b))
+#endif
+
+#ifndef min
+#define min( a, b ) (((a) < (b)) ? (a) : (b))
+#endif
+
/*!
\brief C API portability and convenience.
Modified: MYODBCShell/MYODBCShell.cpp
===================================================================
--- MYODBCShell/MYODBCShell.cpp 2006-04-18 17:39:51 UTC (rev 83)
+++ MYODBCShell/MYODBCShell.cpp 2006-04-18 17:43:35 UTC (rev 84)
@@ -1,221 +1,4 @@
-/*!
- \file A mysql-like console interface which uses ODBC exclusively.
-*/
-#include "MYODBCShell.h"
-#ifdef HAVE_READLINE
- #include <readline/readline.h>
- #include <readline/history.h>
-#endif
-
-#ifdef HAVE_SETLOCALE
- #ifdef HAVE_LOCALE_H
- #include <locale.h>
- #endif
-#endif
-
-int bDriverConnect = 0;
-int bVerbose = 0;
-int nUserWidth = 0;
-SQLHENV hEnv = 0;
-SQLHDBC hDbc = 0;
-
-int main( int argc, char *argv[] )
-{
- int nArg, count;
- int bHTMLTable = 0;
- int bBatch = 0;
- int cDelimiter = 0;
- int bColumnNames = 0;
- TCHAR szDSN[MAX_DATA_WIDTH+1] = L"";
- TCHAR szUID[MAX_DATA_WIDTH+1] = L"";
- TCHAR szPWD[MAX_DATA_WIDTH+1] = L"";
- TCHAR * szSQL;
- TCHAR * pEscapeChar;
- int buffer_size = 9000;
-
- if ( argc < 2 )
- {
- _ftprintf( stderr, szSyntax );
- exit( 1 );
- }
-
-#ifdef HAVE_SETLOCALE
- /*
- * Default locale
- */
- setlocale( LC_ALL, "" );
-#endif
-
- /****************************
- * PARSE ARGS
- ***************************/
- for ( nArg = 1, count = 1 ; nArg < argc; nArg++ )
- {
- if ( argv[nArg][0] == '-' )
- {
- /* Options */
- switch ( argv[nArg][1] )
- {
- case 'd':
- cDelimiter = argv[nArg][2];
- break;
- case 'm':
- nUserWidth = atoi( &(argv[nArg][2]) );
- break;
- case 'r':
- bDriverConnect = 1;
- break;
- case 's':
- buffer_size = atoi( &(argv[nArg][2]) );
- break;
- case 'w':
- bHTMLTable = 1;
- break;
- case 'b':
- bBatch = 1;
- break;
- case 'c':
- bColumnNames = 1;
- break;
- case 'v':
- bVerbose = 1;
- break;
- case '-':
- printf( "MYODBCShell\n" );
- exit(0);
-#ifdef HAVE_STRTOL
- case 'x':
- cDelimiter = strtol( argv[nArg]+2, NULL, 0 );
- break;
-#endif
-#ifdef HAVE_SETLOCALE
- case 'l':
- if ( !setlocale( LC_ALL, argv[nArg]+2 ))
- {
- _ftprintf( stderr, "MYODBCShell: can't set locale to '%s'\n",
argv[nArg]+2 );
- exit ( -1 );
- }
- break;
-#endif
-
- default:
- _ftprintf( stderr, szSyntax );
- exit( 1 );
- }
- continue;
- }
- else if ( count == 1 )
- strncpy( szDSN, argv[nArg], MAX_DATA_WIDTH );
- else if ( count == 2 )
- strncpy( szUID, argv[nArg], MAX_DATA_WIDTH );
- else if ( count == 3 )
- strncpy( szPWD, argv[nArg], MAX_DATA_WIDTH );
- count++;
- }
-
- szSQL = calloc( 1, buffer_size + 1 );
-
- /****************************
- * CONNECT
- ***************************/
-
- if ( !OpenDatabase( &hEnv, &hDbc, szDSN, szUID, szPWD ) )
- exit( 1 );
-
- /****************************
- * EXECUTE
- ***************************/
- if ( !bBatch )
- {
- printf( "+---------------------------------------+\n" );
- printf( "| Connected! |\n" );
- printf( "| |\n" );
- printf( "| sql-statement |\n" );
- printf( "| help [tablename] |\n" );
- printf( "| quit |\n" );
- printf( "| |\n" );
- printf( "+---------------------------------------+\n" );
- }
- do
- {
- if ( !bBatch )
-#ifndef HAVE_READLINE
- printf( "SQL> " );
-#else
- {
- char *line;
- int malloced;
-
- line=readline("SQL> ");
- if ( !line ) /* EOF - ctrl D */
- {
- malloced = 1;
- line = strdup( "quit" );
- }
- else
- {
- malloced = 0;
- }
- strncpy(szSQL, line, buffer_size );
- add_history(line);
- if ( malloced )
- {
- free(line);
- }
- }
- else
-#endif
- {
- char *line;
- int malloced;
-
- line = fgets( szSQL, buffer_size, stdin );
- if ( !line ) /* EOF - ctrl D */
- {
- malloced = 1;
- line = (char*)strdup( "quit" );
- }
- else
- {
- malloced = 0;
- }
- strncpy(szSQL, line, buffer_size );
- if ( malloced )
- {
- free(line);
- }
- }
-
- /* strip away escape chars */
- while ( (pEscapeChar=(char*)strchr(szSQL, '\n')) != NULL ||
(pEscapeChar=(char*)strchr(szSQL, '\r')) != NULL )
- *pEscapeChar = ' ';
-
- if ( szSQL[1] != '\0' )
- {
- /* intercept some keywords or simply send to data source */
- if ( strncmp( szSQL, "quit", 4 ) == 0 )
- szSQL[1] = '\0';
- else if ( strncmp( szSQL, "help", 4 ) == 0 )
- ExecuteHelp( hDbc, szSQL, cDelimiter, bColumnNames, bHTMLTable );
- else if ( strncmp( szSQL, "SQLTables", 9 ) == 0 )
- doTables( hDbc, szSQL, cDelimiter, bColumnNames, bHTMLTable );
- else if ( strncmp( szSQL, "SQLColumns", 10 ) == 0 )
- doColumns( hDbc, szSQL, cDelimiter, bColumnNames, bHTMLTable );
- else if (memcmp(szSQL, "--", 2) != 0)
- ExecuteSQL( hDbc, szSQL, cDelimiter, bColumnNames, bHTMLTable );
- }
-
- } while ( szSQL[1] != '\0' );
-
- /****************************
- * DISCONNECT
- ***************************/
- CloseDatabase( hEnv, hDbc );
-
- exit( 0 );
-}
-
int getElement( char *pszData, char cSeperator, char cTerminator, int nElement, char
*pszElement, int nMaxElement )
{
int nCurElement = 0;
Modified: MYODBCShell/MYODBCShell.h
===================================================================
--- MYODBCShell/MYODBCShell.h 2006-04-18 17:39:51 UTC (rev 83)
+++ MYODBCShell/MYODBCShell.h 2006-04-18 17:43:35 UTC (rev 84)
@@ -40,6 +40,8 @@
#include <MYODBCC.h>
+#define MAX_DATA_WIDTH 300
+
class MYODBCShell : public QObject
{
public:
@@ -71,134 +73,7 @@
// int DumpODBCLog( SQLHENV hEnv, SQLHDBC hDbc, SQLHSTMT hStmt );
};
-++++++++++++++++
-
-char *szSyntax =
-"\n" \
-"**********************************************\n" \
-"* MySQL - MYODBCShell *\n" \
-"**********************************************\n" \
-"* Syntax *\n" \
-"* *\n" \
-"* MYODBCShell DSN | STR [UID [PWD]] [opts] *\n" \
-"* *\n" \
-"* Options *\n" \
-"* *\n" \
-"* -r Use SQLDriverConnect to connect *\n" \
-"* and use STR as connection string*\n" \
-"* -b batch.(no prompting etc) *\n" \
-"* -dx delimit columns with x *\n" \
-"* -x0xXX delimit columns with XX, where *\n" \
-"* x is in hex, ie 0x09 is tab *\n" \
-"* -w wrap results in an HTML table *\n" \
-"* -c column names on first row. *\n" \
-"* (only used when -d) *\n" \
-"* -mn limit column display width to n *\n" \
-"* -v verbose. *\n" \
-"* -lx set locale to x *\n" \
-"* --version version *\n" \
-"* *\n" \
-"* Notes *\n" \
-"* *\n" \
-"* MYODBCShell supports redirection and *\n" \
-"* piping for batch processing. *\n" \
-"* *\n" \
-"* Examples *\n" \
-"* *\n" \
-"* cat My.sql | MYODBCShell db id pwd -w *\n" \
-"* *\n" \
-"* MYODBCShell *\n" \
-"* 'DRIVER=myodbc3;DSN=test;UID=uid;PWD=pwd'*\n" \
-"* -r *\n" \
-"* *\n" \
-"* Each line in My.sql must contain *\n" \
-"* exactly 1 SQL command except for the *\n" \
-"* last line which must be blank. *\n" \
-"* *\n" \
-"* Please visit; *\n" \
-"* *\n" \
-"* http://www.mysql.com *\n" \
-"* pharvey@stripped *\n" \
-"**********************************************\n\n";
-
-#else
-
-char *szSyntax =
-"\n" \
-"**********************************************\n" \
-"* MySQL - MYODBCShell *\n" \
-"**********************************************\n" \
-"* Syntax *\n" \
-"* *\n" \
-"* MYODBCShell DSN | STR [UID [PWD]] [opts] *\n" \
-"* *\n" \
-"* Options *\n" \
-"* *\n" \
-"* -r Use SQLDriverConnect to connect *\n" \
-"* and use STR as connection string*\n" \
-"* -b batch.(no prompting etc) *\n" \
-"* -dx delimit columns with x *\n" \
-"* -x0xXX delimit columns with XX, where *\n" \
-"* x is in hex, ie 0x09 is tab *\n" \
-"* -w wrap results in an HTML table *\n" \
-"* -c column names on first row. *\n" \
-"* (only used when -d) *\n" \
-"* -mn limit column display width to n *\n" \
-"* -v verbose. *\n" \
-"* --version version *\n" \
-"* *\n" \
-"* Notes *\n" \
-"* *\n" \
-"* MYODBCShell supports redirection and *\n" \
-"* piping for batch processing. *\n" \
-"* *\n" \
-"* Examples *\n" \
-"* *\n" \
-"* cat My.sql | MYODBCShell db id pwd -w *\n" \
-"* *\n" \
-"* MYODBCShell *\n" \
-"* 'DRIVER=myodbc3;DSN=test;UID=uid;PWD=pwd'*\n" \
-"* -r *\n" \
-"* *\n" \
-"* Each line in My.sql must contain *\n" \
-"* exactly 1 SQL command except for the *\n" \
-"* last line which must be blank. *\n" \
-"* *\n" \
-"* Please visit; *\n" \
-"* *\n" \
-"* http://www.mysql.com *\n" \
-"* pharvey@stripped *\n" \
-"**********************************************\n\n";
-
#endif
-#define MAX_DATA_WIDTH 300
-#ifndef max
-#define max( a, b ) (((a) > (b)) ? (a) : (b))
-#endif
-#ifndef min
-#define min( a, b ) (((a) < (b)) ? (a) : (b))
-#endif
-
-int OpenDatabase( SQLHENV *phEnv, SQLHDBC *phDbc, char *szDSN, char *szUID, char *szPWD
);
-int ExecuteSQL( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable );
-int ExecuteHelp( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable );
-int CloseDatabase( SQLHENV hEnv, SQLHDBC hDbc );
-
-int doTables( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable );
-int doColumns( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable );
-
-void WriteHeaderHTMLTable( SQLHSTMT hStmt );
-void WriteHeaderNormal( SQLHSTMT hStmt, SQLCHAR *szSepLine );
-void WriteHeaderDelimited( SQLHSTMT hStmt, char cDelimiter );
-void WriteBodyHTMLTable( SQLHSTMT hStmt );
-SQLINTEGER WriteBodyNormal( SQLHSTMT hStmt );
-void WriteBodyDelimited( SQLHSTMT hStmt, char cDelimiter );
-void WriteFooterHTMLTable( SQLHSTMT hStmt );
-void WriteFooterNormal( SQLHSTMT hStmt, SQLCHAR *szSepLine, SQLINTEGER nRows );
-
-int DumpODBCLog( SQLHENV hEnv, SQLHDBC hDbc, SQLHSTMT hStmt );
-
-
Modified: MYODBCShell/main.cpp
===================================================================
--- MYODBCShell/main.cpp 2006-04-18 17:39:51 UTC (rev 83)
+++ MYODBCShell/main.cpp 2006-04-18 17:43:35 UTC (rev 84)
@@ -278,884 +278,4 @@
exit( 0 );
}
-int getElement( char *pszData, char cSeperator, char cTerminator, int nElement, char
*pszElement, int nMaxElement )
-{
- int nCurElement = 0;
- int nChar = 0;
- int nCharInElement = 0;
- memset( pszElement, '\0', nMaxElement );
- for ( ; nCurElement <= nElement && (nCharInElement+1) < nMaxElement;
nChar++ )
- {
- /* check for end of data */
- if ( cSeperator != cTerminator && pszData[nChar] == cTerminator )
- {
- break;
- }
-
- if ( cSeperator == cTerminator && pszData[nChar] == cSeperator &&
pszData[nChar+1] == cTerminator )
- {
- break;
- }
-
- /* check for end of element */
- if ( pszData[nChar] == cSeperator )
- {
- nCurElement++;
- }
- else if ( nCurElement == nElement )
- {
- pszElement[nCharInElement] = pszData[nChar];
- nCharInElement++;
- }
- }
-
- if ( pszElement[0] == '\0' )
- return 0;
-
- return 1;
-}
-
-/****************************
- * OptimalDisplayWidth
- ***************************/
-SQLUINTEGER OptimalDisplayWidth( SQLHSTMT hStmt, SQLINTEGER nCol, int nUserWidth )
-{
- SQLUINTEGER nLabelWidth = 10;
- SQLUINTEGER nDataWidth = 10;
- SQLUINTEGER nOptimalDisplayWidth = 10;
- SQLTCHAR szColumnName[MAX_DATA_WIDTH+1] = "";
-
- SQLColAttribute( hStmt, nCol, SQL_DESC_DISPLAY_SIZE, NULL, 0, NULL, &nDataWidth
);
- SQLColAttribute( hStmt, nCol, SQL_DESC_LABEL, szColumnName, sizeof(szColumnName),
NULL, NULL );
- nLabelWidth = strlen((char*) szColumnName );
-
- nOptimalDisplayWidth = max( nLabelWidth, nDataWidth );
-
- if ( nUserWidth > 0 )
- nOptimalDisplayWidth = min( nOptimalDisplayWidth, (SQLUINTEGER)nUserWidth );
-
- if ( nOptimalDisplayWidth > MAX_DATA_WIDTH )
- nOptimalDisplayWidth = MAX_DATA_WIDTH;
-
- return nOptimalDisplayWidth;
-}
-
-/****************************
- * OpenDatabase - do everything we have to do to get a viable connection to szDSN
- ***************************/
-int OpenDatabase( SQLHENV *phEnv, SQLHDBC *phDbc, char *szDSN, char *szUID, char *szPWD )
-{
- SQLRETURN nReturn;
-
- if ( SQLAllocHandle( SQL_HANDLE_ENV, NULL, phEnv ) != SQL_SUCCESS )
- {
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLAllocEnv\n" );
- return 0;
- }
-
- if ( SQLSetEnvAttr( *phEnv, SQL_ATTR_ODBC_VERSION, (SQLPOINTER)SQL_OV_ODBC3, 0 ) !=
SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, 0, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLSetEnvAttr to set version to
SQL_OV_ODBC3\n" );
- }
-
- if ( SQLAllocHandle( SQL_HANDLE_DBC, *phEnv, phDbc ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, 0, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLAllocConnect\n" );
- SQLFreeEnv( *phEnv );
- return 0;
- }
- if ( bDriverConnect )
- {
- SQLTCHAR szOut[4096];
- SQLSMALLINT nLen;
- nReturn = SQLDriverConnect( *phDbc, 0, (SQLTCHAR*)szDSN, strlen(szDSN), szOut,
4096, &nLen, SQL_DRIVER_NOPROMPT );
- }
- else
- nReturn = SQLConnect( *phDbc, (SQLTCHAR*)szDSN, SQL_NTS, (SQLTCHAR*)szUID,
SQL_NTS, (SQLTCHAR*)szPWD, SQL_NTS );
-
- if ( !SQL_SUCCEEDED( nReturn ) )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLConnect\n" );
- SQLFreeConnect( *phDbc );
- SQLFreeEnv( *phEnv );
- return 0;
- }
-
- return 1;
-}
-
-/****************************
- * ExecuteSQL - create a statement, execute the SQL, and get rid of the statement
- * - show results as per request; bHTMLTable has precedence over other options
- ***************************/
-int ExecuteSQL( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable )
-{
- SQLHSTMT hStmt;
- SQLTCHAR szSepLine[32001] ;
- SQLSMALLINT cols;
- SQLINTEGER nRows = 0;
- SQLINTEGER ret;
-
- /****************************
- * EXECUTE SQL
- ***************************/
- if ( SQLAllocStmt( hDbc, &hStmt ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLAllocStmt\n" );
- return 0;
- }
-
- if ( SQLPrepare( hStmt, (SQLTCHAR*)szSQL, SQL_NTS ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLPrepare\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
-
- ret = SQLExecute( hStmt );
-
- if ( ret == SQL_NO_DATA )
- {
- _ftprintf( stderr, "[MYODBCShell]INFO: SQLExecute returned SQL_NO_DATA\n" );
- }
- else if ( ret == SQL_SUCCESS_WITH_INFO )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]INFO: SQLExecute returned
SQL_SUCCESS_WITH_INFO\n" );
- }
- else if ( ret != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLExecute\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
-
- /*
- * Loop while SQLMoreResults returns success
- */
-
- do
- {
- strcpy ( szSepLine, "" ) ;
-
- /*
- * check to see if it has generated a result set
- */
-
- if ( SQLNumResultCols( hStmt, &cols ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLNunResultCols\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
-
- if ( cols > 0 )
- {
- /****************************
- * WRITE HEADER
- ***************************/
- if ( bHTMLTable )
- WriteHeaderHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteHeaderNormal( hStmt, szSepLine );
- else if ( cDelimiter && bColumnNames )
- WriteHeaderDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE BODY
- ***************************/
- if ( bHTMLTable )
- WriteBodyHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- nRows = WriteBodyNormal( hStmt );
- else
- WriteBodyDelimited( hStmt, cDelimiter );
- }
-
- /****************************
- * WRITE FOOTER
- ***************************/
- if ( bHTMLTable )
- WriteFooterHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteFooterNormal( hStmt, szSepLine, nRows );
- }
- while ( SQL_SUCCEEDED( SQLMoreResults( hStmt )));
-
- /****************************
- * CLEANUP
- ***************************/
- SQLFreeStmt( hStmt, SQL_DROP );
-
- return 1;
-}
-
-/****************************
- * ExecuteHelp - create a statement, execute the SQL, and get rid of the statement
- * - show results as per request; bHTMLTable has precedence over other
options
- ***************************/
-int ExecuteHelp( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable )
-{
- char szTable[250] = "";
- SQLHSTMT hStmt;
- SQLTCHAR szSepLine[32001] = "";
- SQLINTEGER nRows = 0;
- SQLRETURN nReturn;
-
- /****************************
- * EXECUTE SQL
- ***************************/
- if ( SQLAllocStmt( hDbc, &hStmt ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLAllocStmt\n" );
- return 0;
- }
-
- if ( getElement( szSQL, ' ', '\0', 1, szTable, sizeof(szTable) ) )
- {
- /* COLUMNS */
- nReturn = SQLColumns( hStmt, NULL, 0, NULL, 0, (SQLTCHAR*)szTable, SQL_NTS, NULL,
0 );
- if ( (nReturn != SQL_SUCCESS) && (nReturn != SQL_SUCCESS_WITH_INFO) )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLColumns\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
- }
- else
- {
- /* TABLES */
- nReturn = SQLTables( hStmt, NULL, 0, NULL, 0, NULL, 0, NULL, 0 );
- if ( (nReturn != SQL_SUCCESS) && (nReturn != SQL_SUCCESS_WITH_INFO) )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLTables\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
- }
-
- /****************************
- * WRITE HEADER
- ***************************/
- if ( bHTMLTable )
- WriteHeaderHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteHeaderNormal( hStmt, szSepLine );
- else if ( cDelimiter && bColumnNames )
- WriteHeaderDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE BODY
- ***************************/
- if ( bHTMLTable )
- WriteBodyHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- nRows = WriteBodyNormal( hStmt );
- else
- WriteBodyDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE FOOTER
- ***************************/
- if ( bHTMLTable )
- WriteFooterHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteFooterNormal( hStmt, szSepLine, nRows );
-
- /****************************
- * CLEANUP
- ***************************/
- SQLFreeStmt( hStmt, SQL_DROP );
-
- return 1;
-}
-
-/****************************
- *
- ***************************/
-int doTables( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable )
-{
- SQLHSTMT hStmt;
- char szCatalog[250] = "";
- char szSchema[250] = "";
- char szTable[250] = "";
- char szTableTypes[250] = "";
- char * pszCatalog = szCatalog;
- char * pszSchema = szSchema;
- char * pszTable = szTable;
- char * pszTableTypes = szTableTypes;
- SQLSMALLINT nCatalog = SQL_NTS;
- SQLSMALLINT nSchema = SQL_NTS;
- SQLSMALLINT nTable = SQL_NTS;
- SQLSMALLINT nTableTypes = SQL_NTS;
-
- SQLTCHAR szSepLine[32001] = "";
- SQLINTEGER nRows = 0;
- SQLRETURN nReturn;
-
- /****************************
- * EXECUTE SQL
- ***************************/
- if ( SQLAllocStmt( hDbc, &hStmt ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLAllocStmt\n" );
- return 0;
- }
-
- getElement( szSQL, '|', '\0', 1, szCatalog, sizeof(szCatalog) );
- getElement( szSQL, '|', '\0', 2, szSchema, sizeof(szSchema) );
- getElement( szSQL, '|', '\0', 3, szTable, sizeof(szTable) );
- getElement( szSQL, '|', '\0', 4, szTableTypes, sizeof(szTableTypes) );
- if ( !strcmp( szCatalog, "NULL" ) )
- {
- pszCatalog = NULL;
- nCatalog = 0;
- }
- if ( !strcmp( szSchema, "NULL" ) )
- {
- pszSchema = NULL;
- nSchema = 0;
- }
- if ( !strcmp( szTable, "NULL" ) )
- {
- pszTable = NULL;
- nTable = 0;
- }
- if ( !strcmp( szTableTypes, "NULL" ) )
- {
- pszTableTypes = NULL;
- nTableTypes = 0;
- }
-
- nReturn = SQLTables( hStmt, pszCatalog, nCatalog, pszSchema, nSchema, pszTable,
nTable, pszTableTypes, nTableTypes );
- if ( !SQL_SUCCEEDED( nReturn ) )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLTables\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
-
- /****************************
- * WRITE HEADER
- ***************************/
- if ( bHTMLTable )
- WriteHeaderHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteHeaderNormal( hStmt, szSepLine );
- else if ( cDelimiter && bColumnNames )
- WriteHeaderDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE BODY
- ***************************/
- if ( bHTMLTable )
- WriteBodyHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- nRows = WriteBodyNormal( hStmt );
- else
- WriteBodyDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE FOOTER
- ***************************/
- if ( bHTMLTable )
- WriteFooterHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteFooterNormal( hStmt, szSepLine, nRows );
-
- /****************************
- * CLEANUP
- ***************************/
- SQLFreeStmt( hStmt, SQL_DROP );
-
- return 1;
-}
-
-/****************************
- *
- ***************************/
-int doColumns( SQLHDBC hDbc, char *szSQL, char cDelimiter, int bColumnNames, int
bHTMLTable )
-{
- SQLHSTMT hStmt;
- char szCatalog[250] = "";
- char szSchema[250] = "";
- char szTable[250] = "";
- char szColumn[250] = "";
- char * pszCatalog = szCatalog;
- char * pszSchema = szSchema;
- char * pszTable = szTable;
- char * pszColumn = szColumn;
- SQLSMALLINT nCatalog = SQL_NTS;
- SQLSMALLINT nSchema = SQL_NTS;
- SQLSMALLINT nTable = SQL_NTS;
- SQLSMALLINT nColumn = SQL_NTS;
-
- SQLTCHAR szSepLine[32001] = "";
- SQLINTEGER nRows = 0;
- SQLRETURN nReturn;
-
- /****************************
- * EXECUTE SQL
- ***************************/
- if ( SQLAllocStmt( hDbc, &hStmt ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, 0 );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLAllocStmt\n" );
- return 0;
- }
-
- getElement( szSQL, '|', '\0', 1, szCatalog, sizeof(szCatalog) );
- getElement( szSQL, '|', '\0', 2, szSchema, sizeof(szSchema) );
- getElement( szSQL, '|', '\0', 3, szTable, sizeof(szTable) );
- getElement( szSQL, '|', '\0', 4, szColumn, sizeof(szColumn) );
-
- if ( !strcmp( szCatalog, "NULL" ) )
- {
- pszCatalog = NULL;
- nCatalog = 0;
- }
- if ( !strcmp( szSchema, "NULL" ) )
- {
- pszSchema = NULL;
- nSchema = 0;
- }
- if ( !strcmp( szTable, "NULL" ) )
- {
- pszTable = NULL;
- nTable = 0;
- }
- if ( !strcmp( szColumn, "NULL" ) )
- {
- pszColumn = NULL;
- nColumn = 0;
- }
-
- nReturn = SQLColumns( hStmt, pszCatalog, nCatalog, pszSchema, nSchema, pszTable,
nTable, pszColumn, nColumn );
- if ( !SQL_SUCCEEDED( nReturn ) )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- _ftprintf( stderr, "[MYODBCShell]ERROR: Could not SQLColumns\n" );
- SQLFreeStmt( hStmt, SQL_DROP );
- return 0;
- }
-
- /****************************
- * WRITE HEADER
- ***************************/
- if ( bHTMLTable )
- WriteHeaderHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteHeaderNormal( hStmt, szSepLine );
- else if ( cDelimiter && bColumnNames )
- WriteHeaderDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE BODY
- ***************************/
- if ( bHTMLTable )
- WriteBodyHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- nRows = WriteBodyNormal( hStmt );
- else
- WriteBodyDelimited( hStmt, cDelimiter );
-
- /****************************
- * WRITE FOOTER
- ***************************/
- if ( bHTMLTable )
- WriteFooterHTMLTable( hStmt );
- else if ( cDelimiter == 0 )
- WriteFooterNormal( hStmt, szSepLine, nRows );
-
- /****************************
- * CLEANUP
- ***************************/
- if ( !SQL_SUCCEEDED(SQLFreeStmt( hStmt, SQL_DROP )) )
- DumpODBCLog( hEnv, hDbc, hStmt );
-
- return 1;
-}
-
-/****************************
- * CloseDatabase - cleanup in prep for exiting the program
- ***************************/
-int CloseDatabase( SQLHENV hEnv, SQLHDBC hDbc )
-{
- if ( !SQL_SUCCEEDED(SQLDisconnect( hDbc )) )
- DumpODBCLog( hEnv, hDbc, 0 );
- if ( !SQL_SUCCEEDED(SQLFreeConnect( hDbc )) )
- DumpODBCLog( hEnv, hDbc, 0 );
- if ( !SQL_SUCCEEDED(SQLFreeEnv( hEnv )) )
- DumpODBCLog( hEnv, 0, 0 );
- return 1;
-}
-
-
-/****************************
- * WRITE HTML
- ***************************/
-void WriteHeaderHTMLTable( SQLHSTMT hStmt )
-{
- SQLINTEGER nCol = 0;
- SQLSMALLINT nColumns = 0;
- SQLTCHAR szColumnName[MAX_DATA_WIDTH+1] = "";
-
- printf( "<table BORDER>\n" );
- printf( "<tr BGCOLOR=#000099>\n" );
-
- if ( SQLNumResultCols( hStmt, &nColumns ) != SQL_SUCCESS )
- nColumns = -1;
-
- for ( nCol = 1; nCol <= nColumns; nCol++ )
- {
- SQLColAttribute( hStmt, nCol, SQL_DESC_LABEL, szColumnName, sizeof(szColumnName),
NULL, NULL );
- printf( "<td>\n" );
- printf( "<font face=Arial,Helvetica><font color=#FFFFFF>\n" );
- printf( "%s\n", szColumnName );
- printf( "</font></font>\n" );
- printf( "</td>\n" );
- }
- printf( "</tr>\n" );
-}
-
-void WriteBodyHTMLTable( SQLHSTMT hStmt )
-{
- SQLINTEGER nCol = 0;
- SQLSMALLINT nColumns = 0;
- SQLINTEGER nIndicator = 0;
- SQLTCHAR szColumnValue[MAX_DATA_WIDTH+1] = "";
- SQLRETURN nReturn = 0;
- SQLRETURN ret;
-
- if ( SQLNumResultCols( hStmt, &nColumns ) != SQL_SUCCESS )
- nColumns = -1;
-
- while ( (ret = SQLFetch( hStmt )) == SQL_SUCCESS ) /* ROWS */
- {
- printf( "<tr>\n" );
-
- for ( nCol = 1; nCol <= nColumns; nCol++ ) /* COLS */
- {
- printf( "<td>\n" );
- printf( "<font face=Arial,Helvetica>\n" );
-
- nReturn = SQLGetData( hStmt, nCol, SQL_C_CHAR, (SQLPOINTER)szColumnValue,
sizeof(szColumnValue), &nIndicator );
- if ( nReturn == SQL_SUCCESS && nIndicator != SQL_NULL_DATA )
- {
- fputs((char*) szColumnValue, stdout );
- }
- else if ( nReturn == SQL_ERROR )
- {
- ret = SQL_ERROR;
- break;
- }
- else
- printf( "%s\n", "" );
-
- printf( "</font>\n" );
- printf( "</td>\n" );
- }
- if (ret != SQL_SUCCESS)
- break;
- printf( "</tr>\n" );
- }
-}
-
-void WriteFooterHTMLTable( SQLHSTMT hStmt )
-{
- printf( "</table>\n" );
-}
-
-/****************************
- * WRITE DELIMITED
- * - this output can be used by the ODBC Text File driver
- * - last column no longer has a delimit char (it is implicit)...
- * this is consistent with odbctxt
- ***************************/
-void WriteHeaderDelimited( SQLHSTMT hStmt, char cDelimiter )
-{
- SQLINTEGER nCol = 0;
- SQLSMALLINT nColumns = 0;
- SQLTCHAR szColumnName[MAX_DATA_WIDTH+1] = "";
-
- if ( SQLNumResultCols( hStmt, &nColumns ) != SQL_SUCCESS )
- nColumns = -1;
-
- for ( nCol = 1; nCol <= nColumns; nCol++ )
- {
- SQLColAttribute( hStmt, nCol, SQL_DESC_LABEL, szColumnName, sizeof(szColumnName),
NULL, NULL );
- fputs((char*) szColumnName, stdout );
- if ( nCol < nColumns )
- putchar( cDelimiter );
- }
- putchar( '\n' );
-}
-
-void WriteBodyDelimited( SQLHSTMT hStmt, char cDelimiter )
-{
- SQLINTEGER nCol = 0;
- SQLSMALLINT nColumns = 0;
- SQLINTEGER nIndicator = 0;
- SQLTCHAR szColumnValue[MAX_DATA_WIDTH+1] = "";
- SQLRETURN nReturn = 0;
- SQLRETURN ret;
-
- if ( SQLNumResultCols( hStmt, &nColumns ) != SQL_SUCCESS )
- nColumns = -1;
-
- /* ROWS */
- while (( ret = SQLFetch( hStmt )) == SQL_SUCCESS )
- {
- /* COLS */
- for ( nCol = 1; nCol <= nColumns; nCol++ )
- {
- nReturn = SQLGetData( hStmt, nCol, SQL_C_CHAR, (SQLPOINTER)szColumnValue,
sizeof(szColumnValue), &nIndicator );
- if ( nReturn == SQL_SUCCESS && nIndicator != SQL_NULL_DATA )
- {
- fputs((char*) szColumnValue, stdout );
- if ( nCol < nColumns )
- putchar( cDelimiter );
- }
- else if ( nReturn == SQL_ERROR )
- {
- ret = SQL_ERROR;
- break;
- }
- else
- {
- if ( nCol < nColumns )
- putchar( cDelimiter );
- }
- }
- if (ret != SQL_SUCCESS)
- break;
- printf( "\n" );
- }
- if ( ret == SQL_ERROR )
- {
- if ( bVerbose ) DumpODBCLog( 0, 0, hStmt );
- }
-}
-
-/****************************
- * WRITE NORMAL
- ***************************/
-void WriteHeaderNormal( SQLHSTMT hStmt, SQLTCHAR *szSepLine )
-{
- SQLINTEGER nCol = 0;
- SQLSMALLINT nColumns = 0;
- SQLTCHAR szColumn[MAX_DATA_WIDTH+20] = "";
- SQLTCHAR szColumnName[MAX_DATA_WIDTH+1] = "";
- SQLTCHAR szHdrLine[32001] = "";
- SQLUINTEGER nOptimalDisplayWidth = 10;
-
- if ( SQLNumResultCols( hStmt, &nColumns ) != SQL_SUCCESS )
- nColumns = -1;
-
- for ( nCol = 1; nCol <= nColumns; nCol++ )
- {
- nOptimalDisplayWidth = OptimalDisplayWidth( hStmt, nCol, nUserWidth );
- if ( !SQL_SUCCEEDED( SQLColAttribute( hStmt, nCol, SQL_DESC_LABEL, szColumnName,
sizeof(szColumnName), NULL, NULL ) && bVerbose ) )
- {
- DumpODBCLog( 0, 0, hStmt );
- }
- /* SEP */
- memset( szColumn, '\0', sizeof(szColumn) );
- memset( szColumn, '-', nOptimalDisplayWidth + 1 );
- strcat((char*) szSepLine, "+" );
- strcat((char*) szSepLine,(char*) szColumn );
-
- /* HDR */
- sprintf( (char*)szColumn, "| %-*.*s", nOptimalDisplayWidth, nOptimalDisplayWidth,
szColumnName );
- strcat( (char*)szHdrLine,(char*) szColumn );
- }
- strcat((char*) szSepLine, "+\n" );
- strcat((char*) szHdrLine, "|\n" );
-
- printf((char*) szSepLine );
- printf((char*) szHdrLine );
- printf((char*) szSepLine );
-
-}
-
-SQLINTEGER WriteBodyNormal( SQLHSTMT hStmt )
-{
- SQLINTEGER nCol = 0;
- SQLSMALLINT nColumns = 0;
- SQLINTEGER nIndicator = 0;
- SQLTCHAR szColumn[MAX_DATA_WIDTH+20] = "";
- SQLTCHAR szColumnValue[MAX_DATA_WIDTH+1] = "";
- SQLRETURN nReturn = 0;
- SQLINTEGER nRows = 0;
- SQLUINTEGER nOptimalDisplayWidth = 10;
-
- nReturn = SQLNumResultCols( hStmt, &nColumns );
- if ( nReturn != SQL_SUCCESS && nReturn != SQL_SUCCESS_WITH_INFO )
- nColumns = -1;
-
- /* ROWS */
- nReturn = SQLFetch( hStmt );
- while ( nReturn == SQL_SUCCESS || nReturn == SQL_SUCCESS_WITH_INFO )
- {
- /* COLS */
- for ( nCol = 1; nCol <= nColumns; nCol++ )
- {
- nOptimalDisplayWidth = OptimalDisplayWidth( hStmt, nCol, nUserWidth );
- nReturn = SQLGetData( hStmt, nCol, SQL_C_CHAR, (SQLPOINTER)szColumnValue,
sizeof(szColumnValue), &nIndicator );
- szColumnValue[MAX_DATA_WIDTH] = '\0';
-
- if ( nReturn == SQL_SUCCESS && nIndicator != SQL_NULL_DATA )
- {
- sprintf( (char*)szColumn, "| %-*.*s", nOptimalDisplayWidth,
nOptimalDisplayWidth, szColumnValue );
- }
- else if ( nReturn == SQL_ERROR )
- {
- break;
- }
- else
- {
- sprintf( (char*)szColumn, "| %-*s", nOptimalDisplayWidth, "" );
- }
- fputs( (char*)szColumn, stdout );
- } /* for columns */
-
- nRows++;
- printf( "|\n" );
- nReturn = SQLFetch( hStmt );
- } /* while rows */
-
- if ( nReturn == SQL_ERROR )
- {
- if ( bVerbose ) DumpODBCLog( 0, 0, hStmt );
- }
-
- return nRows;
-}
-
-void WriteFooterNormal( SQLHSTMT hStmt, SQLTCHAR *szSepLine, SQLINTEGER nRows )
-{
- SQLLEN nRowsAffected = -1;
-
- printf( (char*)szSepLine );
-
- if ( SQLRowCount( hStmt, &nRowsAffected ) != SQL_SUCCESS )
- {
- if ( bVerbose ) DumpODBCLog( hEnv, hDbc, hStmt );
- }
-
- printf( "%d rows affected\n", nRowsAffected );
- printf( "%d rows fetched\n", nRows );
-}
-
-int DumpODBCLog( SQLHENV hEnv, SQLHDBC hDbc, SQLHSTMT hStmt )
-{
- SQLSMALLINT nRecord = 1;
- SQLTCHAR szState[6];
- SQLINTEGER nNativeError;
- SQLTCHAR szMessageText[SQL_MAX_MESSAGE_LENGTH];
- SQLSMALLINT nTextLength;
-
- if ( hStmt )
- {
- while ( SQL_SUCCEEDED(SQLGetDiagRec( SQL_HANDLE_STMT,
- hStmt,
- nRecord,
- szState,
- &nNativeError,
- szMessageText,
- SQL_MAX_MESSAGE_LENGTH,
- &nTextLength )) )
- {
- szState[5] = '\0';
- _ftprintf( stderr, "[%s]%s\n", szState, szMessageText );
- nRecord++;
- }
- }
-
- if ( hDbc )
- {
- nRecord = 1;
- while ( SQL_SUCCEEDED(SQLGetDiagRec( SQL_HANDLE_DBC,
- hDbc,
- nRecord,
- szState,
- &nNativeError,
- szMessageText,
- SQL_MAX_MESSAGE_LENGTH,
- &nTextLength )) )
- {
- szState[5] = '\0';
- _ftprintf( stderr, "[%s]%s\n", szState, szMessageText );
- nRecord++;
- }
- }
-
- if ( hEnv )
- {
- nRecord = 1;
- while ( SQL_SUCCEEDED(SQLGetDiagRec( SQL_HANDLE_ENV,
- hEnv,
- nRecord,
- szState,
- &nNativeError,
- szMessageText,
- SQL_MAX_MESSAGE_LENGTH,
- &nTextLength )) )
- {
- szState[5] = '\0';
- _ftprintf( stderr, "[%s]%s\n", szState, szMessageText );
- nRecord++;
- }
- }
-
- return 1;
-}
-
-/*
-int DumpODBCLog( SQLHENV hEnv, SQLHDBC hDbc, SQLHSTMT hStmt )
-{
- SQLTCHAR szError[501];
- SQLTCHAR szSqlState[10];
- SQLINTEGER nNativeError;
- SQLSMALLINT nErrorMsg;
-
- if ( hStmt )
- {
- while ( SQLError( hEnv, hDbc, hStmt, szSqlState, &nNativeError, szError, 500,
&nErrorMsg ) == SQL_SUCCESS )
- {
- _ftprintf( stderr, "[%s]%s\n", szSqlState, szError );
- }
- }
-
- if ( hDbc )
- {
- while ( SQLError( hEnv, hDbc, 0, szSqlState, &nNativeError, szError, 500,
&nErrorMsg ) == SQL_SUCCESS )
- {
- _ftprintf( stderr, "[%s]%s\n", szSqlState, szError );
- }
- }
-
- if ( hEnv )
- {
- while ( SQLError( hEnv, 0, 0, szSqlState, &nNativeError, szError, 500,
&nErrorMsg ) == SQL_SUCCESS )
- {
- _ftprintf( stderr, "[%s]%s\n", szSqlState, szError );
- }
- }
-
- return 1;
-}
-*/
| Thread |
|---|
| • Connector/ODBC 5 commit: r84 - MYODBCC/include MYODBCShell | pharvey | 18 Apr |
