src/sql/CopyTblStatement.cxx

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Prabakaran Thirumalai   *
   3  *   praba_tuty@yahoo.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 <os.h>
  17 #include <Statement.h>
  18 #include <TableImpl.h>
  19 #include <OrderTableImpl.h>
  20
  21 CopyTblStatement::CopyTblStatement()
  22 {
  23     innerSelStmt = NULL;
  24     totalFields = 0;
  25     bindFieldValues= NULL;
  26     parsedData = NULL;
  27     dbMgr = NULL;
  28     table = NULL;
  29     params = NULL;
  30     paramValues = NULL;
  31     totalParams = 0;
  32 }
  33
  34 CopyTblStatement::~CopyTblStatement()
  35 {
  36     if (table) { table->close(); table = NULL; }
  37     SelStatement *stmt= (SelStatement *)innerSelStmt;
  38     if (stmt) stmt->close();
  39     if(totalFields){
  40        for(int i=0;i<totalFields;i++)
  41        {
  42            free(bindFieldValues[i]);
  43        }
  44        free(bindFieldValues); bindFieldValues = NULL;
  45     }
  46     totalFields =0;
  47     delete innerSelStmt;
  48 }
  49
  50 DbRetVal CopyTblStatement::resolve()
  51 {
  52     DbRetVal rv = OK;
  53     ListIterator titer = parsedData->getTableNameList().getIterator();
  54     while (titer.hasElement()) {
  55         TableName *t  = (TableName*)titer.nextElement();
  56         if (strcmp(t->tblName, parsedData->getPKTableName())==0) {
  57             printError(ErrBadArg,
  58                             "Source and Target table Names cannot be same.");
  59             return ErrBadArg;
  60         }
  61     }
  62     strcpy(tblName,parsedData->getPKTableName());
  63     SelStatement *sStmt = new SelStatement();
  64     innerSelStmt = ( Statement *) sStmt;
  65     innerSelStmt->setDbMgr(dbMgr);
  66     innerSelStmt->setParsedData(parsedData);
  67     rv = innerSelStmt->resolve();
  68     if (rv != OK) return rv;
  69     if(parsedData->getCreateTbl())
  70         rv = resolveForCreate();
  71     else
  72         rv = resolveForInsert();
  73     return rv;
  74 }
  75
  76 DbRetVal CopyTblStatement::resolveForInsert()
  77 {
  78     table = dbMgr->openTable(tblName);
  79     if (table == NULL)
  80     {
  81         printError(ErrNotExists, "Unable to open the table:Table not exists");
  82         return ErrNotExists;
  83     }
  84     table->setLoading(true);
  85     //bind field
  86     List fNameList = table->getFieldNameList();
  87     ListIterator fNameIter = fNameList.getIterator();
  88     FieldInfo *info = new FieldInfo();
  89     int fcount = 1; void *valBuf; int fieldsize=0;
  90     Identifier *elem = NULL;
  91     totalFields = fNameList.size();
  92     bindFieldValues = (void**) malloc( totalFields * sizeof(char*));
  93     memset(bindFieldValues, 0, totalFields * sizeof(char*));
  94     SelStatement *selStmt = (SelStatement*) innerSelStmt;
  95     while (fNameIter.hasElement())
  96     {
  97         elem = (Identifier*) fNameIter.nextElement();
  98         table->getFieldInfo((const char*)elem->name, info);
  99         //TODO :: Should free
 100         valBuf = AllDataType::alloc(info->type, info->length);
 101         table->bindFld(elem->name, valBuf);
 102         fieldsize=0;
 103         selStmt->setBindField(fcount, valBuf);
 104         bindFieldValues[fcount-1] = valBuf;
 105         fcount++;
 106     }
 107     delete info;
 108     fNameIter.reset();
 109     while (fNameIter.hasElement())
 110     {
 111          delete (Identifier*) fNameIter.nextElement();
 112     }
 113     fNameList.reset();
 114     return OK;
 115 }
 116
 117 DbRetVal CopyTblStatement::resolveForCreate()
 118 {
 119     DbRetVal rv = OK;
 120     int  nop = 0;
 121     SelStatement *stmt = (SelStatement*) innerSelStmt;
 122     nop = stmt->noOfProjFields();
 123     FieldInfo *field = new FieldInfo();
 124     TableDef tabDef;
 125     char name[IDENTIFIER_LENGTH];
 126     for(int i=0; i<nop; i++)
 127     {
 128         rv = stmt->getProjFldInfo(i+1,field);
 129         Table::getFieldNameAlone(field->fldName, name);
 130         tabDef.addField(name,field->type, field->length, field->defaultValueBuf, field->isNull);
 131     }
 132     delete field;
 133     rv = dbMgr->createTable(tblName, tabDef);
 134     if(rv!=OK){
 135        return rv;
 136     }
 137     rv = resolveForInsert();
 138     return rv;
 139 }
 140
 141 DbRetVal CopyTblStatement::execute(int &rowsAffected)
 142 {
 143     DbRetVal rv = OK;
 144     SelStatement *selStmt = (SelStatement*) innerSelStmt;
 145     selStmt->execute(rowsAffected);
 146     while(selStmt->fetch())
 147     {
 148         rv = table->insertTuple();
 149         if(rv!=OK) break;
 150     }
 151     if( (rv!=OK) && ( !parsedData->getCreateTbl()))
 152     {
 153         dbMgr->dropTable(tblName);
 154     }
 155     return rv;
 156 }
 157
 158 DbRetVal CopyTblStatement::setShortParam(int paramNo, short value)
 159 {
 160     SelStatement *stmt = (SelStatement*) innerSelStmt;
 161     return stmt->setShortParam(paramNo,value);
 162 }
 163
 164 DbRetVal CopyTblStatement::setIntParam(int paramNo, int value)
 165 {
 166     SelStatement *stmt = (SelStatement*) innerSelStmt;
 167     return stmt->setIntParam(paramNo,value);
 168 }
 169
 170 DbRetVal CopyTblStatement::setLongParam(int paramNo, long value)
 171 {
 172     SelStatement *stmt = (SelStatement*) innerSelStmt;
 173     return stmt->setLongParam(paramNo,value);
 174 }
 175
 176 DbRetVal CopyTblStatement::setLongLongParam(int paramNo, long long value)
 177 {
 178     SelStatement *stmt = (SelStatement*) innerSelStmt;
 179     return stmt->setLongLongParam(paramNo,value);
 180 }
 181
 182 DbRetVal CopyTblStatement::setByteIntParam(int paramNo, ByteInt value)
 183 {
 184     SelStatement *stmt = (SelStatement*) innerSelStmt;
 185     return stmt->setByteIntParam(paramNo,value);
 186 }
 187
 188 DbRetVal CopyTblStatement::setFloatParam(int paramNo, float value)
 189 {
 190     SelStatement *stmt = (SelStatement*) innerSelStmt;
 191     return stmt->setFloatParam(paramNo,value);
 192 }
 193
 194 DbRetVal CopyTblStatement::setDoubleParam(int paramNo, double value)
 195 {
 196     SelStatement *stmt = (SelStatement*) innerSelStmt;
 197     return stmt->setDoubleParam(paramNo,value);
 198 }
 199
 200 DbRetVal CopyTblStatement::setStringParam(int paramNo, char *value)
 201 {
 202     SelStatement *stmt = (SelStatement*) innerSelStmt;
 203     return stmt->setStringParam(paramNo,value);
 204 }
 205
 206 DbRetVal CopyTblStatement::setDateParam(int paramNo, Date value)
 207 {
 208     SelStatement *stmt = (SelStatement*) innerSelStmt;
 209     return stmt->setDateParam(paramNo,value);
 210 }
 211
 212 DbRetVal CopyTblStatement::setTimeParam(int paramNo, Time value)
 213 {
 214     SelStatement *stmt = (SelStatement*) innerSelStmt;
 215     return stmt->setTimeParam(paramNo,value);
 216 }
 217
 218 DbRetVal CopyTblStatement::setTimeStampParam(int paramNo, TimeStamp value)
 219 {
 220     SelStatement *stmt = (SelStatement*) innerSelStmt;
 221     return stmt->setTimeStampParam(paramNo,value);
 222 }
 223
 224 DbRetVal CopyTblStatement::setBinaryParam(int paramNo, void *value, int length)
 225 {
 226     SelStatement *stmt = (SelStatement*) innerSelStmt;
 227     return stmt->setBinaryParam(paramNo,value,length);
 228 }
 229
 230 void* CopyTblStatement::getParamValuePtr(int pos)
 231 {
 232     SelStatement *stmt = (SelStatement*) innerSelStmt;
 233     return stmt->getParamValuePtr(pos);
 234 }
 235
 236 int CopyTblStatement::getFldPos(char *name)
 237 {
 238    return 0;
 239 }
 240
 241 DbRetVal CopyTblStatement::getParamFldInfo(int paramPos, FieldInfo *&info)
 242 {
 243     DbRetVal rv = OK;
 244     SelStatement *stmt = (SelStatement*) innerSelStmt;
 245     rv = stmt->getParamFldInfo(paramPos, info);
 246     return rv;
 247 }
 248 DbRetVal CopyTblStatement::setParam(int paramNo, void *value)
 249 {
 250     DbRetVal rv = OK;
 251     SelStatement *stmt = (SelStatement*) innerSelStmt;
 252     rv = stmt->setParam(paramNo,value);
 253     return rv;
 254 }