Java获取调用当前方法的类名或方法名(栈堆信息)的四种方式举例_Java

前言

在java代码中，是可以在运行时通过某种方式获取到当前方法被谁调用了（调用链路）。目前我所知道的有四种方式（通过thread、throwable、securitymanager获取），下面逐个列出，附有代码和截图。

thread.getallstacktraces()方式

thread.getallstacktraces()可以获取所有线程的栈堆跟踪信息，返回值是map<thread, stacktraceelement[]>类型，可以通过thread.currentthread()拿到当前thread对象，从map集合中获取栈堆跟踪信息。stacktraceelement对象可以拿到调用者的类名、class对象、调用的代码所在行数等。

代码如下：

public class a {
    public static void main(string[] args) {
        b.print();
    }
}

class b {
    public static void print() {
        system.out.println("调用b#print");
        c.print();
    }
}

class c {
    public static void print() {
        system.out.println("调用c#print");
        // 获取所有线程的栈堆跟踪集合
        map<thread, stacktraceelement[]> allstacktraces = thread.getallstacktraces();
        // 去map集合里面拿当前线程的栈堆跟踪信息数组
        stacktraceelement[] stacktraceelements = allstacktraces.get(thread.currentthread());
        system.out.println(arrays.tostring(stacktraceelements));
    }
}

可以看到，下标为4的stacktraceelement对象就是我们的main方法，之所以出现了下标0、1非自定义类的栈堆信息是因为调用了thread.getallstacktraces()方法，方法以及方法内部也执行了其他代码。

thread.currentthread().getstacktrace()方式

通过thread.currentthread()拿到当前的线程，再通过thread对象的getstacktrace()方法获取线程的栈堆跟踪信息，getstacktrace()非static方法，所以要先拿到thread对象。第一种和第二种方式实际上都是调用了private native static stacktraceelement[][] dumpthreads(thread[] threads)本地方法来获取。这两种方式获取的话，栈堆跟踪信息数组里面都有非自定义类的调用信息（看下标0）。

代码如下：

public class a {
    public static void main(string[] args) {
        b.print();
        system.out.println(arrays.tostring(thread.currentthread().getstacktrace()));
    }
}

class b {
    public static void print() {
        system.out.println("调用b#print");
        c.print();
    }
}

class c {
    public static void print() {
        // 获取当前线程，再获取栈堆跟踪数组
        stacktraceelement[] stacktrace = thread.currentthread().getstacktrace();
        system.out.println(arrays.tostring(stacktrace));
    }
}

new runtimeexception().getstacktrace()方式

实际上，只要是throwable的子类包括他自身都可以，内部也是调用了一个本地方法来获取的栈堆信息。springapplication类中的deducemainapplicationclass()方法就是通过这种方式来拿主类的class对象。拿到了之后就可以通过for循环整个数组通过匹配方法名或者类型来拿想要信息了，或者直接通过整个数组拿到整个调用链路。

代码如下：

public class a {
    public static void main(string[] args) {
        b.print();
    }
}

class b {
    public static void print() {
        system.out.println("调用b#print");
        c.print();
    }
}

class c {
    public static void print() {
        stacktraceelement[] stacktrace = new runtimeexception().getstacktrace();
        system.out.println(arrays.tostring(stacktrace));
    }
}

securitymanager方式（这种可以拿到类信息，但是拿不到栈堆信息）

securitymanager中有一个getclasscontext()方法，这个方法可以获取到调用当前方法的class信息，返回的类型是一个class数组，和前面几种不同的是，这种方式没法获取代码行数、调用的方法等（class对象大家应该很熟悉了，能拿到哪种信息应该都知道）。

由于getclasscontext方法是非static的，所以要通过对象的方式调用，但是不能通过new securitymanager()的对象来获取，因为这个方法是一个protected方法。protected native class[] getclasscontext()。所以需要自定义一个类，继承securitymanager类，然后重写getclasscontext方法，这样我们就能在这个类中使用这个方法了。

这种方式在slf4j中util#getcallingclass()用到了

代码如下：

public class a {
    public static void main(string[] args) {
        b.print();
    }
}

class b {
    public static void print() {
        system.out.println("调用b#print");
        c.print();
    }
}

class c {

    public static void print() {
        // 获取一个安全管理器对象
        classcontextsecuritymanager securitymanager = getsecuritymanager();
        if (securitymanager != null) {
            system.out.println(arrays.tostring(securitymanager.getclasscontext()));
        }
    }

    static classcontextsecuritymanager getsecuritymanager() {
        try {
            return new classcontextsecuritymanager();
        } catch (throwable throwable) {
            return null;
        }
    }

    /**
     * 自定义的安全管理器，继承java.lang.securitymanager类
     */
    private static final class classcontextsecuritymanager extends securitymanager {
    	/**
    	 * 重写getclasscontext方法，不然方法在securitymanager中又被protected修饰，其他类还是调用不了
    	 */
    	@override
        protected class<?>[] getclasscontext() {
            return super.getclasscontext();
        }
    }
}