Your IP : 216.73.216.86
<?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;
/**
* @param string $filePath
* @return void
*/
public function __construct($filePath)
{
$this->_filePath = $filePath;
$this->_ID = '1' . mb_substr(md5($filePath), 1);
}
/**
* @return array[string]string
*/
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;
}
/**
* @return bool
*/
public function isStarted()
{
return is_array($this->GetArray());
}
/**
* @return void
*/
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);
}
/**
* @return void
*/
public function DeleteOld()
{
global $DB;
$DB->Query('DELETE FROM b_clouds_file_upload WHERE TIMESTAMP_X < ' . $DB->CharToDateFunction(ConvertTimeStamp(time() - 24 * 60 * 60)));
}
/**
* @param int $bucket_id
* @param float $fileSize
* @param string $ContentType
* @return bool
*/
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;
}
/**
* @param string $data
* @return bool
*/
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;
}
/**
* @param string $data
* @param int $part_no
* @return bool
*/
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;
}
/**
* @return bool
*/
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;
}
/**
* @return int
*/
public function GetPartCount()
{
$ar = $this->GetArray();
if (is_array($ar))
{
return intval($ar['PART_NO']);
}
else
{
return 0;
}
}
/**
* @return float
*/
public function GetPos()
{
$ar = $this->GetArray();
if (is_array($ar))
{
return intval($ar['PART_NO']) * doubleval($ar['PART_SIZE']);
}
else
{
return 0;
}
}
/**
* @return int
*/
public function getPartSize()
{
$ar = $this->GetArray();
if (is_array($ar))
{
return intval($ar['PART_SIZE']);
}
else
{
return 0;
}
}
/**
* @return bool
*/
public function hasRetries()
{
$ar = $this->GetArray();
return is_array($ar) && (intval($ar['PART_FAIL_COUNTER']) < $this->_max_retries);
}
/**
* @return string
*/
public function getTempFileName()
{
$ar = $this->GetArray();
if (is_array($ar))
{
return $ar['TMP_FILE'];
}
else
{
return '';
}
}
/**
* @param array $arUploadInfo
* @param bool $bSuccess
* @return bool
*/
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']) . "'");
}
}
}
}