前言
針對於我們現在常用的RESTful API
通常我們需要對請求進行唯一標識,也就是每次都要帶上一個請求號,如reqNO
。
對於入庫這種運算元據庫的請求我們一般要保證他的唯一性,一個請求號通常只能用一次,所以需要我們對這種請求加上校驗機制。
該需求的實現思路是通過自定義
annotation
,只給需要進行校驗的介面加上註解。然後通過切面使用了註解的介面將每次請求號存進Redis
,每次都進行判斷是否存在這個請求號即可。
來看下加上本次外掛的實際效果:
自定義註解
首先我們要自定義一個註解:
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface CheckReqNo {
String desc() default "";
}複製程式碼
(ps:這裡並不過多的講解註解相關的知識)。
首先使用@interface
來宣告一個註解。接著利用Java
為我們提供的三個元註解來定義CheckReqNo
註解。
其中@Target
表明這個註解被用於什麼地方,使用ElementType.METHOD
表明被應用到方法上,還有一些其他值可以檢視java.lang.annotation.ElementType
這個列舉型別。
@Retention
註解表明我們的註解在什麼範圍內有效,這裡配置的RetentionPolicy.RUNTIME
表明在執行時可以通過反射來獲取。
@Documented
看字面意思應該也能猜到是用於生成JavaDoc
文件的。
其中定義了一個desc()
的方法其實並沒有用到,但如果需要在使用註解的時候需要自定義一些filed(域)
的需求可以按照這樣的方式寫到這裡,通過反射都可以獲取到具體的值。
如:@CheckReqNo(desc = "abc")
就可以獲取到"abc"
的值。
切面註解
按照之前的想法是在對所有使用了該註解的方法進行切面:
@Aspect
@Component
public class ReqNoDrcAspect {
private static Logger logger = LoggerFactory.getLogger(ReqNoDrcAspect.class);
@Value("${redis.prefixReq:reqNo}")
private String prefixReq ;
@Value("${redis.day:1}")
private long day ;
@Autowired
private RedisTemplate<String, String> redisTemplate;
@PostConstruct
public void init() throws Exception {
logger.info("SSM-REQUEST-CHECK init......");
}
@Pointcut("@annotation(com.crossoverJie.request.anotation.CheckReqNo)")
public void checkRepeat(){
}
@Before("checkRepeat()")
public void before(JoinPoint joinPoint) throws Exception {
BaseRequest request;
request = getBaseRequest(joinPoint);
if(request != null){
final String reqNo = request.getReqNo();
if(StringUtil.isEmpty(reqNo)){
throw new RuntimeException("reqNo不能為空");
}else{
try {
String tempReqNo = redisTemplate.opsForValue().get(prefixReq +reqNo);
logger.debug("tempReqNo="+tempReqNo);
if((StringUtil.isEmpty(tempReqNo))){
redisTemplate.opsForValue().set(prefixReq + reqNo, reqNo, day, TimeUnit.DAYS);
}else{
throw new RuntimeException("請求號重複,reqNo="+reqNo);
}
} catch (RedisConnectionFailureException e){
logger.error("redis操作異常",e);
throw new RuntimeException("need redisService") ;
}
}
}
}
public static BaseRequest getBaseRequest(JoinPoint joinPoint) throws Exception {
BaseRequest returnRequest = null;
Object[] arguments = joinPoint.getArgs();
if(arguments != null && arguments.length > 0){
returnRequest = (BaseRequest) arguments[0];
}
return returnRequest;
}
}複製程式碼
使用@Aspect
來定義了一個切面。
其中prefixReq,day
域可以自定義快取請求號時的key
字首以及快取的時間。
最關鍵的一點是用@Pointcut("@annotation(com.crossoverJie.request.anotation.CheckReqNo)")
定義了一個切入點,這樣所有使用@CheckReqNo
的註解都會被攔截。
接下來的邏輯就比較簡單了,在每次請求之前進行攔截。
先去Redis
中檢視這個請求號(ps:反射獲取
)是否存在,如果不存在則通過並將本次的請求號快取起來。如果存在則丟擲異常。
使用註解
可以在jdbc.properties
配置檔案中自定義字首和快取時間
#redis字首
redis.prefixReq=reqNo
#redis快取時間 預設單位為天
redis.day=1複製程式碼
不定義也可以,會使用預設值。
由於該註解是需要加到controller
層,因此我們得使用CGLIB
代理。
這裡有一個坑,需要將開啟CGLIB
的配置配置到我們web.xml
中的
<!-- Spring MVC servlet -->
<servlet>
<servlet-name>SpringMVC</servlet-name>
<servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
<init-param>
<param-name>contextConfigLocation</param-name>
<param-value>classpath:spring-mvc.xml</param-value>
</init-param>
<load-on-startup>1</load-on-startup>
<async-supported>true</async-supported>
</servlet>複製程式碼
這裡所定義的spring-mvc.xml
檔案中,不然springMVC
所在的子容器是無法被父容器所載入的。
使用例項:
@CheckReqNo
@RequestMapping(value = "/createRedisContent",method = RequestMethod.POST)
@ResponseBody
public BaseResponse<NULLBody> createRedisContent(@RequestBody RedisContentReq redisContentReq){
BaseResponse<NULLBody> response = new BaseResponse<NULLBody>() ;
Rediscontent rediscontent = new Rediscontent() ;
try {
CommonUtil.setLogValueModelToModel(redisContentReq,rediscontent);
rediscontentMapper.insertSelective(rediscontent) ;
response.setReqNo(redisContentReq.getReqNo());
response.setCode(StatusEnum.SUCCESS.getCode());
response.setMessage(StatusEnum.SUCCESS.getMessage());
}catch (Exception e){
logger.error("system error",e);
response.setReqNo(response.getReqNo());
response.setCode(StatusEnum.FAIL.getCode());
response.setMessage(StatusEnum.FAIL.getMessage());
}
return response ;
}複製程式碼
統一異常controller
/**
*
* ClassName: ErrorController <br/>
* Function: 錯誤異常統一處理. <br/>
* @author crossoverJie
* @version
* @since JDK 1.7
*/
@ControllerAdvice
public class ErrorController {
private Logger logger = LoggerFactory.getLogger(this.getClass());
@ExceptionHandler(Exception.class)
@ResponseStatus(HttpStatus.OK)
@ResponseBody
public Object processUnauthenticatedException(NativeWebRequest request, Exception e) {
logger.error("請求出現異常:", e);
BaseResponse<NULLBody> response = new BaseResponse<NULLBody>();
response.setCode(StatusEnum.FAIL.getCode());
if (e instanceof RuntimeException){
response.setMessage(e.getMessage());
} else {
response.setMessage(StatusEnum.FAIL.getMessage());
}
return response ;
}
}複製程式碼
這樣當controller層出現異常之後都會進入這裡進行統一的返回。
總結
至此整個外掛的流程已經全部OK,從中可以看出Spring AOP
在實際開發中的各種好處。
之前的幾篇文章也有應用到:
不知不覺這個小白入門的SSM
系列已經更新了14篇了,在GitHub
也有了500多顆星了,期間也和不少朋友有過交流、探討,感謝大家的支援。
接下來可能不太會更新這個系列了,由於博主現在所在的專案組採用的是目前比較流行的SpringBoot+SpringCloud
和Docker
的方式來進行架構的,所以之後的重心肯定會移到這方面,用過SpringBoot
之後相信大家肯定也回不去了。
所以之後我會繼續更新SpringBoot+SpringCloud
相關的文章,歡迎持續關注,持續拍磚(ps:這個外掛也會用springBoot重寫一遍
)
個人部落格地址:crossoverjie.top。