如何在scala中从yarn客户端获取纱线作业状态

import org.apache.hadoop.conf.Configuration
import org.apache.hadoop.yarn.api.records.{ApplicationId, YarnApplicationState}
import org.apache.hadoop.yarn.client.api.YarnClient
import org.apache.hadoop.yarn.util.ConverterUtils

def getYarnJobStatus(applicationIdStr: String): String = {
  val conf = new Configuration()
  val yarnClient = YarnClient.createYarnClient()
  yarnClient.init(conf)
  yarnClient.start()

  val applicationId = ConverterUtils.toApplicationId(applicationIdStr)
  val applicationReport = yarnClient.getApplicationReport(applicationId)

  val state = applicationReport.getYarnApplicationState
  val status = state match {
    case YarnApplicationState.NEW => "New"
    case YarnApplicationState.NEW_SAVING => "New Saving"
    case YarnApplicationState.SUBMITTED => "Submitted"
    case YarnApplicationState.ACCEPTED => "Accepted"
    case YarnApplicationState.RUNNING => "Running"
    case YarnApplicationState.FINISHED => "Finished"
    case YarnApplicationState.FAILED => "Failed"
    case YarnApplicationState.KILLED => "Killed"
    case _ => "Unknown"
  }

  yarnClient.stop()
  status
}

val applicationId = "application_1234567890_1234"
val jobStatus = getYarnJobStatus(applicationId)
println(s"YARN job status: $jobStatus")