Plasma\Drivers\MySQL\Commands\StatementPrepareCommand

Complexity

Total Complexity

21

Size/Duplication

Total Lines	162
Duplicated Lines	0 %

Test Coverage

Coverage

82%

Importance

Changes	1
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	getEncodedMessage()	0	2	1
A	__construct()	0	9	1
A	resetSequence()	0	2	1
A	createResolve()	0	9	1
C	onNext()	0	49	17

<?php
/**
 * Plasma Driver MySQL component
 * Copyright 2018-2019 PlasmaPHP, All Rights Reserved
 *
 * Website: https://github.com/PlasmaPHP
 * License: https://github.com/PlasmaPHP/driver-mysql/blob/master/LICENSE
*/

namespace Plasma\Drivers\MySQL\Commands;

/**
 * Statement Prepare command.
 * @internal
 */
class StatementPrepareCommand extends PromiseCommand {
    /**
     * The identifier for this command.
     * @var int
     * @source
     */
    const COMMAND_ID = 0x16;
    
    /**
     * @var \Plasma\ClientInterface
     */
    protected $client;
    
    /**
     * @var string
     */
    protected $query;
    
    /**
     * @var string
     */
    protected $rewrittenQuery;
    
    /**
     * @var array
     */
    protected $rewrittenParams;
    
    /**
     * @var \Plasma\Drivers\MySQL\Messages\PrepareStatementOkMessage
     */
    protected $okResponse;
    
    /**
     * @var \Plasma\ColumnDefinitionInterface[]
     */
    protected $params = array();
    
    /**
     * @var \Plasma\ColumnDefinitionInterface[]
     */
    protected $fields = array();
    
    /**
     * @var bool
     */
    protected $paramsDone = false;
    
    /**
     * @var \Plasma\StatementInterface|null
     */
    protected $resolveValue;
    
    /**
     * @var bool
     */
    protected $deprecatedEOF;
    
    /**
     * Constructor.
     * @param \Plasma\ClientInterface  $client
     * @param \Plasma\DriverInterface  $driver
     * @param string                   $query
     */
    function __construct(\Plasma\ClientInterface $client, \Plasma\DriverInterface $driver, string $query) {
        parent::__construct($driver);
        
        $this->client = $client;
        $this->driver = $driver;
        $this->query = $query;
        $this->deprecatedEOF = (($driver->getHandshake()->capability & \Plasma\Drivers\MySQL\CapabilityFlags::CLIENT_DEPRECATE_EOF) !== 0);
        
        [ 'query' => $this->rewrittenQuery, 'parameters' => $this->rewrittenParams ] = \Plasma\Utility::parseParameters($this->query, '?');
    }
    
    /**
     * Get the encoded message for writing to the database connection.
     * @return string
     */
    function getEncodedMessage(): string {
        return \chr(static::COMMAND_ID).$this->rewrittenQuery;
    }
    
    /**
     * Sends the next received value into the command.
     * @param mixed  $value
     * @return void
     * @throws \Plasma\Drivers\MySQL\Messages\ParseException
     */
    function onNext($value): void {
        if($value instanceof \Plasma\Drivers\MySQL\Messages\PrepareStatementOkMessage) {
            $this->okResponse = $value;
            $this->fieldsCount = $this->okResponse->numColumns;
            $this->paramsDone = ($this->okResponse->numParams === 0);
            
            if($this->paramsDone && $this->fieldsCount === 0) {
                $this->createResolve();
            }
        } elseif($value instanceof \Plasma\Drivers\MySQL\ProtocolOnNextCaller) {
            $buffer = $value->getBuffer();
            $parser = $value->getParser();
            
            $parsed = $this->handleQueryOnNextCallerColumns($buffer, $parser);
            
            if($this->paramsDone) {
                $this->fields[$parsed->getName()] = $parsed;
            } else {
                $this->params[] = $parsed;
            }
            
            if($this->deprecatedEOF) {
                if($this->paramsDone && $this->fieldsCount <= \count($this->fields)) {
                    $this->createResolve();
                } elseif($this->okResponse->numParams <= \count($this->params) && $this->resolveValue === null) {
                    $this->paramsDone = true;
                    
                    if($this->fieldsCount === 0) {
                        $this->createResolve();
                    }
                }
            }
        } elseif(
            $value instanceof \Plasma\Drivers\MySQL\Messages\EOFMessage || $value instanceof \Plasma\Drivers\MySQL\Messages\OkResponseMessage
        ) {
            if(!$this->paramsDone) {
                $this->paramsDone = true;
                
                if($this->fieldsCount === 0) {
                    $this->createResolve();
                }
                
                return;
            }
            
            $this->createResolve();
        } else {
            throw new \Plasma\Drivers\MySQL\Messages\ParseException('Command received value of type '
                .(\is_object($value) ? \get_class($value) : \gettype($value)).' it can not handle');
        }
    }
    
    /**
     * Whether the sequence ID should be resetted.
     * @return bool
     */
    function resetSequence(): bool {
        return true;
    }
    
    /**
     * Creates the resolve value.
     * @return void
     */
    protected function createResolve(): void {
        $this->finished = true;
        
        $id = $this->okResponse->statementID;
        $queryr = $this->rewrittenQuery;
        $paramsr = $this->rewrittenParams;
        
        $this->resolveValue = new \Plasma\Drivers\MySQL\Statement($this->client, $this->driver, $id, $this->query, $queryr, $paramsr, $this->params, $this->fields);
        $this->deferred->resolve($this->resolveValue);
    }
}