如何获取 Scala 中当前的脚本或类名？

Question

我希望我的 Scala 程序能够以编程方式确定其脚本文件名或其类名，将字符串存储在变量 program 中并打印出名称。

Java 有几种 方法来实现此目的。

Answer 1

我认为这是最简单的：

val program = new Exception().getStackTrace.head.getFileName

Answer 2

按照 0__ 提示解析堆栈跟踪应该可以处理大多数用例，特别是当具有 main 方法的对象不在同名文件中时：

package utils

trait ProgramInfo {
  val programInfo = try { 
    throw new RuntimeException("x")
  } catch { 
    case e: RuntimeException =>
      val arr = new java.io.CharArrayWriter()
      val buffer = new java.io.PrintWriter(arr)
      e.printStackTrace(buffer)
      val trace = arr.toString 
      val lines = io.Source.fromString(trace)
      val pat = """^.*at.*\.main\(([^:]*)(:.*)?\).*$""".r
      lines.getLines().collectFirst{case pat(n, l) => n}.getOrElse("<none>")
  }
}

object ProgramInfo extends ProgramInfo

然后您可以这样调用：

println(utils.ProgramInfo.programInfo)

或者您混合到您的main object

object A extends utils.ProgramInfo {
   def main(args: Array[String]) {
     println(programInfo)
   }
}

这适用于脚本 scala A.script，无论代码是否包装在对象中。当使用 scalac 编译并作为 scala A 运行时，这也适用。当使用 REPL 运行时，将返回 。

Answer 3

不太确定你在寻找什么......

$ scala -e 'println( "I am " + getClass.getName )'

给了我

"I am Main$anon$1"

并

$ scala -e 'try { sys.error( "" )} catch { case e => println( "I am " + e.getStackTrace()( 3 ))}'

给了我

"I am Main.main(scalacmd2873893687624153305.scala)"

Answer 4

// With help from huynhjl
// http://stackoverflow.com/questions/8129185#8131613

import scala.util.matching.Regex.MatchIterator

object ScriptName {
    val program = {
        val filenames = new RuntimeException("").getStackTrace.map { t => t.getFileName }
        val scala = filenames.indexOf("NativeMethodAccessorImpl.java")

        if (scala == -1)
            "<console>"
        else
            filenames(scala - 1)
    }

    def main(args: Array[String]) {
        val prog = program
        println("Program: " + prog)
    }
}

Rosetta 代码

Answer 5

#!/bin/bash
exec scala "$0" "$0" "$@"
!#

val program = args(0)
println(program)