Your IP : 172.28.240.42
<?
require_once($_SERVER["DOCUMENT_ROOT"]."/bitrix/modules/main/classes/general/user_counter.php");
class CUserCounter extends CAllUserCounter
{
public static function Set($user_id, $code, $value, $site_id = SITE_ID, $tag = '', $sendPull = true)
{
global $DB, $CACHE_MANAGER;
$value = intval($value);
$user_id = intval($user_id);
if ($user_id <= 0 || strlen($code) <= 0)
return false;
$rs = $DB->Query("
SELECT CNT FROM b_user_counter
WHERE USER_ID = ".$user_id."
AND SITE_ID = '".$DB->ForSQL($site_id)."'
AND CODE = '".$DB->ForSQL($code)."'
");
if ($rs->Fetch())
{
$ssql = "";
if ($tag != "")
$ssql = ", TAG = '".$DB->ForSQL($tag)."'";
$DB->Query("
UPDATE b_user_counter SET
CNT = ".$value." ".$ssql."
WHERE USER_ID = ".$user_id."
AND SITE_ID = '".$DB->ForSQL($site_id)."'
AND CODE = '".$DB->ForSQL($code)."'
");
}
else
{
$DB->Query("
INSERT INTO b_user_counter
(CNT, USER_ID, SITE_ID, CODE, TAG)
VALUES
(".$value.", ".$user_id.", '".$DB->ForSQL($site_id)."', '".$DB->ForSQL($code)."', '".$DB->ForSQL($tag)."')
", true);
}
if (self::$counters && self::$counters[$user_id])
{
if ($site_id == '**')
{
foreach(self::$counters[$user_id] as $key => $tmp)
{
self::$counters[$user_id][$key][$code] = $value;
}
}
else
{
if (!isset(self::$counters[$user_id][$site_id]))
self::$counters[$user_id][$site_id] = array();
self::$counters[$user_id][$site_id][$code] = $value;
}
}
$CACHE_MANAGER->Clean("user_counter".$user_id, "user_counter");
if ($sendPull)
self::SendPullEvent($user_id, $code);
return true;
}
public static function Increment($user_id, $code, $site_id = SITE_ID, $sendPull = true)
{
global $DB, $CACHE_MANAGER;
$user_id = intval($user_id);
if ($user_id <= 0 || strlen($code) <= 0)
return false;
$strSQL = "
INSERT INTO b_user_counter (USER_ID, CNT, SITE_ID, CODE)
VALUES (".$user_id.", 1, '".$DB->ForSQL($site_id)."', '".$DB->ForSQL($code)."')
ON DUPLICATE KEY UPDATE CNT = CNT + 1";
$DB->Query($strSQL, false, "FILE: ".__FILE__."<br> LINE: ".__LINE__);
if (self::$counters && self::$counters[$user_id])
{
if ($site_id == '**')
{
foreach(self::$counters[$user_id] as $key => $tmp)
{
if (isset(self::$counters[$user_id][$key][$code]))
self::$counters[$user_id][$key][$code]++;
else
self::$counters[$user_id][$key][$code] = 1;
}
}
else
{
if (!isset(self::$counters[$user_id][$site_id]))
self::$counters[$user_id][$site_id] = array();
if (isset(self::$counters[$user_id][$site_id][$code]))
self::$counters[$user_id][$site_id][$code]++;
else
self::$counters[$user_id][$site_id][$code] = 1;
}
}
$CACHE_MANAGER->Clean("user_counter".$user_id, "user_counter");
if ($sendPull)
self::SendPullEvent($user_id, $code);
return true;
}
public static function Decrement($user_id, $code, $site_id = SITE_ID, $sendPull = true)
{
global $DB, $CACHE_MANAGER;
$user_id = intval($user_id);
if ($user_id <= 0 || strlen($code) <= 0)
return false;
$strSQL = "
INSERT INTO b_user_counter (USER_ID, CNT, SITE_ID, CODE)
VALUES (".$user_id.", -1, '".$DB->ForSQL($site_id)."', '".$DB->ForSQL($code)."')
ON DUPLICATE KEY UPDATE CNT = CNT - 1";
$DB->Query($strSQL, false, "FILE: ".__FILE__."<br> LINE: ".__LINE__);
if (self::$counters && self::$counters[$user_id])
{
if ($site_id == '**')
{
foreach(self::$counters[$user_id] as $key => $tmp)
{
if (isset(self::$counters[$user_id][$key][$code]))
self::$counters[$user_id][$key][$code]--;
else
self::$counters[$user_id][$key][$code] = -1;
}
}
else
{
if (!isset(self::$counters[$user_id][$site_id]))
self::$counters[$user_id][$site_id] = array();
if (isset(self::$counters[$user_id][$site_id][$code]))
self::$counters[$user_id][$site_id][$code]--;
else
self::$counters[$user_id][$site_id][$code] = -1;
}
}
$CACHE_MANAGER->Clean("user_counter".$user_id, "user_counter");
if ($sendPull)
self::SendPullEvent($user_id, $code);
return true;
}
public static function IncrementWithSelect($sub_select, $sendPull = true)
{
global $DB, $CACHE_MANAGER;
if (strlen($sub_select) > 0)
{
$pullInclude = $sendPull && self::CheckLiveMode();
$strSQL = "
INSERT INTO b_user_counter (USER_ID, CNT, SITE_ID, CODE, SENT) (".$sub_select.")
ON DUPLICATE KEY UPDATE CNT = CNT + 1, SENT = ".($pullInclude? 0: 1)."
";
$DB->Query($strSQL, false, "FILE: ".__FILE__."<br> LINE: ".__LINE__);
self::$counters = false;
$CACHE_MANAGER->CleanDir("user_counter");
if ($pullInclude)
{
$arSites = Array();
$res = CSite::GetList(($b = ""), ($o = ""), Array("ACTIVE" => "Y"));
while($row = $res->Fetch())
$arSites[] = $row['ID'];
$strSQL = "
SELECT distinct pc.CHANNEL_ID, uc.USER_ID, uc1.SITE_ID, uc1.CODE, uc1.CNT
FROM b_user_counter uc
INNER JOIN b_user_counter uc1 ON uc1.USER_ID = uc.USER_ID AND uc1.CODE = uc.CODE
INNER JOIN b_pull_channel pc ON pc.USER_ID = uc.USER_ID
WHERE uc.SENT = 0
";
$res = $DB->Query($strSQL, false, "FILE: ".__FILE__."<br> LINE: ".__LINE__);
$updateId = Array();
$pullMessage = Array();
while($row = $res->Fetch())
{
if ($row['SITE_ID'] == '**')
{
foreach($arSites as $siteId)
{
if (isset($pullMessage[$row['CHANNEL_ID']][$siteId][$row['CODE']]))
$pullMessage[$row['CHANNEL_ID']][$siteId][$row['CODE']] += intval($row['CNT']);
else
$pullMessage[$row['CHANNEL_ID']][$siteId][$row['CODE']] = intval($row['CNT']);
}
}
else
{
if (isset($pullMessage[$row['CHANNEL_ID']][$row['SITE_ID']][$row['CODE']]))
$pullMessage[$row['CHANNEL_ID']][$row['SITE_ID']][$row['CODE']] += intval($row['CNT']);
else
$pullMessage[$row['CHANNEL_ID']][$row['SITE_ID']][$row['CODE']] = intval($row['CNT']);
}
$updateId[] = Array(
'USER_ID' => $row['USER_ID'],
'SITE_ID' => $row['SITE_ID'],
'CODE' => $row['CODE'],
);
}
$strSqlValues = "";
$strSqlPrefix = "UPDATE b_user_counter SET SENT = 1 WHERE ";
foreach($updateId as $ar)
{
$strSqlValues .= " OR (USER_ID = '".intval($ar['USER_ID'])."' AND SITE_ID = '".$DB->ForSql($ar['SITE_ID'])."' AND CODE = '".$DB->ForSql($ar['CODE'])."')";
if(strlen($strSqlValues) > 2048)
{
$DB->Query($strSqlPrefix.substr($strSqlValues, 4));
$strSqlValues = "";
}
}
if($strSqlValues <> '')
{
$DB->Query($strSqlPrefix.substr($strSqlValues, 4));
}
foreach ($pullMessage as $channelId => $arMessage)
{
CPullStack::AddByChannel($channelId, Array(
'module_id' => 'main',
'command' => 'user_counter',
'params' => $arMessage,
));
}
}
}
}
public static function Clear($user_id, $code, $site_id = SITE_ID, $sendPull = true)
{
global $DB, $CACHE_MANAGER;
$user_id = intval($user_id);
if ($user_id <= 0 || strlen($code) <= 0)
return false;
if (!is_array($site_id))
$site_id = array($site_id);
$strSQL = "
INSERT INTO b_user_counter (USER_ID, SITE_ID, CODE, CNT, LAST_DATE) VALUES ";
foreach ($site_id as $i => $site_id_tmp)
{
if ($i > 0)
$strSQL .= ",";
$strSQL .= " (".$user_id.", '".$DB->ForSQL($site_id_tmp)."', '".$DB->ForSQL($code)."', 0, ".$DB->CurrentTimeFunction().") ";
}
$strSQL .= " ON DUPLICATE KEY UPDATE CNT = 0, LAST_DATE = ".$DB->CurrentTimeFunction();
$res = $DB->Query($strSQL, false, "FILE: ".__FILE__."<br> LINE: ".__LINE__);
if (self::$counters && self::$counters[$user_id])
{
foreach ($site_id as $site_id_tmp)
{
if ($site_id_tmp == '**')
{
foreach(self::$counters[$user_id] as $key => $tmp)
self::$counters[$user_id][$key][$code] = 0;
break;
}
else
{
if (!isset(self::$counters[$user_id][$site_id_tmp]))
self::$counters[$user_id][$site_id_tmp] = array();
self::$counters[$user_id][$site_id_tmp][$code] = 0;
}
}
}
$CACHE_MANAGER->Clean("user_counter".$user_id, "user_counter");
if ($sendPull)
self::SendPullEvent($user_id, $code);
return true;
}
protected static function dbIF($condition, $yes, $no)
{
return "if(".$condition.", ".$yes.", ".$no.")";
}
// legacy function
public static function ClearByUser($user_id, $site_id = SITE_ID, $code = "**")
{
return self::Clear($user_id, $code, $site_id);
}
}
?>