org.relique.jdbc.csv.CsvMain Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of csvjdbc Show documentation
a Java JDBC driver for reading comma-separated-value files
There is a newer version: 1.0.46
/*
 *  CsvJdbc - a JDBC driver for CSV files
 *  Copyright (C) 2013  Simon Chenery
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *  This library 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
 *  Lesser General Public License for more details.
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
package org.relique.jdbc.csv;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.InputStreamReader;
import java.io.Reader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.util.Properties;

/**
 * Main class for CsvJdbc, so user can easily experiment with SQL statements in a terminal window.
 */
public class CsvMain
{
	public static void main(String []args)
	{
		Properties properties = new Properties();
		int exitStatus = 0;

		try
		{
			Class.forName("org.relique.jdbc.csv.CsvDriver");

			int argIndex = 0;
			if (argIndex + 1 < args.length && args[argIndex].equals("-p"))
			{
				/*
				 * Read directory and properties from a Properties file.
				 */
				try (Reader propertiesReader = new FileReader(args[argIndex + 1]))
				{
					properties.load(propertiesReader);
				}
				argIndex += 2;
			}
			else if (argIndex < args.length && args[argIndex].equals("-h"))
			{
				/*
				 * Print usage/help message.
				 */
				argIndex = args.length;
			}
			else if (argIndex < args.length && args[argIndex].startsWith("-"))
			{
				System.err.println(CsvResources.getString("unknownCommandLine") + ": " + args[argIndex]);
				System.err.println();

				argIndex = args.length;
			}

			if (argIndex < args.length)
			{
				/*
				 * Connect using URL containing directory name and properties.
				 */
				try (Connection connection = DriverManager.getConnection(args[argIndex], properties))
				{
					argIndex++;

					CsvStatement statement = (CsvStatement)connection.createStatement();

					BufferedReader reader;
					boolean isStdin;
					if (argIndex == args.length)
					{
						/*
						 * No files given, so read stdin.
						 */
						reader = new BufferedReader(new InputStreamReader(System.in));
						isStdin = true;
					}
					else
					{
						reader = new BufferedReader(new FileReader(args[argIndex]));
						isStdin = false;
						argIndex++;
					}

					/*
					 * Do not include header line in CSV output, if not included in source CSV file.
					 */
					boolean writeHeaderLine = !((CsvConnection)connection).isSuppressHeaders();

					/*
					 * Read and execute each SQL statement and write results to stdout.
					 */
					do
					{
						String line;
						StringBuilder sb = new StringBuilder();
						while ((line = reader.readLine()) != null)
						{
							sb.append(line);
							sb.append("\n");
						}
						if (!isStdin)
							reader.close();
						reader = null;

						if (statement.execute(sb.toString()))
						{
							ResultSet resultSet = statement.getResultSet();
							CsvDriver.writeToCsv(resultSet, System.out, writeHeaderLine);

							/*
							 * No header line for second, third, ... queries, so that
							 * output of several SQL queries can be concatenated.
							 */
							writeHeaderLine = false;

							while (statement.getMoreResults())
							{
								resultSet = statement.getResultSet();
								CsvDriver.writeToCsv(resultSet, System.out, writeHeaderLine);
							}
						}

						if (argIndex < args.length)
						{
							reader = new BufferedReader(new FileReader(args[argIndex]));
							argIndex++;
						}
					}
					while (reader != null);
				}
			}
			else
			{
				System.err.println(CsvResources.getString("usage"));
				exitStatus = -1;
			}
		}
		catch (Exception e)
		{
			e.printStackTrace();
			exitStatus = -1;
		}
		System.exit(exitStatus);
	}
}