Hive源码阅读--命令的执行与返回--processLocalCmd

处理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,那么意味着程序正常执行,整个过程包括对命令的解析,执行和返回。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

寒 暄

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值