storage_upload.php

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

<?php
/*.
    require_module 'standard';
    require_module 'pcre';
    require_module 'hash';
    require_module 'bitrix_main';
    require_module 'bitrix_clouds_classes_storage_service';
    require_module 'bitrix_clouds_classes_storage_bucket';
.*/
IncludeModuleLangFile(__FILE__);
 
class CCloudStorageUpload
{
    protected /*.string.*/ $_filePath = '';
    protected /*.string.*/ $_ID = '';
    protected /*.CCloudStorageBucket.*/ $obBucket;
    protected /*.int.*/ $_max_retries = 3;
    protected /*.array[string]string.*/ $_cache = null;

    public function __construct($filePath)
    {
        $this->_filePath = $filePath;
        $this->_ID = '1' . mb_substr(md5($filePath), 1);
    }

    public function GetArray()
    {
        global $DB;
 
        if (!isset($this->_cache))
        {
            $rs = $DB->Query("
                SELECT *
                FROM b_clouds_file_upload
                WHERE ID = '" . $this->_ID . "'
            ");
            $this->_cache = $rs->Fetch();
        }
 
        return $this->_cache;
    }

    public function isStarted()
    {
        return is_array($this->GetArray());
    }

    public function Delete()
    {
        global $DB;
        //TODO: clean up temp files in Clodo
        $DB->Query("DELETE FROM b_clouds_file_upload WHERE ID = '" . $this->_ID . "'");
        unset($this->_cache);
    }

    public function DeleteOld()
    {
        global $DB;
        $DB->Query('DELETE FROM b_clouds_file_upload WHERE TIMESTAMP_X < ' . $DB->CharToDateFunction(ConvertTimeStamp(time() - 24 * 60 * 60)));
    }

    public function Start($bucket_id, $fileSize, $ContentType = 'binary/octet-stream', $tmpFileName = false)
    {
        global $DB;
        global $APPLICATION;
 
        if (is_object($bucket_id))
        {
            $obBucket = $bucket_id;
        }
        else
        {
            $obBucket = new CCloudStorageBucket(intval($bucket_id));
        }
 
        if (!$obBucket->Init())
        {
            return false;
        }
 
        if (!$this->isStarted())
        {
            $arUploadInfo = /*.(array[string]string).*/[];
            $bStarted = $obBucket->getService()->InitiateMultipartUpload(
                $obBucket->getBucketArray(),
                $arUploadInfo,
                $this->_filePath,
                $fileSize,
                $ContentType
            );
            if (!$bStarted && $obBucket->RenewToken())
            {
                $bStarted = $obBucket->getService()->InitiateMultipartUpload(
                    $obBucket->getBucketArray(),
                    $arUploadInfo,
                    $this->_filePath,
                    $fileSize,
                    $ContentType
                );
            }
 
            if ($bStarted)
            {
                $bAdded = $DB->Add('b_clouds_file_upload', [
                    'ID' => $this->_ID,
                    '~TIMESTAMP_X' => $DB->CurrentTimeFunction(),
                    'FILE_PATH' => $this->_filePath,
                    'FILE_SIZE' => $fileSize,
                    'TMP_FILE' => $tmpFileName,
                    'BUCKET_ID' => intval($obBucket->ID),
                    'PART_SIZE' => $obBucket->getService()->GetMinUploadPartSize(),
                    'PART_NO' => 0,
                    'PART_FAIL_COUNTER' => 0,
                    'NEXT_STEP' => serialize($arUploadInfo),
                ], ['NEXT_STEP']);
                unset($this->_cache);
 
                return $bAdded !== false;
            }
            else
            {
                $error = $obBucket->getService()->formatError();
                if ($error)
                {
                    $APPLICATION->ThrowException($error);
                }
                else
                {
                    $APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 6]));
                }
            }
        }
 
        return false;
    }

    public function Next($data, $obBucket = null)
    {
        global $APPLICATION;
 
        if ($this->isStarted())
        {
            $ar = $this->GetArray();
 
            if ($obBucket === null)
            {
                $obBucket = new CCloudStorageBucket(intval($ar['BUCKET_ID']));
            }
 
            if (!$obBucket->Init())
            {
                $APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 1]));
                return false;
            }
 
            $arUploadInfo = unserialize($ar['NEXT_STEP'], ['allowed_classes' => false]);
            $bSuccess = $obBucket->getService()->UploadPart(
                $obBucket->getBucketArray(),
                $arUploadInfo,
                $data
            );
 
            if (!$bSuccess)
            {
                $error = $obBucket->getService()->formatError();
                if ($error)
                {
                    $APPLICATION->ThrowException($error);
                }
            }
 
            if (!$this->UpdateProgress($arUploadInfo, $bSuccess))
            {
                $APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 2]));
                return false;
            }
 
            return $bSuccess;
        }
 
        return false;
    }

    public function Part($data, $part_no, $obBucket = null)
    {
        global $APPLICATION;
 
        if ($this->isStarted())
        {
            $ar = $this->GetArray();
 
            if ($obBucket === null)
            {
                $obBucket = new CCloudStorageBucket(intval($ar['BUCKET_ID']));
            }
 
            if (!$obBucket->Init())
            {
                $APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 3]));
                return false;
            }
 
            $arUploadInfo = unserialize($ar['NEXT_STEP'], ['allowed_classes' => false]);
            $bSuccess = $obBucket->getService()->UploadPartNo(
                $obBucket->getBucketArray(),
                $arUploadInfo,
                $data,
                $part_no
            );
 
            if (!$bSuccess)
            {
                $error = $obBucket->getService()->formatError();
                if ($error)
                {
                    $APPLICATION->ThrowException($error);
                }
            }
 
            if (!$this->UpdateProgress($arUploadInfo, $bSuccess))
            {
                $APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 5]));
                return false;
            }
 
            return $bSuccess;
        }
 
        return false;
    }

    public function Finish($obBucket = null)
    {
        global $APPLICATION;
 
        if ($this->isStarted())
        {
            $ar = $this->GetArray();
 
            if ($obBucket === null)
            {
                $obBucket = new CCloudStorageBucket(intval($ar['BUCKET_ID']));
            }
            if (!$obBucket->Init())
            {
                return false;
            }
 
            $arUploadInfo = unserialize($ar['NEXT_STEP'], ['allowed_classes' => false]);
            $bSuccess = $obBucket->getService()->CompleteMultipartUpload(
                $obBucket->getBucketArray(),
                $arUploadInfo
            );
 
            if ($bSuccess)
            {
                $this->Delete();
 
                if ($obBucket->getQueueFlag())
                {
                    CCloudFailover::queueCopy($obBucket, $this->_filePath);
                }
 
                foreach (GetModuleEvents('clouds', 'OnAfterCompleteMultipartUpload', true) as $arEvent)
                {
                    ExecuteModuleEventEx($arEvent, [$obBucket, ['size' => $ar['FILE_SIZE']], $this->_filePath]);
                }
            }
            else
            {
                $error = $obBucket->getService()->formatError();
                if ($error)
                {
                    $APPLICATION->ThrowException($error);
                }
            }
 
            return $bSuccess;
        }
 
        return false;
    }

    public function GetPartCount()
    {
        $ar = $this->GetArray();
 
        if (is_array($ar))
        {
            return intval($ar['PART_NO']);
        }
        else
        {
            return 0;
        }
    }

    public function GetPos()
    {
        $ar = $this->GetArray();
 
        if (is_array($ar))
        {
            return intval($ar['PART_NO']) * doubleval($ar['PART_SIZE']);
        }
        else
        {
            return 0;
        }
    }

    public function getPartSize()
    {
        $ar = $this->GetArray();
 
        if (is_array($ar))
        {
            return intval($ar['PART_SIZE']);
        }
        else
        {
            return 0;
        }
    }

    public function hasRetries()
    {
        $ar = $this->GetArray();
        return is_array($ar) && (intval($ar['PART_FAIL_COUNTER']) < $this->_max_retries);
    }

    public function getTempFileName()
    {
        $ar = $this->GetArray();
        if (is_array($ar))
        {
            return $ar['TMP_FILE'];
        }
        else
        {
            return '';
        }
    }

    protected function UpdateProgress($arUploadInfo, $bSuccess)
    {
        global $DB;
 
        if ($bSuccess)
        {
            $arFields = [
                'NEXT_STEP' => serialize($arUploadInfo),
                '~PART_NO' => 'PART_NO + 1',
                'PART_FAIL_COUNTER' => 0,
            ];
            $arBinds = [
                'NEXT_STEP' => $arFields['NEXT_STEP'],
            ];
        }
        else
        {
            $arFields = [
                '~PART_FAIL_COUNTER' => 'PART_FAIL_COUNTER + 1',
            ];
            $arBinds = [
            ];
        }
 
        $strUpdate = $DB->PrepareUpdate('b_clouds_file_upload', $arFields);
        if ($strUpdate != '')
        {
            $strSql = 'UPDATE b_clouds_file_upload SET ' . $strUpdate . " WHERE ID = '" . $this->_ID . "'";
            if (!$DB->QueryBind($strSql, $arBinds))
            {
                unset($this->_cache);
                return false;
            }
        }
 
        unset($this->_cache);
        return true;
    }
 
    public static function CleanUp($ID = '')
    {
        $connection = \Bitrix\Main\Application::getConnection();
        $helper = $connection->getSqlHelper();
        $rs = false;
 
        if ($ID)
        {
            $rs = $connection->query("
                SELECT ID, BUCKET_ID, NEXT_STEP
                FROM b_clouds_file_upload
                WHERE ID = '" . $helper->forSql($ID) . "'
            ");
        }
        else
        {
            $days = COption::GetOptionInt('clouds', 'multipart_upload_keep_days');
            if ($days > 0)
            {
                $rs = $connection->query('
                    SELECT ID, BUCKET_ID, NEXT_STEP
                    FROM b_clouds_file_upload
                    WHERE TIMESTAMP_X < ' . $helper->addDaysToDateTime(-$days)
                );
            }
        }
 
        if ($rs)
        {
            while ($arBucket = $rs->fetch())
            {
                $obBucket = new CCloudStorageBucket(intval($arBucket['BUCKET_ID']));
                if ($obBucket->Init())
                {
                    $arUploadInfo = unserialize($arBucket['NEXT_STEP'], ['allowed_classes' => false]);
                    $service = $obBucket->getService();
                    $service->CancelMultipartUpload($obBucket->getBucketArray(), $arUploadInfo);
                }
                $connection->query("DELETE FROM b_clouds_file_upload WHERE ID = '" . $helper->forSql($arBucket['ID']) . "'");
            }
        }
    }
}