Java-自定义注解的案例

自定义一个Check注解，实现对一个类中方法运行时bug的检查及捕获

• Calculator

public class Calculator {
    @Check
    public void add() {
        System.out.println("1+0=" + (1 + 0));
    }

    @Check
    public void sub() {
        System.out.println("1-0=" + (1 - 0));
    }

    @Check
    public void mul() {
        System.out.println("1*0=" + (1 * 0));
    }

    @Check
    public void dev() {
        System.out.println("1/0=" + (1 / 0));
    }

    @Check
    public void show() {
        String arr = null;
        System.out.println(arr.toString());
        System.out.println("永无bug");
    }
}

• Check

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

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Check {
}

• CheckTest

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.Method;

public class CheckTest {
    public static void main(String[] args) throws IOException {
        //创建计算器对象
        Calculator calculator = new Calculator();
        //获取字节码文件对象
        Class calculatorClass = calculator.getClass();
        //获取计算器所有方法
        Method[] calculatorClassMethods = calculatorClass.getMethods();
        int number = 0;
        BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter("bug.txt"));
        //遍历计算器的方法
        for (Method calculatorClassMethod : calculatorClassMethods) {
            //查看方法是否有Check注解
            if (calculatorClassMethod.isAnnotationPresent(Check.class)) {
                //有的话，执行方法
                try {
                    calculatorClassMethod.invoke(calculator);
                } catch (Exception e) {
                    //没有的话，捕获异常，并且将异常信息输出到文件中
                    number++;
                    bufferedWriter.write(calculatorClassMethod.getName() + "方法出异常了");
                    bufferedWriter.newLine();
                    bufferedWriter.write("异常的名称：" + e.getCause().getClass().getSimpleName());
                    bufferedWriter.newLine();
                    bufferedWriter.write("异常的原因：" + e.getCause().getMessage());
                    bufferedWriter.newLine();
                    bufferedWriter.write("----------------");
                    bufferedWriter.newLine();
                }
            }
        }
        bufferedWriter.write("总共有" + number + "个bug");
        bufferedWriter.close();
    }
}

运行结果：

show方法出异常了
异常的名称：NullPointerException
异常的原因：null
----------------
dev方法出异常了
异常的名称：ArithmeticException
异常的原因：/ by zero
----------------
总共有2个bug