storage.php

<?php
namespace Bitrix\Iblock\PropertyIndex;
 
class Storage
{
    protected $iblockId = 0;
    protected static $exists = array();
 
    const PRICE = 1;
    const DICTIONARY = 2;
    const STRING = 3;
    const NUMERIC = 4;
    const DATETIME = 5;
 
    public function __construct($iblockId)
    {
        $this->iblockId = intval($iblockId);
    }
 
    public function getIblockId()
    {
        return $this->iblockId;
    }
 
    public function getTableName()
    {
        return "b_iblock_".$this->iblockId."_index";
    }
 
    public function isExists()
    {
        if (!array_key_exists($this->iblockId, self::$exists))
        {
            $connection = \Bitrix\Main\Application::getConnection();
            self::$exists[$this->iblockId] = $connection->isTableExists($this->getTableName());
        }
 
        return self::$exists[$this->iblockId];
    }
 
    public function create()
    {
        $connection = \Bitrix\Main\Application::getConnection();
 
        $connection->createTable($this->getTableName(), array(
            "SECTION_ID" => new \Bitrix\Main\Entity\IntegerField("SECTION_ID", array(
                'required' => true,
            )),
            "ELEMENT_ID" => new \Bitrix\Main\Entity\IntegerField("ELEMENT_ID", array(
                'required' => true,
            )),
            "FACET_ID" => new \Bitrix\Main\Entity\IntegerField("FACET_ID", array(
                'required' => true,
            )),
            "VALUE" => new \Bitrix\Main\Entity\IntegerField("VALUE", array(
                'required' => true,
            )),
            "VALUE_NUM" => new \Bitrix\Main\Entity\FloatField("VALUE_NUM", array(
                'required' => true,
            )),
            "INCLUDE_SUBSECTIONS" => new \Bitrix\Main\Entity\BooleanField("INCLUDE_SUBSECTIONS", array(
                'required' => true,
                'values' => array(0, 1),
            )),
        ), array("SECTION_ID", "FACET_ID", "VALUE", "VALUE_NUM", "ELEMENT_ID"));
 
        $connection->createIndex($this->getTableName(), 'IX_'.$this->getTableName().'_0', array("SECTION_ID", "FACET_ID", "VALUE_NUM", "VALUE", "ELEMENT_ID"));
        $connection->createIndex($this->getTableName(), 'IX_'.$this->getTableName().'_1', array("ELEMENT_ID", "SECTION_ID", "FACET_ID"));
 
        self::$exists[$this->iblockId] = true;
    }
 
    public function drop()
    {
        $connection = \Bitrix\Main\Application::getConnection();
 
        $connection->dropTable($this->getTableName());
 
        self::$exists[$this->iblockId] = false;
    }
 
    public function getLastStoredElementId()
    {
        $connection = \Bitrix\Main\Application::getConnection();
 
        $max =  $connection->queryScalar("select max(ELEMENT_ID) ELEMENT_MAX from ".$this->getTableName());
 
        return $max > 0? $max: 0;
    }
 
    public function addIndexEntry($sectionId, $elementId, $facetId, $value, $valueNum, $includeSubsections)
    {
        $connection = \Bitrix\Main\Application::getConnection();
 
        try
        {
            $connection->query("
                INSERT INTO ".$this->getTableName()." (
                    SECTION_ID
                    ,ELEMENT_ID
                    ,FACET_ID
                    ,VALUE
                    ,VALUE_NUM
                    ,INCLUDE_SUBSECTIONS
                ) VALUES (
                    ".intval($sectionId)."
                    ,".intval($elementId)."
                    ,".intval($facetId)."
                    ,".intval($value)."
                    ,".doubleval($valueNum)."
                    ,".($includeSubsections > 0? 1: 0)."
                )
            ");
        }
        catch (\Bitrix\Main\DB\SqlException $e)
        {
            return false;
        }
 
        return true;
    }
 
    protected $insertBuffer = array();
    protected $insertLength = 0;
    protected $insertMax = 1024000;
 
    public function queueIndexEntry($sectionId, $elementId, $facetId, $value, $valueNum, $includeSubsections)
    {
        $connection = \Bitrix\Main\Application::getConnection();
        $sqlHelper = $connection->getSqlHelper();
 
        $values = "("
            .intval($sectionId).","
            .intval($elementId).","
            .intval($facetId).","
            .intval($value).","
            .doubleval($valueNum).","
            .($includeSubsections > 0? 1: 0)
        .")";
        $this->insertBuffer[] = $values;
        $this->insertLength += mb_strlen($values);
        if ($this->insertLength > $this->insertMax)
        {
            return $this->flushIndexEntries();
        }
 
        return true;
    }
 
    public function flushIndexEntries()
    {
        $connection = \Bitrix\Main\Application::getConnection();
        if ($this->insertBuffer)
        {
            try
            {
                $insertQuery = "
                    INSERT INTO ".$this->getTableName()."
                    (SECTION_ID ,ELEMENT_ID ,FACET_ID ,VALUE ,VALUE_NUM ,INCLUDE_SUBSECTIONS)
                    VALUES ".implode(',', $this->insertBuffer)."
                ";
                $connection->query($insertQuery);
                $this->insertBuffer = array();
                $this->insertLength = 0;
            }
            catch (\Bitrix\Main\DB\SqlException $e)
            {
                return false;
            }
        }
 
        return true;
    }
 
    public function deleteIndexElement($elementId)
    {
        $connection = \Bitrix\Main\Application::getConnection();
 
        $connection->query("DELETE from ".$this->getTableName()." WHERE ELEMENT_ID = ".intval($elementId));
 
        return true;
    }
 
    public static function propertyIdToFacetId($propertyId)
    {
        return intval($propertyId * 2);
    }
 
    public static function priceIdToFacetId($priceId)
    {
        return intval($priceId * 2 + 1);
    }
 
    public static function isPriceId($facetId)
    {
        return ($facetId % 2) != 0;
    }
 
    public static function isPropertyId($facetId)
    {
        return ($facetId % 2) == 0;
    }
 
    public static function facetIdToPropertyId($facetId)
    {
        return intval($facetId / 2);
    }
 
    public static function facetIdToPriceId($facetId)
    {
        return intval(($facetId - 1) / 2);
    }
}