Java注解案例-簡單測試架構

目錄

​​一、需求​​

​​Calculator類：​​

​​Check注解：​​

​​TestCheck類：​​

​​二、運作結果​​

​​控制台：​​

​​記錄異常的檔案(bug.txt)：​​

​​三、結論​​

一、需求

需求：當主方法執行後，會自動執行被檢測的所有方法(加了Check注解的方法)，判斷方法是否有異常，記錄到檔案中

Calculator類：

import demo.Check;

/*
    小明定義的計算機類
 */
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...");
    }
}      

Check注解：

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.METHOD)
public @interface Check {
}      

TestCheck類：

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

/**
 * 簡單的測試架構
 *
 * 當主方法執行後，會自動執行被檢測的所有方法(加了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)) {
                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();
    }
}      

二、運作結果

控制台：

 記錄異常的檔案(bug.txt)：

三、結論