X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/274f38105249c24d355b1f46749c3841ffff6ff0..11153633ffdf9631d411bd6ea1999cd90a5bd98b:/lib/leakybucket.php?ds=sidebyside

diff --git a/lib/leakybucket.php b/lib/leakybucket.php
index 778fe580..6d4e8f29 100644
--- a/lib/leakybucket.php
+++ b/lib/leakybucket.php
@@ -26,7 +26,7 @@
 		        $aCurrentBlock = $m->get($sKey);
 	        	if (!$aCurrentBlock)
 			{
-		                $aCurrentBlock = array($iRequestCost, $t);
+		                $aCurrentBlock = array($iRequestCost, $t, false);
         		}
 			else
 			{
@@ -87,6 +87,36 @@
 		return $iMaxVal;
         }
 
+	function isBucketSleeping($asKey)
+	{
+	        $m = getBucketMemcache();
+		if (!$m) return false;
+
+		foreach($asKey as $sKey)
+		{
+		        $aCurrentBlock = $m->get($sKey);
+			if ($aCurrentBlock[2]) return true;
+		}
+		return false;
+	}
+
+	function setBucketSleeping($asKey, $bVal)
+	{
+	        $m = getBucketMemcache();
+		if (!$m) return false;
+
+		$iMaxVal = 0;
+	        $t = time();
+
+		foreach($asKey as $sKey)
+		{
+		        $aCurrentBlock = $m->get($sKey);
+			$aCurrentBlock[2] = $bVal;
+			$m->set($sKey, $aCurrentBlock, $t + 1 + $aCurrentBlock[0]/CONST_ConnectionBucket_LeakRate);
+		}
+		return true;
+	}
+
 	function byValue1($a, $b)
 	{
 		if ($a[1] == $b[1])
@@ -120,6 +150,7 @@
 			$aBlockedList[$sKey] = array(
 				'totalBlocks' => $aDetails[0],
 				'lastBlockTimestamp' => $aDetails[1],
+				'isSleeping' => (isset($aCurrentBlock[2])?$aCurrentBlock[2]:false),
 				'currentBucketSize' => $iCurrentBucketSize,
 				'currentlyBlocked' => $iCurrentBucketSize + (CONST_ConnectionBucket_Cost_Reverse) >= CONST_ConnectionBucket_BlockLimit,
 				);