mark下关于如何让spark的ThriftServer高可用

1、spark的ThriftServer介绍

spark的thriftServer是在hiveServer2基础上实现的 ， 提供了Thrift服务

，也就是提供了远程的ODBC服务

用户画像，对受众目标进行筛选 ， 数据源是在hive数仓的；

早期直接通过hive on spark的方式去做处理 ， 但是发现性能不好，处理慢

后来改用了thriftServer，采用jdbc的方式，直接去读取数据仓库内容，性能在一定程度得到了提升

2、ThriftServer的缺陷

我们使用了spark的ThriftServer，虽然性能得到了提升，但是安全并没有得到保障

因为spark的ThriftServer并没有 类似hive里面的服务自主发现功能；

所以spark的thriftServer就出现了单点问题

 

3、ThriftServer的解决

解决这个问题，首先要搞懂ThriftServer的启动流程大概 什么样子

3.1、ThriftServer的启动流程

1）:我们在启动thrift server，是通过脚本start-thriftserver.sh

 2）：然后在进入spark-daemon.sh 这个脚本 ， 携带CLASS参数进行启动

 

 3）：然后看bin/spark-submit.sh

 然后sparksubmit，会去执行一个叫做：HiveThriftServer2的类；

3.2、解决单点问题

3.2.1、内部流程说明

1):首先进入HiveThriftServer2，会执行main函数

 

这样，最后在server.start() 这里，就启动了

启动的服务是HiveThriftServer2

 其中的绑定服务addService，就是在ArrayList去维护当前的service

 所以在调用initCompositeService，开始启动服务，会最终调用到HiveServer2

这个init方法是个接口：

直接看实现，是在HiveService2的init方法：

 那么最后我们这个流程是怎么启动的呢？

 然后我们 查看parse方法

 然后查看StartOptionExecutor

 然后查看startHiveServer2

那么我们遇到的问题是sparkThriftServer存在单点问题 ，但是尴尬的是，spark本身没有给出解决方案，

但是hive是有的，hive支持了hiveServer2服务的高可用；

通过配置就可以实现；

所以我们可以通过zookeeper来实现spark的 ThriftServer的高可用

3.2.2、把thriftServer模块导入到工程

源码工程的总pom文件

3.2.3、在HiveThriftServer2启动的时候，让他在zookeeper里面注册一份信息，代表当前第一个实例

所以需要在：

1）：实例化一个全局的hiveConf，用来调用hive的一些回调函数,

主要是可以去hive-site.xml里面拿配置文件

<property>
        <name>hive.server2.support.dynamic.service.discovery</name>
        <value>true</value>
</property>

2）：在启动HiveThriftServer2的时候，把hiveServer2注册到zookeeper中

HiveThriftServer2 实现了HIveServer2

而且是使用反射处理的（早期hive开发的时候，就已经在内部使用private，可能没想过别人 会调用）

3）：在停止HiveThriftServer2的时候，把hiveserrver2从zookeeper中注销掉

3.3、修改hiveServer2源码

1）：添加获取thriftServer的IP:HOST

拿到thriftServer的域名

 拼接ThriftServer的地址

2）：对zookeeper做kuberos认证

3）：添加zookeeper的权限控制

4）：控制是否需要重新在zookeeper上注册hiveServer2

5）：zk的监控者，如果发现注册信息为null，会触发监控，然后关掉当前注册hiveServer2的实例信息

接下来重点来了，上面所有的操作，都是为了给这个方法做铺垫服务的

6）：把服务注册到zookeeper中

 

private CuratorFramework zooKeeperClient;
  private String znodePath;
  /**
   * 把服务注册到zookeeper中（通过反射调用）
   * @param hiveConf
   * @throws Exception
   */
  private void addServerInstanceToZooKeeper(HiveConf hiveConf) throws Exception {
    //从hiveConf中获取zookeeper地址
    String zooKeeperEnsemble = ZooKeeperHiveHelper.getQuorumServers(hiveConf);
    //从hive-site.xml中获取hive.server2.zookeeper.namespace的配置信息
    String rootNamespace = hiveConf.getVar(HiveConf.ConfVars.HIVE_SERVER2_ZOOKEEPER_NAMESPACE);
    //获取用户提供的thriftServer地址(IP:PORT) node01:10001 , node02:10002
    String instanceURI = getServerInstanceURI();
    //做Kerberos认证
    setUpZooKeeperAuth(hiveConf);
    //获取hive连接zookeeper的session超时时间
    int sessionTimeout =
            (int) hiveConf.getTimeVar(HiveConf.ConfVars.HIVE_ZOOKEEPER_SESSION_TIMEOUT,
                    TimeUnit.MILLISECONDS);
    //hive连接zookeeper的等待时间
    int baseSleepTime =
            (int) hiveConf.getTimeVar(HiveConf.ConfVars.HIVE_ZOOKEEPER_CONNECTION_BASESLEEPTIME,
                    TimeUnit.MILLISECONDS);
    //hive连接zookeeper的最大重试次数
    int maxRetries = hiveConf.getIntVar(HiveConf.ConfVars.HIVE_ZOOKEEPER_CONNECTION_MAX_RETRIES);
    // 获取zookeeper客户端
    zooKeeperClient =
            CuratorFrameworkFactory.builder().connectString(zooKeeperEnsemble)
                    .sessionTimeoutMs(sessionTimeout).aclProvider(zooKeeperAclProvider)
                    .retryPolicy(new ExponentialBackoffRetry(baseSleepTime, maxRetries)).build();
    //启动zookeeper客户端
    zooKeeperClient.start();
    //TODO 在zookeeper上根据rootNamespace创建一个空间（用来存储数据的文件夹）
    try {
      zooKeeperClient.create().creatingParentsIfNeeded().withMode(CreateMode.PERSISTENT)
              .forPath(ZooKeeperHiveHelper.ZOOKEEPER_PATH_SEPARATOR + rootNamespace);
      LOG.info("Created the root name space: " + rootNamespace + " on ZooKeeper for HiveServer2");
    } catch (KeeperException e) {
      if (e.code() != KeeperException.Code.NODEEXISTS) {
        LOG.fatal("Unable to create HiveServer2 namespace: " + rootNamespace + " on ZooKeeper", e);
        throw e;
      }
    }
    //TODO 把hiveserver2的信息注册到rootNamespace下：
    // serverUri=cdh1:10001;version=1.2.1.spark2;sequence=0000000005
    try {
      String pathPrefix =
              ZooKeeperHiveHelper.ZOOKEEPER_PATH_SEPARATOR + rootNamespace
                      + ZooKeeperHiveHelper.ZOOKEEPER_PATH_SEPARATOR + "serverUri=" + instanceURI + ";"
                      + "version=" + HiveVersionInfo.getVersion() + ";" + "sequence=";
      byte[] znodeDataUTF8 = instanceURI.getBytes(Charset.forName("UTF-8"));
      //创建一个临时节点
      znode =
              new PersistentEphemeralNode(zooKeeperClient,
                      PersistentEphemeralNode.Mode.EPHEMERAL_SEQUENTIAL, pathPrefix, znodeDataUTF8);
      znode.start();

      //给定临时节点创建的超时时间，如果超过120秒，则抛异常
      long znodeCreationTimeout = 120;
      if (!znode.waitForInitialCreate(znodeCreationTimeout, TimeUnit.SECONDS)) {
        throw new Exception("Max znode creation wait time: " + znodeCreationTimeout + "s exhausted");
      }


      setDeregisteredWithZooKeeper(false);
      znodePath = znode.getActualPath();
      // TODO 添加zk的watch ， 如果服务不见了，需要第一时间watche到
      if (zooKeeperClient.checkExists().usingWatcher(new DeRegisterWatcher()).forPath(znodePath) == null) {
        // No node exists, throw exception
        throw new Exception("Unable to create znode for this HiveServer2 instance on ZooKeeper.");
      }
      LOG.info("Created a znode on ZooKeeper for HiveServer2 uri: " + instanceURI);
    } catch (Exception e) {
      LOG.fatal("Unable to create a znode for this server instance", e);
      if (znode != null) {
        znode.close();
      }
      throw (e);
    }
  }

7）：移除znode，代表当前程序关闭

  //移除znode，代表当前程序关闭（通过反射调用）
  private void removeServerInstanceFromZooKeeper() throws Exception {
    setDeregisteredWithZooKeeper(false);

    if (znode != null) {
      znode.close();
    }
    zooKeeperClient.close();
    LOG.info("Server instance removed from ZooKeeper.");
  }

3：修改spark-daemon.sh脚本

 

4、maven打包

mvn -Pyarn -Phadoop-2.6 -Dhadoop.version=2.6.0 -Phive -Phive-thriftserver -Dscala-2.11 -DskipTests clean package

但是以上操作只是编译出了一份文件，并不是可运行的包，所以需要稍微改一下命令：

./make-distribution.sh --name 2.6.0-cdh5.14.0 --tgz -Pyarn -Phive -Phive-thriftserver -Phadoop-2.6 -Dscala-2.11 -Dhadoop.version=hadoop-2.6.0-cdh5.14.0 -DskipTests clean package

 

5、将新生成的spark-hive-thriftserver_2.11-2.1.0.jar替换spark/jars 下面的同名包

6、修改hive/conf和spark/conf下面的配置文件：hive-site.xml

在里面添加如下配置：

<property>
    <name>hive.server2.support.dynamic.service.discovery</name>
    <value>true</value>
</property>


<property>
    <name>hive.server2.zookeeper.namespace</name>
    <value>hiveserver2_zk</value>
</property>

<property
    <name>hive.zookeeper.quorum</name>
    <value>cdh1:2181,cdh2:2181,cdh3:2181</value>
</property>

<property>
    <name>hive.zookeeper.client.port</name>
    <value>2181</value>
</property>

<property>
    <name>hive.server2.thrift.bind.host</name>
    <value>cdh1</value>
</property>

<property>
    <name>hive.server2.thrift.port</name>
    <value>10003</value>
</property>

7、验证

1）：

分别在某台机器上执行两次：

sbin/start-thriftserver.sh 
--master yarn 
--conf spark.driver.memory=1G 
--executor-memory 512m 
--num-executors 1 
--hiveconf hive.server2.thrift.port=10001/10002

2）：在zookeeper中可以看到如下信息：

 3)：通过beeline模拟jdbc去查询hive

[root@cdh2 spark-2.1.0-bin-2.6.0-cdh5.14.0]# beeline
Beeline version 1.2.1.spark2 by Apache Hive
beeline> !connect jdbc:hive2://cdh1:2181,cdh2:2181,cdh3:2181/default;serviceDiscoveryMode=zooKeeper;zooKeeperNamespace=hiveserver2_zk

去页面观察：

杀掉一个thriftServer

在重新登录beeline，在做查询，可以发现依然可以查询