利用spring-data-redis實現(xiàn)incr自增的操作

更新時間：2020年11月24日 10:40:57 作者：快樂崇拜234

這篇文章主要介紹了利用spring-data-redis實現(xiàn)incr自增的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

應(yīng)該有不少人在使用spring-data-redis時遇到各種各樣的問題。反正我是遇到了。

由于是隔了一段時間才寫的本篇博客，也懶得去重現(xiàn)哪些錯誤場景了，下面憑著記憶寫了幾個我遇到的問題：

redis.clients.jedis.exceptions.JedisDataException: ERR value is not an integer or out of range

使用的RedisTemplate，做讀寫操作時候，都是要經(jīng)過序列化和反序列化。

這時你使用redisTemplate.opsForValue().increment()就可能報錯redis.clients.jedis.exceptions.JedisDataException: ERR value is not an integer or out of range了。

valueOper.get(key) 獲取不到自增的值。

于是我去看了一下redis的官方文檔，找到一個解決方法

使用spring-data-redis實現(xiàn)incr自增
/**
   *
   * @param key
   * @param liveTime
   * @return
   */
  public Long incr(String key, long liveTime) {
    RedisAtomicLong entityIdCounter = new RedisAtomicLong(key, redisTemplate.getConnectionFactory());
    Long increment = entityIdCounter.getAndIncrement();

    if ((null == increment || increment.longValue() == 0) && liveTime > 0) {//初始設(shè)置過期時間
      entityIdCounter.expire(liveTime, TimeUnit.SECONDS);
    }

    return increment;
  }

這樣，上面的increment就是自增后的新知值，然后中間通過entityIdCounter.expire(liveTime, TimeUnit.SECONDS);設(shè)置過期時間。

當(dāng)然這里比較討厭，spring沒有在創(chuàng)建RedisAtomicLong對象的時候一起設(shè)置過期時間。

可以看看其源碼，new RedisAtomicLong最終調(diào)用的是這個方法:

private RedisAtomicLong(String redisCounter, RedisConnectionFactory factory, Long initialValue) {
    Assert.hasText(redisCounter, "a valid counter name is required");
    Assert.notNull(factory, "a valid factory is required");

    RedisTemplate<String, Long> redisTemplate = new RedisTemplate<String, Long>();
    redisTemplate.setKeySerializer(new StringRedisSerializer());
    redisTemplate.setValueSerializer(new GenericToStringSerializer<Long>(Long.class));
    redisTemplate.setExposeConnection(true);
    redisTemplate.setConnectionFactory(factory);
    redisTemplate.afterPropertiesSet();

    this.key = redisCounter;
    this.generalOps = redisTemplate;
    this.operations = generalOps.opsForValue();

    if (initialValue == null) {
      if (this.operations.get(redisCounter) == null) {
        set(0);
      }
    } else {
      set(initialValue);
    }
  }

可以看到，初始值是0。

然后根進(jìn)set方法

public void set(long newValue) {
    operations.set(key, newValue);
  }

可以看到，他是采用的operations.set(key, newValue);但是明明還有一個重載的方法void set(K key, V value, long timeout, TimeUnit unit);可以設(shè)置過期時間，為啥spring不提供呢。

為了解決這個問題，我們可以自己模擬RedisAtomicLong方法，去實現(xiàn)一個帶有過期時間的自增方法。比較簡單，讀者自行擼代碼吧，這里就不寫出了。

補(bǔ)充知識：關(guān)于spring boot使用redis的increment()方法自增問題

需求是限制IP頻繁訪問某接口，用的方案是使用redis記錄訪問IP的值，先設(shè)定好初始值，每次訪問自增，達(dá)到某限定值后，進(jìn)行阻止。

用的是自定義工具類，使用spring封裝的spring-data-redis進(jìn)行操作，在對某key進(jìn)行increment()方法時，報錯：

redis ERR value is not an integer or out of range

代碼邏輯如下：

Integer count = (Integer) redisUtil.get(ipAddress);//取得key的value
    if (count == null){
      redisUtil.set(ipAddress,1,10);
      return false;
    }else if(count == 3){
      return false;
    }else {
      redisUtil.incr(ipAddress,1);
      return false;
    }

第一次進(jìn)來，如果沒有redis中沒有數(shù)據(jù)，則設(shè)置key,value和time，key是ip， value初始值為1，有效時長為10秒。

如果沒達(dá)到限制次數(shù)，則對key自增1。

redisUtil.incr()方法實現(xiàn)如下：

@Resource
  private RedisTemplate<String, Object> redisTemplate; //這里使用的是redisTemplate

  public void setRedisTemplate(RedisTemplate<String, Object> redisTemplate) {
    this.redisTemplate = redisTemplate;
  }

  /**
   * 遞增
   * @param key 鍵
//   * @param by 要增加幾(大于0)
   * @return
   */
  public long incr(String key, long delta){
    if(delta<0){
      throw new RuntimeException("遞增因子必須大于0");
    }
    return redisTemplate.opsForValue().increment(key, delta);
  }

開始以為是incr方法接受的參數(shù)是long型，但我傳入的是INTEGER類型，但轉(zhuǎn)換后還是沒有解決問題，問題不是出在這，后來通過查找資料發(fā)現(xiàn)，Spring對Redis序列化的策略有兩種，分別是StringRedisTemplate和RedisTemplate，其中StringRedisTemplate用于操作字符串，RedisTemplate使用的是JDK默認(rèn)的二進(jìn)制序列化。

大家都知道redis序列化是將key，value值先轉(zhuǎn)換為流的形式，再存儲到redis中。

RedisTemplate是使用的JdkSerializationRedisSerializer序列化，序列化后的值包含了對象信息，版本號，類信息等，是一串字符串，所以無法進(jìn)行數(shù)值自增操作。

而StringRedisTemplate序列化策略是字符串的值直接轉(zhuǎn)為字節(jié)數(shù)組，所以存儲到redis中是數(shù)值，所以可以進(jìn)行自增操作。

StringRedisSerializer源碼：

public class StringRedisSerializer implements RedisSerializer<String> {
  private final Charset charset;

  public StringRedisSerializer() {
    this(StandardCharsets.UTF_8);
  }

  public StringRedisSerializer(Charset charset) {
    Assert.notNull(charset, "Charset must not be null!");
    this.charset = charset;
  }

  public String deserialize(@Nullable byte[] bytes) {
    return bytes == null ? null : new String(bytes, this.charset);
  }

  public byte[] serialize(@Nullable String string) {
    return string == null ? null : string.getBytes(this.charset); //注意這里是字節(jié)數(shù)組
  }
}

所以問題出在這里，我們需要自定義序列化策略，在application啟動類中添加如下：

 @Bean
  public RedisTemplate<String, String> redisTemplate(RedisConnectionFactory factory) {
    StringRedisTemplate template = new StringRedisTemplate(factory);
    //定義key序列化方式
    //RedisSerializer<String> redisSerializer = new StringRedisSerializer();//Long類型會出現(xiàn)異常信息;需要我們上面的自定義key生成策略，一般沒必要
    //定義value的序列化方式
    Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer(Object.class);
    ObjectMapper om = new ObjectMapper();
    om.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
    om.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
    jackson2JsonRedisSerializer.setObjectMapper(om);

    // template.setKeySerializer(redisSerializer);
    template.setValueSerializer(jackson2JsonRedisSerializer);
    template.setHashValueSerializer(jackson2JsonRedisSerializer);
    template.afterPropertiesSet();
    return template;

費(fèi)了2多小時才成功解決問題，RedisUtil.incr()能夠成功對key進(jìn)行自增了，如有錯誤之處請歡迎指出。

以上這篇利用spring-data-redis實現(xiàn)incr自增的操作就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: