处理sql命令,根据不同的命令处理器处理不同sql命令,主要区分处理器是否是Driver类,如果是,处理除了(set,reset,dfs,add,list,delete,reload,crypto)命令外的其他sql ,如果不是调用各自处理器的run方法.
int processLocalCmd(String cmd, CommandProcessor proc, CliSessionState ss) {
int tryCount = 0;
int ret = 0;
boolean needRetry;
do {
try {
needRetry = false;
if (proc != null) {
// 如果CommandProcessor是Driver实例
if (proc instanceof Driver) {
Driver qp = (Driver)proc;
// 获取标准输出流,打印结果信息
PrintStream out = ss.out;
long start = System.currentTimeMillis();
// 输出命令原文
if (ss.getIsVerbose()) {
out.println(cmd);
}
// 获取运行的命令
qp.setTryCount(tryCount);
// Driver实例运行用户指令,获取运行结果响应码
ret = qp.run(cmd).getResponseCode();
// 如果执行失败,直接返回ret
if (ret != 0) {
qp.close();
return ret;
}
// 统计指令的运行时间
long end = System.currentTimeMillis();
double timeTaken = (double)(end - start) / 1000.0D;
ArrayList<String> res = new ArrayList();
// 打印查询结果的列名称
this.printHeader(qp, out);
int counter = 0;
// 打印查询结果
try {
if (out instanceof FetchConverter) {
((FetchConverter)out).fetchStarted();
}
while(qp.getResults(res)) {
Iterator var17 = res.iterator();
while(var17.hasNext()) {
String r = (String)var17.next();
out.println(r);
}
counter += res.size();
res.clear();
if (out.checkError()) {
break;
}
}
} catch (IOException var19) {
this.console.printError("Failed with exception " + var19.getClass().getName() + ":" + var19.getMessage(), "\n" + StringUtils.stringifyException(var19));
ret = 1;
}
// 关闭结果
int cret = qp.close();
if (ret == 0) {
ret = cret;
}
if (out instanceof FetchConverter) {
((FetchConverter)out).fetchFinished();
}
this.console.printInfo("Time taken: " + timeTaken + " seconds" + (counter == 0 ? "" : ", Fetched: " + counter + " row(s)"));
} else {
// 如果proc不是Driver,也就是用户执行的是非SQL查询操作,直接执行语句,不执行FetchResult的操作
String firstToken = this.tokenizeCmd(cmd.trim())[0];
String cmd_1 = this.getFirstCmd(cmd.trim(), firstToken.length());
if (ss.getIsVerbose()) {
ss.out.println(firstToken + " " + cmd_1);
}
CommandProcessorResponse res = proc.run(cmd_1);
if (res.getResponseCode() != 0) {
ss.out.println("Query returned non-zero code: " + res.getResponseCode() + ", cause: " + res.getErrorMessage());
}
if (res.getConsoleMessages() != null) {
Iterator var10 = res.getConsoleMessages().iterator();
while(var10.hasNext()) {
String consoleMsg = (String)var10.next();
this.console.printInfo(consoleMsg);
}
}
ret = res.getResponseCode();
}
}
} catch (CommandNeedRetryException var20) {
// 如果执行过程中出现异常,修改needRetry标志,下次循环是retry。
this.console.printInfo("Retry query with a different approach...");
++tryCount;
needRetry = true;
}
} while(needRetry);
return ret;
}
如果程序的needRetry为false,那么意味着程序正常执行,整个过程包括对命令的解析,执行和返回。