blockinserter.php

См. документацию.

<?
 
namespace Bitrix\Sale\Location\DB;
 
use Bitrix\Main;
use Bitrix\Sale\Location\DB\Helper;
 
class BlockInserter
{
    protected $tableName =        '';
    protected $tableMap =      array();
    protected $fldVector =        array();
 
    protected $insertHead =  '';
    protected $insertTail =  '';
    protected $index =            0;
    protected $bufferSize =  0;
    protected $buffer =      '';
    protected $map =            array();
 
    protected $autoIncFld =  false;
    protected $dbType =      false;
    protected $mtu =            0;
 
    protected $dbConnection =  null;
    protected $dbHelper =      null;
 
    protected $callbacks =        array();
 
    const RED_LINE =            100;
 
    public function __construct($parameters = array())
    {
        $this->dbConnection = Main\HttpApplication::getConnection();
        $this->dbHelper = $this->dbConnection->getSqlHelper();
 
        $map = array();
        $entityName = trim((string)($parameters['entityName'] ?? ''));
        $tableName = trim((string)($parameters['tableName'] ?? ''));
        if ($entityName !== '' && is_a($entityName, Main\ORM\Data\DataManager::class, true))
        {
            $table = $entityName;

            $this->tableName = $table::getTableName();
            $this->tableMap = $table::getMap();
 
            // filter map throught $parameters['exactFields']
            if (!empty($parameters['exactFields']) && is_array($parameters['exactFields']))
            {
                foreach($parameters['exactFields'] as $fld)
                {
                    if(!isset($this->tableMap[$fld]))
                    {
                        throw new Main\SystemException('Field does not exist in ORM class, but present in "exactFields" parameter: '.$fld, 0, __FILE__, __LINE__);
                    }
 
                    $map[] = $fld;
                    $this->fldVector[$fld] = true;
                }
            }
            else
            {
                foreach($this->tableMap as $fld => $params)
                {
                    $map[] = $fld;
                    $this->fldVector[$fld] = true;
                }
            }
        }
        elseif ($tableName !== '')
        {
            $this->tableName = $this->dbHelper->forSql($parameters['tableName']);
            $this->tableMap = [];
            if (!empty($parameters['exactFields']) && is_array($parameters['exactFields']))
            {
                $this->tableMap = $parameters['exactFields'];
            }
 
            // $this->tableMap as $fld => $params - is the right way!
            /*
            required for
 
                $loc2site = new DB\BlockInserter(array(
                    'tableName' => 'b_sale_loc_2site',
                    'exactFields' => array(
                        'LOCATION_ID' => array('data_type' => 'integer'),
                        'SITE_ID' => array('data_type' => 'string')
                    ),
                    'parameters' => array(
                        'mtu' => 9999,
                        'autoIncrementFld' => 'ID'
                    )
                ));
            */
            foreach($this->tableMap as $fld => $params)
            {
                $map[] = $fld;
                $this->fldVector[$fld] = true;
            }
        }
 
        // automatically insert to this field an auto-increment value
        // beware of TransactSQL`s IDENTITY_INSERT when setting autoIncrementFld to a database-driven auto-increment field
        $autoIncrementFld = trim((string)($parameters['parameters']['autoIncrementFld'] ?? ''));
        if ($autoIncrementFld !== '')
        {
            $this->autoIncFld = $autoIncrementFld;
            if (!isset($this->fldVector[$this->autoIncFld]))
            {
                $map[] = $this->autoIncFld;
                $this->fldVector[$this->autoIncFld] = true;
                $this->tableMap[$this->autoIncFld] = [
                    'data_type' => 'integer',
                ];
            }
 
            $this->initIndexFromField();
        }
 
        $this->dbType = Main\HttpApplication::getConnection()->getType();
 
        $this->mtu = (int)($parameters['parameters']['mtu'] ?? 0);
        if ($this->mtu <= 0)
        {
            $this->mtu = 9999;
        }
 
        $this->mtu = min($this->mtu, (int)Helper::getMaxTransferUnit());
 
        $this->insertHead = Helper::getBatchInsertHead($this->tableName, $map);
        $this->insertTail = Helper::getBatchInsertTail();
 
        if (
            isset($parameters['parameters']['CALLBACKS']['ON_BEFORE_FLUSH'])
            && is_callable($parameters['parameters']['CALLBACKS']['ON_BEFORE_FLUSH'])
        )
        {
            $this->callbacks['ON_BEFORE_FLUSH'] = $parameters['parameters']['CALLBACKS']['ON_BEFORE_FLUSH'];
        }
 
        $this->map = $map;
    }
 
    // this method is buggy when table is empty
    public function initIndexFromField($fld = 'ID')
    {
        if($fld == '')
            throw new Main\SystemException('Field is not set');
 
        $fld = $this->dbHelper->forSql($fld);
 
        $sql = 'select MAX('.$fld.') as VAL from '.$this->tableName;
 
        $res = $this->dbConnection->query($sql)->fetch();
        $this->index = intval($res['VAL']);
 
        /*
        $sql = 'select '.$fld.' from '.$this->tableName.' order by '.$fld.' desc';
        $sql = $this->dbHelper->getTopSql($sql, 1);
 
        $res = $this->dbConnection->query($sql)->fetch();
        $this->index = intval($res[$this->autoIncFld]);
        */
 
        return $this->index;
    }
 
    public function getIndex()
    {
        return $this->index;
    }
 
    public function dropAutoIncrementRestrictions()
    {
        if($this->autoIncFld === false)
            return false;
 
        return Helper::dropAutoIncrementRestrictions($this->tableName);
    }
 
    public function restoreAutoIncrementRestrictions()
    {
        if($this->autoIncFld === false)
            return false;
 
        return Helper::restoreAutoIncrementRestrictions($this->tableName);
    }
 
    public function resetAutoIncrementFromIndex()
    {
        Helper::resetAutoIncrement($this->tableName, $this->getIndex() + 1);
    }
 
    public function insert($row)
    {
        if(!is_array($row) || empty($row))
            return;
 
        $this->index++;
        $this->bufferSize++;
 
        if($this->autoIncFld !== false)
        {
            $row[$this->autoIncFld] = $this->index;
            Helper::incrementSequenceForTable($this->tableName); // if this is oracle and we insert auto increment key directly, we must provide sequence increment manually
        }
 
        $sql = Helper::getBatchInsertValues($row, $this->tableName, $this->fldVector, $this->map);
 
        /*
        MySQL & MsSQL: insert into b_test (F1,F2) values ('one','two'),('one1','two1'),('one2','two2')
        Oracle: insert all into b_test (F1,F2) values ('one','two') into b_test (F1,F2) values ('one1','two1') into b_test (F1,F2) values ('one2','two2')  select * from dual
        */
 
        $nextBuffer = (empty($this->buffer) ? $this->insertHead : $this->buffer.Helper::getBatchInsertSeparator()).$sql;
 
        if(($this->mtu - (mb_strlen($nextBuffer) + 100)) < self::RED_LINE)
        {
            $this->flush(); // flushing the previous buffer (now $this->buffer == '')
            $this->buffer = $this->insertHead.$sql;
        }
        else
            $this->buffer = $nextBuffer;
 
        return $this->index;
    }
 
    public function flush()
    {
        if($this->buffer == '')
            return;
 
        if(isset($this->callbacks['ON_BEFORE_FLUSH']))
            call_user_func($this->callbacks['ON_BEFORE_FLUSH']);
 
        $this->buffer .= ' '.$this->insertTail;
 
        if ($this->dbType === 'pgsql' && $this->autoIncFld)
        {
            $this->buffer .= '; ALTER SEQUENCE ' . $this->tableName . '_' . $this->autoIncFld . '_seq RESTART WITH ' . $this->index + 1 . ';';
        }
 
        $restrDropped = $this->dropAutoIncrementRestrictions();
 
        Main\HttpApplication::getConnection()->query($this->buffer);
 
        if($restrDropped)
            $this->restoreAutoIncrementRestrictions();
 
        $this->buffer = '';
        $this->bufferSize = 0;
    }
}