Merge pull request #1 from zx5026605/feature/sms-update-allowance

新增立即提升短信报警阀接口

Author: lileiseven

src/RateLimiter.php

@@ -67,6 +67,22 @@ public function getAllowance($id)
         return $this->maxAllowance;
     }
 
+    public function updateAllowance($id, $threshold)
+    {
+        $key = $this->getKey($id);
+        $value = $this->storage->get($key);
+        if ($value !== false) {
+            list($allowance, $lastCheckTime) = $this->unpackValue($value);
+            $updatedAllowance = ($allowance + $threshold) > 0 ? ($allowance + $threshold) : 0;
+        } else {
+            $updatedAllowance = $threshold > 0 ? $threshold : 0;
+        }
+
+        $this->storage->set($this->getKey($id), $this->packValue($updatedAllowance, time()), $this->period);
+
+        return $updatedAllowance;
+    }
+
     public function getMaxAllowance()
     {
         return $this->maxAllowance;

tests/RateLimitTest.php

@@ -30,6 +30,26 @@ public function testCheckArray()
         $this->check($storage);
     }
 
+    public function testUpdateAllowanceRedis()
+    {
+        $storage = new \Codeages\RateLimiter\Storage\RedisStorage();
+        $this->updateAllowance($storage);
+    }
+
+    public function testUpdateAllowanceMySQLPDO()
+    {
+        $pdo = new PDO(getenv('MYSQL_DSN'), getenv('MYSQL_USER'), getenv('MYSQL_MYSQL_PASSWORD'));
+        $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+        $storage = new \Codeages\RateLimiter\Storage\MySQLPDOStorage($pdo);
+        $this->updateAllowance($storage);
+    }
+
+    public function testUpdateAllowanceArray()
+    {
+        $storage = new \Codeages\RateLimiter\Storage\ArrayStorage();
+        $this->updateAllowance($storage);
+    }
+
     private function check($storage)
     {
         $ip = '127.0.0.1';
@@ -59,6 +79,23 @@ private function check($storage)
         $rateLimit->purge($ip);
     }
 
+    private function updateAllowance($storage)
+    {
+        $name = 'testUpdateAllowance';
+        $rateLimit = $this->getRateLimiter($storage);
+
+        $this->assertEquals(self::MAX_REQUESTS, $rateLimit->updateAllowance($name, self::MAX_REQUESTS));
+        $this->assertEquals(0, $rateLimit->updateAllowance($name, -self::MAX_REQUESTS));
+        $this->assertEquals(0, $rateLimit->updateAllowance($name, -self::MAX_REQUESTS));
+
+        for ($i = 0, $j = 0; $i < self::MAX_REQUESTS; ++$i) {
+            $j += $i;
+            $this->assertEquals($j, $rateLimit->updateAllowance($name, $i));
+        }
+
+        $rateLimit->purge($name);
+    }
+
     private function getRateLimiter(Storage $storage)
     {
         return new RateLimiter(self::NAME, self::MAX_REQUESTS, self::PERIOD, $storage);