Number of tuples inserted/affected by INSERT/UPDATE/DELETE...

src/backend/executor/execMain.c

@@ -26,7 +26,7 @@
  *
  *
  * IDENTIFICATION
- *    $Header: /cvsroot/pgsql/src/backend/executor/execMain.c,v 1.19 1997/08/22 14:28:20 vadim Exp $
+ *    $Header: /cvsroot/pgsql/src/backend/executor/execMain.c,v 1.20 1997/08/27 09:02:24 vadim Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -65,7 +65,7 @@ static TupleTableSlot *ExecutePlan(EState *estate, Plan *plan,
 				   int numberTuples, ScanDirection direction,
 				   void (*printfunc)());
 static void ExecRetrieve(TupleTableSlot *slot, void (*printfunc)(),
-			 Relation intoRelationDesc);
+			 EState *estate);
 static void ExecAppend(TupleTableSlot *slot,ItemPointer tupleid,
 		       EState *estate);
 static void ExecDelete(TupleTableSlot *slot, ItemPointer tupleid,
@@ -171,6 +171,8 @@ ExecutorRun(QueryDesc *queryDesc, EState *estate, int feature, int count)
     plan =	  queryDesc->plantree;
     dest =	  queryDesc->dest;
     destination = (void (*)()) DestToFunction(dest);
+    estate->es_processed = 0;
+    estate->es_lastoid = InvalidOid;
 
 #if 0
     /*
@@ -665,7 +667,6 @@ ExecutePlan(EState *estate,
 	    ScanDirection direction,
 	    void (*printfunc)())
 {
-    Relation		intoRelationDesc;
     JunkFilter		*junkfilter;
 
     TupleTableSlot	*slot;
@@ -674,12 +675,6 @@ ExecutePlan(EState *estate,
     int	 		current_tuple_count;
     TupleTableSlot	*result;	
 
-    /* ----------------
-     *  get information
-     * ----------------
-     */
-    intoRelationDesc =	estate->es_into_relation_descriptor;
-
     /* ----------------
      *	initialize local variables
      * ----------------
@@ -780,9 +775,9 @@ ExecutePlan(EState *estate,
 	
 	switch(operation) {
 	case CMD_SELECT:
-	    ExecRetrieve(slot, 	  /* slot containing tuple */
-			 printfunc,	  /* print function */
-			 intoRelationDesc); /* "into" relation */
+	    ExecRetrieve(slot,		/* slot containing tuple */
+			 printfunc,	/* print function */
+			 estate);	/*  */
 	    result = slot;
 	    break;
 	    
@@ -853,7 +848,7 @@ ExecutePlan(EState *estate,
 static void
 ExecRetrieve(TupleTableSlot *slot,
 	     void (*printfunc)(),
-	     Relation intoRelationDesc)
+	     EState *estate)
 {
     HeapTuple	tuple;
     TupleDesc 	attrtype;
@@ -869,9 +864,10 @@ ExecRetrieve(TupleTableSlot *slot,
      *	insert the tuple into the "into relation"
      * ----------------
      */
-    if (intoRelationDesc != NULL) {
-      heap_insert (intoRelationDesc, tuple);
-      IncrAppended();
+    if ( estate->es_into_relation_descriptor != NULL )
+    {
+    	heap_insert (estate->es_into_relation_descriptor, tuple);
+    	IncrAppended();
     }
 
     /* ----------------
@@ -880,6 +876,7 @@ ExecRetrieve(TupleTableSlot *slot,
      */
     (*printfunc)(tuple, attrtype);
     IncrRetrieved();
+    (estate->es_processed)++;
 }
 
 /* ----------------------------------------------------------------
@@ -947,7 +944,6 @@ ExecAppend(TupleTableSlot *slot,
     newId = heap_insert(resultRelationDesc, /* relation desc */
 			tuple);		    /* heap tuple */
     IncrAppended();
-    UpdateAppendOid(newId);
 
     /* ----------------
      *	process indices
@@ -961,6 +957,8 @@ ExecAppend(TupleTableSlot *slot,
     if (numIndices > 0) {
 	ExecInsertIndexTuples(slot, &(tuple->t_ctid), estate, false);
     }
+    (estate->es_processed)++;
+    estate->es_lastoid = newId;
 }
 
 /* ----------------------------------------------------------------
@@ -989,10 +987,12 @@ ExecDelete(TupleTableSlot *slot,
      *	delete the tuple
      * ----------------
      */
-    heap_delete(resultRelationDesc, /* relation desc */
-			tupleid);	    /* item pointer to tuple */
-
+    if ( heap_delete(resultRelationDesc,	/* relation desc */
+				tupleid) )	/* item pointer to tuple */
+    	return;
+    
     IncrDeleted();
+    (estate->es_processed)++;
 
     /* ----------------
      *	Note: Normally one would think that we have to
@@ -1094,6 +1094,7 @@ ExecReplace(TupleTableSlot *slot,
     }
 
     IncrReplaced();
+    (estate->es_processed)++;
 
     /* ----------------
      *	Note: instead of having to update the old index tuples

src/backend/tcop/dest.c

@@ -7,7 +7,7 @@
  *
  *
  * IDENTIFICATION
- *    $Header: /cvsroot/pgsql/src/backend/tcop/dest.c,v 1.7 1997/08/19 21:34:02 momjian Exp $
+ *    $Header: /cvsroot/pgsql/src/backend/tcop/dest.c,v 1.8 1997/08/27 09:03:14 vadim Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -43,8 +43,7 @@
 
 #include "commands/async.h"
 
-static Oid GetAppendOid(void);
-static void ResetAppendOid(void);
+static char CommandInfo[32] = {0};
 
 /* ----------------
  *	output functions
@@ -87,8 +86,6 @@ void (*DestToFunction(CommandDest dest))(HeapTuple, TupleDesc)
     return donothing;
 }
 
-#define IS_INSERT_TAG(tag) (*tag == 'I' && *(tag+1) == 'N')
-
 /* ----------------
  * 	EndCommand - tell destination that no more tuples will arrive
  * ----------------
@@ -106,14 +103,8 @@ EndCommand(char *commandTag, CommandDest dest)
 	 * ----------------
 	 */
 	pq_putnchar("C", 1);
-/*	pq_putint(0, 4); */
-	if (IS_INSERT_TAG(commandTag))
-	    {
-		sprintf(buf, "%s %d", commandTag, GetAppendOid());
-		pq_putstr(buf);
-	    }
-	else
-	    pq_putstr(commandTag);
+	sprintf(buf, "%s%s", commandTag, CommandInfo);
+	pq_putstr(buf);
 	pq_flush();
 	break;
 	
@@ -239,7 +230,7 @@ BeginCommand(char *pname,
 	 *	because nothing needs to be sent to the fe.
 	 * ----------------
 	 */
-        ResetAppendOid();
+    	CommandInfo[0] = 0;
 	if (isIntoPortal)
 	    return;
 	
@@ -318,41 +309,22 @@ BeginCommand(char *pname,
     }
 }
 
-static Oid AppendOid;
-
-static void
-ResetAppendOid(void)
-{
-    AppendOid = InvalidOid;
-}
-
-#define MULTI_TUPLE_APPEND -1
-
 void
-UpdateAppendOid(Oid newoid)
+UpdateCommandInfo (int operation, Oid lastoid, uint32 tuples)
 {
-    /*
-     * First update after AppendOid was reset (at command beginning).
-     */
-    if (AppendOid == InvalidOid)
-	AppendOid = newoid;
-    /*
-     * Already detected a multiple tuple append, return a void oid ;)
-     */
-    else if (AppendOid == MULTI_TUPLE_APPEND)
-	return;
-    /*
-     * Oid has been assigned once before, tag this as a multiple tuple
-     * append.
-     */
-    else
-	AppendOid = MULTI_TUPLE_APPEND;
-}
-
-static Oid
-GetAppendOid(void)
-{
-    if (AppendOid == MULTI_TUPLE_APPEND)
-	return InvalidOid;
-    return AppendOid;
+    switch (operation)
+    {
+    	case CMD_INSERT :
+    	    if ( tuples > 1 )
+    	    	lastoid = InvalidOid;
+	    sprintf (CommandInfo, " %u %u", lastoid, tuples);
+	    break;
+    	case CMD_DELETE :
+    	case CMD_UPDATE :
+	    sprintf (CommandInfo, " %u", tuples);
+	    break;
+	default :
+	    CommandInfo[0] = 0;
+    }
+    return;
 }

src/backend/tcop/pquery.c

@@ -7,7 +7,7 @@
  *
  *
  * IDENTIFICATION
- *    $Header: /cvsroot/pgsql/src/backend/tcop/pquery.c,v 1.6 1997/08/19 21:34:07 momjian Exp $
+ *    $Header: /cvsroot/pgsql/src/backend/tcop/pquery.c,v 1.7 1997/08/27 09:03:15 vadim Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -322,6 +322,9 @@ ProcessQueryDesc(QueryDesc *queryDesc)
      */
     ExecutorRun(queryDesc, state, EXEC_RUN, 0);
     
+    /* save infos for EndCommand */
+    UpdateCommandInfo (operation, state->es_lastoid, state->es_processed);
+    
     /* ----------------
      *   now, we close down all the scans and free allocated resources...
      * with ExecutorEnd()