src/tools/isql.cxx

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by www.databasecache.com                           *
   3  *   Contact: praba_tuty@databasecache.com                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  ***************************************************************************/
  16 #include<CSql.h>
  17 #include <Statement.h>
  18 #include <SqlStatement.h>
  19 #define SQL_STMT_LEN 1024
  20 enum STMT_TYPE
  21 {
  22     SELECT =0,
  23     DDL ,
  24     OTHER
  25 };
  26 STMT_TYPE stmtType = SELECT;
  27 FILE *fp;
  28 SqlConnection *conn;
  29 SqlStatement *stmt;
  30 bool getInput(bool);
  31 int main(int argc, char **argv)
  32 {
  33     char username[IDENTIFIER_LENGTH];
  34     username [0] = '\0';
  35     char password[IDENTIFIER_LENGTH];
  36     password [0] = '\0';
  37     char filename[512];
  38     filename [0] ='\0';
  39     int c = 0;
  40     while ((c = getopt(argc, argv, "u:p:s")) != EOF)
  41     {
  42         switch (c)
  43         {
  44             case 'u' : strcpy(username , argv[optind - 1]); break;
  45             case 'p' : strcpy(password , argv[optind - 1]); break;
  46             case 's' : strcpy(filename , argv[optind ]); break;
  47             default: printf("Wrong args\n"); exit(1);
  48
  49         }
  50     }//while options
  51     //printf("%s %s %s", username, password, filename);
  52     if (username[0] == '\0' )
  53     {
  54         strcpy(username, "root");
  55         strcpy(password, "manager");
  56     }
  57     bool fileFlag = false;
  58     if (filename [0] !='\0')
  59     {
  60         fp = fopen(filename,"r");
  61         if (fp == NULL)
  62         {
  63             printf("Unable to open the file %s\n", filename);
  64             return 1;
  65         }
  66         fileFlag = true;
  67     }
  68
  69     DbRetVal rv = OK;
  70     conn = new SqlConnection();
  71     rv = conn->connect("root", "manager");
  72     if (rv != OK) return 1;
  73     stmt = new SqlStatement();
  74     stmt->setConnection(conn);
  75     rv = conn->beginTrans();
  76     if (rv != OK) return 2;
  77
  78     while (getInput(fileFlag) == true) continue;
  79
  80     //TODO::conn should provide method telling status of the transaction.
  81     //if running, then abort it
  82     conn->rollback();
  83     if (filename [0] !='\0')
  84     {
  85        fclose(fp);
  86     }
  87     delete stmt;
  88     delete conn;
  89     return 0;
  90 }
  91 bool handleTransaction(char *st)
  92 {
  93     while (isspace (*st)|| *st == '(' ) st++; // Skip white spaces
  94     if (strncasecmp (st, "COMMIT", 6) == 0 ||
  95         strncasecmp (st, "commit", 6) == 0 )
  96     {
  97         conn->commit();
  98         conn->beginTrans();
  99         return true;
 100     }
 101     else if (strncasecmp (st, "ROLLBACK", 8) == 0||
 102         strncasecmp (st, "rollback", 8) == 0)
 103     {
 104         conn->rollback();
 105         conn->beginTrans();
 106         return true;
 107     }
 108     return false;
 109 }
 110 bool handleEchoAndComment(char *st)
 111 {
 112     while (isspace (*st)|| *st == '(' ) st++; // Skip white spaces
 113     if (strncasecmp (st, "ECHO", 4) == 0 ||
 114         strncasecmp (st, "echo", 4) == 0 )
 115     {
 116         printf("%s\n", st);
 117         return true;
 118     }else if (strncmp(st, "--", 2) == 0)
 119     {
 120         return true;
 121     }
 122     return false;
 123 }
 124 void setStmtType(char *st)
 125 {
 126    if (strncasecmp (st, "SELECT", 6) == 0)  {stmtType=SELECT; return; }
 127    else if (strncasecmp (st, "CREATE", 6) == 0) {stmtType=DDL; return; }
 128    else if (strncasecmp (st, "DROP", 4) == 0) { stmtType=DDL; return; }
 129    stmtType = OTHER;
 130    return ;
 131 }
 132
 133 char getQueryFromStdIn(char *buf)
 134 {
 135     char    c;
 136     int    ln;
 137
 138     ln=1;
 139     printf("CSQL>");
 140     while( (c=(char ) getchar()) != EOF && c != ';')
 141     {
 142         *buf++ = c;
 143         if(c=='\n')
 144             //printf("%1d>",ln++);
 145             ln++;
 146     }
 147     *buf++ = ';';
 148     *buf = '\0';
 149     return c;
 150 }
 151 char getQueryFromFile(char *buf)
 152 {
 153     char    c;
 154     while( (c=(char ) fgetc(fp)) != EOF && c != ';')
 155     {
 156         *buf++ = c;
 157     }
 158     *buf++ = ';';
 159     *buf = '\0';
 160     return c;
 161 }
 162
 163 bool getInput(bool fromFile)
 164 {
 165     char buffer [SQL_STMT_LEN + 1];
 166
 167     char eof;
 168     if (fromFile == false)
 169         eof = getQueryFromStdIn(buffer);
 170     else
 171         eof = getQueryFromFile(buffer);
 172
 173     char *buf = buffer;
 174     while(*buf == ' ' || *buf == '\t' || *buf == '\n') buf++;
 175     if (eof == EOF || strncasecmp (buf, "quit", 4) == 0)
 176         return false;
 177     if (handleTransaction(buf)) return true;
 178     if (handleEchoAndComment(buf)) return true;
 179
 180     setStmtType(buf);
 181
 182     DbRetVal rv = stmt->prepare(buf);
 183     if (rv != OK)
 184     {
 185         printf("Statement prepare failed with error %d\n", rv);
 186         return true;
 187     }
 188     int rows =0;
 189     rv = stmt->execute(rows);
 190     if (rv != OK)
 191     {
 192         printf("Statement execute failed with error %d\n", rv);
 193         return true;
 194     }
 195     if (stmtType == OTHER)
 196     {
 197         printf("Statement Executed: Rows Affected = %d\n", rows);
 198     }
 199     else if (stmtType == DDL)
 200     {
 201         printf("Statement Executed\n");
 202     }
 203     else
 204     {
 205         FieldInfo *info = new FieldInfo();
 206         printf("---------------------------------------------------------\n");
 207         printf("\t");
 208         for (int i = 0 ; i < stmt->noOfProjFields() ; i++)
 209         {
 210             stmt->getProjFldInfo(i, info);
 211             printf("%s\t", info->fldName);
 212         }
 213         printf("\n---------------------------------------------------------\n");
 214         delete info;
 215         void *tuple = NULL;
 216         while(true)
 217         {
 218             printf("\t");
 219             tuple = (char*)stmt->fetchAndPrint();
 220             printf("\n");
 221             if (tuple == NULL) { break; }
 222         }
 223     }
 224     stmt->free();
 225     return true;
 226
 227 }