spark提交应用的全流程分析
@(SPARK)[spark]
本文分析一下spark的应用通过spark-submit后，如何提交到集群中并开始运行。
先介绍一下spark从提交到运行的全流程，下面再详细分析。
- 1、用户通过spark-submit脚本提交应用。
- 2、spark-submit根据用户代码及配置确定使用哪个资源管理器，以及在合适的位置启动driver。
- 3、driver与集群管理器(如YARN)通信，申请资源以启动executor。
- 4、集群管理器启动executor。
- 5、driver进程执行用户的代码，根据程序中定义的transformation和action，进行stage的划分，然后以task的形式发送到executor。（通过DAGScheduler划分stage，通过TaskScheduler和TaskSchedulerBackend来真正申请资源运行task）
- 6、task在executor中进行计算并保存结果。
- 7、如果driver中的main()方法执行完成退出，或者调用了SparkContext#stop()，driver会终止executor进程，并且通过集群管理器释放资源。
一、提交前准备
（一）脚本调用
1、spark-submit
spark通过spark-submit脚本来向集群提交应用，举个例子：
/home/hadoop/spark/bin/spark-submit --master yarn-client --num-executors 10 --class com.lujinhong.spark.ml.TrainModel myusml-0.0.1-SNAPSHOT.jar args1 args2 args3
- 1
- 2
我们看看spark-submit脚本，很简单，只有3行：
SPARK_HOME="$(cd "`dirname "$0"`"/..; pwd)"
# disable randomized hash for string in Python 3.3+
export PYTHONHASHSEED=0
exec "$SPARK_HOME"/bin/spark-class org.apache.spark.deploy.SparkSubmit "$@"
- 1
- 2
- 3
- 4
- 5
- 6
- 7
其实就是调用spark-class这个脚本。
2、spark-class
spark-class完成了配置的加载：
. "$SPARK_HOME"/bin/load-spark-env.sh
- 1
- 2
以及调用上面说的SparkSubmit类
3、load-spark-env
有兴趣的可以看看如何加载配置，主要是spark-evn.sh文件，以及scala的版本等。
（二）SparkSubmit
1、main函数
很简单，appArgs解释命令行中的参数，然后判断action是什么，并执行相应的操作。
def main(args: Array[String]): Unit = {
val appArgs = new SparkSubmitArguments(args)
if (appArgs.verbose) {
// scalastyle:off println
printStream.println(appArgs)
// scalastyle:on println
}
appArgs.action match {
case SparkSubmitAction.SUBMIT => submit(appArgs)
case SparkSubmitAction.KILL => kill(appArgs)
case SparkSubmitAction.REQUEST_STATUS => requestStatus(appArgs)
}
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
verbose是一个布尔值，用于确定是否打印一些JVM的信息，默认为false。
action的默认值是submit，我们这里也只分析submit的过程，因此下面将进入submit函数看目的地。
2、submit(args: SparkSubmitArguments): Unit
submit函数先是将参数转化为一个4元组的形式：
val (childArgs, childClasspath, sysProps, childMainClass) = prepareSubmitEnvironment(args)
- 1
- 2
然后就使用这些参数调用runMain函数了：
runMain(childArgs, childClasspath, sysProps, childMainClass, args.verbose)
- 1
- 2
其它内容分别考虑了使用proxy以及standalone的情形。下一步：runMain函数。
3、runMain函数
runMain函数开始执行Client类中的main函数了。
首先是一大堆的环境变量及参数的加载，判断类是否存在等，最后的目的是执行Client类中的main函数。
找到主类：
mainClass = Utils.classForName(childMainClass)
- 1
- 2
然后是主函数：
val mainMethod = mainClass.getMethod("main", new Array[String](0).getClass)
- 1
- 2
最后调用main方法：
mainMethod.invoke(null, childArgs.toArray)
- 1
- 2
那mainClass是哪个类呢？对于yarn-cluster来说，是：
if (isYarnCluster)
childMainClass = "org.apache.spark.deploy.yarn.Client"
- 1
- 2
- 3
如果是yarn-client，是：
// In client mode, launch the application main class directly
// In addition, add the main application jar and any added jars (if any) to the classpath
if (deployMode == CLIENT)
childMainClass = args.mainClass
- 1
- 2
- 3
- 4
- 5
即，client类就是用户定义的主类，直接开始运行主类即可。
二、提交应用
（一）yarn-cluster方式
我们先看一下yarn-cluster方式，由上面的分析可知，yarn-cluster使用的是org.apache.spark.deploy.yarn.Client这个类进行任务提交，先看一下流程图： 
图片来自于spark技术内幕P84，下同。
先说一下总体的流程步骤：
＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝
步骤一：Client类提交应用到YARN ResourceManager，向RM申请资源作为AM
步骤二：在申请到的机器中启动driver，注册成为AM，并调用用户代码，然后创建SparkContext。（driver是一个逻辑概念，并不实际存在，通过抽象出driver这一层，所有的运行模式都可以说是在driver中调用用户代码了）
步骤三：SparkContext中创建DAGScheduler与YarnClusterScheduler与YarnClusterSchedulerBackend。当在用户代码中遇到action时，即会调用DAGScheduler的runJob，任务开始调度执行。
步骤四：YarnClusterSchedulerBackend在NodeManager上启动Executor
步骤五：Executor启动Task，开始执行任务
＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝
简单的说就是：
向RM申请资源建立driver——->在driver中执行用户代码，并创建AM——->遇到action时调用runJob——->开始调度、执行的过程了
因此3个比较复杂的流程分别为：
* 1、如何向YARN中申请资源，这涉及YARN的源码
* 2、如何调度，涉及DAGScheduler、YarnClusterScheduler与YarnClusterSchedulerBackend
* 3、如何执行任务，涉及Executor与Task。这3个部分会有专门的章节来讨论，我们这里先把整个流程理顺。
下面按按被调用的类来详细分析一下：
1、Client
Client类作为向YARN提交应用的客户端
步骤一：Client类提交应用到YARN ResourceManager，向RM申请资源作为 AM
（1）main函数
我们从main函数开始入手：
def main(argStrings: Array[String]) {
.....
new Client(args, sparkConf).run()
}
- 1
- 2
- 3
- 4
- 5
将不关键代码去掉后，就剩下一行，它调用run方法，继续看run方法
（2）run方法
好吧，它的主要内容也只是一行：
def run(): Unit = {
val appId = submitApplication()
.......
}
- 1
- 2
- 3
- 4
- 5
它调用了submitApplication方法。
（3）submitApplication方法
def submitApplication(): ApplicationId = {
var appId: ApplicationId = null
try {
// Setup the credentials before doing anything else,
// so we have don't have issues at any point.
setupCredentials()
yarnClient.init(yarnConf)
yarnClient.start()
// Get a new application from our RM
val newApp = yarnClient.createApplication()
val newAppResponse = newApp.getNewApplicationResponse()
appId = newAppResponse.getApplicationId()
// Verify whether the cluster has enough resources for our AM
verifyClusterResources(newAppResponse)
// Set up the appropriate contexts to launch our AM
val containerContext = createContainerLaunchContext(newAppResponse)
val appContext = createApplicationSubmissionContext(newApp, containerContext)
// Finally, submit and monitor the application
yarnClient.submitApplication(appContext)
appId
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
在submitApplication方法中，先对yarnClient进行了初始化，并从RM中申请到一个application，设置合适的AM（见下一点），最后就向RM提交应用了，并返回应用的ID。
（4）createContainerLaunchContext方法
上面在启动一个应用前，调用了createContainerLaunchContext方法，用于指定的appContext使用哪个AM：
val amClass =
if (isClusterMode) {
Utils.classForName("org.apache.spark.deploy.yarn.ApplicationMaster").getName
} else {
Utils.classForName("org.apache.spark.deploy.yarn.ExecutorLauncher").getName
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
上面代码中指定了当yarn-cluster模式和yarn-client时，分别使用哪个类作为AM。
当向RM提交应用后，RM就会开始启动AM。YARN中启动AM的源码分析以后再补充。
步骤二：在申请到的机器中启动driver，注册成为AM，并调用用户代码，然后创建SparkContext。
2、ApplicationMaster
（1）main函数
当RM启动AM后，AM就开始执行main函数了
def main(args: Array[String]): Unit = {
val amArgs = new ApplicationMasterArguments(args)
SparkHadoopUtil.get.runAsSparkUser { () =>
master = new ApplicationMaster(amArgs, new YarnRMClient(amArgs))
System.exit(master.run())
}
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
关键是调用了run方法，我们继续看run方法。
（2）run方法
先是设置了一些参数，并加载yarn的配置文件。然后设置了一些钩子
最后关键是执行了这2个方法：
if (isClusterMode) {
runDriver(securityMgr)
} else {
runExecutorLauncher(securityMgr)
}
- 1
- 2
- 3
- 4
- 5
- 6
分别对应yarn-cluster模式和yarn-client模式。
（3）runDriver方法
定义了如何启动driver，这也是yarn-cluster和yarn-client最大的区别，前者在yarn分配一台机器启动driver，并注册成为AM，而后者在本地上启动driver，再注册成为AM。
private def runDriver(securityMgr: SecurityManager): Unit = {
addAmIpFilter()
userClassThread = startUserApplication()
// This a bit hacky, but we need to wait until the spark.driver.port property has
// been set by the Thread executing the user class.
val sc = waitForSparkContextInitialized()
// If there is no SparkContext at this point, just fail the app.
if (sc == null) {
finish(FinalApplicationStatus.FAILED,
ApplicationMaster.EXIT_SC_NOT_INITED,
"Timed out waiting for SparkContext.")
} else {
rpcEnv = sc.env.rpcEnv
val driverRef = runAMEndpoint(
sc.getConf.get("spark.driver.host"),
sc.getConf.get("spark.driver.port"),
isClusterMode = true)
registerAM(rpcEnv, driverRef, sc.ui.map(_.appUIAddress).getOrElse(""), securityMgr)
userClassThread.join()
}
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
startUserApplication主要执行了调用用户的代码，以及创建了一个spark driver的进程。
Start the user class, which contains the spark driver, in a separate Thread.
registerAM向RM中正式注册AM。有了AM以后，用户代码就可以执行了，开始将任务切分、调度、执行。我们继续往下看。
然后，用户代码中的action会调用SparkContext的runJob，SparkContext中有很多个runJob，但最后都是调用DAGScheduler的runJob
步骤三：SparkContext中创建DAGScheduler与YarnClusterScheduler与YarnClusterSchedulerBackend
val (sched, ts) = SparkContext.createTaskScheduler(this, master)
_schedulerBackend = sched
_taskScheduler = ts
_dagScheduler = new DAGScheduler(this)
- 1
- 2
- 3
- 4
- 5
然后调用DAGScheduler的runJob：
* Run a function on a given set of partitions in an RDD and pass the results to the given handler function. This is the main entry point for all actions in Spark.*
def runJob[T, U: ClassTag](
rdd: RDD[T],
func: (TaskContext, Iterator[T]) => U,
partitions: Seq[Int],
resultHandler: (Int, U) => Unit): Unit = {
if (stopped.get()) {
throw new IllegalStateException("SparkContext has been shutdown")
}
val callSite = getCallSite
val cleanedFunc = clean(func)
logInfo("Starting job: " + callSite.shortForm)
if (conf.getBoolean("spark.logLineage", false)) {
logInfo("RDD's recursive dependencies:\n" + rdd.toDebugString)
}
dagScheduler.runJob(rdd, cleanedFunc, partitions, callSite, resultHandler, localProperties.get)
progressBar.foreach(_.finishAll())
rdd.doCheckpoint()
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
至此，应用就正式提交到集群准备运行了。
然后就开始DAGScheduler调用YarnClusterScheduler，YarnClusterScheduler调用YarnClusterSchedulerBackend，Executor启动Task开始执行任务的具体流程了。* 这些内容在之后的专题中详细分析。*
步骤四：YarnClusterSchedulerBackend在NodeManager上启动Executor
步骤五：Executor启动Task
（二）yarn-cluster方式
yarn-client的流程与yarn-cluster类似，主要区别在于它在本地运行driver，而cluster是在AM上运行driver。
先看一下流程图： 
1、区别一：主类入口不同
如果是yarn-client，是：
// In client mode, launch the application main class directly
// In addition, add the main application jar and any added jars (if any) to the classpath
if (deployMode == CLIENT)
childMainClass = args.mainClass
- 1
- 2
- 3
- 4
- 5
即，client类就是用户定义的主类，直接开始运行主类即可。
cluster是在专门的Client类中开始执行的，而yarn-client是在用户代码中开始执行的。
2、启动driver的方式不一样
client模式将在本机启动进程，并注册成为AM。
if (isClusterMode) {
runDriver(securityMgr)
} else {
runExecutorLauncher(securityMgr)
}
private def runExecutorLauncher(securityMgr: SecurityManager): Unit = {
val port = sparkConf.getInt("spark.yarn.am.port", 0)
rpcEnv = RpcEnv.create("sparkYarnAM", Utils.localHostName, port, sparkConf, securityMgr)
val driverRef = waitForSparkDriver()
addAmIpFilter()
registerAM(rpcEnv, driverRef, sparkConf.get("spark.driver.appUIAddress", ""), securityMgr)
// In client mode the actor will stop the reporter thread.
reporterThread.join()
}
扫一扫
950
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
