mirror of
https://github.com/sqlite/sqlite.git
synced 2025-11-11 01:42:22 +03:00
:-) (CVS 1697)
FossilOrigin-Name: 5d773b5d4e9c23b81f53afd0cef7bd2300fff329
This commit is contained in:
drh
142
src/insert.c
142
src/insert.c
@@ -24,7 +24,7 @@
|
||||
** This file contains C code routines that are called by the parser
|
||||
** to handle INSERT statements.
|
||||
**
|
||||
** $Id: insert.c,v 1.6 2000/06/05 18:54:46 drh Exp $
|
||||
** $Id: insert.c,v 1.7 2000/06/07 14:42:27 drh Exp $
|
||||
*/
|
||||
#include "sqliteInt.h"
|
||||
|
||||
@@ -39,12 +39,18 @@ void sqliteInsert(
|
||||
Parse *pParse, /* Parser context */
|
||||
Token *pTableName, /* Name of table into which we are inserting */
|
||||
ExprList *pList, /* List of values to be inserted */
|
||||
Select *pSelect, /* A SELECT statement to use as the data source */
|
||||
IdList *pField /* Field name corresponding to pList. Might be NULL */
|
||||
){
|
||||
Table *pTab;
|
||||
char *zTab;
|
||||
int i, j, idx;
|
||||
Vdbe *v;
|
||||
Table *pTab; /* The table to insert into */
|
||||
char *zTab; /* Name of the table into which we are inserting */
|
||||
int i, j, idx; /* Loop counters */
|
||||
Vdbe *v; /* Generate code into this virtual machine */
|
||||
Index *pIdx; /* For looping over indices of the table */
|
||||
int srcTab; /* Date comes from this temporary cursor if >=0 */
|
||||
int nField; /* Number of columns in the data */
|
||||
int base; /* First available cursor */
|
||||
int iCont, iBreak; /* Beginning and end of the loop over srcTab */
|
||||
|
||||
zTab = sqliteTableNameFromToken(pTableName);
|
||||
pTab = sqliteFindTable(pParse->db, zTab);
|
||||
@@ -61,10 +67,28 @@ void sqliteInsert(
|
||||
pParse->nErr++;
|
||||
goto insert_cleanup;
|
||||
}
|
||||
if( pField==0 && pList->nExpr!=pTab->nCol ){
|
||||
v = pParse->pVdbe;
|
||||
if( v==0 ){
|
||||
v = pParse->pVdbe = sqliteVdbeCreate(pParse->db->pBe);
|
||||
}
|
||||
if( v==0 ) goto insert_cleanup;
|
||||
if( pSelect ){
|
||||
int rc;
|
||||
srcTab = pParse->nTab++;
|
||||
sqliteVdbeAddOp(v, OP_Open, srcTab, 1, 0, 0);
|
||||
rc = sqliteSelect(pParse, pSelect, SRT_Table, srcTab);
|
||||
if( rc ) goto insert_cleanup;
|
||||
assert( pSelect->pEList );
|
||||
nField = pSelect->pEList->nExpr;
|
||||
}else{
|
||||
srcTab = -1;
|
||||
assert( pList );
|
||||
nField = pList->nExpr;
|
||||
}
|
||||
if( pField==0 && nField!=pTab->nCol ){
|
||||
char zNum1[30];
|
||||
char zNum2[30];
|
||||
sprintf(zNum1,"%d", pList->nExpr);
|
||||
sprintf(zNum1,"%d", nField);
|
||||
sprintf(zNum2,"%d", pTab->nCol);
|
||||
sqliteSetString(&pParse->zErrMsg, "table ", pTab->zName,
|
||||
" has ", zNum2, " columns but ",
|
||||
@@ -72,10 +96,10 @@ void sqliteInsert(
|
||||
pParse->nErr++;
|
||||
goto insert_cleanup;
|
||||
}
|
||||
if( pField!=0 && pList->nExpr!=pField->nId ){
|
||||
if( pField!=0 && nField!=pField->nId ){
|
||||
char zNum1[30];
|
||||
char zNum2[30];
|
||||
sprintf(zNum1,"%d", pList->nExpr);
|
||||
sprintf(zNum1,"%d", nField);
|
||||
sprintf(zNum2,"%d", pField->nId);
|
||||
sqliteSetString(&pParse->zErrMsg, zNum1, " values for ",
|
||||
zNum2, " columns", 0);
|
||||
@@ -101,73 +125,81 @@ void sqliteInsert(
|
||||
}
|
||||
}
|
||||
}
|
||||
v = pParse->pVdbe;
|
||||
if( v==0 ){
|
||||
v = pParse->pVdbe = sqliteVdbeCreate(pParse->db->pBe);
|
||||
base = pParse->nTab;
|
||||
sqliteVdbeAddOp(v, OP_Open, base, 1, pTab->zName, 0);
|
||||
for(idx=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, idx++){
|
||||
sqliteVdbeAddOp(v, OP_Open, idx+base, 1, pIdx->zName, 0);
|
||||
}
|
||||
if( v ){
|
||||
Index *pIdx;
|
||||
sqliteVdbeAddOp(v, OP_Open, 0, 1, pTab->zName, 0);
|
||||
for(idx=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, idx++){
|
||||
sqliteVdbeAddOp(v, OP_Open, idx, 1, pIdx->zName, 0);
|
||||
if( srcTab>=0 ){
|
||||
sqliteVdbeAddOp(v, OP_Rewind, srcTab, 0, 0, 0);
|
||||
iBreak = sqliteVdbeMakeLabel(v);
|
||||
iCont = sqliteVdbeAddOp(v, OP_Next, srcTab, iBreak, 0, 0);
|
||||
}
|
||||
sqliteVdbeAddOp(v, OP_New, 0, 0, 0, 0);
|
||||
if( pTab->pIndex ){
|
||||
sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
|
||||
}
|
||||
for(i=0; i<pTab->nCol; i++){
|
||||
if( pField==0 ){
|
||||
j = i;
|
||||
}else{
|
||||
for(j=0; j<pField->nId; j++){
|
||||
if( pField->a[j].idx==i ) break;
|
||||
}
|
||||
}
|
||||
sqliteVdbeAddOp(v, OP_New, 0, 0, 0, 0);
|
||||
if( pTab->pIndex ){
|
||||
if( pField && j>=pField->nId ){
|
||||
char *zDflt = pTab->aCol[i].zDflt;
|
||||
if( zDflt==0 ){
|
||||
sqliteVdbeAddOp(v, OP_Null, 0, 0, 0, 0);
|
||||
}else{
|
||||
sqliteVdbeAddOp(v, OP_String, 0, 0, zDflt, 0);
|
||||
}
|
||||
}else if( srcTab>=0 ){
|
||||
sqliteVdbeAddOp(v, OP_Field, srcTab, i, 0, 0);
|
||||
}else{
|
||||
sqliteExprCode(pParse, pList->a[j].pExpr);
|
||||
}
|
||||
}
|
||||
sqliteVdbeAddOp(v, OP_MakeRecord, pTab->nCol, 0, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_Put, base, 0, 0, 0);
|
||||
/* sqliteVdbeAddOp(v, OP_Close, 0, 0, 0, 0); */
|
||||
for(idx=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, idx++){
|
||||
if( pIdx->pNext ){
|
||||
sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
|
||||
}
|
||||
for(i=0; i<pTab->nCol; i++){
|
||||
for(i=0; i<pIdx->nField; i++){
|
||||
int idx = pIdx->aiField[i];
|
||||
if( pField==0 ){
|
||||
j = i;
|
||||
j = idx;
|
||||
}else{
|
||||
for(j=0; j<pField->nId; j++){
|
||||
if( pField->a[j].idx==i ) break;
|
||||
if( pField->a[j].idx==idx ) break;
|
||||
}
|
||||
}
|
||||
if( pField && j>=pField->nId ){
|
||||
char *zDflt = pTab->aCol[i].zDflt;
|
||||
char *zDflt = pTab->aCol[idx].zDflt;
|
||||
if( zDflt==0 ){
|
||||
sqliteVdbeAddOp(v, OP_Null, 0, 0, 0, 0);
|
||||
}else{
|
||||
sqliteVdbeAddOp(v, OP_String, 0, 0, zDflt, 0);
|
||||
}
|
||||
}else if( srcTab>=0 ){
|
||||
sqliteVdbeAddOp(v, OP_Field, srcTab, i, 0, 0);
|
||||
}else{
|
||||
sqliteExprCode(pParse, pList->a[j].pExpr);
|
||||
}
|
||||
}
|
||||
sqliteVdbeAddOp(v, OP_MakeRecord, pTab->nCol, 0, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_Put, 0, 0, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_Close, 0, 0, 0, 0);
|
||||
for(idx=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, idx++){
|
||||
if( pIdx->pNext ){
|
||||
sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
|
||||
}
|
||||
for(i=0; i<pIdx->nField; i++){
|
||||
int idx = pIdx->aiField[i];
|
||||
if( pField==0 ){
|
||||
j = idx;
|
||||
}else{
|
||||
for(j=0; j<pField->nId; j++){
|
||||
if( pField->a[j].idx==idx ) break;
|
||||
}
|
||||
}
|
||||
if( pField && j>=pField->nId ){
|
||||
char *zDflt = pTab->aCol[idx].zDflt;
|
||||
if( zDflt==0 ){
|
||||
sqliteVdbeAddOp(v, OP_Null, 0, 0, 0, 0);
|
||||
}else{
|
||||
sqliteVdbeAddOp(v, OP_String, 0, 0, zDflt, 0);
|
||||
}
|
||||
}else{
|
||||
sqliteExprCode(pParse, pList->a[j].pExpr);
|
||||
}
|
||||
}
|
||||
sqliteVdbeAddOp(v, OP_MakeKey, pIdx->nField, 0, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_PutIdx, idx, 0, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_Close, idx, 0, 0, 0);
|
||||
}
|
||||
sqliteVdbeAddOp(v, OP_MakeKey, pIdx->nField, 0, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_PutIdx, idx+base, 0, 0, 0);
|
||||
/* sqliteVdbeAddOp(v, OP_Close, idx, 0, 0, 0); */
|
||||
}
|
||||
if( srcTab>=0 ){
|
||||
sqliteVdbeAddOp(v, OP_Goto, 0, iCont, 0, 0);
|
||||
sqliteVdbeAddOp(v, OP_Noop, 0, 0, 0, iBreak);
|
||||
}
|
||||
|
||||
insert_cleanup:
|
||||
sqliteExprListDelete(pList);
|
||||
if( pList ) sqliteExprListDelete(pList);
|
||||
if( pSelect ) sqliteSelectDelete(pSelect);
|
||||
sqliteIdListDelete(pField);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user