HiveServer连接优化 - yurun

 

问题：HiveServer如何在执行语句之前、之后发出相应的通知？

　　　　　　　　　QueryPlan queryPlan = hookContext.getQueryPlan();

HiveConf conf = hookContext.getConf();

String queryId = queryPlan.getQueryId();

if (StringUtils.isEmpty(queryId)) {
LOGGER.warn("queryId is null or empty, return");

return;
}

LOGGER.info("queryId: " + queryId);

String queryStr = URLEncoder.encode(queryPlan.getQueryStr(),
CharEncoding.UTF_8);

if (StringUtils.isEmpty(queryStr)) {
LOGGER.warn("queryStr is null or empty, return");

return;
}

LOGGER.info("queryStr: " + queryStr);

String jobName = conf.getVar(HiveConf.ConfVars.HADOOPJOBNAME);

LOGGER.info("jobName: " + jobName);

if (StringUtils.isEmpty(jobName)) {
LOGGER.warn("jobName is null or empty, return");

return;
}

String server = conf.get("hiveserver.execute.hook.server");

if (StringUtils.isEmpty(server)) {
LOGGER.warn("server is null or empty, return");

return;
}

LOGGER.info("server: " + server);

String rest = conf.get("hiveserver.execute.hook.rest");

LOGGER.info("rest: " + rest);

if (StringUtils.isEmpty(rest)) {
LOGGER.warn("rest is null or empty, return");

return;
}

Map<String, String> params = new HashMap<String, String>();

params.put("server", server);

params.put("hook", hookContext.getHookType().toString());

params.put("queryId", queryId);

params.put("queryStr", queryStr);

params.put("jobName", jobName);

params.put("timestamp", String.valueOf(DatetimeUtil.now()));

try {
HttpClientUtil.doPost(rest, params);
} catch (Exception e) {
LOGGER.error("do post error: "
+ ExceptionUtils.getFullStackTrace(e));
}