| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.tajo.cli; |
| |
| import org.apache.commons.lang.exception.ExceptionUtils; |
| import org.apache.tajo.QueryId; |
| import org.apache.tajo.catalog.TableDesc; |
| import org.apache.tajo.catalog.statistics.TableStats; |
| import org.apache.tajo.client.QueryStatus; |
| import org.apache.tajo.conf.TajoConf; |
| import org.apache.tajo.util.FileUtil; |
| |
| import java.io.InputStream; |
| import java.io.PrintWriter; |
| import java.sql.ResultSet; |
| import java.sql.ResultSetMetaData; |
| |
| public class DefaultTajoCliOutputFormatter implements TajoCliOutputFormatter { |
| private TajoConf tajoConf; |
| private int printPauseRecords; |
| private boolean printPause; |
| private boolean printErrorTrace; |
| |
| @Override |
| public void init(TajoConf tajoConf) { |
| this.tajoConf = tajoConf; |
| |
| this.printPause = tajoConf.getBoolVar(TajoConf.ConfVars.CLI_PRINT_PAUSE); |
| this.printPauseRecords = tajoConf.getIntVar(TajoConf.ConfVars.CLI_PRINT_PAUSE_NUM_RECORDS); |
| this.printErrorTrace = tajoConf.getBoolVar(TajoConf.ConfVars.CLI_PRINT_ERROR_TRACE); |
| } |
| |
| @Override |
| public void setScirptMode() { |
| this.printPause = false; |
| } |
| |
| private String getQuerySuccessMessage(TableDesc tableDesc, float responseTime, int totalPrintedRows, String postfix) { |
| TableStats stat = tableDesc.getStats(); |
| String volume = stat == null ? "0 B" : FileUtil.humanReadableByteCount(stat.getNumBytes(), false); |
| long resultRows = stat == null ? 0 : stat.getNumRows(); |
| |
| long realNumRows = resultRows != 0 ? resultRows : totalPrintedRows; |
| return "(" + realNumRows + " rows, " + getResponseTimeReadable(responseTime) + ", " + volume + " " + postfix + ")"; |
| } |
| |
| protected String getResponseTimeReadable(float responseTime) { |
| return responseTime + " sec"; |
| } |
| |
| @Override |
| public void printResult(PrintWriter sout, InputStream sin, TableDesc tableDesc, |
| float responseTime, ResultSet res) throws Exception { |
| long resultRows = tableDesc.getStats() == null ? 0 : tableDesc.getStats().getNumRows(); |
| if (resultRows == 0) { |
| resultRows = Integer.MAX_VALUE; |
| } |
| |
| if (res == null) { |
| sout.println(getQuerySuccessMessage(tableDesc, responseTime, 0, "inserted")); |
| return; |
| } |
| ResultSetMetaData rsmd = res.getMetaData(); |
| int numOfColumns = rsmd.getColumnCount(); |
| for (int i = 1; i <= numOfColumns; i++) { |
| if (i > 1) sout.print(", "); |
| String columnName = rsmd.getColumnName(i); |
| sout.print(columnName); |
| } |
| sout.println("\n-------------------------------"); |
| |
| int numOfPrintedRows = 0; |
| int totalPrintedRows = 0; |
| while (res.next()) { |
| for (int i = 1; i <= numOfColumns; i++) { |
| if (i > 1) sout.print(", "); |
| String columnValue = res.getObject(i).toString(); |
| if(res.wasNull()){ |
| sout.print("null"); |
| } else { |
| sout.print(columnValue); |
| } |
| } |
| sout.println(); |
| sout.flush(); |
| numOfPrintedRows++; |
| totalPrintedRows++; |
| if (printPause && printPauseRecords > 0 && totalPrintedRows < resultRows && numOfPrintedRows >= printPauseRecords) { |
| if (resultRows < Integer.MAX_VALUE) { |
| sout.print("(" + totalPrintedRows + "/" + resultRows + " rows, continue... 'q' is quit)"); |
| } else { |
| sout.print("(" + totalPrintedRows + " rows, continue... 'q' is quit)"); |
| } |
| sout.flush(); |
| if (sin != null) { |
| if (sin.read() == 'q') { |
| sout.println(); |
| break; |
| } |
| } |
| numOfPrintedRows = 0; |
| sout.println(); |
| } |
| } |
| sout.println(getQuerySuccessMessage(tableDesc, responseTime, totalPrintedRows, "selected")); |
| sout.flush(); |
| } |
| |
| @Override |
| public void printNoResult(PrintWriter sout) { |
| sout.println("(0 rows)"); |
| } |
| |
| @Override |
| public void printProgress(PrintWriter sout, QueryStatus status) { |
| sout.println("Progress: " + (int)(status.getProgress() * 100.0f) |
| + "%, response time: " |
| + getResponseTimeReadable((float)((status.getFinishTime() - status.getSubmitTime()) / 1000.0))); |
| sout.flush(); |
| } |
| |
| @Override |
| public void printMessage(PrintWriter sout, String message) { |
| sout.println(message); |
| } |
| |
| @Override |
| public void printErrorMessage(PrintWriter sout, Throwable t) { |
| sout.println(parseErrorMessage(t.getMessage())); |
| if (printErrorTrace) { |
| sout.println(ExceptionUtils.getStackTrace(t)); |
| } |
| } |
| |
| @Override |
| public void printErrorMessage(PrintWriter sout, String message) { |
| sout.println(parseErrorMessage(message)); |
| } |
| |
| @Override |
| public void printKilledMessage(PrintWriter sout, QueryId queryId) { |
| sout.println(TajoCli.KILL_PREFIX + queryId); |
| } |
| |
| @Override |
| public void printErrorMessage(PrintWriter sout, QueryStatus status) { |
| if (status.getErrorMessage() != null && !status.getErrorMessage().isEmpty()) { |
| printErrorMessage(sout, parseErrorMessage(status.getErrorMessage())); |
| } else { |
| printErrorMessage(sout, "No error message"); |
| } |
| if (printErrorTrace && status.getErrorTrace() != null && !status.getErrorTrace().isEmpty()) { |
| sout.println(status.getErrorTrace()); |
| } |
| } |
| |
| public static String parseErrorMessage(String message) { |
| if (message == null) { |
| return TajoCli.ERROR_PREFIX + "No error message"; |
| } |
| String[] lines = message.split("\n"); |
| message = lines[0]; |
| |
| int index = message.lastIndexOf(TajoCli.ERROR_PREFIX); |
| if (index < 0) { |
| message = TajoCli.ERROR_PREFIX + message; |
| } else { |
| message = message.substring(index); |
| } |
| |
| return message; |
| } |
| } |