美文网首页大数据，机器学习，人工智能大数据大数据 爬虫Python AI Sql
Hive Query生命周期--钩子函数篇

Hive Query生命周期--钩子函数篇

作者: 大数据首席数据师 | 来源:发表于2019-02-13 10:57 被阅读20次

    前言

    无论Hive Cli还是HiveServer2，一个HQl语句都要经过Driver进行解析和执行，粗略如下图：

    hive arch|center

    Driver处理的流程如下：

    HQL解析(生成AST语法树) => 语法分析(得到QueryBlock) => 生成逻辑执行计划(Operator) => 逻辑优化(Logical Optimizer Operator) => 生成物理执行计划(Task Plan) => 物理优化(Task Tree) => 构建执行计划(QueryPlan) => 表以及操作鉴权 => 执行引擎执行

    流程涉及HQL解析，HQL编译(语法分析，逻辑计划和物理计划，鉴权)，执行器执行三个大的方面，在整个生命周期中，按执行顺序会有如下钩子函数：

    Driver.run()之前的preDriverRun

    该钩子函数由配置 hive.exec.driver.run.hooks 控制，多个钩子实现类以逗号间隔，钩子需实现 org.apache.hadoop.hive.ql.HiveDriverRunHook 接口，该接口描述如下：

    public interface HiveDriverRunHook extends Hook {
  /**
   * Invoked before Hive begins any processing of a command in the Driver,
   * notably before compilation and any customizable performance logging.
   */
  public void preDriverRun(
    HiveDriverRunHookContext hookContext) throws Exception;

  /**
   * Invoked after Hive performs any processing of a command, just before a
   * response is returned to the entity calling the Driver.
   */
  public void postDriverRun(
    HiveDriverRunHookContext hookContext) throws Exception;
}

    可以看出钩子还提供了 postDriverRun 方法供HQL执行完，数据返回前调用，这个在后面会说到

    其参数在Hive里使用的是 HiveDriverRunHookContext 的默认实现类 org.apache.hadoop.hive.ql.HiveDriverRunHookContextImpl，里面提供了两个有用的参数，分别是HiveConf和要执行的Command，其调用信息如下：

    HiveDriverRunHookContext hookContext = new HiveDriverRunHookContextImpl(conf, command);
// Get all the driver run hooks and pre-execute them.
List<HiveDriverRunHook> driverRunHooks;
try {
  driverRunHooks = getHooks(HiveConf.ConfVars.HIVE_DRIVER_RUN_HOOKS,
      HiveDriverRunHook.class);
  for (HiveDriverRunHook driverRunHook : driverRunHooks) {
      driverRunHook.preDriverRun(hookContext);
  }
} catch (Exception e) {
  errorMessage = "FAILED: Hive Internal Error: " + Utilities.getNameMessage(e);
  SQLState = ErrorMsg.findSQLState(e.getMessage());
  downstreamError = e;
  console.printError(errorMessage + "\n"
      + org.apache.hadoop.util.StringUtils.stringifyException(e));
  return createProcessorResponse(12);
}

    语法分析之前的preAnalyze

    在Driver开始run之后，HQL经过解析会进入编译阶段的语法分析，而在语法分析前会经过钩子 HiveSemanticAnalyzerHook 的 preAnalyze 方法，该钩子函数由 hive.semantic.analyzer.hook 配置，钩子需实现 org.apache.hadoop.hive.ql.parse.HiveSemanticAnalyzerHook 接口，接口描述如下：

    public interface HiveSemanticAnalyzerHook extends Hook {
  public ASTNode preAnalyze(
    HiveSemanticAnalyzerHookContext context,
    ASTNode ast) throws SemanticException;

  public void postAnalyze(
    HiveSemanticAnalyzerHookContext context,
    List<Task<? extends Serializable>> rootTasks) throws SemanticException;
}

    可以看出钩子类还提供了 postAnalyze 方法供语法分析完后调用，这个在后面会提到

    其参数在Hive里使用的是 HiveSemanticAnalyzerHookContext 的默认实现类 org.apache.hadoop.hive.ql.parse.HiveSemanticAnalyzerHookContextImpl，里面提供了HQL对应的输入，输出，提交用户，HiveConf和客户端IP等信息，输入和输出的表及分区信息需要做完语法分析后才能得到，在 preAnalyze 里获取不到，其调用信息如下：

    List<HiveSemanticAnalyzerHook> saHooks =
    getHooks(HiveConf.ConfVars.SEMANTIC_ANALYZER_HOOK,
        HiveSemanticAnalyzerHook.class);

// Do semantic analysis and plan generation
if (saHooks != null) {
  HiveSemanticAnalyzerHookContext hookCtx = new HiveSemanticAnalyzerHookContextImpl();
  hookCtx.setConf(conf);
  hookCtx.setUserName(userName);
  hookCtx.setIpAddress(SessionState.get().getUserIpAddress());
  hookCtx.setCommand(command);
  for (HiveSemanticAnalyzerHook hook : saHooks) {
    tree = hook.preAnalyze(hookCtx, tree);
  }
  // 此处开始进行语法分析，会涉及逻辑执行计划和物理执行计划的生成和优化
  sem.analyze(tree, ctx);
  // 更新分析器以便后续的postAnalyzer钩子执行
  hookCtx.update(sem);
  for (HiveSemanticAnalyzerHook hook : saHooks) {
    hook.postAnalyze(hookCtx, sem.getRootTasks());
  }
} else {
  sem.analyze(tree, ctx);
}

    语法分析之后的postAnalyze

    从 preAnalyze 的分析可以看出，postAnalyze 与其属于同一个钩子类，因此配置也相同，不同的是它位于Hive的语法分析之后，因此可以获取到HQL的输入和输出表及分区信息，以及语法分析得到的Task信息，由此可以判断是否是需要分布式执行的任务，以及执行引擎是什么，具体的代码和配置可见上面的 preAnalyze 分析

    生成执行计划之前的redactor钩子

    这个钩子函数是在语法分析之后，生成QueryPlan之前，所以执行它的时候语法分析已完成，具体要跑的任务已定，这个钩子的目的在于完成QueryString的替换，比如QueryString中包含敏感的表或字段信息，在这里都可以完成替换，从而在Yarn的RM界面或其他方式查询该任务的时候，会显示经过替换后的HQL

    该钩子由 hive.exec.query.redactor.hooks 配置，多个实现类以逗号间隔，钩子需继承 org.apache.hadoop.hive.ql.hooks.Redactor 抽象类，并替换 redactQuery 方法，接口描述如下：

    public abstract class Redactor implements Hook, Configurable {

  private Configuration conf;

  public void setConf(Configuration conf) {
    this.conf = conf;
  }

  public Configuration getConf() {
    return conf;
  }

  /**
   * Implementations may modify the query so that when placed in the job.xml
   * and thus potenially exposed to admin users, the query does not expose
   * sensitive information.
   */
  public String redactQuery(String query) {
    return query;
  }
}

    其调用信息如下：

    public static String redactLogString(HiveConf conf, String logString)
    throws InstantiationException, IllegalAccessException, ClassNotFoundException {

  String redactedString = logString;

  if (conf != null && logString != null) {
    List<Redactor> queryRedactors = getHooks(conf, ConfVars.QUERYREDACTORHOOKS, Redactor.class);
    for (Redactor redactor : queryRedactors) {
      redactor.setConf(conf);
      redactedString = redactor.redactQuery(redactedString);
    }
  }

  return redactedString;
}

    Task执行之前的preExecutionHook

    在执行计划QueryPlan生成完，并通过鉴权后，就会进行具体Task的执行，而Task执行之前会经过一个钩子函数，钩子函数由 hive.exec.pre.hooks 配置，多个钩子实现类以逗号间隔，该钩子的实现方式有两个，分别是：

    一、实现 org.apache.hadoop.hive.ql.hooks.ExecuteWithHookContext 接口

    该接口会传入 org.apache.hadoop.hive.ql.hooks.HookContext 的实例作为参数，而参数类里带有执行计划，HiveConf，Lineage信息，UGI，提交用户名，输入输出表及分区信息等私有变量，为我们实现自己的功能提供了很多帮助

    接口描述如下：

    public interface ExecuteWithHookContext extends Hook {

  void run(HookContext hookContext) throws Exception;
}

    二、实现 org.apache.hadoop.hive.ql.hooks.PreExecute 接口

    该接口传入参数包括SessionState，UGI和HQL的输入输出表及分区信息，目前该接口被标为已过时的接口，相比上面的ExecuteWithHookContext，该接口提供的信息可能不完全能满足我们的需求
    这里向大家推荐一个大数据交流圈q裙：894951460。
    其接口描述如下：

    public interface PreExecute extends Hook {

  /**
   * The run command that is called just before the execution of the query.
   *
   * @param sess
   *          The session state.
   * @param inputs
   *          The set of input tables and partitions.
   * @param outputs
   *          The set of output tables, partitions, local and hdfs directories.
   * @param ugi
   *          The user group security information.
   */
  @Deprecated
  public void run(SessionState sess, Set<ReadEntity> inputs,
      Set<WriteEntity> outputs, UserGroupInformation ugi)
    throws Exception;
}

    该钩子的调用信息如下：

    SessionState ss = SessionState.get();
HookContext hookContext = new HookContext(plan, conf, ctx.getPathToCS(), ss.getUserName(), ss.getUserIpAddress());
hookContext.setHookType(HookContext.HookType.PRE_EXEC_HOOK);

for (Hook peh : getHooks(HiveConf.ConfVars.PREEXECHOOKS)) {
  if (peh instanceof ExecuteWithHookContext) {
    perfLogger.PerfLogBegin(CLASS_NAME, PerfLogger.PRE_HOOK + peh.getClass().getName());

    ((ExecuteWithHookContext) peh).run(hookContext);

    perfLogger.PerfLogEnd(CLASS_NAME, PerfLogger.PRE_HOOK + peh.getClass().getName());
  } else if (peh instanceof PreExecute) {
    perfLogger.PerfLogBegin(CLASS_NAME, PerfLogger.PRE_HOOK + peh.getClass().getName());

    ((PreExecute) peh).run(SessionState.get(), plan.getInputs(), plan.getOutputs(),
        Utils.getUGI());

    perfLogger.PerfLogEnd(CLASS_NAME, PerfLogger.PRE_HOOK + peh.getClass().getName());
  }
}

    Task执行失败时的ON_FAILURE_HOOKS

    Task执行完后，如果执行失败了，那么Hive会调用这个失败的Hook。该钩子由参数 hive.exec.failure.hooks 配置，多个钩子实现类以逗号间隔，钩子需实现 org.apache.hadoop.hive.ql.hooks.ExecuteWithHookContext 接口，该接口在上文已有描述。该钩子主要用于在任务执行失败时执行一些措施，比如统计等等

    该钩子的调用信息如下：

    hookContext.setHookType(HookContext.HookType.ON_FAILURE_HOOK);
// Get all the failure execution hooks and execute them.
for (Hook ofh : getHooks(HiveConf.ConfVars.ONFAILUREHOOKS)) {
  perfLogger.PerfLogBegin(CLASS_NAME, PerfLogger.FAILURE_HOOK + ofh.getClass().getName());

  ((ExecuteWithHookContext) ofh).run(hookContext);

  perfLogger.PerfLogEnd(CLASS_NAME, PerfLogger.FAILURE_HOOK + ofh.getClass().getName());
}

    Task执行完毕的postExecutionHook

    这个钩子是在Task任务执行完毕后执行，如果Task失败，会先执行ON_FAILURE_HOOKS这个钩子，之后执行postExecutionHook，该钩子由参数 hive.exec.post.hooks 配置，多个钩子实现类以逗号间隔，该钩子的实现方式也有两个

    一、实现 org.apache.hadoop.hive.ql.hooks.ExecuteWithHookContext 接口

    这个与preExecutionHook一致

    二、实现 org.apache.hadoop.hive.ql.hooks.PostExecute 接口

    该接口传入参数包括SessionState，UGI，列级的LineageInfo和HQL的输入输出表及分区信息，目前该接口被标为已过时的接口，相比上面的ExecuteWithHookContext，该接口提供的信息可能不完全能满足我们的需求

    其接口描述如下：

    public interface PostExecute extends Hook {

  /**
   * The run command that is called just before the execution of the query.
   *
   * @param sess
   *          The session state.
   * @param inputs
   *          The set of input tables and partitions.
   * @param outputs
   *          The set of output tables, partitions, local and hdfs directories.
   * @param lInfo
   *           The column level lineage information.
   * @param ugi
   *          The user group security information.
   */
  @Deprecated
  void run(SessionState sess, Set<ReadEntity> inputs,
      Set<WriteEntity> outputs, LineageInfo lInfo,
      UserGroupInformation ugi) throws Exception;
}

    该钩子的调用信息如下：

    hookContext.setHookType(HookContext.HookType.POST_EXEC_HOOK);
// Get all the post execution hooks and execute them.
for (Hook peh : getHooks(HiveConf.ConfVars.POSTEXECHOOKS)) {
  if (peh instanceof ExecuteWithHookContext) {
    perfLogger.PerfLogBegin(CLASS_NAME, PerfLogger.POST_HOOK + peh.getClass().getName());

    ((ExecuteWithHookContext) peh).run(hookContext);

    perfLogger.PerfLogEnd(CLASS_NAME, PerfLogger.POST_HOOK + peh.getClass().getName());
  } else if (peh instanceof PostExecute) {
    perfLogger.PerfLogBegin(CLASS_NAME, PerfLogger.POST_HOOK + peh.getClass().getName());

    ((PostExecute) peh).run(SessionState.get(), plan.getInputs(), plan.getOutputs(),
        (SessionState.get() != null ? SessionState.get().getLineageState().getLineageInfo()
            : null), Utils.getUGI());

    perfLogger.PerfLogEnd(CLASS_NAME, PerfLogger.POST_HOOK + peh.getClass().getName());
  }
}

    Task执行完毕结果返回之前的postDriverRun

    该钩子在Task执行完毕，而结果尚未返回之前执行，与preDriverRun相对应，由于是同一个接口，这里不做详细描述

    最后

    至此，整个HQL执行生命周期中的钩子函数都讲完了，执行顺序和流程可梳理如下：

    Driver.run()

    HiveDriverRunHook.preDriverRun()(hive.exec.driver.run.hooks)

    Driver.compile()

    HiveSemanticAnalyzerHook.preAnalyze()(hive.semantic.analyzer.hook)

    SemanticAnalyze(QueryBlock, LogicalPlan, PhyPlan, TaskTree)

    HiveSemanticAnalyzerHook.postAnalyze()(hive.semantic.analyzer.hook)

    QueryString redactor(hive.exec.query.redactor.hooks)

    QueryPlan Generation

    Authorization

    Driver.execute()

    ExecuteWithHookContext.run() || PreExecute.run() (hive.exec.pre.hooks)

    TaskRunner

    if failed, ExecuteWithHookContext.run()(hive.exec.failure.hooks)

    ExecuteWithHookContext.run() || PostExecute.run() (hive.exec.post.hooks)

    HiveDriverRunHook.postDriverRun()(hive.exec.driver.run.hooks)

    结语

    感谢您的观看，如有不足之处，欢迎批评指正。

    获取资料

    本次给大家推荐一个免费的学习群，里面概括数据仓库/源码解析/Python/Hadoop/Flink/Spark/Storm/Hive以及面试资源等。
    对大数据开发技术感兴趣的同学，欢迎加入Q群：894951460，不管你是小白还是大牛我都欢迎，还有大牛整理的一套高效率学习路线和教程与您免费分享，同时每天更新视频资料。
    最后，祝大家早日学有所成，拿到满意offer，快速升职加薪，走上人生巅峰。

    原文出处：https://my.oschina.net/u/2539801/blog/1514648

    相关文章

      网友评论

        本文标题：Hive Query生命周期--钩子函数篇

        本文链接：https://www.haomeiwen.com/subject/xrpreqtx.html

        延伸阅读

        深度阅读

        • 您也可以注册成为美文阅读网的作者，发表您的原创作品、分享您的心情！

        栏目导航

        热点阅读

        大数据，机器学习，人工智能

        大数据

        大数据 爬虫Python AI Sql

        玩转大数据

        关于我们|服务条款|联系我们|Hive Query生命周期--钩子函数篇|投稿指南|网站地图|RSS订阅|排版工具|手机版

        提供经典美文摘抄,优美散文欣赏,现代诗歌精选,短篇小说,心情随笔,表白情书范文,故事会在线阅读欣赏

        Copyright © 2014-2023 Haomeiwen.com All Rights Reserved. 好美文阅读网 版权所有

        备案信息：桂公网安备 45052102000051号 · 桂ICP备13007215号-3

        本站所收录作品、热点评论等信息部分来源互联网，目的只是为了系统归纳学习和传递资讯

        所有作品版权归原创作者所有，与本站立场无关，如不慎侵犯了你的权益，请联系我们告知，我们将做删除处理！