[CARMEL-7402] [CARMEL-6525] Support tag different drivers in the queue (apache#141)

Author: wangyum

sql/hive-thriftserver/src/main/java/org/apache/hive/service/server/HiveServer2.java

@@ -203,9 +203,13 @@ private void addServerInstanceToZooKeeper(HiveConf hiveConf) throws Exception {
         ZOOKEEPER_PATH_SEPARATOR + queue + ZOOKEEPER_PATH_SEPARATOR + ephemeralPath;
     LOG.info("Add server instance to zookeeper on {}", zkPath);
     try {
+      String data = "{\"tag\":[]}";
+      String queueTag = hiveConf.get("spark.yarn.queue.tag", "");
+      if (StringUtils.isNotBlank(queueTag)) {
+        data = "{\"tag\":[\"" + queueTag + "\"]}"; // schedule
+      }
       znode = new PersistentEphemeralNode(zooKeeperClient,
-          PersistentEphemeralNode.Mode.EPHEMERAL, zkPath,
-          ephemeralPath.getBytes(StandardCharsets.UTF_8));
+          PersistentEphemeralNode.Mode.EPHEMERAL, zkPath, data.getBytes(StandardCharsets.UTF_8));
       znode.start();
       // We'll wait for 120s for node creation
       long znodeCreationTimeout = 120;