Return to copy.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / php / ext / sqlite / libsqlite / src

Annotation of embedaddon/php/ext/sqlite/libsqlite/src/copy.c, revision 1.1.1.1

1.1       misho       1: /*
                      2: ** 2003 April 6
                      3: **
                      4: ** The author disclaims copyright to this source code.  In place of
                      5: ** a legal notice, here is a blessing:
                      6: **
                      7: **    May you do good and not evil.
                      8: **    May you find forgiveness for yourself and forgive others.
                      9: **    May you share freely, never taking more than you give.
                     10: **
                     11: *************************************************************************
                     12: ** This file contains code used to implement the COPY command.
                     13: **
                     14: ** $Id: copy.c 195361 2005-09-07 15:11:33Z iliaa $
                     15: */
                     16: #include "sqliteInt.h"
                     17: 
                     18: /*
                     19: ** The COPY command is for compatibility with PostgreSQL and specificially
                     20: ** for the ability to read the output of pg_dump.  The format is as
                     21: ** follows:
                     22: **
                     23: **    COPY table FROM file [USING DELIMITERS string]
                     24: **
                     25: ** "table" is an existing table name.  We will read lines of code from
                     26: ** file to fill this table with data.  File might be "stdin".  The optional
                     27: ** delimiter string identifies the field separators.  The default is a tab.
                     28: */
                     29: void sqliteCopy(
                     30:   Parse *pParse,       /* The parser context */
                     31:   SrcList *pTableName, /* The name of the table into which we will insert */
                     32:   Token *pFilename,    /* The file from which to obtain information */
                     33:   Token *pDelimiter,   /* Use this as the field delimiter */
                     34:   int onError          /* What to do if a constraint fails */
                     35: ){
                     36:   Table *pTab;
                     37:   int i;
                     38:   Vdbe *v;
                     39:   int addr, end;
                     40:   char *zFile = 0;
                     41:   const char *zDb;
                     42:   sqlite *db = pParse->db;
                     43: 
                     44: 
                     45:   if( sqlite_malloc_failed  ) goto copy_cleanup;
                     46:   assert( pTableName->nSrc==1 );
                     47:   pTab = sqliteSrcListLookup(pParse, pTableName);
                     48:   if( pTab==0 || sqliteIsReadOnly(pParse, pTab, 0) ) goto copy_cleanup;
                     49:   zFile = sqliteStrNDup(pFilename->z, pFilename->n);
                     50:   sqliteDequote(zFile);
                     51:   assert( pTab->iDb<db->nDb );
                     52:   zDb = db->aDb[pTab->iDb].zName;
                     53:   if( sqliteAuthCheck(pParse, SQLITE_INSERT, pTab->zName, 0, zDb)
                     54:       || sqliteAuthCheck(pParse, SQLITE_COPY, pTab->zName, zFile, zDb) ){
                     55:     goto copy_cleanup;
                     56:   }
                     57:   v = sqliteGetVdbe(pParse);
                     58:   if( v ){
                     59:     sqliteBeginWriteOperation(pParse, 1, pTab->iDb);
                     60:     addr = sqliteVdbeOp3(v, OP_FileOpen, 0, 0, pFilename->z, pFilename->n);
                     61:     sqliteVdbeDequoteP3(v, addr);
                     62:     sqliteOpenTableAndIndices(pParse, pTab, 0);
                     63:     if( db->flags & SQLITE_CountRows ){
                     64:       sqliteVdbeAddOp(v, OP_Integer, 0, 0);  /* Initialize the row count */
                     65:     }
                     66:     end = sqliteVdbeMakeLabel(v);
                     67:     addr = sqliteVdbeAddOp(v, OP_FileRead, pTab->nCol, end);
                     68:     if( pDelimiter ){
                     69:       sqliteVdbeChangeP3(v, addr, pDelimiter->z, pDelimiter->n);
                     70:       sqliteVdbeDequoteP3(v, addr);
                     71:     }else{
                     72:       sqliteVdbeChangeP3(v, addr, "\t", 1);
                     73:     }
                     74:     if( pTab->iPKey>=0 ){
                     75:       sqliteVdbeAddOp(v, OP_FileColumn, pTab->iPKey, 0);
                     76:       sqliteVdbeAddOp(v, OP_MustBeInt, 0, 0);
                     77:     }else{
                     78:       sqliteVdbeAddOp(v, OP_NewRecno, 0, 0);
                     79:     }
                     80:     for(i=0; i<pTab->nCol; i++){
                     81:       if( i==pTab->iPKey ){
                     82:         /* The integer primary key column is filled with NULL since its
                     83:         ** value is always pulled from the record number */
                     84:         sqliteVdbeAddOp(v, OP_String, 0, 0);
                     85:       }else{
                     86:         sqliteVdbeAddOp(v, OP_FileColumn, i, 0);
                     87:       }
                     88:     }
                     89:     sqliteGenerateConstraintChecks(pParse, pTab, 0, 0, pTab->iPKey>=0, 
                     90:                                    0, onError, addr);
                     91:     sqliteCompleteInsertion(pParse, pTab, 0, 0, 0, 0, -1);
                     92:     if( (db->flags & SQLITE_CountRows)!=0 ){
                     93:       sqliteVdbeAddOp(v, OP_AddImm, 1, 0);  /* Increment row count */
                     94:     }
                     95:     sqliteVdbeAddOp(v, OP_Goto, 0, addr);
                     96:     sqliteVdbeResolveLabel(v, end);
                     97:     sqliteVdbeAddOp(v, OP_Noop, 0, 0);
                     98:     sqliteEndWriteOperation(pParse);
                     99:     if( db->flags & SQLITE_CountRows ){
                    100:       sqliteVdbeAddOp(v, OP_ColumnName, 0, 1);
                    101:       sqliteVdbeChangeP3(v, -1, "rows inserted", P3_STATIC);
                    102:       sqliteVdbeAddOp(v, OP_Callback, 1, 0);
                    103:     }
                    104:   }
                    105:   
                    106: copy_cleanup:
                    107:   sqliteSrcListDelete(pTableName);
                    108:   sqliteFree(zFile);
                    109:   return;
                    110: }