Kyuubi服务源码解析:KyuubiServer

  Kyuubi服务与HiveServer2服务非常相似,在Kyuubi中很多类的设计和代码逻辑都参照了HiveServer2(Spark SQL Thrift Server也是同样的道理)。
  HiveServer2服务启动的源码解析参见:
  Hive源码剖析之HiveServer2服务启动过程
  Spark SQL Thrift Server服务启动的源码解析参见以下两处文章:
  SparkSQL Hive ThriftServer 源码解析:Intro
  Spark SQL源码走读(一):HiveThriftServer2:Intro

(注:以下代码模块中,英文注释皆为作者注释,中文注释是我自己加上的)

KuubiServer.scala

  • KuubiServer类声明及成员
/**
 * Main entrance of Kyuubi Server
 */
private[kyuubi] class KyuubiServer private(name: String)
  extends CompositeService(name) with Logging {
  //BackendService服务
  private[this] var _beService: BackendService = _
  def beService: BackendService = _beService
  //FrontendService服务
  private[this] var _feService: FrontendService = _
  def feService: FrontendService = _feService
  ***后续代码省略***
}

  简单介绍一下KyuubiServer中两个很重要的成员FrontendServiceBackendService:FrontendService负责维护与客户端的连接,与客户端进行交互,将客户端的SQL请求转发至FrontendService;BackendService负责执行SQL并将执行结果返回给FrontendService。FrontendService最后将结果返回至客户端。

  • main方法
def main(args: Array[String]): Unit = {
    SparkUtils.initDaemon(logger)
    //加载配置
    val conf = new SparkConf(loadDefaults = true)
    setupCommonConfig(conf)

    try {
      val server = new KyuubiServer()
      //对各种服务进行初始化
      server.init(conf)
      //启动各种服务
      server.start()
      info(server.getName + " started!")
      if (HighAvailabilityUtils.isSupportDynamicServiceDiscovery(conf)) {
        info(s"HA mode: start to add this ${server.getName} instance to Zookeeper...")
        HighAvailabilityUtils.addServerInstanceToZooKeeper(server)
      }
    } catch {
      case e: Exception =>
        error("Error starting Kyuubi Server", e)
        System.exit(-1)
    }
  • init方法
  override def init(conf: SparkConf): Unit = synchronized {
    this.conf = conf
    _beService = new BackendService()
    _feService = new FrontendService(_beService)
    //将BackendService和FrontendService服务加入serviceList中
    addService(_beService)
    addService(_feService)
    //调用父类CompositeService的init方法
    super.init(conf)
    SparkUtils.addShutdownHook {
      () => this.stop()
    }
  }

CompositeService的init方法:

  override def init(conf: SparkConf): Unit = {
    //依次调用serviceList中各个服务的init方法
    for (service <- serviceList) {
      service.init(conf)
    }
    super.init(conf)
  }
  • start方法
  override def start(): Unit = {
    //遍历serviceList中所有的服务并依次启动
    serviceList.zipWithIndex.foreach { case (service, i) =>
      try {
        service.start()
      } catch {
        case e: Throwable =>
          error("Error starting services " + getName, e)
          stop(i)
          throw new ServiceException("Failed to Start " + getName, e)
      }
    }
    super.start()
  }

  调用各个服务的init和start方法时,最终都会调用AbstractService的init和start方法(这些服务类要么直接继承AbstractService,要么继承CompositeService。而CompositeService又继承自AbstractService)。

AbstractService的init和start方法:

  override def init(conf: SparkConf): Unit = {
    ensureCurrentState(State.NOT_INITED)
    this.conf = conf
    changeState(State.INITED)
    info("Service: [" + getName + "] is initialized.")
  }

  override def start(): Unit = {
    startTime = System.currentTimeMillis
    ensureCurrentState(State.INITED)
    changeState(State.STARTED)
    info("Service: [" + getName + "] is started.")
  }

  对照Kyuubi服务启动日志,可以看到KyuubiServer依次启动的服务有:KyuubiServer、OperationManager、SessionManager、BackendService和FrontendService(至于为什么也启动了OperationManager和SessionManager服务,在后续解析BackendService的源码时会提及)。

Kyuubi服务启动日志

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容