Hive SQL运行状态监控(HiveSQLMonitor)

引言


 


目前数据平台使用Hadoop构建,为了方便数据分析师的工作,使用Hive对Hadoop MapReduce任务进行封装,我们面对的不再是一个个的MR任务,而是一条条的SQL语句。数据平台内部通过类似JDBC的接口与HiveServer进行交互,仅仅能够感知到一条SQL的开始与结束,而中间的这个过程通常是漫长的(两个因素:数据量、SQL复杂度),某些场景下用户需要了解这条SQL语句的执行进度,从而为我们引入以下几个问题:


 


(1)通过JDBC接口执行一条SQL语句时,这条SQL语句被转换成几个MR任务,每个MR任务的JobId是多少,如何维护这条SQL语句与MR任务的对应关系?


(2)如何获取MR任务的运行状态,通过JobClient?


(3)通过HiveServer是否可以获取到上述信息?


 


思路


 


当我们在终端下执行命令“hive”后,会看到有如下输出:


 

Hive Metrics 监控指标_Hive Metrics 监控指标




 


Hive有会话(Session)的概念,而 这次会话中的所有日志消息将会输出到这个日志文件中,包含SQL语句的执行日志,查看这个日志文件可以看到以下信息:


 


 

Hive Metrics 监控指标_SQL_02




 


QueryStart行日志包含QUERY_STRING、QUERY_ID。


 



Hive Metrics 监控指标_数据_03




 


TaskStart行日志包含TASK_ID、QUERY_ID。


 



Hive Metrics 监控指标_数据_04




 


TaskProgress行日志包含TASK_HADOOP_PROGRESS、TASK_ID、QUERY_ID、TASK_HADOOP_ID,其中TASK_HADOOP_PROGRESS中可以获取到map、reduce进度。


 



Hive Metrics 监控指标_Hive_05




 


TaskEnd行日志包含TASK_HADOOP_PROGRESS、TASK_ID、QUERY_ID、TASK_HADOOP_ID。


 



Hive Metrics 监控指标_Hive Metrics 监控指标_06




 


QueryEnd行日志包含QUERY_STRING、QUERY_ID。


 


由上可知,QueryStart、TaskStart、TaskProgress、TaskEnd(一个复杂的Query可能会产生多个Task)、QueryEnd覆盖整个查询的执行过程,通过对这些行日志的解析,我们就可以获取到Hive SQL的执行状态。


 


此外,还有SessionStart、SessionEnd,由于使用过程中发现SessionEnd日志有时不被输出,因此没有使用这两个状态。


 


会话的日志文件存储在HiveServer的本地磁盘中,而实际应用中我们有多台HiveServer提供服务,因此我们需要能够统一收集所有HiveServer的会话日志。


 


通过对Hive源码的分析发现,每次Hive执行语句时都会执行一些“Hook”(PreHook),代码如下:


 



Hive Metrics 监控指标_Hive Metrics 监控指标_07




 


通过会话日志、PreHook,我们基本可以整理出以下思路:


 


在PreHook中启动线程监听会话日志的输出(类型Linux的tailf),将这些日志信息统一收集到某一服务中,统一处理后做进度展示。


 


实现


 


我们构建了一个Rest API服务,一部分用于接收由PreHook发送的会话日志信息,另一部分用于对外提供进度展示。


 


PreHook要求实现接口ExecuteWithHookContext,如下:


 



Hive Metrics 监控指标_Hive_08




 


通过hookContext我们可以获取到以下信息:


 


QueryId:


 



Hive Metrics 监控指标_SQL_09




 


QueryStr:


 



Hive Metrics 监控指标_Hive Metrics 监控指标_10




 


HadoopJobName:


 



Hive Metrics 监控指标_SQL_11




 


Jobs:


 



Hive Metrics 监控指标_SQL_12




 


HistFileName:


 



Hive Metrics 监控指标_SQL_13




 


为了保证后续对会话日志的接收,我们需要在查询执行伊始就将上述信息发送给Rest API服务,如下:


 



Hive Metrics 监控指标_Hive_14


 




 


然后就是对会话日志的输出监听(即tailer),我们使用Apache Commons IO中的Tailer完成些功能,如下:


 



Hive Metrics 监控指标_Hive_15




 


Tailer实际上启动一个后台线程,并通过listener完成数据行的处理,而一次会话中可能执行多条查询语句,而每一次执行查询语句时都会导致PreHook的执行,因此我们需要避免同一会话中对histFileName多次“tailf”,需要维护已被“tailf”的文件,而且Tailer实例是需要被“stop”的,多数时候无法获取到SessionEnd数据行,需要通过其它方式能够终止会话已经消失的Tailer线程。为此专门设计了TailerTracker(单例,即TAILER_TRACKER)。


 


TailerTracker维护着一个记录列表:


 




 

Hive Metrics 监控指标_Hive Metrics 监控指标_16


 



Hive Metrics 监控指标_Hive_17




 


维护着成对的tailer与listener实例,其中listener实例中维护着对应tailer实例中最后一次新数据产生的时间,如果tailer实例在设定的时间内都没有新数据产生,则应该对其执行stop,核心代码如下:


 



Hive Metrics 监控指标_Hive Metrics 监控指标_18




 


判断某一个会话文件是否已经被“tailer”,代码如下:


 



Hive Metrics 监控指标_Hive_19




 


标记一个会话文件已经被“tailer”,代码如下:


 



Hive Metrics 监控指标_Hive_20




 


会话日志数据行的输出实际由FileTailerListener(继承自TailerListenerAdapter)完成,代码如下:


 



Hive Metrics 监控指标_SQL_21




 


每处理一行数据,都要更新一下时间戳lastHandleTime,而QueryStart、QueryEnd、TaskStart、TaskProgress、TaskEnd的数据行会通过不同的Rest API Post。


 


至此,HiveServer的会话日志收集过程完毕,而Rest服务则需要通过这些收集到的数据完成Hive SQL进度跟踪。


 


我们在通过JDBC接口与HiveServer交互时,是无法获取到QueryId的,但是我们可以通过属性mapred.job.name设置Hive SQL执行时的MR JobName,JobName代表查询名称,需要唯一,同时我们需要维护JobName与QueryId的对应关系。


 


在Rest服务内部设计实现ProgressController,用以维护JobName与QueryId的对应关系,同时使用QueryId跟踪Hive SQL执行进度,核心变量如下:


 



Hive Metrics 监控指标_Hive Metrics 监控指标_22




 


目前Hive SQL的进度记录仅仅在内存里维护(超过一定时间后,这些进度信息便不再有价值),因此需要控制内存中进度记录的数量,这一点是通过记录每一条SQL相关进度信息的最后更新时间(lastUpdateTime)来实现的,过期即被清除。


 


lastUpdateTime:维护JobName(即某个查询)记录最后更新时间;


 


jobNameToQueryId:维护JobName与QueryId的对应关系;


 


querys:维护QueryId与Hive SQL执行进度(QueryProgress)的对应关系。


 


QueryProgress内部结构如下:


 



Hive Metrics 监控指标_Hive Metrics 监控指标_23




 


queryId:查询ID;


 


sql:查询语句;


 


jobs:查询被转换成MapRecude Job的数量;


 


taskProgresses:维护TaskId与MapReduce的执行进度的对应关系;


 


startTime:查询的起始时间;


 


stopTime:查询的终止时间;


 


state:查询状态。


 


TaskProgress内部结构如下:


 



Hive Metrics 监控指标_数据_24




 


taskId:TaskId(Stage-1、Stage-2、...);


 


taskHadoopId:Task对应的Hadoop MapReduce Job Id;


 


map:Hadoop MapReduce map进度百分比值;


 


reduce:Hadoop MapReduce reduce进度百分比值;


 


startTime:Task起始时间;


 


stopTime:Task截止时间;


 


state:Task运行状态。


 


当收到query/init的请求时,执行ProgressController queryInit方法,代码如下:


 



Hive Metrics 监控指标_Hive_25




 


当收到query/start的请求时,执行ProgressController queryStart方法,代码如下:


 




 



Hive Metrics 监控指标_数据_26


 


当收到task/start的请求时,执行ProgressController taskStart方法,代码如下:


 



Hive Metrics 监控指标_SQL_27




 


当收到task/progress的请求时,执行ProgressController taskProgress方法,代码如下:


 



Hive Metrics 监控指标_Hive_28



Hive Metrics 监控指标_SQL_29






 


当收到task/end的请求时,执行ProgressController taskEnd方法,代码如下:


 



Hive Metrics 监控指标_Hive_30



Hive Metrics 监控指标_Hive_31






 


当收到query/end的请求时,执行ProgressController queryEnd方法,代码如下:


 



Hive Metrics 监控指标_Hive_32




 


其中ProgressController还承担着定时清理的工作,代码如下:


 



Hive Metrics 监控指标_Hive_33




 


进度示例


 



Hive Metrics 监控指标_Hive Metrics 监控指标_34


 



Hive Metrics 监控指标_SQL_35


 


不足


 

Hive SQL执行进度数据维护在内存中,而且Rest服务为单点。