Home > Backend Development > PHP Tutorial > Usage example of php frequency limit class

Usage example of php frequency limit class

不言
Release: 2023-04-05 18:34:02
forward
2855 people have browsed it

This article brings you usage examples of the PHP frequency limit class. It has certain reference value. Friends in need can refer to it. I hope it will be helpful to you.

For example:

Realize a single IP limit once every 60 seconds

A single keyword, such as a mobile phone number, is limited once every 60 seconds, and 10 times every 3600 seconds

<?php
class Sina_Mail_WebAntispam {

    const PREFIX_WHITELIST = &#39;w:&#39;;
    const PREFIX_KILL = &#39;k:&#39;;
    const PREFIX_VERIFYCODE = &#39;c:&#39;;
    const PREFIX_VERIFIED = &#39;v:&#39;;
    const STATUS_UPDATE = &#39;[U]&#39;;

    private $mc = null;
    private $config = null;
    private $whitelist = array();
    private $keyPrefix = &#39;&#39;;
    private $intervals = array();
    private $updates = array();
    private $status = array();

    public function __construct($mc, $config) {
        $this->mc = $mc;
        $this->config = $config;
        if (isset($this->config->prefix)) {
            $this->keyPrefix = $this->config->prefix;
        }
        if (isset($this->config->whitelistKey)) {
            $wls = $this->mc->get($this->config->whitelistKey);
            if (!empty($wls)) {
                $this->whitelist = & $wls;
            }
        }
    }

    public function setWhitelist(&$whitelist) {
        $this->whitelist = & $whitelist;
    }
    /*验证限制规则*/
    public function check($ip = null, $key = null) {
        if (!$ip && !$key) {
            return false;
        }

        if ($key) {
            if (!is_array($key)) {
                $keys = array($key);
            } else {
                $keys = $key;
            }
        }

        // first filter by whitelist
        if (!empty($this->whitelist)) {
            if ($ip && $this->filterByWhitelist($ip, &#39;ip&#39;)) {
                $this->status[self::PREFIX_WHITELIST . $ip] = 1;
                return true;
            }
            if ($keys) {
                foreach ($keys as $key) {
                    if ($this->filterByWhitelist($key, &#39;key&#39;)) {
                        $this->status[self::PREFIX_WHITELIST . $key] = 1;
                        return true;
                    }
                }
            }
        }

        if ($ip) {
            $ip = $this->keyPrefix . $ip;
        }

        // second, check verified ok
        if (!empty($this->config->verified)) {
            if ($ip && $this->mc->get(self::PREFIX_VERIFIED . $ip)) {
                $this->status[self::PREFIX_VERIFIED . $ip] = 1;
                return true;
            }
            if ($keys) {
                foreach ($keys as $key) {
                    $verifiedKey = self::PREFIX_VERIFIED . $this->keyPrefix . $key;
                    if ($this->mc->get($verifiedKey)) {
                        $this->status[$verifiedKey] = 1;
                        return true;
                    }
                }
            }
        }

        $kos = !empty($this->config->kill);
        // check killed
        if ($kos) {
            if ($ip && $this->mc->get(self::PREFIX_KILL . $ip)) {
                $this->status[self::PREFIX_KILL . $ip] = 1;
                return false;
            }
            if ($keys) {
                foreach ($keys as $key) {
                    $killKey = self::PREFIX_KILL . $this->keyPrefix . $key;
                    if ($this->mc->get($killKey)) {
                        $this->status[$killKey] = 1;
                        return false;
                    }
                }
            }
        }

        // check ip rule
        if ($ip && isset($this->config->ip)) {
            if (!$this->checkRule($ip, $this->config->ip)) {
                if ($kos && $this->mc->set(self::PREFIX_KILL . $ip, 1, intval($this->config->kill))) {
                    $this->status[self::PREFIX_KILL . $ip] = 1;
                }
                return false;
            }
        }

        // check keys rule
        if ($keys && isset($this->config->key)) {
            foreach ($keys as $key) {
                if (!$this->checkRule($this->keyPrefix . $key, $this->config->key)) {
                    $killKey = self::PREFIX_KILL . $this->keyPrefix . $key;
                    if ($kos && $this->mc->set($killKey, 1, intval($this->config->kill))) {
                        $this->status[$killKey] = 1;
                    }
                    return false;
                }
            }
        }

        return true;
    }
    /*更新限制规则*/
    public function update($c = 1, $ip = null, $key = null) {
        if (is_null($ip) && is_null($key)) {
            if (!empty($this->updates)) {
                foreach ($this->updates as $k => $v) {
                    if (!$v && isset($this->intervals[$k])) {
                        if ($this->mc->add($k, $c, false,$this->intervals[$k])) {
                            $this->status[self::STATUS_UPDATE . $k] = $c;
                            continue;
                        }
                    }
                    $r = $this->mc->increment($k, $c);
                    $this->status[self::STATUS_UPDATE . $k] = $r;
                }
            }
        } else {
            if (!is_null($ip) && isset($this->config->ip)) {
                $rule = $this->config->ip;
                foreach ($rule as $interval => $limit) {
                    $k = $this->keyPrefix . $ip . &#39;_&#39; . $interval;
                    if ($this->mc->add($k, $c,false,$interval)) {
                        $this->status[self::STATUS_UPDATE . $k] = true;
                        continue;
                    }
                    $r = $this->mc->increment($k, $c);
                    $this->status[self::STATUS_UPDATE . $k] = $r;
                }
            }
            if (!is_null($key) && isset($this->config->key)) {
                $rule = $this->config->key;
                if (!is_array($key)) {
                    $keys = array($key);
                } else {
                    $keys = $key;
                }
                foreach ($keys as $key) {
                    foreach ($rule as $interval => $limit) {
                        $k = $this->keyPrefix . $key . &#39;_&#39; . $interval;
                        if ($this->mc->add($k, $c,false,$interval)) {
                            $this->status[self::STATUS_UPDATE . $k] = true;
                            continue;
                        }
                        $r = $this->mc->increment($k, $c);
                        $this->status[self::STATUS_UPDATE . $k] = $r;
                    }
                }
            }
        }
    }

    public function checkVerifyCode($key, $code) {
        $servcode = $this->mc->get(self::PREFIX_VERIFYCODE . $this->keyPrefix . $key);
        if (strcasecmp($servcode, $code) == 0) {
            $verified = intval($this->config->verified);
            if ($verified > 0) {
                $r = $this->mc->set(self::PREFIX_VERIFIED . $this->keyPrefix . $key, 1, false, $verified);
            } else {
                $r = true;
            }
            if ($r) {
                $this->mc->delete(self::PREFIX_VERIFYCODE . $this->keyPrefix . $key);
            }
            return $r;
        }
        return false;
    }

    public function isVerified($key) {
        $r = $this->mc->get(self::PREFIX_VERIFIED . $this->keyPrefix . $key);
        if (!empty($r)) {
            return true;
        } else {
            return false;
        }
    }

    public function setVerifyCode($key, $code) {
        $verifytime = intval($this->config->verifytime);
        if ($verifytime < 1) {
            return false;
        }
        return $this->mc->set(self::PREFIX_VERIFYCODE . $this->keyPrefix . $key, $code, false, $verifytime);
    }

    public function getStatus() {
        return $this->status;
    }

    private function filterByWhitelist($value, $key) {
//        if (empty($this->whitelist[$key])) {
//            return false;
//        }
//        $ls = & $this->whitelist[$key];
        $ls = & $this->whitelist;
        foreach ($ls as $i) {
            if ($i[strlen($i) - 1] == &#39;.&#39;) { // ip segment
                if (strpos($value, $i) === 0) {
                    return true;
                }
            } else {
                if (strcmp($i, $value) === 0) {
                    return true;
                }
            }
        }
        return false;
    }

    private function checkRule($key, $rule) {
        $flag = true;
        if (!empty($rule)) {
            foreach ($rule as $interval => $limit) {
                $k = $key . &#39;_&#39; . $interval;
                $c = $this->mc->get($k);
                if (!$c) {
                    $this->updates[$k] = 0;
                    $this->intervals[$k] = $interval;
                    $this->status[$k] = 0;
                } else {
                    $this->updates[$k] = $c;
                    $this->status[$k] = $c;
                    if ($c >= $limit) {
                        $flag = false;
                    }
                }
            }
        }
        return $flag;
    }

    public static function getInstance($conf) {
        $mc = new Memcache();
        $mc->connect("115.159.28.112");
        $conf=json_decode(json_encode($conf));
        return new self($mc, $conf);
    }

}
/*
单个ip限制60秒1次
单个关键字,比如手机号,限制60秒1次,3600秒10次
*/
$conf=array(
            &#39;prefix&#39; => &#39;selfservice:&#39;,
            &#39;key&#39; => array(60 => 1,3600=>10),
            &#39;ip&#39; => array(60 => 1),
        );
$spam=Sina_Mail_WebAntispam::getInstance($conf);
if(!$spam->check(&#39;127.25.12.123&#39;,17610725730)){
	echo "limit...";
	exit;
}

//更新频率限制
$spam->update();
Copy after login

The final storage key in memache

##[Recommended course:

PHP video tutorial]

The above is the detailed content of Usage example of php frequency limit class. For more information, please follow other related articles on the PHP Chinese website!

Related labels:
php
source:segmentfault.com
Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn
Popular Tutorials
More>
Latest Downloads
More>
Web Effects
Website Source Code
Website Materials
Front End Template