如何确定在运行时调用了哪个类的“main”方法？

Question

我想动态确定调用哪个类的主要方法，以便更容易消化组合日志文件。

目前，单个（轮换）日志文件聚合了多个守护进程的所有日志输出，但没有明显的方法来确定日志条目源自哪个守护进程，因为所有守护进程都使用共享代码库，并且记录器是使用 log4j 的 getLogger(Something.class) 创建

由于我们首先使用自定义 Layout 类，因此实际上输出信息不是问题，但查找信息才是问题。

一种可以作为后备的方法是在调用时定义一个属性并读取该属性。

java -cp ... -Dmain.program=<WHATEVER> MainProgram

但是，如果该功能已经存在，则无需创建新的约定。

更新：就我的目的而言，以下内容似乎工作正常：

import org.apache.log4j.PatternLayout;
import org.apache.log4j.spi.LoggingEvent;
public class MyLayout extends PatternLayout {
  private static String _mainClass = null;
  public String format( LoggingEvent event ) {
    String mesg = super.format( event );
    if (mesg.indexOf("$main") > -1) {
      mesg = mesg.replaceAll("\\$main", getMainClass());
    }
    return mesg;
  }
  private static String getMainClass() {
    if (_mainClass == null) {
      StackTraceElement[] elem = new Exception().getStackTrace();
      int offset = elem.length - 1;
      if (elem[offset].getMethodName().equals("main")) {
        _mainClass = elem[offset].getClassName();
      }
      else {
        _mainClass = "<Unknown_Main_Class>";
      }
    }
    return _mainClass;
  }
}

感谢您的建议！

Answer 1

如果您只需要执行一次，您可以遍历异常堆栈并查看最后一个类/方法调用。

    StackTraceElement[] elem = new Exception().getStackTrace();
    elem[elem.length - 1].getClassName();

但很容易出错。如果我通过反射加载你的类，你会在顶部看到一个完全不同的方法。

您可以尝试 M. Jessup 变体（匹配 main 方法签名），但如果我也从代码中调用 main 方法，它将失败。

Answer 2

这有点 hackish，但您可以使用静态方法 Thread.getAllStackTraces()。这将为您提供虚拟机中每个活动线程的堆栈跟踪，并假设启动应用程序的线程仍然活动，您可以检查跟踪并查找其方法签名与 main(String[] args) 匹配的底部元素。

Answer 3

Veera Sundar 撰写了两篇关于如何使用 Log4j 的映射诊断上下文（Servlet 过滤器的源代码）的文章 http://veerasundar.com/blog/2009/11/log4j-mdc-mapped-diagnostic-context-example-code/ 这可能会根据您的用例进行修改。

让您的应用程序类（使用 main 方法）将其类名添加为可以写入日志文件的变量。

是不是比在运行时添加属性要少一些工作？不，不是真的，但是，更优雅