Java注解 案例——简单的测试框架

需求：简单的测试框架 判断calculator类的方法是否有问题 当主方法被执行后，会自动执行被检测的所有方法（加了check的注解方法），判断方法是否有异常，记录到文件中，效果如下图所示

package cn.cast.annotation.demo;

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


//执行testcheck框架，来对加了@check的方法测试，来看这个calculator类是否有异常，有几个异常，发生了什么异常
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Check {

}

package cn.cast.annotation.demo;

public class Calculator {
    //加法
    @Check
    public void add(){
        String str=null;
        str.toString();
        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 div(){
        System.out.println("1/0="+(1/0));
    }

    public void show(){
        System.out.println("永无bug...");
    }
}

package cn.cast.annotation.demo;

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

/**
 *简单的测试框架 判断calculator类的方法是否有问题
 *
 * 当主方法被执行后，会自动执行被检测的所有方法（加了check的注解方法），判断方法是否有异常，记录到文件中
 */
public class TestCheck {
    public static void main(String[] args) throws IOException {
        //1.创建计算器对象
        Calculator c=new Calculator();
        //2.获取字节码文件对象
        Class cls=c.getClass();
        //3.获取所有方法
        Method[] methods=cls.getMethods();
        int number=0;//出现异常的次数
        BufferedWriter bw=new BufferedWriter(new FileWriter("bug.txt"));

        for (Method method:methods){
            //4.判断方法上是否有Check注解
            if(method.isAnnotationPresent(Check.class)){
                //5.有，执行
                try {
                    method.invoke(c);
                } catch (Exception e) {
                    //6.捕获异常
                    //记录到文件中
                    number++;
                    bw.write(method.getName()+"方法出异常了");
                    bw.newLine();
                    bw.write("异常的名称"+e.getCause().getClass().getSimpleName());
                    bw.newLine();
                    bw.write("异常的原因"+e.getCause().getMessage());
                    bw.newLine();
                    bw.write("-----------------------------");
                    bw.newLine();
                }
            }
        }
        bw.write("本次测试一共出现"+number+"次异常");
        bw.flush();
        bw.close();



    }

}