一、背景需求

最近做了一个小需求，由于系统对接，导致我们的系统在高峰的时候CPU飙升，所以需要在高峰的时候保护系统进程不受影响。
而且还需要我们知道当前对接的数据总量，并且可以实时释放，如果释放失败了，还需要定时释放，减少系统卡顿的同时，不能对第三方产生影响。

此处肯定有人想到令牌桶算法，或者redission。
SpringBoot基于guava集成令牌桶算法：https://blog.csdn.net/qq_38254635/article/details/126398730
SpringBoot集成Redisson：https://blog.csdn.net/qq_38254635/article/details/126398716
集成的虽好，但我们用的场景单一，集成还需要引入jar导致部署包变大，所以我们需要手动搞一个。

二、具体要求

1、可以知道当前存在的总量
2、可实时对每个变量进行调整
3、变量的关闭失败后，有自修复机制

三、代码实现

了解redisson的应该知道，可以参考其做一个简易版。

package com;import com.alibaba.fastjson.JSONObject;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.util.StringUtils;
import javax.annotation.Resource;
import java.util.Map;
import java.util.concurrent.TimeUnit;public class Test {@Resourceprivate StringRedisTemplate redisTemplate;private final static Integer SWITCH_COUNT = 3;public static final String BUSINESS = "business:";public static final String CHECK = "check";public static final String BUSINESS_CHECK = BUSINESS + CHECK;public Result doLink(String strJson) {if(!getCheck(strJson)) return Result.error("当前调用超上限，请稍后重试！");return Result.success();}private boolean getCheck(String code){Map<Object, Object> objectMap = redisTemplate.opsForHash().entries(BUSINESS_CHECK);//如果为空，则视为第一次使用，直接放行if(null == objectMap || objectMap.size() == 0) {this.setKey(code);return true;}//清洗数据，清理过期redisInteger count = 0;//当前有效总量for(Map.Entry<Object, Object> entry : objectMap.entrySet()){String user = entry.getKey().toString();if(StringUtils.isEmpty(redisTemplate.opsForValue().get(BUSINESS_CHECK + "_" + user))) {redisTemplate.opsForHash().delete(BUSINESS_CHECK, user);continue;}count++;}//校验是否超过上限if(count >= SWITCH_COUNT) return false;this.setKey(code);return true;}private void setKey(String code){redisTemplate.opsForHash().put(BUSINESS_CHECK, code, JSONObject.toJSONString(code));redisTemplate.opsForValue().set(BUSINESS_CHECK + "_" + code, code,1, TimeUnit.MINUTES);}//清理当前调用线程redisprivate void after(String code){redisTemplate.opsForHash().delete(BUSINESS_CHECK, code);redisTemplate.delete(code);}}