org.verdictdb.commons.DBTablePrinter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of verdictdb-core Show documentation
Show all versions of verdictdb-core Show documentation
Platform-independent, interactive-speed data analytics system
The newest version!
/*
Database Table Printer
Copyright (C) 2014 Hami Galip Torun
Email: [email protected]
Project Home: https://github.com/htorun/dbtableprinter
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see ResultSet to standard out, formatted to look
* like a table with rows and columns with borders.
*
*
Stack Overflow website
* (stackoverflow.com)
* was the primary source of inspiration and help to put this
* code together. Especially the questions and answers of
* the following people were very useful:
*
*
Question:
* How to display or
* print the contents of a database table as is
* People: sky scraper
*
*
Question:
* System.out.println()
* from database into a table
* People: Simon Cottrill, Tony Toews, Costis Aivali, Riggy, corsiKa
*
*
Question:
* Simple way to repeat
* a string in java
* People: Everybody who contributed but especially user102008
*/
public class DBTablePrinter {
/**
* Default maximum number of rows to query and print.
*/
private static final int DEFAULT_MAX_ROWS = 10;
/**
* Default maximum width for text columns
* (like a VARCHAR) column.
*/
private static final int DEFAULT_MAX_TEXT_COL_WIDTH = 150;
/**
* Column type category for CHAR, VARCHAR
* and similar text columns.
*/
public static final int CATEGORY_STRING = 1;
/**
* Column type category for TINYINT, SMALLINT,
* INT and BIGINT columns.
*/
public static final int CATEGORY_INTEGER = 2;
/**
* Column type category for REAL, DOUBLE,
* and DECIMAL columns.
*/
public static final int CATEGORY_DOUBLE = 3;
/**
* Column type category for date and time related columns like
* DATE, TIME, TIMESTAMP etc.
*/
public static final int CATEGORY_DATETIME = 4;
/**
* Column type category for BOOLEAN columns.
*/
public static final int CATEGORY_BOOLEAN = 5;
/**
* Column type category for types for which the type name
* will be printed instead of the content, like BLOB,
* BINARY, ARRAY etc.
*/
public static final int CATEGORY_OTHER = 0;
/**
* Represents a database table column.
*/
private static class Column {
/**
* Column label.
*/
private String label;
/**
* Generic SQL type of the column as defined in
*
* java.sql.Types
* .
*/
private int type;
/**
* Generic SQL type name of the column as defined in
*
* java.sql.Types
* .
*/
private String typeName;
/**
* Width of the column that will be adjusted according to column label
* and values to be printed.
*/
private int width = 0;
/**
* Column values from each row of a ResultSet.
*/
private List values = new ArrayList<>();
/**
* Flag for text justification using String.format.
* Empty string ("") to justify right,
* dash (-) to justify left.
*
* @see #justifyLeft()
*/
private String justifyFlag = "";
/**
* Column type category. The columns will be categorised according
* to their column types and specific needs to print them correctly.
*/
private int typeCategory = 0;
/**
* Constructs a new Column with a column label,
* generic SQL type and type name (as defined in
*
* java.sql.Types
* )
*
* @param label Column label or name
* @param type Generic SQL type
* @param typeName Generic SQL type name
*/
public Column(String label, int type, String typeName) {
this.label = label;
this.type = type;
this.typeName = typeName;
}
/**
* Returns the column label
*
* @return Column label
*/
public String getLabel() {
return label;
}
/**
* Returns the generic SQL type of the column
*
* @return Generic SQL type
*/
public int getType() {
return type;
}
/**
* Returns the generic SQL type name of the column
*
* @return Generic SQL type name
*/
public String getTypeName() {
return typeName;
}
/**
* Returns the width of the column
*
* @return Column width
*/
public int getWidth() {
return width;
}
/**
* Sets the width of the column to width
*
* @param width Width of the column
*/
public void setWidth(int width) {
this.width = width;
}
/**
* Adds a String representation (value)
* of a value to this column object's {@link #values} list.
* These values will come from each row of a
*
* ResultSet
* of a database query.
*
* @param value The column value to add to {@link #values}
*/
public void addValue(String value) {
values.add(value);
}
/**
* Returns the column value at row index i.
* Note that the index starts at 0 so that getValue(0)
* will get the value for this column from the first row
* of a
* ResultSet.
*
* @param i The index of the column value to get
* @return The String representation of the value
*/
public String getValue(int i) {
return values.get(i);
}
/**
* Returns the value of the {@link #justifyFlag}. The column
* values will be printed using String.format and
* this flag will be used to right or left justify the text.
*
* @return The {@link #justifyFlag} of this column
* @see #justifyLeft()
*/
public String getJustifyFlag() {
return justifyFlag;
}
/**
* Sets {@link #justifyFlag} to "-" so that
* the column value will be left justified when printed with
* String.format. Typically numbers will be right
* justified and text will be left justified.
*/
public void justifyLeft() {
this.justifyFlag = "-";
}
/**
* Returns the generic SQL type category of the column
*
* @return The {@link #typeCategory} of the column
*/
public int getTypeCategory() {
return typeCategory;
}
/**
* Sets the {@link #typeCategory} of the column
*
* @param typeCategory The type category
*/
public void setTypeCategory(int typeCategory) {
this.typeCategory = typeCategory;
}
}
/**
* Overloaded method that prints rows from table tableName
* to standard out using the given database connection
* conn. Total number of rows will be limited to
* {@link #DEFAULT_MAX_ROWS} and
* {@link #DEFAULT_MAX_TEXT_COL_WIDTH} will be used to limit
* the width of text columns (like a VARCHAR column).
*
* @param conn Database connection object (java.sql.Connection)
* @param tableName Name of the database table
*/
public static void printTable(Connection conn, String tableName) {
printTable(conn, tableName, DEFAULT_MAX_ROWS, DEFAULT_MAX_TEXT_COL_WIDTH);
}
/**
* Overloaded method that prints rows from table tableName
* to standard out using the given database connection
* conn. Total number of rows will be limited to
* maxRows and
* {@link #DEFAULT_MAX_TEXT_COL_WIDTH} will be used to limit
* the width of text columns (like a VARCHAR column).
*
* @param conn Database connection object (java.sql.Connection)
* @param tableName Name of the database table
* @param maxRows Number of max. rows to query and print
*/
public static void printTable(Connection conn, String tableName, int maxRows) {
printTable(conn, tableName, maxRows, DEFAULT_MAX_TEXT_COL_WIDTH);
}
/**
* Overloaded method that prints rows from table tableName
* to standard out using the given database connection
* conn. Total number of rows will be limited to
* maxRows and
* maxStringColWidth will be used to limit
* the width of text columns (like a VARCHAR column).
*
* @param conn Database connection object (java.sql.Connection)
* @param tableName Name of the database table
* @param maxRows Number of max. rows to query and print
* @param maxStringColWidth Max. width of text columns
*/
public static void printTable(Connection conn, String tableName, int maxRows, int maxStringColWidth) {
if (conn == null) {
System.err.println("DBTablePrinter Error: No connection to database (Connection is null)!");
return;
}
if (tableName == null) {
System.err.println("DBTablePrinter Error: No table name (tableName is null)!");
return;
}
if (tableName.length() == 0) {
System.err.println("DBTablePrinter Error: Empty table name!");
return;
}
if (maxRows < 1) {
System.err.println("DBTablePrinter Info: Invalid max. rows number. Using default!");
maxRows = DEFAULT_MAX_ROWS;
}
Statement stmt = null;
ResultSet rs = null;
try {
if (conn.isClosed()) {
System.err.println("DBTablePrinter Error: Connection is closed!");
return;
}
String sqlSelectAll = "SELECT * FROM " + tableName + " LIMIT " + maxRows;
stmt = conn.createStatement();
rs = stmt.executeQuery(sqlSelectAll);
printResultSet(rs, maxStringColWidth);
} catch (SQLException e) {
System.err.println("SQL exception in DBTablePrinter. Message:");
System.err.println(e.getMessage());
} finally {
try {
if (stmt != null) {
stmt.close();
}
if (rs != null) {
rs.close();
}
} catch (SQLException ignore) {
// ignore
}
}
}
/**
* Overloaded method to print rows of a
* ResultSet to standard out using {@link #DEFAULT_MAX_TEXT_COL_WIDTH}
* to limit the width of text columns.
*
* @param rs The ResultSet to print
*/
public static void printResultSet(ResultSet rs) {
printResultSet(rs, DEFAULT_MAX_TEXT_COL_WIDTH);
}
/**
* Overloaded method to print rows of a
* ResultSet to standard out using maxStringColWidth
* to limit the width of text columns.
*
* @param rs The ResultSet to print
* @param maxStringColWidth Max. width of text columns
*/
public static void printResultSet(ResultSet rs, int maxStringColWidth) {
try {
if (rs == null) {
System.err.println("DBTablePrinter Error: Result set is null!");
return;
}
if (rs.isClosed()) {
System.err.println("DBTablePrinter Error: Result Set is closed!");
return;
}
if (maxStringColWidth < 1) {
System.err.println("DBTablePrinter Info: Invalid max. varchar column width. Using default!");
maxStringColWidth = DEFAULT_MAX_TEXT_COL_WIDTH;
}
// Get the meta data object of this ResultSet.
ResultSetMetaData rsmd;
rsmd = rs.getMetaData();
// Total number of columns in this ResultSet
int columnCount = rsmd.getColumnCount();
// List of Column objects to store each columns of the ResultSet
// and the String representation of their values.
List columns = new ArrayList<>(columnCount);
// Get the columns and their meta data.
// NOTE: columnIndex for rsmd.getXXX methods STARTS AT 1 NOT 0
for (int i = 1; i <= columnCount; i++) {
Column c = new Column(rsmd.getColumnLabel(i),
rsmd.getColumnType(i), rsmd.getColumnTypeName(i));
c.setWidth(c.getLabel().length());
c.setTypeCategory(whichCategory(c.getType()));
columns.add(c);
}
// Go through each row, get values of each column and adjust
// column widths.
int rowCount = 0;
while (rs.next()) {
// NOTE: columnIndex for rs.getXXX methods STARTS AT 1 NOT 0
for (int i = 0; i < columnCount; i++) {
Column c = columns.get(i);
String value;
int category = c.getTypeCategory();
if (category == CATEGORY_OTHER) {
// Use generic SQL type name instead of the actual value
// for column types BLOB, BINARY etc.
value = "(" + c.getTypeName() + ")";
} else {
value = rs.getString(i + 1) == null ? "NULL" : rs.getString(i + 1);
}
switch (category) {
case CATEGORY_DOUBLE:
// For real numbers, format the string value to have 3 digits
// after the point. THIS IS TOTALLY ARBITRARY and can be
// improved to be CONFIGURABLE.
if (!value.equals("NULL")) {
Double dValue = rs.getDouble(i + 1);
value = String.format("%.3f", dValue);
}
break;
case CATEGORY_STRING:
// Left justify the text columns
c.justifyLeft();
// and apply the width limit
if (value.length() > maxStringColWidth) {
value = value.substring(0, maxStringColWidth - 3) + "...";
}
break;
}
// Adjust the column width
c.setWidth(value.length() > c.getWidth() ? value.length() : c.getWidth());
c.addValue(value);
} // END of for loop columnCount
rowCount++;
} // END of while (rs.next)
/*
At this point we have gone through meta data, get the
columns and created all Column objects, iterated over the
ResultSet rows, populated the column values and adjusted
the column widths.
We cannot start printing just yet because we have to prepare
a row separator String.
*/
// For the fun of it, I will use StringBuilder
StringBuilder strToPrint = new StringBuilder();
StringBuilder rowSeparator = new StringBuilder();
/*
Prepare column labels to print as well as the row separator.
It should look something like this:
+--------+------------+------------+-----------+ (row separator)
| EMP_NO | BIRTH_DATE | FIRST_NAME | LAST_NAME | (labels row)
+--------+------------+------------+-----------+ (row separator)
*/
// Iterate over columns
for (Column c : columns) {
int width = c.getWidth();
// Center the column label
String toPrint;
String name = c.getLabel();
int diff = width - name.length();
if ((diff % 2) == 1) {
// diff is not divisible by 2, add 1 to width (and diff)
// so that we can have equal padding to the left and right
// of the column label.
width++;
diff++;
c.setWidth(width);
}
int paddingSize = diff / 2; // InteliJ says casting to int is redundant.
// Cool String repeater code thanks to user102008 at stackoverflow.com
// (http://tinyurl.com/7x9qtyg) "Simple way to repeat a string in java"
String padding = new String(new char[paddingSize]).replace("\0", " ");
toPrint = "| " + padding + name + padding + " ";
// END centering the column label
strToPrint.append(toPrint);
rowSeparator.append("+");
rowSeparator.append(new String(new char[width + 2]).replace("\0", "-"));
}
String lineSeparator = System.getProperty("line.separator");
// Is this really necessary ??
lineSeparator = lineSeparator == null ? "\n" : lineSeparator;
rowSeparator.append("+").append(lineSeparator);
strToPrint.append("|").append(lineSeparator);
strToPrint.insert(0, rowSeparator);
strToPrint.append(rowSeparator);
// Print out the formatted column labels
System.out.print(strToPrint.toString());
String format;
// Print out the rows
for (int i = 0; i < rowCount; i++) {
for (Column c : columns) {
// This should form a format string like: "%-60s"
format = String.format("| %%%s%ds ", c.getJustifyFlag(), c.getWidth());
System.out.print(
String.format(format, c.getValue(i))
);
}
System.out.println("|");
System.out.print(rowSeparator);
}
System.out.println();
/*
Hopefully this should have printed something like this:
+--------+------------+------------+-----------+--------+-------------+
| EMP_NO | BIRTH_DATE | FIRST_NAME | LAST_NAME | GENDER | HIRE_DATE |
+--------+------------+------------+-----------+--------+-------------+
| 10001 | 1953-09-02 | Georgi | Facello | M | 1986-06-26 |
+--------+------------+------------+-----------+--------+-------------+
| 10002 | 1964-06-02 | Bezalel | Simmel | F | 1985-11-21 |
+--------+------------+------------+-----------+--------+-------------+
*/
} catch (SQLException e) {
System.err.println("SQL exception in DBTablePrinter. Message:");
System.err.println(e.getMessage());
}
}
/**
* Takes a generic SQL type and returns the category this type
* belongs to. Types are categorized according to print formatting
* needs:
*
* Integers should not be truncated so column widths should
* be adjusted without a column width limit. Text columns should be
* left justified and can be truncated to a max. column width etc...
*
* See also:
* java.sql.Types
*
* @param type Generic SQL type
* @return The category this type belongs to
*/
private static int whichCategory(int type) {
switch (type) {
case Types.BIGINT:
case Types.TINYINT:
case Types.SMALLINT:
case Types.INTEGER:
return CATEGORY_INTEGER;
case Types.REAL:
case Types.DOUBLE:
case Types.DECIMAL:
return CATEGORY_DOUBLE;
case Types.DATE:
case Types.TIME:
//case Types.TIME_WITH_TIMEZONE:
case Types.TIMESTAMP:
//case Types.TIMESTAMP_WITH_TIMEZONE:
// return CATEGORY_DATETIME;
case Types.BOOLEAN:
return CATEGORY_BOOLEAN;
case Types.VARCHAR:
case Types.NVARCHAR:
case Types.LONGVARCHAR:
case Types.LONGNVARCHAR:
case Types.CHAR:
case Types.NCHAR:
return CATEGORY_STRING;
default:
return CATEGORY_OTHER;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy