function Insert::execute

Overrides Insert::execute

File

core/modules/sqlite/src/Driver/Database/sqlite/Insert.php, line 30

Class

Insert

SQLite implementation of \Drupal\Core\Database\Query\Insert.

Namespace

Code

public function execute() {
    if (!$this->preExecute()) {
        return NULL;
    }
    // If we're selecting from a SelectQuery, finish building the query and
    // pass it back, as any remaining options are irrelevant.
    if (!empty($this->fromQuery)) {
        // The SelectQuery may contain arguments, load and pass them through.
        return $this->connection
            ->query((string) $this, $this->fromQuery
            ->getArguments(), $this->queryOptions);
    }
    // If there are any fields in the query, execute normal INSERT statements.
    if (count($this->insertFields)) {
        $stmt = $this->connection
            ->prepareStatement((string) $this, $this->queryOptions);
        if (count($this->insertValues) === 1) {
            // Inserting a single row does not require a transaction to be atomic,
            // and executes faster without a transaction wrapper.
            $insert_values = $this->insertValues[0];
            try {
                $stmt->execute($insert_values, $this->queryOptions);
            } catch (\Exception $e) {
                $this->connection
                    ->exceptionHandler()
                    ->handleExecutionException($e, $stmt, $insert_values, $this->queryOptions);
            }
        }
        else {
            // Inserting multiple rows requires a transaction to be atomic, and
            // executes faster as a single transaction.
            try {
                $transaction = $this->connection
                    ->startTransaction();
            } catch (\PDOException $e) {
                // $this->connection->exceptionHandler()->handleExecutionException()
                // requires a $statement argument, so we cannot use that.
                throw new DatabaseExceptionWrapper($e->getMessage(), 0, $e);
            }
            foreach ($this->insertValues as $insert_values) {
                try {
                    $stmt->execute($insert_values, $this->queryOptions);
                } catch (\Exception $e) {
                    // One of the INSERTs failed, rollback the whole batch.
                    $transaction->rollBack();
                    $this->connection
                        ->exceptionHandler()
                        ->handleExecutionException($e, $stmt, $insert_values, $this->queryOptions);
                }
            }
        }
        // Re-initialize the values array so that we can re-use this query.
        $this->insertValues = [];
    }
    else {
        $stmt = $this->connection
            ->prepareStatement("INSERT INTO {{$this->table}} DEFAULT VALUES", $this->queryOptions);
        try {
            $stmt->execute(NULL, $this->queryOptions);
        } catch (\Exception $e) {
            $this->connection
                ->exceptionHandler()
                ->handleExecutionException($e, $stmt, [], $this->queryOptions);
        }
    }
    return $this->connection
        ->lastInsertId();
}