Spring中如何借助Redis設(shè)計(jì)一個(gè)簡單訪問計(jì)數(shù)器

這篇文章主要介紹Spring中如何借助Redis設(shè)計(jì)一個(gè)簡單訪問計(jì)數(shù)器，文中介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們一定要看完！

I. 設(shè)計(jì)

一個(gè)簡單的訪問計(jì)數(shù)器，主要利用redis的hash結(jié)構(gòu)，對(duì)應(yīng)的存儲(chǔ)結(jié)構(gòu)如下:

存儲(chǔ)結(jié)構(gòu)比較簡單，為了擴(kuò)展，每個(gè)應(yīng)用（or站點(diǎn)）對(duì)應(yīng)一個(gè)APP，然后根據(jù)path路徑進(jìn)行分頁統(tǒng)計(jì)，最后有一個(gè)特殊的用于統(tǒng)計(jì)全站的訪問計(jì)數(shù)

II. 實(shí)現(xiàn)

主要就是利用Redis的hash結(jié)構(gòu)，然后實(shí)現(xiàn)數(shù)據(jù)統(tǒng)計(jì)，并沒有太多的難度，Spring環(huán)境下搭建redis環(huán)境可以參考:

Spring之RedisTemplate配置與使用

1. Redis封裝類

針對(duì)幾個(gè)常用的做了簡單的封裝，直接使用RedisTemplate的excute方法進(jìn)行的操作，當(dāng)然也是可以使用 template.opsForValue() 等便捷方式，這里采用JSON方式進(jìn)行對(duì)象的序列化和反序列化

public class QuickRedisClient {
  private static final Charset CODE = Charset.forName("UTF-8");
  private static RedisTemplate<String, String> template;

  public static void register(RedisTemplate<String, String> template) {
    QuickRedisClient.template = template;
  }

  public static void nullCheck(Object... args) {
    for (Object obj : args) {
      if (obj == null) {
        throw new IllegalArgumentException("redis argument can not be null!");
      }
    }
  }

  public static byte[] toBytes(String key) {
    nullCheck(key);
    return key.getBytes(CODE);
  }

  public static byte[][] toBytes(List<String> keys) {
    byte[][] bytes = new byte[keys.size()][];
    int index = 0;
    for (String key : keys) {
      bytes[index++] = toBytes(key);
    }
    return bytes;
  }

  public static String getStr(String key) {
    return template.execute((RedisCallback<String>) con -> {
      byte[] val = con.get(toBytes(key));
      return val == null ? null : new String(val);
    });
  }

  public static void putStr(String key, String value) {
    template.execute((RedisCallback<Void>) con -> {
      con.set(toBytes(key), toBytes(value));
      return null;
    });
  }

  public static Long incr(String key, long add) {
    return template.execute((RedisCallback<Long>) con -> {
      Long record = con.incrBy(toBytes(key), add);
      return record == null ? 0L : record;
    });
  }

  public static Long hIncr(String key, String field, long add) {
    return template.execute((RedisCallback<Long>) con -> {
      Long record = con.hIncrBy(toBytes(key), toBytes(field), add);
      return record == null ? 0L : record;
    });
  }

  public static <T> T hGet(String key, String field, Class<T> clz) {
    return template.execute((RedisCallback<T>) con -> {
      byte[] records = con.hGet(toBytes(key), toBytes(field));
      if (records == null) {
        return null;
      }

      return JSON.parseObject(records, clz);
    });
  }

  public static <T> Map<String, T> hMGet(String key, List<String> fields, Class<T> clz) {
    List<byte[]> list =
        template.execute((RedisCallback<List<byte[]>>) con -> con.hMGet(toBytes(key), toBytes(fields)));
    if (CollectionUtils.isEmpty(list)) {
      return Collections.emptyMap();
    }

    Map<String, T> result = new HashMap<>();
    for (int i = 0; i < fields.size(); i++) {
      if (list.get(i) == null) {
        continue;
      }

      result.put(fields.get(i), JSON.parseObject(list.get(i), clz));
    }
    return result;
  }
}

對(duì)應(yīng)的配置類

package com.git.hui.story.cache.redis;

import com.git.hui.story.cache.redis.serializer.DefaultStrSerializer;
import org.springframework.cache.CacheManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.PropertySource;
import org.springframework.core.env.Environment;
import org.springframework.data.redis.cache.RedisCacheManager;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.connection.RedisPassword;
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;

/**
 * Created by yihui in 18:45 18/6/11.
 */
@Configuration
@PropertySource(value = "classpath:application.yml")
public class RedisConf {

  private final Environment environment;

  public RedisConf(Environment environment) {
    this.environment = environment;
  }

  @Bean
  public CacheManager cacheManager() {
    return RedisCacheManager.RedisCacheManagerBuilder.fromConnectionFactory(redisConnectionFactory()).build();
  }

  @Bean
  public RedisTemplate<String, String> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
    RedisTemplate<String, String> redisTemplate = new RedisTemplate<>();
    redisTemplate.setConnectionFactory(redisConnectionFactory);

    DefaultStrSerializer serializer = new DefaultStrSerializer();
    redisTemplate.setValueSerializer(serializer);
    redisTemplate.setHashValueSerializer(serializer);
    redisTemplate.setKeySerializer(serializer);
    redisTemplate.setHashKeySerializer(serializer);

    redisTemplate.afterPropertiesSet();

    QuickRedisClient.register(redisTemplate);
    return redisTemplate;
  }


  @Bean
  public RedisConnectionFactory redisConnectionFactory() {
    LettuceConnectionFactory fac = new LettuceConnectionFactory();
    fac.getStandaloneConfiguration().setHostName(environment.getProperty("spring.redis.host"));
    fac.getStandaloneConfiguration().setPort(Integer.parseInt(environment.getProperty("spring.redis.port")));
    fac.getStandaloneConfiguration()
        .setPassword(RedisPassword.of(environment.getProperty("spring.redis.password")));
    fac.afterPropertiesSet();
    return fac;
  }
}

2. Controller 支持

首先是定義請(qǐng)求參數(shù):

@Data
public class WebCountReqDO implements Serializable {
  private String appKey;
  private String referer;
}

其次是實(shí)現(xiàn)Controller接口，稍稍注意下，根據(jù)path進(jìn)行計(jì)數(shù)的邏輯：

1. 如果請(qǐng)求參數(shù)顯示指定了referer參數(shù)，則用傳入的參數(shù)進(jìn)行統(tǒng)計(jì)

2. 如果沒有顯示指定referer，則根據(jù)header獲取referer

3. 解析referer，分別對(duì)path和host進(jìn)行統(tǒng)計(jì)+1，這樣站點(diǎn)的統(tǒng)計(jì)計(jì)數(shù)就是根據(jù)host來的，而頁面的統(tǒng)計(jì)計(jì)數(shù)則是根據(jù)path路徑來的

@Slf4j
@RestController
@RequestMapping(path = "/count")
public class WebCountController {

  @RequestMapping(path = "cc", method = {RequestMethod.GET})
  public ResponseWrapper<CountDTO> addCount(WebCountReqDO webCountReqDO) {
    String appKey = webCountReqDO.getAppKey();
    if (StringUtils.isBlank(appKey)) {
      return ResponseWrapper.errorReturnMix(Status.StatusEnum.ILLEGAL_PARAMS_MIX, "請(qǐng)指定APPKEY!");
    }

    String referer = ReqInfoContext.getReqInfo().getReferer();
    if (StringUtils.isBlank(referer)) {
      referer = webCountReqDO.getReferer();
    }

    if (StringUtils.isBlank(referer)) {
      return ResponseWrapper.errorReturnMix(Status.StatusEnum.FAIL_MIX, "無法獲取請(qǐng)求referer!");
    }

    return ResponseWrapper.successReturn(doUpdateCnt(appKey, referer));
  }


  private CountDTO doUpdateCnt(String appKey, String referer) {
    try {
      if (!referer.startsWith("http")) {
        referer = "https://" + referer;
      }

      URI uri = new URI(referer);
      String host = uri.getHost();
      String path = uri.getPath();
      long count = QuickRedisClient.hIncr(appKey, path, 1);
      long total = QuickRedisClient.hIncr(appKey, host, 1);
      return new CountDTO(count, total);
    } catch (Exception e) {
      log.error("get referer path error! referer: {}, e: {}", referer, e);
      return new CountDTO(1L, 1L);
    }
  }
}

以上是“Spring中如何借助Redis設(shè)計(jì)一個(gè)簡單訪問計(jì)數(shù)器”這篇文章的所有內(nèi)容，感謝各位的閱讀！希望分享的內(nèi)容對(duì)大家有幫助，更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！