1. 程式人生 > 實用技巧 >談談 Java 中自定義註解及使用場景

談談 Java 中自定義註解及使用場景

www.jianshu.com/p/a7bedc771204

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定義,常用的包括:

  • METHOD:用於描述方法
  • PACKAGE:用於描述包
  • PARAMETER:用於描述方法變數
  • TYPE:用於描述類、介面或enum型別

Retention: 表示註解保留時間長短。取值在java.lang.annotation.RetentionPolicy中,取值為:

  • SOURCE:在原始檔中有效,編譯過程中會被忽略
  • CLASS:隨原始檔一起編譯在class檔案中,執行時忽略
  • RUNTIME:在執行時有效

只有定義為RetentionPolicy.RUNTIME時,我們才能通過註解反射獲取到註解。

所以,假設我們要自定義一個註解,它用在欄位上,並且可以通過反射獲取到,功能是用來描述欄位的長度和作用。

@Target(ElementType.FIELD)//註解用於欄位上
@Retention(RetentionPolicy.RUNTIME)//保留到執行時,可通過註解獲取
public@interfaceMyField{
Stringdescription();
intlength();
}

四. 示例-反射獲取註解

先定義一個註解:

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public@interfaceMyField{
Stringdescription();
intlength();
}

通過反射獲取註解

publicclassMyFieldTest{

//使用我們的自定義註解
@MyField(description="使用者名稱",length=12)
privateStringusername;

@Test
publicvoidtestMyField(){

//獲取類模板
Classc=MyFieldTest.class;

//獲取所有欄位
for(Fieldf:c.getDeclaredFields()){
//判斷這個欄位是否有MyField註解
if(f.isAnnotationPresent(MyField.class)){
MyFieldannotation=f.getAnnotation(MyField.class);
System.out.println("欄位:["+f.getName()+"],描述:["+annotation.description()+"],長度:["+annotation.length()+"]");
}
}

}
}

執行結果

應用場景一:自定義註解+攔截器 實現登入校驗

接下來,我們使用springboot攔截器實現這樣一個功能,如果方法上加了@LoginRequired,則提示使用者該介面需要登入才能訪問,否則不需要登入。

首先定義一個LoginRequired註解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public@interfaceLoginRequired{

}

然後寫兩個簡單的介面,訪問sourceA,sourceB資源

@RestController
publicclassIndexController{

@GetMapping("/sourceA")
publicStringsourceA(){
return"你正在訪問sourceA資源";
}

@GetMapping("/sourceB")
publicStringsourceB(){
return"你正在訪問sourceB資源";
}

}

沒新增攔截器之前成功訪問

實現spring的HandlerInterceptor 類先實現攔截器,但不攔截,只是簡單列印日誌,如下:

publicclassSourceAccessInterceptorimplementsHandlerInterceptor{
@Override
publicbooleanpreHandle(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler)throwsException{
System.out.println("進入攔截器了");
returntrue;
}

@Override
publicvoidpostHandle(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler,ModelAndViewmodelAndView)throwsException{

}

@Override
publicvoidafterCompletion(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler,Exceptionex)throwsException{

}
}

實現spring類WebMvcConfigurer,建立配置類把攔截器新增到攔截器鏈中

@Configuration
publicclassInterceptorTrainConfigurerimplementsWebMvcConfigurer{
@Override
publicvoidaddInterceptors(InterceptorRegistryregistry){
registry.addInterceptor(newSourceAccessInterceptor()).addPathPatterns("/**");
}
}

攔截成功如下

在sourceB方法上新增我們的登入註解@LoginRequired

@RestController
publicclassIndexController{

@GetMapping("/sourceA")
publicStringsourceA(){
return"你正在訪問sourceA資源";
}

@LoginRequired
@GetMapping("/sourceB")
publicStringsourceB(){
return"你正在訪問sourceB資源";
}

}

簡單實現登入攔截邏輯

@Override
publicbooleanpreHandle(HttpServletRequestrequest,HttpServletResponseresponse,Objecthandler)throwsException{
System.out.println("進入攔截器了");

//反射獲取方法上的LoginRequred註解
HandlerMethodhandlerMethod=(HandlerMethod)handler;
LoginRequiredloginRequired=handlerMethod.getMethod().getAnnotation(LoginRequired.class);
if(loginRequired==null){
returntrue;
}

//有LoginRequired註解說明需要登入,提示使用者登入
response.setContentType("application/json;charset=utf-8");
response.getWriter().print("你訪問的資源需要登入");
returnfalse;
}

執行成功,訪問sourceB時需要登入了,訪問sourceA則不用登入

應用場景二:自定義註解+AOP 實現日誌列印

先匯入切面需要的依賴包

<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>

定義一個註解@MyLog

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public@interfaceMyLog{

}

定義一個切面類,見如下程式碼註釋理解:

@Aspect//1.表明這是一個切面類
@Component
publicclassMyLogAspect{

//2.PointCut表示這是一個切點,@annotation表示這個切點切到一個註解上,後面帶該註解的全類名
//切面最主要的就是切點,所有的故事都圍繞切點發生
//logPointCut()代表切點名稱
@Pointcut("@annotation(me.zebin.demo.annotationdemo.aoplog.MyLog)")
publicvoidlogPointCut(){};

//3.環繞通知
@Around("logPointCut()")
publicvoidlogAround(ProceedingJoinPointjoinPoint){
//獲取方法名稱
StringmethodName=joinPoint.getSignature().getName();
//獲取入參
Object[]param=joinPoint.getArgs();

StringBuildersb=newStringBuilder();
for(Objecto:param){
sb.append(o+";");
}
System.out.println("進入["+methodName+"]方法,引數為:"+sb.toString());

//繼續執行方法
try{
joinPoint.proceed();
}catch(Throwablethrowable){
throwable.printStackTrace();
}
System.out.println(methodName+"方法執行結束");

}
}

在步驟二中的IndexController寫一個sourceC進行測試,加上我們的自定義註解:

@MyLog
@GetMapping("/sourceC/{source_name}")
publicStringsourceC(@PathVariable("source_name")StringsourceName){
return"你正在訪問sourceC資源";
}

啟動springboot web專案,輸入訪問地址

瑣碎時間想看一些技術文章,可以去公眾號選單欄翻一翻我分類好的內容,應該對部分童鞋有幫助。同時看的過程中發現問題歡迎留言指出,