Java自定義註解一般使用場景為:自定義註解+攔截器或者AOP,使用自定義註解來自己設計框架,使得代碼看起來非常優雅。本文將先從自定義註解的基礎概念說起,然後開始實戰,寫小段代碼實現自定義註解+攔截器,自定義註解+AOP。
一. 什麼是註解(Annotation)Java註解是什麼,以下是引用自維基百科的內容
Java註解又稱Java標註,是JDK5.0版本開始支持加入原始碼的特殊語法元數據。
Java語言中的類、方法、變量、參數和包等都可以被標註。和Javadoc不同,Java標註可以通過反射獲取標註內容。在編譯器生成類文件時,標註可以被嵌入到字節碼中。Java虛擬機可以保留標註內容,在運行時可以獲取到標註內容。當然它也支持自定義Java標註。
二. 註解體系圖元註解:java.lang.annotation中提供了元註解,可以使用這些註解來定義自己的註解。主要使用的是Target和Retention註解
註解處理類:既然上面定義了註解,那得有辦法拿到我們定義的註解啊。java.lang.reflect.AnnotationElement接口則提供了該功能。註解的處理是通過java反射來處理的。
如下,反射相關的類Class, Method, Field都實現了AnnotationElement接口。
因此,只要我們通過反射拿到Class, Method, Field類,就能夠通過getAnnotation(Class<T>)拿到我們想要的註解並取值。
搜索Java知音公眾號,回復「後端面試」,送你一份Java面試題寶典
三. 常用元註解Target:描述了註解修飾的對象範圍,取值在java.lang.annotation.ElementType定義,常用的包括:
Retention: 表示註解保留時間長短。取值在java.lang.annotation.RetentionPolicy中,取值為:
CLASS:隨源文件一起編譯在class文件中,運行時忽略只有定義為RetentionPolicy.RUNTIME時,我們才能通過註解反射獲取到註解。
所以,假設我們要自定義一個註解,它用在欄位上,並且可以通過反射獲取到,功能是用來描述欄位的長度和作用。
@Target(ElementType.FIELD) // 註解用於欄位上
@Retention(RetentionPolicy.RUNTIME) // 保留到運行時,可通過註解獲取
public @interface MyField {
String description();
int length();
}
四. 示例-反射獲取註解先定義一個註解:
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyField {
String description();
int length();
}通過反射獲取註解
public class MyFieldTest {
//使用我們的自定義註解
@MyField(description = "用戶名", length = 12)
private String username;
@Test
public void testMyField(){
// 獲取類模板
Class c = MyFieldTest.class;
// 獲取所有欄位
for(Field f : c.getDeclaredFields()){
// 判斷這個欄位是否有MyField註解
if(f.isAnnotationPresent(MyField.class)){
MyField annotation = f.getAnnotation(MyField.class);
System.out.println("欄位:[" + f.getName() + "], 描述:[" + annotation.description() + "], 長度:[" + annotation.length() +"]");
}
}
}
}運行結果
應用場景一:自定義註解+攔截器 實現登錄校驗接下來,我們使用springboot攔截器實現這樣一個功能,如果方法上加了@LoginRequired,則提示用戶該接口需要登錄才能訪問,否則不需要登錄。
首先定義一個LoginRequired註解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface LoginRequired {
}然後寫兩個簡單的接口,訪問sourceA,sourceB資源
@RestController
public class IndexController {
@GetMapping("/sourceA")
public String sourceA(){
return "你正在訪問sourceA資源";
}
@GetMapping("/sourceB")
public String sourceB(){
return "你正在訪問sourceB資源";
}
}沒添加攔截器之前成功訪問
實現spring的HandlerInterceptor 類先實現攔截器,但不攔截,只是簡單列印日誌,如下:
public class SourceAccessInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
System.out.println("進入攔截器了");
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 {
}
}實現spring類WebMvcConfigurer,創建配置類把攔截器添加到攔截器鏈中
@Configuration
public class InterceptorTrainConfigurer implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new SourceAccessInterceptor()).addPathPatterns("/**");
}
}攔截成功如下
在sourceB方法上添加我們的登錄註解@LoginRequired
@RestController
public class IndexController {
@GetMapping("/sourceA")
public String sourceA(){
return "你正在訪問sourceA資源";
}
@LoginRequired
@GetMapping("/sourceB")
public String sourceB(){
return "你正在訪問sourceB資源";
}
}簡單實現登錄攔截邏輯
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
System.out.println("進入攔截器了");
// 反射獲取方法上的LoginRequred註解
HandlerMethod handlerMethod = (HandlerMethod)handler;
LoginRequired loginRequired = handlerMethod.getMethod().getAnnotation(LoginRequired.class);
if(loginRequired == null){
return true;
}
// 有LoginRequired註解說明需要登錄,提示用戶登錄
response.setContentType("application/json; charset=utf-8");
response.getWriter().print("你訪問的資源需要登錄");
return false;
}運行成功,訪問sourceB時需要登錄了,訪問sourceA則不用登錄
應用場景二:自定義註解+AOP 實現日誌列印先導入切面需要的依賴包
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>定義一個註解@MyLog
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MyLog {
}定義一個切面類,見如下代碼注釋理解:
@Aspect // 1.表明這是一個切面類
@Component
public class MyLogAspect {
// 2. PointCut表示這是一個切點,@annotation表示這個切點切到一個註解上,後面帶該註解的全類名
// 切面最主要的就是切點,所有的故事都圍繞切點發生
// logPointCut()代表切點名稱
@Pointcut("@annotation(me.zebin.demo.annotationdemo.aoplog.MyLog)")
public void logPointCut(){};
// 3. 環繞通知
@Around("logPointCut()")
public void logAround(ProceedingJoinPoint joinPoint){
// 獲取方法名稱
String methodName = joinPoint.getSignature().getName();
// 獲取入參
Object[] param = joinPoint.getArgs();
StringBuilder sb = new StringBuilder();
for(Object o : param){
sb.append(o + "; ");
}
System.out.println("進入[" + methodName + "]方法,參數為:" + sb.toString());
// 繼續執行方法
try {
joinPoint.proceed();
} catch (Throwable throwable) {
throwable.printStackTrace();
}
System.out.println(methodName + "方法執行結束");
}
}在步驟二中的IndexController寫一個sourceC進行測試,加上我們的自定義註解:
@MyLog
@GetMapping("/sourceC/{source_name}")
public String sourceC(@PathVariable("source_name") String sourceName){
return "你正在訪問sourceC資源";
}啟動springboot web項目,輸入訪問地址