【工作随笔：五】记一个通过线程堆栈控制方法调用的思路

最新推荐文章于 2024-09-15 19:46:31 发布

RiceZongzi

最新推荐文章于 2024-09-15 19:46:31 发布

阅读量209

点赞数

分类专栏：工作随笔文章标签： java

本文链接：https://blog.csdn.net/weixin_44049867/article/details/120339765

版权

工作随笔专栏收录该内容

5 篇文章 0 订阅

订阅专栏

本文介绍了如何在Java中通过线程堆栈（Thread.getStackTrace()）来控制方法调用，以防调用者绕过指定API直接调用底层代码。文章通过模拟工具类和调用者，展示了如何增加判断并在运行时检查调用路径，确保正确使用API。

摘要由CSDN通过智能技术生成

文章目录

前言

有这样一种场景，某些代码存在多层封装而只对外暴露一个调用API。但要考虑调用者绕过这个API调用的可能。

模拟工具类

public class ThreadUtils {

    public static String hello() {
        return "Hello";
    }
}

模拟调用者

public class TestA {

    public static void main(String[] args) {
        System.out.println(ThreadUtils.hello());
    }
}

public class TestB {

    public static void main(String[] args) {
        System.out.println(ThreadUtils.hello());
    }
}

增加判断

考虑通过增加一个isAuth方法，来控制调用

    public static String hello() throws Exception {
        if (!isAuth()) {
            throw new Exception("NOT ALLOW！");
        }
        return "Hello";
    }

    private static boolean isAuth() {
        return false;
    }

Java Thread getStackTrace()方法

方法返回一个代表该线程的堆栈转储堆栈跟踪元素的数组。
这将返回一个零长度数组，如果该线程尚未启动或已经终止。如果返回的数组是非零长度则该数组的第一个元素代表堆栈，这是该序列中的最新的方法调用的顶部。所述阵列的最后一个元素代表堆栈，这是该序列中的至少最近的方法调用的底部。

所以

    private static boolean isAuth() {
        Thread thread = Thread.currentThread();
        StackTraceElement[] stackTrace = thread.getStackTrace();
        for (int i = 3; i < stackTrace.length; i++) {
            if (allow.contains(stackTrace[i].getClassName())) {
                return true;
            }
        }
        return false;
    }

效果展示

TestA.java执行
Hello

Process finished with exit code 0

TestB.java执行
Exception in thread "main" java.lang.Exception: NOT ALLOW！
	at com.rice.util.ThreadUtils.hello(ThreadUtils.java:18)
	at com.rice.test.TestB.main(TestB.java:11)

Process finished with exit code 1