Final Class Yiisoft\Db\Mssql\DMLQueryBuilder

                public function insertReturningPks(string $table, array|QueryInterface $columns, array &$params = []): string
{
    $primaryKeys = $this->schema->getTableSchema($table)?->getPrimaryKey() ?? [];
    return $this->insertReturning($table, $columns, $primaryKeys, $params);
}

            

                public function resetSequence(string $table, int|string|null $value = null): string
{
    $tableSchema = $this->schema->getTableSchema($table);
    if ($tableSchema === null) {
        throw new InvalidArgumentException("Table not found: '$table'.");
    }
    $sequenceName = $tableSchema->getSequenceName();
    if ($sequenceName === null) {
        throw new InvalidArgumentException("There is not sequence associated with table '$table'.'");
    }
    $tableName = $this->quoter->quoteTableName($table);
    if ($value === null) {
        return "DBCC CHECKIDENT ('$tableName', RESEED, 0) WITH NO_INFOMSGS;DBCC CHECKIDENT ('$tableName', RESEED)";
    }
    return "DBCC CHECKIDENT ('$tableName', RESEED, $value)";
}

            

                public function update(
    string $table,
    array $columns,
    array|string|ExpressionInterface $condition,
    array|string|ExpressionInterface|null $from = null,
    array &$params = [],
): string {
    if ($from !== null) {
        throw new NotSupportedException(__METHOD__ . '() does not support UPDATE with FROM clause with SQL Server.');
    }
    return parent::update($table, $columns, $condition, null, $params);
}

            

                public function upsert(
    string $table,
    array|QueryInterface $insertColumns,
    array|bool $updateColumns = true,
    array &$params = [],
): string {
    return implode('', $this->prepareUpsertParts($table, $insertColumns, $updateColumns, $params)) . ';';
}

            

                public function upsertReturning(
    string $table,
    array|QueryInterface $insertColumns,
    array|bool $updateColumns = true,
    ?array $returnColumns = null,
    array &$params = [],
): string {
    [$uniqueNames] = $this->prepareUpsertColumns($table, $insertColumns, $updateColumns);
    if (empty($uniqueNames)) {
        return $this->insertReturning($table, $insertColumns, $returnColumns, $params);
    }
    $tableSchema = $this->schema->getTableSchema($table);
    $returnColumns ??= $tableSchema?->getColumnNames();
    if (empty($returnColumns)) {
        return $this->upsert($table, $insertColumns, $updateColumns, $params);
    }
    /** @var TableSchema $tableSchema */
    [$declareSql, $outputSql, $selectSql] = $this->prepareReturningParts($tableSchema, $returnColumns);
    [$mergeSql, $updateSql, $insertSql] = $this->prepareUpsertParts($table, $insertColumns, $updateColumns, $params);
    return $declareSql
        . (!empty($insertSql) && empty($updateSql) ? 'DECLARE @temp int;' : '')
        . $mergeSql
        . (!empty($insertSql) && empty($updateSql) ? ' WHEN MATCHED THEN UPDATE SET @temp=1' : $updateSql)
        . $insertSql
        . $outputSql . ';'
        . $selectSql;
}