天天看點

利用redis限制單個時間内某個mac位址的通路次數

一、思路

  使用者mac位址唯一,可以作為redis中的key,每次請求進來,利用ttl指令,判斷redis中key的剩餘時間,如果大于零,則利用incr進行+1操作,然後再與總的限制次數作對比。

二、代碼

RedisUtill.java

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import redis.clients.jedis.BinaryClient;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

import javax.annotation.PostConstruct;
import java.util.List;
import java.util.Map;
import java.util.Set;

@Component
public class RedisUtil {
    private static final Logger LOGGER = LoggerFactory.getLogger(RedisUtil.class);
    private static JedisPool pool = null;

    private static RedisUtil ru = null;

    @Value("${redis.ip}")
    private String redisIp;

    @Value("${redis.password}")
    private String password;

    @PostConstruct
    private void  initRedis() {
        if (pool == null) {
            JedisPoolConfig config = new JedisPoolConfig();
            // 控制一個pool可配置設定多少個jedis執行個體,通過pool.getResource()來擷取;
            // 如果指派為-1,則表示不限制;如果pool已經配置設定了maxActive個jedis執行個體,則此時pool的狀态為exhausted(耗盡)。
            config.setMaxActive(1000);
            // 控制一個pool最多有多少個狀态為idle(空閑的)的jedis執行個體。
            config.setMaxIdle(1000);
            // 表示當borrow(引入)一個jedis執行個體時,最大的等待時間,如果超過等待時間,則直接抛出JedisConnectionException;
            config.setMaxWait(1000 * 100);
            config.setTestOnBorrow(true);
            pool = new JedisPool(config,redisIp, 6379, 100000,password);
            ru = new RedisUtil();
        }

    }

    /**
     * <p>通過key擷取儲存在redis中的value的過期時間,機關為 秒 </p>
     * <p>并釋放連接配接</p>
     * @param key
     * @return 成功傳回value 失敗傳回null
     */
    public Long ttl(String key){
        Jedis jedis = null;
        Long value = null;
        try {
            jedis = pool.getResource();
            value = jedis.ttl(key);
        }
        catch (Exception e) {
            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return value;
    }
    /**
     * <p>通過key擷取儲存在redis中的value</p>
     * <p>并釋放連接配接</p>
     * @param key
     * @return 成功傳回value 失敗傳回null
     */
    public String get(String key){
        Jedis jedis = null;
        String value = null;
        try {
            jedis = pool.getResource();
            value = jedis.get(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return value;
    }

    /**
     * <p>向redis存入key和value,并釋放連接配接資源</p>
     * <p>如果key已經存在 則覆寫</p>
     * @param key
     * @param value
     * @return 成功 傳回OK 失敗傳回 0
     */
    public String set(String key, String value){
        Jedis jedis = null;
        try {
            jedis = pool.getResource();
            return jedis.set(key, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
            return "0";
        } finally {
            returnResource(pool, jedis);
        }
    }


    /**
     * <p>删除指定的key,也可以傳入一個包含key的數組</p>
     * @param keys 一個key  也可以使 string 數組
     * @return 傳回删除成功的個數
     */
    public Long del(String...keys){
        Jedis jedis = null;
        try {
            jedis = pool.getResource();
            return jedis.del(keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
            return 0L;
        } finally {
            returnResource(pool, jedis);
        }
    }

    /**
     * <p>通過key向指定的value值追加值</p>
     * @param key
     * @param str
     * @return 成功傳回 添加後value的長度 失敗 傳回 添加的 value 的長度  異常傳回0L
     */
    public Long append(String key , String str){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.append(key, str);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
            return 0L;
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>判斷key是否存在</p>
     * @param key
     * @return true OR false
     */
    public Boolean exists(String key){
        Jedis jedis = null;
        try {
            jedis = pool.getResource();
            return jedis.exists(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
            return false;
        } finally {
            returnResource(pool, jedis);
        }
    }

    /**
     * <p>設定key value,如果key已經存在則傳回0,nx==> not exist</p>
     * @param key
     * @param value
     * @return 成功傳回1 如果存在 和 發生異常 傳回 0
     */
    public Long setnx(String key , String value){
        Jedis jedis = null;
        try {
            jedis = pool.getResource();
            return jedis.setnx(key, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
            return 0L;
        } finally {
            returnResource(pool, jedis);
        }
    }

    /**
     * <p>設定key value并制定這個鍵值的有效期</p>
     * @param key
     * @param value
     * @param seconds 機關:秒
     * @return 成功傳回OK 失敗和異常傳回null
     */
    public String setex(String key, String value, int seconds){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.setex(key, seconds, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }


    /**
     * <p>通過key 和offset 從指定的位置開始将原先value替換</p>
     * <p>下标從0開始,offset表示從offset下标開始替換</p>
     * <p>如果替換的字元串長度過小則會這樣</p>
     * <p>example:</p>
     * <p>value : [email protected]</p>
     * <p>str : abc </p>
     * <P>從下标7開始替換  則結果為</p>
     * <p>RES : bigsea.abc.cn</p>
     * @param key
     * @param str
     * @param offset 下标位置
     * @return 傳回替換後  value 的長度
     */
    public Long setrange(String key, String str, int offset){
        Jedis jedis = null;
        try {
            jedis = pool.getResource();
            return jedis.setrange(key, offset, str);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
            return 0L;
        } finally {
            returnResource(pool, jedis);
        }
    }



    /**
     * <p>通過批量的key擷取批量的value</p>
     * @param keys string數組 也可以是一個key
     * @return 成功傳回value的集合, 失敗傳回null的集合 ,異常傳回空
     */
    public List<String> mget(String...keys){
        Jedis jedis = null;
        List<String> values = null;
        try {
            jedis = pool.getResource();
            values = jedis.mget(keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return values;
    }

    /**
     * <p>批量的設定key:value,可以一個</p>
     * <p>example:</p>
     * <p>  obj.mset(new String[]{"key2","value1","key2","value2"})</p>
     * @param keysvalues
     * @return 成功傳回OK 失敗 異常 傳回 null
     *
     */
    public String mset(String...keysvalues){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.mset(keysvalues);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>批量的設定key:value,可以一個,如果key已經存在則會失敗,操作會復原</p>
     * <p>example:</p>
     * <p>  obj.msetnx(new String[]{"key2","value1","key2","value2"})</p>
     * @param keysvalues
     * @return 成功傳回1 失敗傳回0
     */
    public Long msetnx(String...keysvalues){
        Jedis jedis = null;
        Long res = 0L;
        try {
            jedis = pool.getResource();
            res =jedis.msetnx(keysvalues);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>設定key的值,并傳回一個舊值</p>
     * @param key
     * @param value
     * @return 舊值 如果key不存在 則傳回null
     */
    public String getset(String key, String value){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.getSet(key, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過下标 和key 擷取指定下标位置的 value</p>
     * @param key
     * @param startOffset 開始位置 從0 開始 負數表示從右邊開始截取
     * @param endOffset
     * @return 如果沒有傳回null
     */
    public String getrange(String key, int startOffset , int endOffset){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.getrange(key, startOffset, endOffset);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key 對value進行加值+1操作,當value不是int類型時會傳回錯誤,當key不存在是則value為1</p>
     * @param key
     * @return 加值後的結果
     */
    public Long incr(String key){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.incr(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key給指定的value加值,如果key不存在,則這是value為該值</p>
     * @param key
     * @param integer
     * @return
     */
    public Long incrBy(String key, Long integer){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.incrBy(key, integer);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>對key的值做減減操作,如果key不存在,則設定key為-1</p>
     * @param key
     * @return
     */
    public Long decr(String key) {
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.decr(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>減去指定的值</p>
     * @param key
     * @param integer
     * @return
     */
    public Long decrBy(String key, Long integer){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.decrBy(key, integer);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取value值的長度</p>
     * @param key
     * @return 失敗傳回null
     */
    public Long serlen(String key){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.strlen(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key給field設定指定的值,如果key不存在,則先建立</p>
     * @param key
     * @param field 字段
     * @param value
     * @return 如果存在傳回0 異常傳回null
     */
    public Long hset(String key, String field, String value) {
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hset(key, field, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key給field設定指定的值,如果key不存在則先建立,如果field已經存在,傳回0</p>
     * @param key
     * @param field
     * @param value
     * @return
     */
    public Long hsetnx(String key, String field, String value){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hsetnx(key, field, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key同時設定 hash的多個field</p>
     * @param key
     * @param hash
     * @return 傳回OK 異常傳回null
     */
    public String hmset(String key, Map<String, String> hash){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hmset(key, hash);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key 和 field 擷取指定的 value</p>
     * @param key
     * @param field
     * @return 沒有傳回null
     */
    public String hget(String key, String field){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hget(key, field);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key 和 fields 擷取指定的value 如果沒有對應的value則傳回null</p>
     * @param key
     * @param fields 可以使 一個String 也可以是 String數組
     * @return
     */
    public List<String> hmget(String key, String...fields){
        Jedis jedis = null;
        List<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hmget(key, fields);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key給指定的field的value加上給定的值</p>
     * @param key
     * @param field
     * @param value
     * @return
     */
    public Long hincrby(String key , String field , Long value){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hincrBy(key, field, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key和field判斷是否有指定的value存在</p>
     * @param key
     * @param field
     * @return
     */
    public Boolean hexists(String key , String field){
        Jedis jedis = null;
        Boolean res = false;
        try {
            jedis = pool.getResource();
            res = jedis.hexists(key, field);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回field的數量</p>
     * @param key
     * @return
     */
    public Long hlen(String key){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hlen(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;

    }

    /**
     * <p>通過key 删除指定的 field </p>
     * @param key
     * @param fields 可以是 一個 field 也可以是 一個數組
     * @return
     */
    public Long hdel(String key , String...fields){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hdel(key, fields);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回所有的field</p>
     * @param key
     * @return
     */
    public Set<String> hkeys(String key){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hkeys(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回所有和key有關的value</p>
     * @param key
     * @return
     */
    public List<String> hvals(String key){
        Jedis jedis = null;
        List<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hvals(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取所有的field和value</p>
     * @param key
     * @return
     */
    public Map<String, String> hgetall(String key){
        Jedis jedis = null;
        Map<String, String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.hgetAll(key);
        } catch (Exception e) {
            // TODO
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key向list頭部添加字元串</p>
     * @param key
     * @param strs 可以使一個string 也可以使string數組
     * @return 傳回list的value個數
     */
    public Long lpush(String key , String...strs){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.lpush(key, strs);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key向list尾部添加字元串</p>
     * @param key
     * @param strs 可以使一個string 也可以使string數組
     * @return 傳回list的value個數
     */
    public Long rpush(String key , String...strs){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.rpush(key, strs);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key在list指定的位置之前或者之後 添加字元串元素</p>
     * @param key
     * @param where LIST_POSITION枚舉類型
     * @param pivot list裡面的value
     * @param value 添加的value
     * @return
     */
    public Long linsert(String key, BinaryClient.LIST_POSITION where,
                        String pivot, String value){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.linsert(key, where, pivot, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key設定list指定下标位置的value</p>
     * <p>如果下标超過list裡面value的個數則報錯</p>
     * @param key
     * @param index 從0開始
     * @param value
     * @return 成功傳回OK
     */
    public String lset(String key , Long index, String value){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.lset(key, index, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key從對應的list中删除指定的count個 和 value相同的元素</p>
     * @param key
     * @param count 當count為0時删除全部
     * @param value
     * @return 傳回被删除的個數
     */
    public Long lrem(String key, long count, String value){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.lrem(key, count, value);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key保留list中從strat下标開始到end下标結束的value值</p>
     * @param key
     * @param start
     * @param end
     * @return 成功傳回OK
     */
    public String ltrim(String key , long start , long end){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.ltrim(key, start, end);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key從list的頭部删除一個value,并傳回該value</p>
     * @param key
     * @return
     */
    synchronized public String lpop(String key){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.lpop(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key從list尾部删除一個value,并傳回該元素</p>
     * @param key
     * @return
     */
    synchronized public String rpop(String key){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.rpop(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key從一個list的尾部删除一個value并添加到另一個list的頭部,并傳回該value</p>
     * <p>如果第一個list為空或者不存在則傳回null</p>
     * @param srckey
     * @param dstkey
     * @return
     */
    public String rpoplpush(String srckey, String dstkey){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.rpoplpush(srckey, dstkey);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取list中指定下标位置的value</p>
     * @param key
     * @param index
     * @return 如果沒有傳回null
     */
    public String lindex(String key, long index){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.lindex(key, index);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回list的長度</p>
     * @param key
     * @return
     */
    public Long llen(String key){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.llen(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取list指定下标位置的value</p>
     * <p>如果start 為 0 end 為 -1 則傳回全部的list中的value</p>
     * @param key
     * @param start
     * @param end
     * @return
     */
    public List<String> lrange(String key, long start, long end){
        Jedis jedis = null;
        List<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.lrange(key, start, end);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key向指定的set中添加value</p>
     * @param key
     * @param members 可以是一個String 也可以是一個String數組
     * @return 添加成功的個數
     */
    public Long sadd(String key, String...members){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sadd(key, members);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key删除set中對應的value值</p>
     * @param key
     * @param members 可以是一個String 也可以是一個String數組
     * @return 删除的個數
     */
    public Long srem(String key, String...members){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.srem(key, members);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key随機删除一個set中的value并傳回該值</p>
     * @param key
     * @return
     */
    public String spop(String key){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.spop(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取set中的差集</p>
     * <p>以第一個set為标準</p>
     * @param keys 可以使一個string 則傳回set中所有的value 也可以是string數組
     * @return
     */
    public Set<String> sdiff(String...keys){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sdiff(keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取set中的差集并存入到另一個key中</p>
     * <p>以第一個set為标準</p>
     * @param dstkey 差集存入的key
     * @param keys 可以使一個string 則傳回set中所有的value 也可以是string數組
     * @return
     */
    public Long sdiffstore(String dstkey, String... keys){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sdiffstore(dstkey, keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取指定set中的交集</p>
     * @param keys 可以使一個string 也可以是一個string數組
     * @return
     */
    public Set<String> sinter(String...keys){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sinter(keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取指定set中的交集 并将結果存入新的set中</p>
     * @param dstkey
     * @param keys 可以使一個string 也可以是一個string數組
     * @return
     */
    public Long sinterstore(String dstkey, String...keys){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sinterstore(dstkey, keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回所有set的并集</p>
     * @param keys 可以使一個string 也可以是一個string數組
     * @return
     */
    public Set<String> sunion(String... keys){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sunion(keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回所有set的并集,并存入到新的set中</p>
     * @param dstkey
     * @param keys 可以使一個string 也可以是一個string數組
     * @return
     */
    public Long sunionstore(String dstkey, String...keys){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sunionstore(dstkey, keys);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key将set中的value移除并添加到第二個set中</p>
     * @param srckey 需要移除的
     * @param dstkey 添加的
     * @param member set中的value
     * @return
     */
    public Long smove(String srckey, String dstkey, String member){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.smove(srckey, dstkey, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取set中value的個數</p>
     * @param key
     * @return
     */
    public Long scard(String key){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.scard(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key判斷value是否是set中的元素</p>
     * @param key
     * @param member
     * @return
     */
    public Boolean sismember(String key, String member){
        Jedis jedis = null;
        Boolean res = null;
        try {
            jedis = pool.getResource();
            res = jedis.sismember(key, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取set中随機的value,不删除元素</p>
     * @param key
     * @return
     */
    public String srandmember(String key){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.srandmember(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取set中所有的value</p>
     * @param key
     * @return
     */
    public Set<String> smembers(String key){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.smembers(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }


    /**
     * <p>通過key向zset中添加value,score,其中score就是用來排序的</p>
     * <p>如果該value已經存在則根據score更新元素</p>
     * @param key
     * @param score
     * @param member
     * @return
     */
    public Long zadd(String key, double score, String member){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zadd(key, score, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key删除在zset中指定的value</p>
     * @param key
     * @param members 可以使一個string 也可以是一個string數組
     * @return
     */
    public Long zrem(String key, String...members){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zrem(key, members);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key增加該zset中value的score的值</p>
     * @param key
     * @param score
     * @param member
     * @return
     */
    public Double zincrby(String key , double score , String member){
        Jedis jedis = null;
        Double res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zincrby(key, score, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回zset中value的排名</p>
     * <p>下标從小到大排序</p>
     * @param key
     * @param member
     * @return
     */
    public Long zrank(String key, String member){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zrank(key, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回zset中value的排名</p>
     * <p>下标從大到小排序</p>
     * @param key
     * @param member
     * @return
     */
    public Long zrevrank(String key, String member){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zrevrank(key, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key将擷取score從start到end中zset的value</p>
     * <p>socre從大到小排序</p>
     * <p>當start為0 end為-1時傳回全部</p>
     * @param key
     * @param start
     * @param end
     * @return
     */
    public Set<String> zrevrange(String key , long start , long end){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zrevrange(key, start, end);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回指定score内zset中的value</p>
     * @param key
     * @param max
     * @param min
     * @return
     */
    public Set<String> zrangebyscore(String key, String max, String min){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zrevrangeByScore(key, max, min);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回指定score内zset中的value</p>
     * @param key
     * @param max
     * @param min
     * @return
     */
    public Set<String> zrangeByScore(String key , double max, double min){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zrevrangeByScore(key,max,min);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>傳回指定區間内zset中value的數量</p>
     * @param key
     * @param min
     * @param max
     * @return
     */
    public Long zcount(String key, String min, String max){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zcount(key, min, max);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key傳回zset中的value個數</p>
     * @param key
     * @return
     */
    public Long zcard(String key){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zcard(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key擷取zset中value的score值</p>
     * @param key
     * @param member
     * @return
     */
    public Double zscore(String key, String member){
        Jedis jedis = null;
        Double res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zscore(key, member);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key删除給定區間内的元素</p>
     * @param key
     * @param start
     * @param end
     * @return
     */
    public Long zremrangeByRank(String key , long start, long end){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zremrangeByRank(key, start, end);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key删除指定score内的元素</p>
     * @param key
     * @param start
     * @param end
     * @return
     */
    public Long zremrangeByScore(String key, double start, double end){
        Jedis jedis = null;
        Long res = null;
        try {
            jedis = pool.getResource();
            res = jedis.zremrangeByScore(key, start, end);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }
    /**
     * <p>傳回滿足pattern表達式的所有key</p>
     * <p>keys(*)</p>
     * <p>傳回所有的key</p>
     * @param pattern
     * @return
     */
    public Set<String> keys(String pattern){
        Jedis jedis = null;
        Set<String> res = null;
        try {
            jedis = pool.getResource();
            res = jedis.keys(pattern);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * <p>通過key判斷值得類型</p>
     * @param key
     * @return
     */
    public String type(String key){
        Jedis jedis = null;
        String res = null;
        try {
            jedis = pool.getResource();
            res = jedis.type(key);
        } catch (Exception e) {

            LOGGER.error(e.getMessage());
        } finally {
            returnResource(pool, jedis);
        }
        return res;
    }

    /**
     * 返還到連接配接池
     *
     * @param pool
     * @param jedis
     */
    public static void returnResource(JedisPool pool, Jedis jedis) {
        if (jedis != null) {
            pool.returnResource(jedis);
        }
    }

    public static RedisUtil getRu() {
        return ru;
    }

    public static void setRu(RedisUtil ru) {
        RedisUtil.ru = ru;
    }
}      

工具方法:

/**
     * timeOut 秒内限制調用臨時放行接口totalCount次
     * @param userMac 使用者mac 位址
     * @param timeOut 多少秒内
     * @param totalCount 總的次數
     * @return
     */
    public  boolean allowLimitTimes(String userMac, int timeOut, int totalCount){

        //次數key
        String key="allowTimes_"+userMac;
        boolean result=false;
        try{
            Long ttl = redisUtil.ttl(key);
            if (ttl > 0) {

                //第幾次通路
                Long nowTimes = redisUtil.incr(key);
                if (nowTimes > totalCount) {
                    logger.error("key:{},超出{}秒内允許通路{}次的限制,這是第{}次通路", new Object[] { key, timeOut, totalCount, nowTimes});
                }
                else {
                    result = true;
                }
            }
            else if (ttl == -1 || ttl == -2 || ttl == 0) {//逾時不存在key
                redisUtil.setex(key,"1",timeOut);
                result=true;
            }
        }
        catch(Exception e){
            logger.error("OpenPlatformService.allowLimitTimes error:",e);
        }
        return result;
    }      

使用:

30分鐘内限制5次請求進來

if(allowLimitTimes("fc:e9:98:3e:24:37",1800,5)){
  
  //業務邏輯    
}      

作者:no-npe

出處:https://www.cnblogs.com/geekdc

本文版權歸作者和部落格園共有,歡迎轉載,但未經作者同意必須保留此聲明,且在文章頁面明顯位置給出原文連結,否則保留追究法律責任的權利。

由于作者個人水準有限,如果文中有什麼錯誤,歡迎指出。以免更多的人被誤導。