fix test cases

babelfish-for-postgresql · Nov 10, 2023 · 402f81e · 402f81e
1 parent 23f97a7
commit 402f81e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/src/backend/commands/functioncmds.c b/src/backend/commands/functioncmds.c
@@ -2489,6 +2489,7 @@ ExecuteCallStmt(CallStmt *stmt, ParamListInfo params, bool atomic, DestReceiver
 		rsinfo.isDone = ExprSingleResult;
 		rsinfo.setResult = NULL;
 		rsinfo.setDesc = NULL;
+		rsinfo.insertExecStmt = NULL;
 	}
 
 	/* Get rid of temporary snapshot for arguments, if we made one */

diff --git a/src/backend/parser/analyze.c b/src/backend/parser/analyze.c
@@ -706,21 +706,17 @@ transformInsertStmt(ParseState *pstate, InsertStmt *stmt)
 					((CallStmt *) stmt->execStmt)->retdesc = NULL;
 					((CallStmt *) stmt->execStmt)->dest = NULL;
 					qry->utilityStmt = callStmtQry->utilityStmt;
-					oldContext = MemoryContextSwitchTo(TopMemoryContext);
-					((CallStmt *) stmt->execStmt)->exec_insert_icolumns = copyObject(icolumns);
-					((CallStmt *) stmt->execStmt)->exec_insert_stmt_cols = copyObject(stmt->cols);
-					MemoryContextSwitchTo(oldContext);
+					((CallStmt *) stmt->execStmt)->exec_insert_icolumns = icolumns;
+					((CallStmt *) stmt->execStmt)->exec_insert_stmt_cols = stmt->cols;
 					((CallStmt *) stmt->execStmt)->rd_length = list_length(pstate->p_rtable);
 				}
 				break;
 			case T_DoStmt:
 				((DoStmt *) stmt->execStmt)->relation = pstate->p_target_relation->rd_id;
 				((DoStmt *) stmt->execStmt)->attrnos = attrnos;
 				qry->utilityStmt = stmt->execStmt;
-                oldContext = MemoryContextSwitchTo(TopMemoryContext);
-                ((DoStmt *) stmt->execStmt)->exec_insert_icolumns = copyObject(icolumns);
-                ((DoStmt *) stmt->execStmt)->exec_insert_stmt_cols = copyObject(stmt->cols);
-                MemoryContextSwitchTo(oldContext);
+                ((DoStmt *) stmt->execStmt)->exec_insert_icolumns = icolumns;
+                ((DoStmt *) stmt->execStmt)->exec_insert_stmt_cols = stmt->cols;
                 ((DoStmt *) stmt->execStmt)->rd_length = list_length(pstate->p_rtable);
 				break;
 			default: