<?php
IncludeModuleLangFile(__FILE__);

class CPerfCluster
{
	public static function Add($arFields)
	{
		global $DB;
		$ID = $DB->Add('b_perf_cluster', $arFields);
		return $ID;
	}

	public static function Truncate()
	{
		global $DB;
		$res = $DB->Query('DELETE FROM b_perf_cluster');
		return $res;
	}

	/**
	 * @param bool|array[] $arOrder
	 * @param bool|array[] $arFilter
	 * @param bool|array[] $arSelect
	 *
	 * @return bool|CDBResult
	 */
	public static function GetList($arOrder = false, $arFilter = false, $arSelect = false)
	{
		global $DB;

		if (!is_array($arSelect))
		{
			$arSelect = [];
		}
		if (count($arSelect) < 1)
		{
			$arSelect = [
				'ID',
				'TIMESTAMP_X',
				'THREADS',
				'HITS',
				'ERRORS',
				'PAGES_PER_SECOND',
				'PAGE_EXEC_TIME',
				'PAGE_RESP_TIME',
			];
		}

		if (!is_array($arOrder))
		{
			$arOrder = [];
		}

		$arQueryOrder = [];
		foreach ($arOrder as $strColumn => $strDirection)
		{
			$strColumn = mb_strtoupper($strColumn);
			$strDirection = mb_strtoupper($strDirection) === 'ASC' ? 'ASC' : 'DESC';
			if ($strColumn === 'ID')
			{
				$arSelect[] = $strColumn;
				$arQueryOrder[$strColumn] = $strColumn . ' ' . $strDirection;
			}
		}

		$arQuerySelect = [];
		foreach ($arSelect as $strColumn)
		{
			$strColumn = mb_strtoupper($strColumn);
			switch ($strColumn)
			{
			case 'ID':
			case 'TIMESTAMP_X':
			case 'THREADS':
			case 'HITS':
			case 'ERRORS':
			case 'PAGES_PER_SECOND':
			case 'PAGE_EXEC_TIME':
			case 'PAGE_RESP_TIME':
				$arQuerySelect[$strColumn] = 'p.' . $strColumn;
				break;
			}
		}
		if (count($arQuerySelect) < 1)
		{
			$arQuerySelect = ['ID' => 'p.ID'];
		}

		$obQueryWhere = new CSQLWhere;
		$arFields = [
			'ID' => [
				'TABLE_ALIAS' => 'p',
				'FIELD_NAME' => 'p.ID',
				'FIELD_TYPE' => 'int',
				'JOIN' => false,
			],
		];
		$obQueryWhere->SetFields($arFields);

		if (!is_array($arFilter))
		{
			$arFilter = [];
		}
		$strQueryWhere = $obQueryWhere->GetQuery($arFilter);

		$bDistinct = $obQueryWhere->bDistinctReqired;

		$strSql = '
			SELECT ' . ($bDistinct ? 'DISTINCT' : '') . '
			' . implode(', ', $arQuerySelect) . '
			FROM
				b_perf_cluster p
			' . $obQueryWhere->GetJoins() . '
		';

		if ($strQueryWhere)
		{
			$strSql .= '
				WHERE
				' . $strQueryWhere . '
			';
		}

		if (count($arQueryOrder) > 0)
		{
			$strSql .= '
				ORDER BY
				' . implode(', ', $arQueryOrder) . '
			';
		}

		return $DB->Query($strSql, false, '', ['fixed_connection' => true]);
	}

	public function Measure($host, $port, $url, $threads, $iterations = 3, $arOptions = [])
	{
		global $DB;

		$strRequest = 'GET ' . $url . " HTTP/1.0\r\n";
		$strRequest .= "User-Agent: BitrixSMCluster (thread #thread#)\r\n";
		$strRequest .= "Accept: */*\r\n";
		$strRequest .= 'Host: ' . $host . "\r\n";
		$strRequest .= "Accept-Language: en\r\n";

		$socket_timeout = intval($arOptions['socket_timeout']);
		if ($socket_timeout <= 0)
		{
			$socket_timeout = 20;
		}

		$rw_timeout = intval($arOptions['rw_timeout']);
		if ($rw_timeout <= 0)
		{
			$rw_timeout = 20;
		}

		$iteration_timeout = intval($arOptions['iteration_timeout']);
		if ($iteration_timeout <= 0)
		{
			$iteration_timeout = 30;
		}

		if ($port == 443)
		{
			$proto = 'ssl://';
		}
		else
		{
			$proto = '';
		}

		$start = microtime(1);
		$end = $start + $iterations;
		$end_after_end = $start + $iteration_timeout;

		$errors = 0;
		$arConnections = [];
		$arCookie = [];
		$arStartTimes = [];
		$Pages = 0;
		$arPageExecTime = [];
		$arResponseTime = [];
		while (microtime(1) < $end)
		{
			//Open new connection if needed
			if (count($arConnections) < $threads)
			{
				//Find first free slot
				for ($j = 0; $j < $threads; $j++)
				{
					if (!isset($arConnections[$j]))
					{
						$arStartTimes[$j] = microtime(1);
						$socket = fsockopen($proto . $host, $port, $errno, $errstr, $socket_timeout);
						if ($socket)
						{
							$request = str_replace('#thread#', $j, $strRequest);
							if (isset($arCookie[$j]))
							{
								$request .= 'Cookie: ' . implode(';', $arCookie[$j]) . "\r\n";
							}
							$request .= "\r\n";

							stream_set_blocking($socket, true);
							stream_set_timeout($socket, $rw_timeout);
							fputs($socket, $request);
							stream_set_blocking($socket, false);
							$arConnections[$j] = $socket;
							$Pages++;
						}
						else
						{
							$arConnections[$j] = false;
							$errors++;
						}
						break;
					}
				}
			}

			//Try to read connections
			foreach ($arConnections as $j => $socket)
			{
				if ($socket)
				{
					if (feof($socket))
					{
						$arResponseTime[] = microtime(1) - $arStartTimes[$j];
						fclose($socket);
						unset($arConnections[$j]);
					}
					else
					{
						$line = fgets($socket);
						if ($line !== false)
						{
							if (preg_match('/^Set-Cookie: (.*?)=(.*?);/', $line, $match))
							{
								$arCookie[$j][$match[1]] = $match[1] . '=' . $match[2];
							}
							elseif (preg_match('/<span id="bx_main_exec_time">(\\d+\\.\\d+)<\\/span>/', $line, $match))
							{
								$arPageExecTime[] = $match[1];
							}
							elseif (preg_match('/^HTTP\\/\\d+\\.\\d+\\s+(\\d+)\\s/', $line, $match))
							{
								if ($match[1] !== '200')
								{
									$errors++;
								}
							}
							elseif (preg_match('/^Status:\\s+(\\d+)\\s/', $line, $match))
							{
								if ($match[1] !== '200')
								{
									$errors++;
								}
							}
						}
					}
				}
			}
		}

		//Finish all connections
		while (count($arConnections) > 0)
		{
			//Try to read connections
			foreach ($arConnections as $j => $socket)
			{
				if ($socket)
				{
					if (feof($socket))
					{
						$arResponseTime[] = microtime(1) - $arStartTimes[$j];
						fclose($socket);
						unset($arConnections[$j]);
					}
					else
					{
						$line = fgets($socket);
						if ($line !== false)
						{
							if (preg_match('/<span id="bx_main_exec_time">(\\d+\\.\\d+)<\\/span>/', $line, $match))
							{
								$arPageExecTime[] = $match[1];
							}
							elseif (preg_match('/^HTTP\\/\\d+\\.\\d+\\s+(\\d+)\\s/', $line, $match))
							{
								if ($match[1] !== '200')
								{
									$errors++;
								}
							}
							elseif (preg_match('/^Status:\\s+(\\d+)\\s/', $line, $match))
							{
								if ($match[1] !== '200')
								{
									$errors++;
								}
							}
						}
					}
				}
				else
				{
					unset($arConnections[$j]);
				}
				if (microtime(1) > $end_after_end)
				{
					break;
				}
			}
			if (microtime(1) > $end_after_end)
			{
				break;
			}
		}

		static::Add([
			'~TIMESTAMP_X' => $DB->CurrentTimeFunction(),
			'THREADS' => $threads,
			'HITS' => $Pages,
			'ERRORS' => $errors,
			'PAGES_PER_SECOND' => $Pages / $iterations,
			'PAGE_EXEC_TIME' => count($arPageExecTime) ? array_sum($arPageExecTime) / count($arPageExecTime) : 0,
			'PAGE_RESP_TIME' => count($arResponseTime) ? array_sum($arResponseTime) / count($arResponseTime) : 0,
		]);
	}
}