這篇文章將為大家詳細講解有關(guān)Springboot+redis+Interceptor+自定義annotation如何實現(xiàn)接口自動冪等,小編覺得挺實用的,因此分享給大家做個參考,希望大家閱讀完這篇文章后可以有所收獲。
略陽ssl適用于網(wǎng)站、小程序/APP、API接口等需要進行數(shù)據(jù)傳輸應(yīng)用場景,ssl證書未來市場廣闊!成為成都創(chuàng)新互聯(lián)的ssl證書銷售渠道,可以享受市場價格4-6折優(yōu)惠!如果有意向歡迎電話聯(lián)系或者加微信:13518219792(備注:SSL證書合作)期待與您的合作!
一:搭建redis的服務(wù)Api
1:首先是搭建redis服務(wù)器,這個之前搭過了,就不贅述了。詳情可參考:https://www.cnblogs.com/wyq178/p/10340234.html
2:引入springboot中到的redis的stater,或者Spring封裝的jedis也可以,后面主要用到的api就是它的set方法和exists方法,這里我們使用springboot的封裝好的redisTemplate
/** * redis工具類 */@Componentpublic class RedisService { @Autowired private RedisTemplate redisTemplate; /** * 寫入緩存 * @param key * @param value * @return */ public boolean set(final String key, Object value) { boolean result = false; try { ValueOperations
二:自定義注解AutoIdempotent
自定義一個注解,定義此注解的主要目的是把它添加在需要實現(xiàn)冪等的方法上,凡是某個方法注解了它,都會實現(xiàn)自動冪等。后臺利用反射如果掃描到這個注解,就會處理這個方法實現(xiàn)自動冪等,使用元注解ElementType.METHOD表示它只能放在方法上,etentionPolicy.RUNTIME表示它在運行時
@Target({ElementType.METHOD})@Retention(RetentionPolicy.RUNTIME)public @interface AutoIdempotent { }
三:token創(chuàng)建和檢驗
1:token服務(wù)接口
我們新建一個接口,創(chuàng)建token服務(wù),里面主要是兩個方法,一個用來創(chuàng)建token,一個用來驗證token。創(chuàng)建token主要產(chǎn)生的是一個字符串,檢驗token的話主要是傳達request對象,為什么要傳request對象呢?主要作用就是獲取header里面的token,然后檢驗,通過拋出的Exception來獲取具體的報錯信息返回給前端
public interface TokenService { /** * 創(chuàng)建token * @return */ public String createToken(); /** * 檢驗token * @param request * @return */ public boolean checkToken(HttpServletRequest request) throws Exception;}
2:token的服務(wù)實現(xiàn)類
token引用了redis服務(wù),創(chuàng)建token采用隨機算法工具類生成隨機uuid字符串,然后放入到redis中,如果放入成功,最后返回這個token值。checkToken方法就是從header中獲取token到值(如果header中拿不到,就從paramter中獲取),如若不存在,直接拋出異常。這個異常信息可以被攔截器捕捉到,然后返回給前端。
@Servicepublic class TokenServiceImpl implements TokenService { @Autowired private RedisService redisService; /** * 創(chuàng)建token * * @return */ @Override public String createToken() { String str = RandomUtil.randomUUID(); StrBuilder token = new StrBuilder(); try { token.append(Constant.Redis.TOKEN_PREFIX).append(str); redisService.setEx(token.toString(), token.toString(),1000L); boolean notEmpty = StrUtil.isNotEmpty(token.toString()); if (notEmpty) { return token.toString(); } }catch (Exception ex){ ex.printStackTrace(); } return null; } /** * 檢驗token * * @param request * @return */ @Override public boolean checkToken(HttpServletRequest request) throws Exception { String token = request.getHeader(Constant.TOKEN_NAME); if (StrUtil.isBlank(token)) {// header中不存在token token = request.getParameter(Constant.TOKEN_NAME); if (StrUtil.isBlank(token)) {// parameter中也不存在token throw new ServiceException(Constant.ResponseCode.ILLEGAL_ARGUMENT, 100); } } if (!redisService.exists(token)) { throw new ServiceException(Constant.ResponseCode.REPETITIVE_OPERATION, 200); } boolean remove = redisService.remove(token); if (!remove) { throw new ServiceException(Constant.ResponseCode.REPETITIVE_OPERATION, 200); } return true; }}
四:攔截器的配置
1:web配置類,實現(xiàn)WebMvcConfigurerAdapter,主要作用就是添加autoIdempotentInterceptor到配置類中,這樣我們到攔截器才能生效,注意使用@Configuration注解,這樣在容器啟動是時候就可以添加進入context中
@Configurationpublic class WebConfiguration extends WebMvcConfigurerAdapter { @Resource private AutoIdempotentInterceptor autoIdempotentInterceptor; /** * 添加攔截器 * @param registry */ @Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(autoIdempotentInterceptor); super.addInterceptors(registry); }}
2:攔截處理器:主要的功能是攔截掃描到AutoIdempotent到注解到方法,然后調(diào)用tokenService的checkToken()方法校驗token是否正確,如果捕捉到異常就將異常信息渲染成json返回給前端
/** * 攔截器 */@Componentpublic class AutoIdempotentInterceptor implements HandlerInterceptor { @Autowired private TokenService tokenService; /** * 預(yù)處理 * * @param request * @param response * @param handler * @return * @throws Exception */ @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { if (!(handler instanceof HandlerMethod)) { return true; } HandlerMethod handlerMethod = (HandlerMethod) handler; Method method = handlerMethod.getMethod(); //被ApiIdempotment標記的掃描 AutoIdempotent methodAnnotation = method.getAnnotation(AutoIdempotent.class); if (methodAnnotation != null) { try { return tokenService.checkToken(request);// 冪等性校驗, 校驗通過則放行, 校驗失敗則拋出異常, 并通過統(tǒng)一異常處理返回友好提示 }catch (Exception ex){ ResultVo failedResult = ResultVo.getFailedResult(101, ex.getMessage()); writeReturnJson(response, JSONUtil.toJsonStr(failedResult)); throw ex; } } //必須返回true,否則會被攔截一切請求 return true; } @Override public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { } @Override public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { } /** * 返回的json值 * @param response * @param json * @throws Exception */ private void writeReturnJson(HttpServletResponse response, String json) throws Exception{ PrintWriter writer = null; response.setCharacterEncoding("UTF-8"); response.setContentType("text/html; charset=utf-8"); try { writer = response.getWriter(); writer.print(json); } catch (IOException e) { } finally { if (writer != null) writer.close(); } }}
五:測試用例
1:模擬業(yè)務(wù)請求類
首先我們需要通過/get/token路徑通過getToken()方法去獲取具體的token,然后我們調(diào)用testIdempotence方法,這個方法上面注解了@AutoIdempotent,攔截器會攔截所有的請求,當判斷到處理的方法上面有該注解的時候,就會調(diào)用TokenService中的checkToken()方法,如果捕獲到異常會將異常拋出調(diào)用者,下面我們來模擬請求一下:
@RestControllerpublic class BusinessController { @Resource private TokenService tokenService; @Resource private TestService testService; @PostMapping("/get/token") public String getToken(){ String token = tokenService.createToken(); if (StrUtil.isNotEmpty(token)) { ResultVo resultVo = new ResultVo(); resultVo.setCode(Constant.code_success); resultVo.setMessage(Constant.SUCCESS); resultVo.setData(token); return JSONUtil.toJsonStr(resultVo); } return StrUtil.EMPTY; } @AutoIdempotent @PostMapping("/test/Idempotence") public String testIdempotence() { String businessResult = testService.testIdempotence(); if (StrUtil.isNotEmpty(businessResult)) { ResultVo successResult = ResultVo.getSuccessResult(businessResult); return JSONUtil.toJsonStr(successResult); } return StrUtil.EMPTY; }}
2:使用postman請求
首先訪問get/token路徑獲取到具體到token:
利用獲取到到token,然后放到具體請求到header中,可以看到第一次請求成功,接著我們請求第二次:
第二次請求,返回到是重復(fù)性操作,可見重復(fù)性驗證通過,再多次請求到時候我們只讓其第一次成功,第二次就是失?。?/p>
關(guān)于“Springboot+redis+Interceptor+自定義annotation如何實現(xiàn)接口自動冪等”這篇文章就分享到這里了,希望以上內(nèi)容可以對大家有一定的幫助,使各位可以學(xué)到更多知識,如果覺得文章不錯,請把它分享出去讓更多的人看到。