自定义注解的实现

如何实现自定义注解

在我们实际开发过程中如果能合理的运用自定义注解,则会大大减少我们代码的开发量。

一、创建注解

这一步呢,我们可以理解成对应的实体类,我们要自定义注解,也需要这么一个东西,注解的名称,有哪些属性等等。

package com.disp.mindmatrix.annotations;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD, ElementType.TYPE, ElementType.PARAMETER, ElementType.METHOD})
public @interface MindMatrix{
    String value() default "";
}
  • @Retention: 表示该注解的生命周期,是RetentionPolicy类型的,该类型是一个枚举类型,可提供三个值选择,分别是:CLASS、RUNTIME、SOURCE

RetentionPolicy.CLASS: 注解被保留到class文件,但jvm加载class文件时候被遗弃,这是默认的生命周期;
RetentionPolicy.RUNTIME: 注解不仅被保存到class文件中,jvm加载class文件之后,仍然存在;
RetentionPolicy.SOURCE: 注解只保留在源文件,当Java文件编译成class文件的时候,注解被遗弃;
由此可见生命周期关系:SOURCE < CLASS < RUNTIME,我们一般用RUNTIME

  • @Target: 表示该注解的作用范围,是ElementType类型的,该类型是一个枚举类型,一共提供了10个值选择,我们最常用的几个:FIELD、TYPE、PARAMETER、METHOD

ElementType.FIELD:用于字段、枚举的常量
ElementType.TYPE:用于接口、类、枚举、注解
ElementType.PARAMETER:用于方法参数
ElementType.METHOD:用于方法
在KingMouse这个类里面可以根据实际需求写上所需要的属性,到这里,我们的第一步就完成!

二、定义注解行为

这一步就是我们需要如何去处理我们的注解,这里面有四个方法,分别是@Before、@after、@Around、AfterReturning、AfterThrowing,我们常用的一般是前三个,看具体需求选择适合自己的就行

@Before: 前置通知, 在方法执行之前执行,这个通知不能阻止连接点前的执行(除非它抛出一个异常)。
@After: 后置通知, 在方法执行之后执行(不论是正常返回还是异常退出)。
@Around: 包围一个连接点(join point)的通知,如方法调用。这是最强大的一种通知类型。 环绕通知可以在方法调用前后完成自定义的行为。它也会选择是否继续执行连接点或直接返回它们自己的返回值或抛出异常来结束执行。
@AfterRunning:返回通知, 在方法正常返回结果之后执行 。
@AfterThrowing: 异常通知, 在方法抛出异常之后。

这里演示@Before、@after

package com.disp.mindmatrix.annotations;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class MindMatrixAspect {

    private final static Logger LOGGER = LoggerFactory.getLogger(MindMatrixAspect.class);

    @Before("@annotation(mindMatrix)")
    public void doBefore(MindMatrix mindMatrix) {
        String value = mindMatrix.value();
        LOGGER.info("@MindMatrix before msg: [{}]", value);
        //do something
    }


    @After(value = "@annotation(mindMatrix)")
    public void testAround(MindMatrix mindMatrix) {
        String value = mindMatrix.value();
        LOGGER.info("@MindMatrix around msg: [{}]", value);
        //do something
    }
}

这个类里面我们就可以做我们想做的事情,我这里就直接打印mindMatrix中的值代替了,上面这两步做完,我们的自定义注解就完成了,剩下的就是我们如何去使用了

三、验证测试

我们直接将mindMatrix这个注解写到方法上,代码!

package com.disp.mindmatrix.controller;


import lombok.extern.java.Log;
import com.disp.mindmatrix.annotations.MindMatrix;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * Created with IntelliJ IDEA
 *
 * @auther MindMatrix
 * @date 2023/1/11 11:26
 * Description:
 */
@Log
@Controller
@RequestMapping("/peisn")
public class TestController {
    @RequestMapping("/test")
    @KingMouse(value = "自定义注解-controller")
    public void test(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse){
        log.info("MindMatrix- test");
        test2();
    }

    @KingMouse(value = "自定义注解-method")
    public void test2(){
        log.info("MindMatrix- test2");
    }
}

运行日志:

INFO  o.d.k.annotations.MindMatrixAspect[21] [TxId :  , SpanId : ]- @KingMouse before msg: [自定义注解-controller]
INFO  o.d.k.controller.TestController[26] [TxId :  , SpanId : ]- MindMatrix - test
INFO  o.d.k.controller.TestController[32] [TxId :  , SpanId : ]- MindMatrix - test2
INFO  o.d.k.annotations.MindMatrixAspect[29] [TxId :  , SpanId : ]- @MindMatrix around msg: [自定义注解-controller]

这里面我们可以清楚的看到@KingMouse before msg: [自定义注解-controller]和@KingMouse around msg: [自定义注解-controller]分别对应@Before、@After两个方法,看到这里,有不少小伙伴应该会发现,为什么自定义注解-method这个内容没有打印,但是看日志,test2方法命名是执行了的,这是为什么?其实,注解的本质还是aop,所有当我们发起调用的时候,是可以拦截的,但是我们通过类直接调用方法是不行的,如果需要调用方法也可以被拦截,有两种方法:

  • 添加切面. 代码:
package com.disp.mindmatrix.annotations;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class MindMatrixAspect {

    private final static Logger LOGGER = LoggerFactory.getLogger(MindMatrixAspect.class);

   @Pointcut("@annotation(com.disp.mindmatrix.annotations.MindMatrix)")
   public void syncMindMatrix() {
   }

    @Before("syncMindMatrix()")
    public void doBefore(JoinPoint joinPoint) {
        MindMatrix mindMatrix = getMindMatrixAnnotation(joinPoint);
        String value = mindMatrix.value();
        LOGGER.info("@MindMatrix before msg: [{}]", value);
        //do something
    }


    @After(value = "syncMindMatrix() && @annotation(mindMatrix)")
    public void testAround(MindMatrix mindMatrix) {
        String value = mindMatrix.value();
        LOGGER.info("@MindMatrix around msg: [{}]", value);
        //do something
    }

 @AfterReturning(pointcut = "syncMindMatrix()", returning = "rst")
    public void afterRunning(JoinPoint joinPoint,String rst) {
       
        System.out.println("方法执行完执行...afterRunning");
        System.out.println("返回数据:{}" + rst);

        System.out.println("方法执行完执行...afterRunning,执行完毕了:" + rst);
    }

    @AfterThrowing(value = "syncMindMatrix()")
    public void afterThrowing(JoinPoint joinPoint) {
        System.out.println("异常出现之后...afterThrowing");
    }

    private MindMatrix getMindMatrixAnnotation(JoinPoint joinPoint) {
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        Method method = methodSignature.getMethod();

        Signature sig = joinPoint.getSignature();
        if (!(sig instanceof MethodSignature)) {
            throw new IllegalArgumentException("该注解只能用于方法");
        }

        MindMatrix mindMatrix = method.getAnnotation(MindMatrix.class);
        return mindMatrix;
    }
}
  • 改变方法引用. 代码:
 * Created with IntelliJ IDEA
 *
 * @auther MindMatrix
 * @date 2023/1/11 11:26
 * Description:
 */
@Log
@Controller
@RequestMapping("/peisn")
public class TestController {
    @RequestMapping("/test")
    @KingMouse(value = "自定义注解-controller")
    public void test(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse){
        log.info("MindMatrix- test");
        ((TestController) AopContext.currentProxy()).test2();
    }

    @KingMouse(value = "自定义注解-method")
    public void test2(){
        log.info("MindMatrix- test2");
    }
}

这步改完,我们重启服务,测试,结果!!!!报错了!!
报错信息:

ERROR o.a.c.c.C.[.[.[.[dispatcherServlet][175] [TxId :  , SpanId : ]- Servlet.service() for servlet [dispatcherServlet] in context with path [] threw exception [Request processing failed; nested exception is java.lang.IllegalStateException: Cannot find current proxy: Set 'exposeProxy' property on Advised to 'true' to make it available.] with root cause
java.lang.IllegalStateException: Cannot find current proxy: Set 'exposeProxy' property on Advised to 'true' to make it available.
    at org.springframework.aop.framework.AopContext.currentProxy(AopContext.java:69)
    at com.disp.mindmatrix.controller.TestController.test(TestController.java:28)

通过代码我们不难看出,被禁用了,其实springboot默认是关闭的,不允许我们这么用,打开就好了,我们只需要在我们的启动类上增加@EnableAspectJAutoProxy(exposeProxy=true,proxyTargetClass=true),上代码!

package com.disp.mindmatrix;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.EnableAspectJAutoProxy;

@SpringBootApplication
@EnableAspectJAutoProxy(exposeProxy = true, proxyTargetClass = true)
public class MindMatrixApplication {

    public static void main(String[] args) {
        SpringApplication.run(MindMatrixApplication.class, args);
    }

}

改完重启服务测试,运行结果:

INFO  o.d.k.annotations.MindMatrixAspect[22] [TxId :  , SpanId : ]- @MindMatrix before msg: [自定义注解-controller]
INFO  o.d.k.controller.TestController[27] [TxId :  , SpanId : ]- MindMatrix - test
INFO  o.d.k.annotations.MindMatrixAspect[22] [TxId :  , SpanId : ]- @MindMatrix before msg: [自定义注解-method]
INFO  o.d.k.controller.TestController[33] [TxId :  , SpanId : ]- MindMatrix - test2
INFO  o.d.k.annotations.MindMatrixAspect[30] [TxId :  , SpanId : ]- @MindMatrix around msg: [自定义注解-method]
INFO  o.d.k.annotations.MindMatrixAspect[30] [TxId :  , SpanId : ]- @MindMatrix around msg: [自定义注解-controller]

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容