Function dsqlProcessReturning

src/dsql/StmtNodes.cpp:10372–10507

Compile a RETURNING clause.

// Compile a RETURNING clause.
static ReturningClause* dsqlProcessReturning(DsqlCompilerScratch* dsqlScratch, dsql_rel* relation,
	ReturningClause* input, bool singleton)
{
	if (!input)
		return nullptr;

	auto& pool = dsqlScratch->getPool();

	auto inputFirst = input->first;

	if (inputFirst->items.isEmpty())
	{
		// Process RETURNING *
		inputFirst = FB_NEW_POOL(pool) ValueListNode(pool, 0u);
		dsqlExplodeFields(relation, inputFirst->items, true);
	}
	else
	{
		// Process alias.* items.
		inputFirst = PASS1_expand_select_list(dsqlScratch, inputFirst, nullptr);
	}

	const auto node = FB_NEW_POOL(pool) ReturningClause(pool);

	node->first = Node::doDsqlPass(dsqlScratch, inputFirst, false);

	dsqlScratch->flags |= DsqlCompilerScratch::FLAG_RETURNING_INTO;
	node->second = dsqlPassArray(dsqlScratch,
		dsqlScratch->returningClause ? dsqlScratch->returningClause->second : input->second);
	dsqlScratch->flags &= ~DsqlCompilerScratch::FLAG_RETURNING_INTO;

	if (!dsqlScratch->isPsql() && input->second)
	{
		// RETURNING INTO is not allowed syntax for DSQL
		ERRD_post(Arg::Gds(isc_sqlerr) << Arg::Num(-104) <<
				  // Token unknown
				  Arg::Gds(isc_token_err) <<
				  Arg::Gds(isc_random) << Arg::Str("INTO"));
	}
	else if (dsqlScratch->isPsql() && !input->second)
	{
		// This trick because we don't copy lexer positions when copying lists.
		const ValueListNode* errSrc = input->first;
		// RETURNING without INTO is not allowed for PSQL
		ERRD_post(Arg::Gds(isc_sqlerr) << Arg::Num(-104) <<
				  // Unexpected end of command
				  Arg::Gds(isc_command_end_err2) << Arg::Num(errSrc->line) <<
													Arg::Num(errSrc->column));
	}

	const unsigned count = node->first->items.getCount();
	fb_assert(count);

	if (input->second)
	{
		fb_assert(dsqlScratch->isPsql());

		if (count != node->second->items.getCount())