numeratorsequence.php

<?php
namespace Bitrix\Main\Numerator\Model;
 
use Bitrix\Main\DB\SqlQueryException;
use Bitrix\Main\Entity\DataManager;
use Bitrix\Main\Entity\IntegerField;
use Bitrix\Main\Entity\StringField;
use Bitrix\Main\Application;
use Bitrix\Main\Result;
 
class NumeratorSequenceTable extends DataManager
{
    public static function getTableName()
    {
        return 'b_numerator_sequence';
    }
 
    public static function getMap()
    {
        return [
            (new IntegerField('NUMERATOR_ID'))
                ->configureRequired(true)
                ->configurePrimary(true)
            ,
            (new StringField('KEY'))
                ->configureRequired(true)
                ->configurePrimary(true)
            ,
            (new StringField('TEXT_KEY'))
                ->configureRequired(true)
            ,
            (new IntegerField('NEXT_NUMBER')),
            (new IntegerField('LAST_INVOCATION_TIME'))
                ->configureRequired(true)
            ,
        ];
    }
 
    public static function updateSettings($numeratorId, $numberHash, $fields, $whereNextNumber = null)
    {
        if ($whereNextNumber)
        {
            $conn = Application::getConnection();
            $helper = $conn->getSqlHelper();
            $update = $helper->prepareUpdate(static::getTableName(), $fields);
            $query = 'UPDATE ' . $helper->quote(static::getTableName())
                     . ' SET ' . $update[0]
                     . ' WHERE NUMERATOR_ID = ' . intval($numeratorId)
                     . ' AND ' . $helper->quote('KEY') . ' = ' . $helper->convertToDbString(md5($numberHash))
                     . ' AND NEXT_NUMBER = ' . intval($whereNextNumber);
            $conn->query($query);
 
            return $conn->getAffectedRowsCount();
        }
        $res = static::update([
            'NUMERATOR_ID' => intval($numeratorId),
            'KEY'          => md5($numberHash),
        ], $fields);
        if ($res->isSuccess())
        {
            return $res->getAffectedRowsCount();
        }
 
        return 0;
    }
 
    public static function deleteByNumeratorId($id)
    {
        $result = new Result();
 
        $list = static::getList([
            'filter' => [
                '=NUMERATOR_ID' => (int)$id,
            ]
        ]);
        while($row = $list->fetchObject())
        {
            $deleteResult = $row->delete();
            if (!$deleteResult->isSuccess())
            {
                $result->addErrors($deleteResult->getErrors());
            }
        }
 
        return $result;
    }
 
    public static function getSettings($numeratorId, $numberHash)
    {
        $sequenceSettings = static::getList([
            'select' => ['NEXT_NUMBER', 'LAST_INVOCATION_TIME'],
            'filter' => ['=NUMERATOR_ID' => $numeratorId, '=KEY' => md5($numberHash),],
        ])->fetch();
        return $sequenceSettings;
    }
 
    public static function setSettings($numeratorId, $numberHash, $defaultNumber, $lastInvocationTime)
    {
        try
        {
            $result = static::add([
                'NUMERATOR_ID'         => $numeratorId,
                'KEY'                  => md5($numberHash),
                'TEXT_KEY'             => mb_substr($numberHash, 0, 50),
                'LAST_INVOCATION_TIME' => $lastInvocationTime,
                'NEXT_NUMBER'          => $defaultNumber,
            ]);
            if ($result->isSuccess())
            {
                return $result->getData();
            }
            return [];
        }
        catch (SqlQueryException $exc)
        {
            if (mb_stripos($exc->getMessage(), "Duplicate entry") !== false)
            {
                return [];
            }
            throw $exc;
        }
    }
}