Method internalDsqlPass

src/dsql/StmtNodes.cpp:6559–6740

StmtNode* ModifyNode::internalDsqlPass(DsqlCompilerScratch* dsqlScratch, bool updateOrInsert)
{
	auto& pool = dsqlScratch->getPool();

	// Separate old and new context references.

	Array<NestConst<ValueExprNode>> orgValues, newValues;

	const auto assignments = nodeAs<CompoundStmtNode>(statement);
	fb_assert(assignments);

	for (FB_SIZE_T i = 0; i < assignments->statements.getCount(); ++i)
	{
		AssignmentNode* const assign = nodeAs<AssignmentNode>(assignments->statements[i]);
		fb_assert(assign);
		orgValues.add(assign->asgnFrom);
		newValues.add(assign->asgnTo);
	}

	NestConst<RelationSourceNode> relation = nodeAs<RelationSourceNode>(dsqlRelation);
	fb_assert(relation);

	NestConst<ValueExprNode>* ptr;

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

	if (dsqlReturning && !dsqlScratch->isPsql() && dsqlCursorName.isEmpty())
	{
		node->dsqlReturningLocalTableNumber = updateOrInsert ?
			dsqlReturningLocalTableNumber.value :
			dsqlScratch->localTableNumber++;
	}

	node->dsqlCursorName = dsqlCursorName;
	node->dsqlSkipLocked = dsqlSkipLocked;

	if (dsqlCursorName.hasData() && dsqlScratch->isPsql())
	{
		node->dsqlContext = dsqlPassCursorContext(dsqlScratch, dsqlCursorName, relation);
		node->marks |= StmtNode::MARK_POSITIONED;

		// Process old context values.
		dsqlScratch->context->push(node->dsqlContext);
		++dsqlScratch->scopeLevel;

		for (ptr = orgValues.begin(); ptr != orgValues.end(); ++ptr)
			*ptr = doDsqlPass(dsqlScratch, *ptr, false);

		--dsqlScratch->scopeLevel;
		dsqlScratch->context->pop();

		// Process relation.
		doDsqlPass(dsqlScratch, node->dsqlRelation, relation, false);

		// Process new context values.
		for (ptr = newValues.begin(); ptr != newValues.end(); ++ptr)
			*ptr = doDsqlPass(dsqlScratch, *ptr, false);