Mercurial > hg > monetdb-java
annotate src/main/java/org/monetdb/client/JdbcClient.java @ 723:7200f7942b16
For JdbcClient enlarge the fetchsize at connection moment.
This allows statements such as: call sys.eval('SELECT * from sys.keywords order by 1 desc;');
to be executed succesfully.
Before this change it would list only the first 250 rows and fail message:
Error [M0M10] resultBlocks[1] should have been fetched by now
author | Martin van Dinther <martin.van.dinther@monetdbsolutions.com> |
---|---|
date | Thu, 19 Jan 2023 19:55:10 +0100 (2023-01-19) |
parents | aeb268156580 |
children | a80c21fe7bb2 |
rev | line source |
---|---|
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1 /* |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
2 * This Source Code Form is subject to the terms of the Mozilla Public |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
3 * License, v. 2.0. If a copy of the MPL was not distributed with this |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
5 * |
716
aeb268156580
Updated Copyright year.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
709
diff
changeset
|
6 * Copyright 1997 - July 2008 CWI, August 2008 - 2023 MonetDB B.V. |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
7 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
8 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
9 package org.monetdb.client; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
10 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
11 import org.monetdb.jdbc.MonetDriver; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
12 import org.monetdb.jdbc.MonetConnection; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
13 import org.monetdb.util.CmdLineOpts; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
14 import org.monetdb.util.Exporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
15 import org.monetdb.util.FileTransferHandler; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
16 import org.monetdb.util.MDBvalidator; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
17 import org.monetdb.util.OptionsException; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
18 import org.monetdb.util.SQLExporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
19 import org.monetdb.util.XMLExporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
20 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
21 import java.io.BufferedReader; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
22 import java.io.BufferedWriter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
23 import java.io.Console; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
24 import java.io.IOException; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
25 import java.io.InputStreamReader; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
26 import java.io.File; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
27 import java.io.PrintWriter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
28 import java.net.HttpURLConnection; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
29 import java.net.URL; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
30 import java.nio.charset.Charset; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
31 import java.sql.Connection; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
32 import java.sql.DatabaseMetaData; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
33 import java.sql.DriverManager; // this import is required as it will trigger loading the org.monetdb.jdbc.MonetDriver class |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
34 import java.sql.ResultSet; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
35 import java.sql.Statement; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
36 import java.sql.SQLException; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
37 import java.sql.SQLWarning; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
38 import java.util.ArrayList; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
39 import java.util.LinkedList; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
40 import java.util.List; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
41 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
42 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
43 * This program acts like an extended client program for MonetDB. Its |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
44 * look and feel is very much like PostgreSQL's interactive terminal |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
45 * program. Although it looks like this client is designed for MonetDB, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
46 * it demonstrates the power of the JDBC interface since it built on top |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
47 * of JDBC only. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
48 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
49 * @author Fabian Groffen |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
50 * @author Martin van Dinther |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
51 * @version 1.7 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
52 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
53 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
54 public class JdbcClient { /* cannot (yet) be final as nl.cwi.monetdb.client.JdbcClient extends this class */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
55 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
56 private static Connection con; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
57 private static DatabaseMetaData dbmd; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
58 private static Statement stmt; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
59 private static BufferedReader in; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
60 private static PrintWriter out; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
61 private static Exporter exporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
62 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
63 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
64 * JdbcClient is a command line query tool for MonetDB, similar to mclient. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
65 * It uses the JDBC API and the MonetDB JDBC driver to communicate with a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
66 * MonetDB server. The MonetDB JDBC driver is included in the jdbcclient.jre8.jar |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
67 * for ease of use, so only 1 jar file is needed to use it. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
68 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
69 * <pre>Usage java -jar jdbcclient.jre8.jar |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
70 * [-h host[:port]] [-p port] [-f file] [-u user] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
71 * [-l language] [-d database] [-e] [-D [table]] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
72 * [--csvdir /path/to/csvfiles] [-X<opt>] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
73 * | [--help] | [--version] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
74 * or using long option equivalents --host --port --file --user --language |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
75 * --dump --echo --database. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
76 * Arguments may be written directly after the option like -p50000. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
77 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
78 * If no host and port are given, localhost and 50000 are assumed. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
79 * An .monetdb file may exist in the user's home directory. This file can contain |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
80 * preferences to use each time JdbcClient is started. Options given on the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
81 * command line override the preferences file. The .monetdb file syntax is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
82 * <option>=<value> where option is one of the options host, port, file, mode |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
83 * debug, or password. Note that the last one is perilous and therefore not |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
84 * available as command line option. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
85 * If no input file is given using the -f flag, an interactive session is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
86 * started on the terminal. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
87 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
88 * OPTIONS |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
89 * -h --host The hostname of the host that runs the MonetDB database. A port |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
90 * number can be supplied by use of a colon, i.e. -h somehost:12345. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
91 * -p --port The port number to connect to. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
92 * -f --file A file name to use either for reading or writing. The file will |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
93 * be used for writing when dump mode is used (-D --dump). In read |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
94 * mode, the file can also be an URL pointing to a plain text file |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
95 * that is optionally gzip compressed. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
96 * -u --user The username to use when connecting to the database. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
97 * -d --database Try to connect to the given database (only makes sense if |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
98 * connecting to monetdbd). |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
99 * -l --language Use the given language, defaults to 'sql'. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
100 * --help This help screen. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
101 * --version Display driver version and exit. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
102 * -e --echo Also outputs the contents of the input file, if any. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
103 * -q --quiet Suppress printing the welcome header. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
104 * -D --dump Dumps the given table(s), or the complete database if none given. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
105 * --csvdir The directory path where csv data files wil be read from or |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
106 * written to when COPY ... ON CLIENT commands are executed. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
107 * -Xoutput The output mode when dumping. Default is sql, xml may be used for |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
108 * an experimental XML output. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
109 * -Xhash Use the given hash algorithm during challenge response. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
110 * Supported algorithm names: SHA512, SHA384, SHA256 and SHA1. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
111 * -Xdebug Writes a transmission log to disk for debugging purposes. If a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
112 * file name is given, it is used, otherwise a file called |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
113 * monet<timestamp>.log is created. A given file never be |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
114 * overwritten; instead a unique variation of the file is used. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
115 * -Xbatching Indicates that a batch should be used instead of direct |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
116 * communication with the server for each statement. If a number is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
117 * given, it is used as batch size. i.e. 8000 would execute the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
118 * contents on the batch after each 8000 statements read. Batching |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
119 * can greatly speedup the process of restoring a database dump.</pre> |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
120 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
121 * @param args optional list of startup arguments |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
122 * @throws Exception if uncaught exception is thrown |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
123 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
124 public final static void main(String[] args) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
125 final CmdLineOpts copts = new CmdLineOpts(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
126 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
127 // arguments which take exactly one argument |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
128 copts.addOption("h", "host", CmdLineOpts.CAR_ONE, "localhost", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
129 "The hostname of the host that runs the MonetDB database. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
130 "A port number can be supplied by use of a colon, i.e. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
131 "-h somehost:12345."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
132 copts.addOption("p", "port", CmdLineOpts.CAR_ONE, "50000", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
133 "The port number to connect to."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
134 // todo make it CAR_ONE_MANY |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
135 copts.addOption("f", "file", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
136 "A file name to use either for reading or writing. The " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
137 "file will be used for writing when dump mode is used " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
138 "(-D --dump). In read mode, the file can also be an URL " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
139 "pointing to a plain text file that is optionally gzip " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
140 "compressed."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
141 copts.addOption("u", "user", CmdLineOpts.CAR_ONE, System.getProperty("user.name"), |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
142 "The username to use when connecting to the database."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
143 // this one is only here for the .monetdb file parsing, it is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
144 // removed before the command line arguments are parsed |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
145 copts.addOption(null, "password", CmdLineOpts.CAR_ONE, null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
146 copts.addOption("d", "database", CmdLineOpts.CAR_ONE, "", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
147 "Try to connect to the given database (only makes sense " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
148 "if connecting to monetdbd)."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
149 copts.addOption("l", "language", CmdLineOpts.CAR_ONE, "sql", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
150 "Use the given language, defaults to 'sql'."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
151 copts.addOption(null, "csvdir", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
152 "The directory path where csv data files are read or " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
153 "written when using ON CLIENT clause of COPY command."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
154 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
155 // arguments which have no argument(s) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
156 copts.addOption(null, "help", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
157 "This help screen."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
158 copts.addOption(null, "version", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
159 "Display driver version and exit."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
160 copts.addOption("e", "echo", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
161 "Also outputs the contents of the input file, if any."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
162 copts.addOption("q", "quiet", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
163 "Suppress printing the welcome header."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
164 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
165 // arguments which have zero to many arguments |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
166 copts.addOption("D", "dump", CmdLineOpts.CAR_ZERO_MANY, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
167 "Dumps the given table(s), or the complete database if " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
168 "none given."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
169 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
170 // extended options |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
171 copts.addOption(null, "Xoutput", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
172 "The output mode when dumping. Default is sql, xml may " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
173 "be used for an experimental XML output."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
174 copts.addOption(null, "Xhash", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
175 "Use the given hash algorithm during challenge response. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
176 "Supported algorithm names: SHA512, SHA384, SHA256 and SHA1."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
177 // arguments which can have zero or one argument(s) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
178 copts.addOption(null, "Xdebug", CmdLineOpts.CAR_ZERO_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
179 "Writes a transmission log to disk for debugging purposes. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
180 "If a file name is given, it is used, otherwise a file " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
181 "called monet<timestamp>.log is created. A given file " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
182 "never be overwritten; instead a unique variation of the " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
183 "file is used."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
184 copts.addOption(null, "Xbatching", CmdLineOpts.CAR_ZERO_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
185 "Indicates that a batch should be used instead of direct " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
186 "communication with the server for each statement. If a " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
187 "number is given, it is used as batch size. i.e. 8000 " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
188 "would execute the contents on the batch after each 8000 " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
189 "statements read. Batching can greatly speedup the " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
190 "process of restoring a database dump."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
191 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
192 // we store user and password in separate variables in order to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
193 // be able to properly act on them like forgetting the password |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
194 // from the user's file if the user supplies a username on the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
195 // command line arguments |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
196 String pass = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
197 String user = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
198 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
199 // look for a file called .monetdb in the current dir or in the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
200 // user's homedir and read its preferences |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
201 File pref = new File(".monetdb"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
202 if (!pref.exists()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
203 pref = new File(System.getProperty("user.home"), ".monetdb"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
204 if (pref.exists()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
205 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
206 copts.processFile(pref); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
207 } catch (OptionsException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
208 System.err.println("Error in " + pref.getAbsolutePath() + ": " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
209 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
210 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
211 user = copts.getOption("user").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
212 pass = copts.getOption("password").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
213 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
214 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
215 // process the command line arguments, remove password option |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
216 // first, and save the user we had at this point |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
217 copts.removeOption("password"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
218 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
219 copts.processArgs(args); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
220 } catch (OptionsException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
221 System.err.println("Error: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
222 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
223 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
224 // we can actually compare pointers (objects) here |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
225 if (user != copts.getOption("user").getArgument()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
226 pass = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
227 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
228 if (copts.getOption("help").isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
229 System.out.print( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
230 "Usage java -jar jdbcclient.jre8.jar\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
231 "\t\t[-h host[:port]] [-p port] [-f file] [-u user]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
232 "\t\t[-l language] [-d database] [-e] [-D [table]]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
233 "\t\t[--csvdir /path/to/csvfiles]] [-X<opt>]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
234 "\t\t| [--help] | [--version]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
235 "or using long option equivalents --host --port --file --user --language\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
236 "--dump --echo --database.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
237 "Arguments may be written directly after the option like -p50000.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
238 "\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
239 "If no host and port are given, localhost and 50000 are assumed.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
240 "An .monetdb file may exist in the user's home directory. This file can contain\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
241 "preferences to use each time JdbcClient is started. Options given on the\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
242 "command line override the preferences file. The .monetdb file syntax is\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
243 "<option>=<value> where option is one of the options host, port, file, mode\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
244 "debug, or password. Note that the last one is perilous and therefore not\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
245 "available as command line option.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
246 "If no input file is given using the -f flag, an interactive session is\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
247 "started on the terminal.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
248 "\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
249 "OPTIONS\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
250 copts.produceHelpMessage() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
251 ); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
252 System.exit(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
253 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
254 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
255 if (copts.getOption("version").isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
256 // We cannot use the DatabaseMetaData here, because we |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
257 // cannot get a Connection. So instead, we just get the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
258 // values we want out of the Driver directly. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
259 System.out.println("JDBC Driver: v" + MonetDriver.getDriverVersion()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
260 System.exit(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
261 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
262 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
263 // whether the semi-colon at the end of a String terminates the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
264 // query or not (default = yes => SQL) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
265 final boolean scolonterm = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
266 final boolean xmlMode = "xml".equals(copts.getOption("Xoutput").getArgument()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
267 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
268 // we need the password from the user, fetch it with a pseudo |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
269 // password protector |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
270 if (pass == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
271 final Console syscon = System.console(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
272 char[] tmp = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
273 if (syscon != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
274 tmp = syscon.readPassword("password: "); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
275 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
276 if (tmp == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
277 System.err.println("Invalid password!"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
278 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
279 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
280 pass = String.valueOf(tmp); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
281 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
282 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
283 user = copts.getOption("user").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
284 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
285 // build the hostname |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
286 String host = copts.getOption("host").getArgument(); |
698
6beecac8aa1b
Use indexOf(char) instead of indexOf(String) where possible.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
675
diff
changeset
|
287 if (host.indexOf(':') == -1) { |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
288 host = host + ":" + copts.getOption("port").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
289 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
290 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
291 // build the extra arguments of the JDBC connect string |
723
7200f7942b16
For JdbcClient enlarge the fetchsize at connection moment.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
716
diff
changeset
|
292 // increase the fetchsize from the default 250 to 10000 |
7200f7942b16
For JdbcClient enlarge the fetchsize at connection moment.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
716
diff
changeset
|
293 String attr = "?fetchsize=10000&"; |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
294 CmdLineOpts.OptionContainer oc = copts.getOption("language"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
295 final String lang = oc.getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
296 if (oc.isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
297 attr += "language=" + lang + "&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
298 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
299 /* Xquery is no longer functional or supported |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
300 // set some behaviour based on the language XQuery |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
301 if (lang.equals("xquery")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
302 scolonterm = false; // no ; to end a statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
303 if (!copts.getOption("Xoutput").isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
304 xmlMode = true; // the user will like xml results, most probably |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
305 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
306 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
307 oc = copts.getOption("Xdebug"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
308 if (oc.isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
309 attr += "debug=true&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
310 if (oc.getArgumentCount() == 1) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
311 attr += "logfile=" + oc.getArgument() + "&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
312 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
313 oc = copts.getOption("Xhash"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
314 if (oc.isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
315 attr += "hash=" + oc.getArgument() + "&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
316 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
317 // request a connection suitable for MonetDB from the driver |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
318 // manager note that the database specifier is only used when |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
319 // connecting to a proxy-like service, since MonetDB itself |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
320 // can't access multiple databases. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
321 con = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
322 final String database = copts.getOption("database").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
323 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
324 // make sure the driver class is loaded (and thus register itself with the DriverManager) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
325 Class.forName("org.monetdb.jdbc.MonetDriver"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
326 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
327 con = DriverManager.getConnection( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
328 "jdbc:monetdb://" + host + "/" + database + attr, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
329 user, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
330 pass |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
331 ); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
332 SQLWarning warn = con.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
333 while (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
334 System.err.println("Connection warning: " + warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
335 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
336 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
337 con.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
338 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
339 System.err.println("Database connect failed: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
340 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
341 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
342 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
343 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
344 dbmd = con.getMetaData(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
345 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
346 // we ignore this because it's probably because we don't use |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
347 // SQL language |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
348 dbmd = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
349 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
350 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
351 oc = copts.getOption("csvdir"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
352 if (oc.isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
353 final String csvdir = oc.getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
354 if (csvdir != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
355 // check if provided csvdir is an existing dir |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
356 // else a download of data into file will terminate the JDBC connection!! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
357 if (java.nio.file.Files.isDirectory(java.nio.file.Paths.get(csvdir))) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
358 final FileTransferHandler FThandler = new FileTransferHandler(csvdir, Charset.defaultCharset()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
359 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
360 // register file data uploadHandler to allow support |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
361 // for: COPY INTO mytable FROM 'data.csv' ON CLIENT; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
362 ((MonetConnection) con).setUploadHandler(FThandler); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
363 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
364 // register file data downloadHandler to allow support |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
365 // for: COPY select_query INTO 'data.csv' ON CLIENT; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
366 ((MonetConnection) con).setDownloadHandler(FThandler); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
367 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
368 System.err.println("Warning: provided csvdir \"" + csvdir + "\" does not exist. Ignoring csvdir setting."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
369 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
370 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
371 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
372 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
373 stmt = con.createStatement(); // is used by processInteractive(), processBatch(), doDump() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
374 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
375 in = new BufferedReader(new InputStreamReader(System.in)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
376 out = new PrintWriter(new BufferedWriter(new java.io.OutputStreamWriter(System.out))); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
377 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
378 // see if we will have to perform a database dump (only in SQL mode) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
379 if ("sql".equals(lang) && copts.getOption("dump").isPresent() && dbmd != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
380 final int argcount = copts.getOption("dump").getArgumentCount(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
381 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
382 // use the given file for writing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
383 oc = copts.getOption("file"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
384 if (oc.isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
385 out = new PrintWriter(new BufferedWriter(new java.io.FileWriter(oc.getArgument()))); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
386 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
387 // we only want user tables and views to be dumped (DDL and optional data), unless a specific table is requested |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
388 final String[] types = {"TABLE","VIEW","MERGE TABLE","REMOTE TABLE","REPLICA TABLE","STREAM TABLE"}; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
389 // Future: fetch all type names using dbmd.getTableTypes() and construct String[] with all |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
390 // table type names excluding the SYSTEM ... ones and LOCAL TEMPORARY TABLE ones. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
391 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
392 // request the list of tables/views available in the current schema in the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
393 ResultSet tbl = dbmd.getTables(null, con.getSchema(), null, (argcount == 0) ? types : null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
394 // fetch all tables and store them in a LinkedList of Table objects |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
395 final LinkedList<Table> tables = new LinkedList<Table>(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
396 while (tbl.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
397 tables.add(new Table( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
398 tbl.getString(2), // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
399 tbl.getString(3), // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
400 tbl.getString(4))); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
401 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
402 tbl.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
403 tbl = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
404 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
405 if (xmlMode) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
406 exporter = new XMLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
407 exporter.setProperty(XMLExporter.TYPE_NIL, XMLExporter.VALUE_XSI); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
408 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
409 exporter = new SQLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
410 // stick with SQL INSERT INTO commands for now |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
411 // in the future we might do COPY INTO's here using VALUE_COPY |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
412 exporter.setProperty(SQLExporter.TYPE_OUTPUT, SQLExporter.VALUE_INSERT); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
413 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
414 exporter.useSchemas(true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
415 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
416 // start SQL output |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
417 if (!xmlMode) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
418 out.println("START TRANSACTION;\n"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
419 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
420 // dump specific table(s) or not? |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
421 if (argcount > 0) { // yes we do |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
422 final String[] dumpers = copts.getOption("dump").getArguments(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
423 for (int i = 0; i < tables.size(); i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
424 Table ttmp = tables.get(i); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
425 for (int j = 0; j < dumpers.length; j++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
426 String dumptblnm = dumpers[j].toString(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
427 if (ttmp.getName().equalsIgnoreCase(dumptblnm) || |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
428 ttmp.getFqname().equalsIgnoreCase(dumptblnm)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
429 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
430 // dump the table |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
431 doDump(out, ttmp); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
432 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
433 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
434 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
435 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
436 /* this returns everything, so including SYSTEM TABLE constraints */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
437 tbl = dbmd.getImportedKeys(null, null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
438 while (tbl.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
439 // find FK table object 6 = "FKTABLE_SCHEM", 7 = "FKTABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
440 Table fk = Table.findTable(tbl.getString(6), tbl.getString(7), tables); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
441 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
442 // find PK table object 2 = "PKTABLE_SCHEM", 3 = "PKTABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
443 Table pk = Table.findTable(tbl.getString(2), tbl.getString(3), tables); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
444 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
445 // this happens when a system table has referential constraints |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
446 if (fk == null || pk == null) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
447 continue; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
448 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
449 // add PK table dependency to FK table |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
450 fk.addDependency(pk); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
451 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
452 tbl.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
453 tbl = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
454 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
455 // search for cycles of type a -> (x ->)+ b probably not |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
456 // the most optimal way, but it works by just scanning |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
457 // every table for loops in a recursive manor |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
458 for (Table t : tables) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
459 Table.checkForLoop(t, new ArrayList<Table>()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
460 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
461 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
462 // find the graph, at this point we know there are no |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
463 // cycles, thus a solution exists |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
464 for (int i = 0; i < tables.size(); i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
465 final List<Table> needs = tables.get(i).requires(tables.subList(0, i + 1)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
466 if (needs.size() > 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
467 tables.removeAll(needs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
468 tables.addAll(i, needs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
469 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
470 // re-evaluate this position, for there is a new |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
471 // table now |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
472 i--; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
473 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
474 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
475 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
476 // we now have the right order to dump tables |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
477 for (Table t : tables) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
478 // dump the table |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
479 doDump(out, t); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
480 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
481 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
482 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
483 if (!xmlMode) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
484 out.println("COMMIT;"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
485 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
486 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
487 // free resources, close the statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
488 stmt.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
489 // close the connection with the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
490 con.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
491 // completed database dump |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
492 System.exit(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
493 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
494 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
495 if (xmlMode) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
496 exporter = new XMLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
497 exporter.setProperty(XMLExporter.TYPE_NIL, XMLExporter.VALUE_XSI); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
498 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
499 exporter = new SQLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
500 // we want nice table formatted output |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
501 exporter.setProperty(SQLExporter.TYPE_OUTPUT, SQLExporter.VALUE_TABLE); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
502 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
503 exporter.useSchemas(false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
504 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
505 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
506 // use the given file for reading |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
507 final boolean hasFile = copts.getOption("file").isPresent(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
508 final boolean doEcho = hasFile && copts.getOption("echo").isPresent(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
509 if (hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
510 final String tmp = copts.getOption("file").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
511 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
512 in = getReader(tmp); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
513 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
514 System.err.println("Error: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
515 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
516 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
517 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
518 // check for batch mode |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
519 int batchSize = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
520 oc = copts.getOption("Xbatching"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
521 if (oc.isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
522 if (oc.getArgumentCount() == 1) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
523 // parse the number |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
524 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
525 batchSize = Integer.parseInt(oc.getArgument()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
526 } catch (NumberFormatException ex) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
527 // complain to the user |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
528 throw new IllegalArgumentException("Illegal argument for Xbatching: " + oc.getArgument() + " is not a parseable number!"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
529 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
530 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
531 processBatch(batchSize); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
532 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
533 processInteractive(true, doEcho, scolonterm, user); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
534 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
535 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
536 if (!copts.getOption("quiet").isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
537 // print welcome message |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
538 out.println("Welcome to the MonetDB interactive JDBC terminal!"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
539 if (dbmd != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
540 out.println("JDBC Driver: " + dbmd.getDriverName() + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
541 " v" + dbmd.getDriverVersion()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
542 out.println("Database Server: " + dbmd.getDatabaseProductName() + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
543 " v" + dbmd.getDatabaseProductVersion()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
544 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
545 out.println("Current Schema: " + con.getSchema()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
546 out.println("Type \\q to quit (you can also use: quit or exit), \\? or \\h for a list of available commands"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
547 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
548 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
549 processInteractive(false, doEcho, scolonterm, user); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
550 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
551 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
552 // free resources, close the statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
553 stmt.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
554 // close the connection with the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
555 con.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
556 // close the file (if we used a file) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
557 in.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
558 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
559 System.err.println("A fatal exception occurred: " + e.toString()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
560 e.printStackTrace(System.err); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
561 // at least try to close the connection properly, since it will |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
562 // close all statements associated with it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
563 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
564 con.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
565 } catch (SQLException ex) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
566 // ok... nice try |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
567 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
568 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
569 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
570 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
571 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
572 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
573 * Tries to interpret the given String as URL or file. Returns the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
574 * assigned BufferedReader, or throws an Exception if the given |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
575 * string couldn't be identified as a valid URL or file. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
576 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
577 * @param uri URL or filename as String |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
578 * @return a BufferedReader for the uri |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
579 * @throws Exception if uri cannot be identified as a valid URL or file |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
580 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
581 static BufferedReader getReader(final String uri) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
582 BufferedReader ret = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
583 URL u = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
584 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
585 // Try and parse as URL first |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
586 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
587 u = new URL(uri); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
588 } catch (java.net.MalformedURLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
589 // no URL, try as file |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
590 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
591 ret = new BufferedReader(new java.io.FileReader(uri)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
592 } catch (java.io.FileNotFoundException fnfe) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
593 // the message is descriptive enough, adds "(No such file |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
594 // or directory)" itself. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
595 throw new Exception(fnfe.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
596 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
597 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
598 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
599 if (ret == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
600 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
601 HttpURLConnection.setFollowRedirects(true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
602 final HttpURLConnection con = (HttpURLConnection)u.openConnection(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
603 con.setRequestMethod("GET"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
604 final String ct = con.getContentType(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
605 if ("application/x-gzip".equals(ct)) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
606 // open gzip stream |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
607 ret = new BufferedReader(new InputStreamReader( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
608 new java.util.zip.GZIPInputStream(con.getInputStream()))); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
609 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
610 // text/plain otherwise just attempt to read as is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
611 ret = new BufferedReader(new InputStreamReader(con.getInputStream())); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
612 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
613 } catch (IOException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
614 // failed to open the url |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
615 throw new Exception("No such host/file: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
616 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
617 // this is an exception that comes from deep ... |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
618 throw new Exception("Invalid URL: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
619 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
620 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
621 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
622 return ret; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
623 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
624 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
625 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
626 * Starts an interactive processing loop, where output is adjusted to an |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
627 * user session. This processing loop is not suitable for bulk processing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
628 * as in executing the contents of a file, since processing on the given |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
629 * input is done after each row that has been entered. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
630 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
631 * @param hasFile a boolean indicating whether a file is used as input |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
632 * @param doEcho a boolean indicating whether to echo the given input |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
633 * @param scolonterm whether a ';' makes this query part complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
634 * @param user a String representing the username of the current user |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
635 * @throws IOException if an IO exception occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
636 * @throws SQLException if a database related error occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
637 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
638 private static void processInteractive( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
639 final boolean hasFile, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
640 final boolean doEcho, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
641 final boolean scolonterm, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
642 final String user) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
643 throws IOException, SQLException |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
644 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
645 // an SQL stack keeps track of ( " and ' |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
646 final SQLStack stack = new SQLStack(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
647 boolean lastac = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
648 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
649 if (!hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
650 lastac = con.getAutoCommit(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
651 out.println("auto commit mode: " + (lastac ? "on" : "off")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
652 out.print(getPrompt(stack, true)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
653 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
654 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
655 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
656 String curLine; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
657 String query = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
658 boolean doProcess; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
659 boolean wasComplete = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
660 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
661 // the main (interactive) process loop |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
662 for (long i = 1; true; i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
663 // Manually read a line, because we want to detect an EOF |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
664 // (ctrl-D). Doing so allows to have a terminator for query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
665 // which is not based on a special character, as is the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
666 // problem for XQuery |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
667 curLine = in.readLine(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
668 if (curLine == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
669 out.println(""); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
670 if (!query.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
671 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
672 executeQuery(query, stmt, out, !hasFile); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
673 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
674 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
675 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
676 if (hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
677 System.err.println("Error on line " + i + ": [" + e.getSQLState() + "] " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
678 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
679 System.err.println("Error [" + e.getSQLState() + "]: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
680 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
681 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
682 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
683 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
684 query = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
685 wasComplete = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
686 if (!hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
687 final boolean ac = con.getAutoCommit(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
688 if (ac != lastac) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
689 out.println("auto commit mode: " + (ac ? "on" : "off")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
690 lastac = ac; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
691 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
692 out.print(getPrompt(stack, wasComplete)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
693 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
694 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
695 // try to read again |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
696 continue; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
697 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
698 // user did ctrl-D without something in the buffer, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
699 // so terminate |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
700 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
701 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
702 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
703 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
704 if (doEcho) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
705 out.println(curLine); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
706 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
707 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
708 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
709 // a query part is a line of an SQL query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
710 QueryPart qp = scanQuery(curLine, stack, scolonterm); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
711 if (!qp.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
712 final String command = qp.getQuery(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
713 doProcess = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
714 if (wasComplete) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
715 doProcess = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
716 // check for commands only when the previous row was complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
717 if (command.equals("\\q") || command.equals("quit") || command.equals("exit")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
718 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
719 } else if (dbmd != null && command.startsWith("\\d")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
720 processDescribeCmd(command, scolonterm); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
721 } else if (command.startsWith("\\v")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
722 if (command.equals("\\vsci")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
723 MDBvalidator.validateSqlCatalogIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
724 } else if (command.equals("\\vsci_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
725 MDBvalidator.validateSqlCatalogIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
726 } else if (command.equals("\\vsni")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
727 MDBvalidator.validateSqlNetcdfTablesIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
728 } else if (command.equals("\\vsni_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
729 MDBvalidator.validateSqlNetcdfTablesIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
730 } else if (command.equals("\\vsgi")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
731 MDBvalidator.validateSqlGeomTablesIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
732 } else if (command.equals("\\vsgi_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
733 MDBvalidator.validateSqlGeomTablesIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
734 } else if (command.startsWith("\\vsi ")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
735 String schema_nm = command.substring(5); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
736 MDBvalidator.validateSchemaIntegrity(con, schema_nm, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
737 } else if (command.startsWith("\\vsi_noheader ")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
738 String schema_nm = command.substring(14); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
739 MDBvalidator.validateSchemaIntegrity(con, schema_nm, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
740 } else if (command.equals("\\vdbi")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
741 MDBvalidator.validateDBIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
742 } else if (command.equals("\\vdbi_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
743 MDBvalidator.validateDBIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
744 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
745 showCommands(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
746 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
747 } else if (command.startsWith("\\l") || command.startsWith("\\i")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
748 String object = command.substring(2).trim(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
749 if (scolonterm && object.endsWith(";")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
750 object = object.substring(0, object.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
751 if (object.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
752 System.err.println("Usage: '" + command.substring(0, 2) + "<uri>' where <uri> is a file or URL"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
753 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
754 // temporarily redirect input from in |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
755 final BufferedReader console = in; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
756 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
757 in = getReader(object); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
758 if (command.startsWith("\\l")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
759 processInteractive(true, doEcho, scolonterm, user); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
760 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
761 processBatch(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
762 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
763 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
764 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
765 System.err.println("Error: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
766 } finally { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
767 // put back in redirection |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
768 in = console; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
769 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
770 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
771 } else if (command.equals("\\?") || command.equals("\\h") || command.startsWith("\\")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
772 showCommands(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
773 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
774 doProcess = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
775 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
776 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
777 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
778 if (doProcess) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
779 query += command + (qp.hasOpenQuote() ? "\\n" : " "); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
780 if (qp.isComplete()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
781 // strip off trailing ';' |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
782 query = query.substring(0, query.length() - 2); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
783 // execute query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
784 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
785 executeQuery(query, stmt, out, !hasFile); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
786 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
787 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
788 final String startmsg = (hasFile ? ("Error on line " + i + ": [") : "Error ["); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
789 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
790 System.err.println(startmsg + e.getSQLState() + "] " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
791 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
792 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
793 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
794 query = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
795 wasComplete = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
796 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
797 wasComplete = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
798 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
799 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
800 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
801 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
802 if (!hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
803 final boolean ac = con.getAutoCommit(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
804 if (ac != lastac) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
805 out.println("auto commit mode: " + (ac ? "on" : "off")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
806 lastac = ac; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
807 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
808 out.print(getPrompt(stack, wasComplete)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
809 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
810 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
811 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
812 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
813 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
814 private static void showCommands() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
815 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
816 out.println("Available commands:"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
817 out.println("\\q quits this program (you can also use: quit or exit)"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
818 if (dbmd != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
819 out.println("\\d list available user tables and views in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
820 out.println("\\dt list available user tables in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
821 out.println("\\dv list available user views in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
822 out.println("\\df list available user functions in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
823 out.println("\\dp list available user procedures in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
824 out.println("\\ds list available user sequences in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
825 out.println("\\dn list available user schemas"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
826 out.println("\\dS list available system tables and views in sys/tmp schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
827 out.println("\\dSt list available system tables in sys/tmp schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
828 out.println("\\dSv list available system views in sys/logging schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
829 out.println("\\dSf list available system functions in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
830 out.println("\\dSp list available system procedures in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
831 out.println("\\dSn list available system schemas"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
832 out.println("\\d <obj> describes the given table or view"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
833 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
834 out.println("\\l<uri> executes the contents of the given file or URL"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
835 out.println("\\i<uri> batch executes the inserts from the given file or URL"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
836 out.println("\\vsci validate sql system catalog integrity"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
837 // out.println("\\vsni validate sql system netcdf tables integrity"); // do not print as it depends on availability of netcdf library on server |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
838 // out.println("\\vsgi validate sql system geom tables integrity"); // do not print as it depends on availability of geom library on server |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
839 out.println("\\vsi <schema> validate integrity of data in the given schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
840 out.println("\\vdbi validate integrity of data in all user schemas in the database"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
841 out.println("\\? or \\h this help screen"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
842 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
843 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
844 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
845 * Process the commands that start with \d |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
846 * These commands list the available user or system tables, views, functions, procedures, schemas |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
847 * or user sequences |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
848 * or describe a specific table or view by printing the CREATE DDL statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
849 * See also showCommands() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
850 * These commands are almost the same as in mclient program |
709
bdeabbd46ec6
Resolve javac and javadoc warnings when compiled with JDK19.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
698
diff
changeset
|
851 * |
bdeabbd46ec6
Resolve javac and javadoc warnings when compiled with JDK19.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
698
diff
changeset
|
852 * @param dcommand the command starting with \d |
bdeabbd46ec6
Resolve javac and javadoc warnings when compiled with JDK19.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
698
diff
changeset
|
853 * @param scolonterm whether a ';' makes this query part complete |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
854 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
855 private static void processDescribeCmd(final String dcommand, final boolean scolonterm) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
856 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
857 final String command = dcommand.substring(1); // remove the leading \ character first |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
858 ResultSet rs = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
859 Statement st = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
860 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
861 if (command.equals("d") || command.equals("dt") || command.equals("dv")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
862 // list available user tables and/or views in current schema (maybe tmp schema to show temporary tables) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
863 boolean tablesOnly = command.equals("dt"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
864 boolean viewsOnly = command.equals("dv"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
865 rs = dbmd.getTables(null, con.getSchema(), null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
866 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
867 final String tableType = rs.getString(4); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
868 boolean include = (tableType != null && !tableType.startsWith("SYSTEM ")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
869 if (include && tablesOnly && !tableType.contains("TABLE")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
870 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
871 if (include && viewsOnly && !tableType.contains("VIEW")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
872 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
873 if (include) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
874 out.println(tableType + "\t" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
875 optAddDqs(rs.getString(2)) + "." + // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
876 optAddDqs(rs.getString(3)) ); // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
877 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
878 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
879 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
880 if (command.equals("dS") || command.equals("dSt") || command.equals("dSv") || command.equals("dtS") || command.equals("dvS")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
881 // list available system tables and/or views in sys/tmp/logging schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
882 boolean tablesOnly = command.contains("t"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
883 boolean viewsOnly = command.contains("v"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
884 String curSchema = con.getSchema(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
885 // only schemas: sys, tmp and logging contain system tables and views. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
886 if (!("sys".equals(curSchema) || "tmp".equals(curSchema) || "logging".equals(curSchema))) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
887 // when currently in another schema, default to the sys tables/views else nothing will be listed |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
888 curSchema = "sys"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
889 rs = dbmd.getTables(null, curSchema, null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
890 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
891 final String tableType = rs.getString(4); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
892 boolean include = (tableType != null && tableType.startsWith("SYSTEM ")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
893 if (include && tablesOnly && !tableType.contains("TABLE")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
894 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
895 if (include && viewsOnly && !tableType.contains("VIEW")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
896 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
897 if (include) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
898 out.println(tableType + "\t" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
899 optAddDqs(rs.getString(2)) + "." + // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
900 optAddDqs(rs.getString(3)) ); // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
901 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
902 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
903 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
904 if (command.equals("df") || command.equals("dp") || command.equals("dSf") || command.equals("dSp") || command.equals("dfS") || command.equals("dpS")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
905 // list available user/system functions or procedures in current schema (maybe tmp schema) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
906 boolean proceduresOnly = command.contains("p"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
907 boolean systemOnly = command.contains("S"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
908 // Note we cannot use: dbmd.getFunctions(null, con.getSchema(), null); or dbmd.getProcedures(null, con.getSchema(), null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
909 // as they do not return the full function type name, nor provide information whether it is a system function or system procedure. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
910 // Instead we query the MonetDB system catog tables directly using: |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
911 String qy = "SELECT DISTINCT " + (systemOnly ? "'SYSTEM ' || " : "") + "\"function_type_keyword\" as func_type, " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
912 "\"schemas\".\"name\" as schem_name, \"functions\".\"name\" as func_name" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
913 " FROM sys.\"functions\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
914 " JOIN sys.\"function_types\" ON \"functions\".\"type\" = \"function_types\".\"function_type_id\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
915 " JOIN sys.\"schemas\" ON \"functions\".\"schema_id\" = \"schemas\".\"id\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
916 " WHERE \"functions\".\"system\" = " + (systemOnly ? "true" : "false") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
917 " AND \"functions\".\"type\" " + (proceduresOnly ? "= 2" : "<> 2") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
918 " AND \"schemas\".\"name\" = current_schema" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
919 " ORDER BY 2, 3"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
920 st = con.createStatement(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
921 rs = st.executeQuery(qy); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
922 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
923 out.println(rs.getString(1) + " \t" + // 1 = func_type |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
924 optAddDqs(rs.getString(2)) + "." + // 2 = schem_name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
925 optAddDqs(rs.getString(3)) ); // 3 = func_name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
926 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
927 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
928 if (command.equals("ds")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
929 // list available (user) sequences in current schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
930 String qy = "SELECT \"schemas\".\"name\", \"sequences\".\"name\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
931 " FROM sys.\"sequences\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
932 " JOIN sys.\"schemas\" ON \"sequences\".\"schema_id\" = \"schemas\".\"id\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
933 " WHERE \"schemas\".\"name\" = current_schema" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
934 " ORDER BY 1, 2"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
935 st = con.createStatement(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
936 rs = st.executeQuery(qy); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
937 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
938 out.println("SEQUENCE " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
939 optAddDqs(rs.getString(1)) + "." + // 1 = schemas.name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
940 optAddDqs(rs.getString(2)) ); // 2 = sequences.name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
941 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
942 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
943 if (command.equals("dn") || command.equals("dSn") || command.equals("dnS")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
944 // list available user/system schemas in the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
945 boolean systemOnly = command.contains("S"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
946 String qy = "SELECT \"name\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
947 " FROM sys.\"schemas\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
948 " WHERE \"system\" = " + (systemOnly ? "true" : "false") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
949 " ORDER BY 1"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
950 String schema_type = (systemOnly ? "SYSTEM SCHEMA\t" : "SCHEMA\t"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
951 st = con.createStatement(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
952 rs = st.executeQuery(qy); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
953 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
954 out.println(schema_type + optAddDqs(rs.getString(1)) ); // 1 = schemas.name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
955 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
956 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
957 if (command.startsWith("d ")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
958 // describes the given table or view name. It may be a fully qualified name such as: sys."keys" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
959 String object = command.substring(2).trim(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
960 if (scolonterm && object.endsWith(";")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
961 object = object.substring(0, object.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
962 if (object.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
963 System.err.println("Missing name of object to describe"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
964 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
965 // extract the name of the schema first. If none found use current schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
966 String schema; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
967 String obj_nm = object; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
968 int len; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
969 boolean found = false; |
698
6beecac8aa1b
Use indexOf(char) instead of indexOf(String) where possible.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
675
diff
changeset
|
970 final int dot = object.indexOf('.'); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
971 if (dot > 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
972 // use specified schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
973 schema = object.substring(0, dot); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
974 obj_nm = object.substring(dot + 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
975 // remove potential surrounding double quotes around schema name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
976 len = schema.length(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
977 if (len > 2 && schema.charAt(0) == '"' && schema.charAt(len -1) == '"') |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
978 schema = schema.substring(1, len -1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
979 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
980 // use current schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
981 schema = con.getSchema(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
982 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
983 // remove potential surrounding double quotes around table or view name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
984 len = obj_nm.length(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
985 if (len > 2 && obj_nm.charAt(0) == '"' && obj_nm.charAt(len -1) == '"') |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
986 obj_nm = obj_nm.substring(1, len -1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
987 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
988 // System.err.println("calling dbmd.getTables(" + schema + ", " + obj_nm + ")"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
989 rs = dbmd.getTables(null, schema, obj_nm, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
990 while (rs.next() && !found) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
991 final String schemaName = rs.getString(2); // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
992 final String tableName = rs.getString(3); // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
993 if (obj_nm.equals(tableName) && schema.equals(schemaName)) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
994 String tableType = rs.getString(4); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
995 if (tableType.startsWith("SYSTEM ")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
996 tableType = tableType.substring(7); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
997 // we found it, describe it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
998 exporter.dumpSchema(dbmd, tableType, schemaName, tableName); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
999 found = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1000 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1001 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1002 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1003 if (!found) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1004 System.err.println("table or view: " + schema + "." + obj_nm + " does not exist"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1005 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1006 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1007 if (command.startsWith("d")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1008 System.err.println("unknown sub-command for \\d: " + command.substring(1)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1009 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1010 showCommands(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1011 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1012 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1013 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1014 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1015 System.err.println("Error [" + e.getSQLState() + "]: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1016 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1017 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1018 } finally { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1019 if (rs != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1020 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1021 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1022 } catch (SQLException e) { /* ignore */ } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1023 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1024 if (st != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1025 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1026 st.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1027 } catch (SQLException e) { /* ignore */ } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1028 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1029 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1030 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1031 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1032 private static String optAddDqs(final String name) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1033 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1034 if (name.contains(" ") || name.contains("\t") || name.contains("\n")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1035 return Exporter.dq(name); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1036 return name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1037 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1038 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1039 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1040 * Executes the given query and prints the result tabularised to the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1041 * given PrintWriter stream. The result of this method is the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1042 * default output of a query: tabular data. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1043 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1044 * @param query the query to execute |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1045 * @param stmt the Statement to execute the query on |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1046 * @param out the PrintWriter to write to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1047 * @param showTiming flag to specify if timing information nees to be printed |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1048 * @throws SQLException if a database related error occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1049 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1050 private static void executeQuery(final String query, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1051 final Statement stmt, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1052 final PrintWriter out, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1053 final boolean showTiming) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1054 throws SQLException |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1055 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1056 // warnings generated during querying |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1057 SQLWarning warn; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1058 long startTime = (showTiming ? System.currentTimeMillis() : 0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1059 long finishTime = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1060 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1061 // execute the query, let the driver decide what type it is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1062 int aff = -1; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1063 boolean nextRslt = stmt.execute(query, Statement.RETURN_GENERATED_KEYS); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1064 if (!nextRslt) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1065 aff = stmt.getUpdateCount(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1066 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1067 if (nextRslt) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1068 // we have a ResultSet, print it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1069 final ResultSet rs = stmt.getResultSet(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1070 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1071 exporter.dumpResultSet(rs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1072 if (showTiming) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1073 finishTime = System.currentTimeMillis(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1074 out.println("Elapsed Time: " + (finishTime - startTime) + " ms"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1075 startTime = finishTime; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1076 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1077 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1078 // if there were warnings for this result, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1079 // show them! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1080 warn = rs.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1081 if (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1082 // force stdout to be written so the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1083 // warning appears below it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1084 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1085 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1086 System.err.println("ResultSet warning: " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1087 warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1088 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1089 } while (warn != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1090 rs.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1091 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1092 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1093 } else if (aff != -1) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1094 String timingoutput = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1095 if (showTiming) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1096 finishTime = System.currentTimeMillis(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1097 timingoutput = ". Elapsed Time: " + (finishTime - startTime) + " ms"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1098 startTime = finishTime; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1099 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1100 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1101 if (aff == Statement.SUCCESS_NO_INFO) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1102 out.println("Operation successful" + timingoutput); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1103 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1104 // we have an update count |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1105 // see if a key was generated |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1106 final ResultSet rs = stmt.getGeneratedKeys(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1107 final boolean hasGeneratedKeyData = rs.next(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1108 out.println(aff + " affected row" + (aff != 1 ? "s" : "") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1109 (hasGeneratedKeyData ? ", last generated key: " + rs.getString(1) : "") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1110 timingoutput); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1111 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1112 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1113 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1114 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1115 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1116 } while ((nextRslt = stmt.getMoreResults()) || |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1117 (aff = stmt.getUpdateCount()) != -1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1118 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1119 // if there were warnings for this statement show them! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1120 warn = stmt.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1121 while (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1122 System.err.println("Statement warning: " + warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1123 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1124 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1125 stmt.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1126 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1127 // if there were warnings for this connection show them! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1128 warn = con.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1129 while (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1130 // suppress warning when issueing a "set schema xyz;" command |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1131 // if ( !(warn.getMessage()).equals("Server enabled auto commit mode while local state already was auto commit.") ) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1132 System.err.println("Connection warning: " + warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1133 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1134 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1135 con.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1136 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1137 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1138 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1139 * Starts a processing loop optimized for processing (large) chunks of |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1140 * continuous data, such as input from a file. Unlike in the interactive |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1141 * loop above, queries are sent only to the database if a certain batch |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1142 * amount is reached. No client side query checks are made, but everything |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1143 * is sent to the server as-is. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1144 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1145 * @param batchSize the number of items to store in the batch before |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1146 * sending them to the database for execution. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1147 * @throws IOException if an IO exception occurs. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1148 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1149 private static void processBatch(final int batchSize) throws IOException { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1150 final StringBuilder query = new StringBuilder(2048); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1151 int i = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1152 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1153 String curLine; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1154 // the main loop |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1155 for (i = 1; (curLine = in.readLine()) != null; i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1156 query.append(curLine); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1157 if (curLine.endsWith(";")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1158 // lousy check for end of statement, but in batch mode it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1159 // is not very important to catch all end of statements... |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1160 stmt.addBatch(query.toString()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1161 query.setLength(0); // clear the buffer |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1162 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1163 query.append('\n'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1164 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1165 if (batchSize > 0 && i % batchSize == 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1166 stmt.executeBatch(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1167 // stmt.clearBatch(); // this is no longer needed after call executeBatch(), see https://www.monetdb.org/bugzilla/show_bug.cgi?id=6953 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1168 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1169 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1170 stmt.addBatch(query.toString()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1171 stmt.executeBatch(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1172 // stmt.clearBatch(); // this is no longer needed after call executeBatch(), see https://www.monetdb.org/bugzilla/show_bug.cgi?id=6953 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1173 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1174 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1175 System.err.println("Error at line " + i + ": [" + e.getSQLState() + "] " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1176 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1177 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1178 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1179 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1180 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1181 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1182 * Wrapper method that decides to dump SQL or XML. In the latter case, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1183 * this method does the XML data generation. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1184 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1185 * @param out a Writer to write the data to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1186 * @param table the table to dump |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1187 * @throws SQLException if a database related error occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1188 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1189 private static void doDump(final PrintWriter out, final Table table) throws SQLException { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1190 final String tableType = table.getType(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1191 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1192 // dump CREATE definition of this table/view |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1193 exporter.dumpSchema(dbmd, tableType, table.getSchem(), table.getName()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1194 out.println(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1195 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1196 // only dump data from real tables, not from VIEWs / MERGE / REMOTE / REPLICA / STREAM tables |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1197 if (tableType.contains("TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1198 && !tableType.equals("MERGE TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1199 && !tableType.equals("REMOTE TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1200 && !tableType.equals("REPLICA TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1201 && !tableType.equals("STREAM TABLE")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1202 final ResultSet rs = stmt.executeQuery("SELECT * FROM " + table.getFqnameQ()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1203 if (rs != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1204 exporter.dumpResultSet(rs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1205 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1206 out.println(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1207 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1208 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1209 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1210 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1211 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1212 * Simple helper method that generates a prompt. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1213 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1214 * @param stack the current SQLStack |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1215 * @param compl whether the statement is complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1216 * @return a prompt which consist of "sql" plus the top of the stack |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1217 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1218 private static String getPrompt(final SQLStack stack, final boolean compl) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1219 return (compl ? "sql" : "more") + (stack.empty() ? ">" : stack.peek()) + " "; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1220 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1221 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1222 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1223 * Scans the given string and tries to discover if it is a complete query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1224 * or that there needs something to be added. If a string doesn't end with |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1225 * a ; it is considered not to be complete. SQL string quotation using ' and |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1226 * SQL identifier quotation using " is taken into account when scanning a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1227 * string this way. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1228 * Additionally, this method removes comments from the SQL statements, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1229 * identified by -- and removes white space where appropriate. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1230 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1231 * @param query the query to parse |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1232 * @param stack query stack to work with |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1233 * @param scolonterm whether a ';' makes this query part complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1234 * @return a QueryPart object containing the results of this parse |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1235 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1236 private static QueryPart scanQuery( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1237 final String query, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1238 final SQLStack stack, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1239 final boolean scolonterm) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1240 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1241 // examine string, char for char |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1242 final boolean wasInString = (stack.peek() == '\''); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1243 final boolean wasInIdentifier = (stack.peek() == '"'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1244 boolean escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1245 int len = query.length(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1246 for (int i = 0; i < len; i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1247 switch(query.charAt(i)) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1248 case '\\': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1249 escaped = !escaped; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1250 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1251 default: |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1252 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1253 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1254 case '\'': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1255 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1256 * We can not be in a string if we are in an identifier. So |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1257 * If we find a ' and are not in an identifier, and not in |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1258 * a string we can safely assume we will be now in a string. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1259 * If we are in a string already, we should stop being in a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1260 * string if we find a quote which is not prefixed by a \, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1261 * for that would be an escaped quote. However, a nasty |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1262 * situation can occur where the string is like 'test \\'. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1263 * As obvious, a test for a \ in front of a ' doesn't hold |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1264 * here. Because 'test \\\'' can exist as well, we need to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1265 * know if a quote is prefixed by an escaping slash or not. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1266 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1267 if (!escaped && stack.peek() != '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1268 if (stack.peek() != '\'') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1269 // although it makes no sense to escape a quote |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1270 // outside a string, it is escaped, thus not meant |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1271 // as quote for us, apparently |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1272 stack.push('\''); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1273 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1274 stack.pop(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1275 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1276 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1277 // reset escaped flag |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1278 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1279 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1280 case '"': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1281 if (!escaped && stack.peek() != '\'') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1282 if (stack.peek() != '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1283 stack.push('"'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1284 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1285 stack.pop(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1286 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1287 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1288 // reset escaped flag |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1289 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1290 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1291 case '-': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1292 if (!escaped && stack.peek() != '\'' && stack.peek() != '"' && i + 1 < len && query.charAt(i + 1) == '-') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1293 len = i; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1294 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1295 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1296 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1297 case '(': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1298 if (!escaped && stack.peek() != '\'' && stack.peek() != '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1299 stack.push('('); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1300 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1301 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1302 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1303 case ')': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1304 if (!escaped && stack.peek() == '(') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1305 stack.pop(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1306 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1307 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1308 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1309 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1310 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1311 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1312 int start = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1313 if (!wasInString && !wasInIdentifier && len > 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1314 // trim spaces at the start of the string |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1315 for (; start < len && Character.isWhitespace(query.charAt(start)); start++); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1316 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1317 int stop = len - 1; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1318 if (stack.peek() != '\'' && !wasInIdentifier && stop > start) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1319 // trim spaces at the end of the string |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1320 for (; stop >= start && Character.isWhitespace(query.charAt(stop)); stop--); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1321 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1322 stop++; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1323 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1324 if (start == stop) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1325 // we have an empty string |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1326 return new QueryPart(false, null, stack.peek() == '\'' || stack.peek() == '"'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1327 } else if (stack.peek() == '\'' || stack.peek() == '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1328 // we have an open quote |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1329 return new QueryPart(false, query.substring(start, stop), true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1330 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1331 // see if the string is complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1332 if (scolonterm && query.charAt(stop - 1) == ';') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1333 return new QueryPart(true, query.substring(start, stop), false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1334 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1335 return new QueryPart(false, query.substring(start, stop), false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1336 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1337 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1338 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1339 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1340 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1341 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1342 * A QueryPart is (a part of) a SQL query. In the QueryPart object information |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1343 * like the actual SQL query string, whether it has an open quote and the like |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1344 * is stored. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1345 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1346 final class QueryPart { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1347 private final boolean complete; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1348 private final String query; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1349 private final boolean open; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1350 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1351 QueryPart(final boolean complete, final String query, final boolean open) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1352 this.complete = complete; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1353 this.query = query; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1354 this.open = open; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1355 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1356 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1357 boolean isEmpty() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1358 return query == null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1359 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1360 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1361 boolean isComplete() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1362 return complete; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1363 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1364 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1365 String getQuery() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1366 return query; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1367 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1368 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1369 boolean hasOpenQuote() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1370 return open; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1371 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1372 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1373 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1374 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1375 * An SQLStack is a simple stack that keeps track of open brackets and |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1376 * (single and double) quotes in an SQL query. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1377 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1378 final class SQLStack { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1379 final StringBuilder stack = new StringBuilder(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1380 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1381 char peek() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1382 if (empty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1383 return '\0'; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1384 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1385 return stack.charAt(stack.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1386 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1387 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1388 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1389 char pop() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1390 final char tmp = peek(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1391 if (tmp != '\0') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1392 stack.setLength(stack.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1393 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1394 return tmp; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1395 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1396 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1397 char push(char item) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1398 stack.append(item); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1399 return item; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1400 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1401 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1402 boolean empty() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1403 return stack.length() == 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1404 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1405 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1406 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1407 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1408 * A Table represents an SQL table. All data required to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1409 * generate a fully qualified name is stored, as well as dependency |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1410 * data. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1411 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1412 final class Table { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1413 final String schem; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1414 final String name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1415 final String type; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1416 final String fqname; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1417 final ArrayList<Table> needs = new ArrayList<Table>(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1418 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1419 Table(final String schem, final String name, final String type) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1420 this.schem = schem; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1421 this.name = name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1422 this.type = type; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1423 this.fqname = schem + "." + name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1424 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1425 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1426 void addDependency(final Table dependsOn) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1427 if (this.fqname.equals(dependsOn.fqname)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1428 throw new Exception("Cyclic dependency graphs are not supported (foreign key relation references self)"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1429 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1430 if (dependsOn.needs.contains(this)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1431 throw new Exception("Cyclic dependency graphs are not supported (foreign key relation a->b and b->a)"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1432 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1433 if (!needs.contains(dependsOn)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1434 needs.add(dependsOn); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1435 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1436 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1437 List<Table> requires(final List<Table> existingTables) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1438 if (existingTables == null || existingTables.isEmpty()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1439 return new ArrayList<Table>(needs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1440 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1441 final ArrayList<Table> req = new ArrayList<Table>(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1442 for (Table n : needs) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1443 if (!existingTables.contains(n)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1444 req.add(n); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1445 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1446 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1447 return req; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1448 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1449 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1450 final String getSchem() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1451 return schem; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1452 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1453 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1454 final String getName() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1455 return name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1456 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1457 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1458 final String getType() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1459 return type; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1460 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1461 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1462 final String getFqname() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1463 return fqname; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1464 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1465 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1466 final String getFqnameQ() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1467 return Exporter.dq(schem) + "." + Exporter.dq(name); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1468 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1469 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1470 public final String toString() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1471 return fqname; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1472 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1473 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1474 static final Table findTable(final String schname, final String tblname, final List<Table> list) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1475 for (Table t : list) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1476 if (t.schem.equals(schname) && t.name.equals(tblname)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1477 return t; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1478 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1479 // not found |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1480 return null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1481 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1482 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1483 static final void checkForLoop(final Table table, final List<Table> parents) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1484 parents.add(table); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1485 for (int i = 0; i < table.needs.size(); i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1486 Table child = table.needs.get(i); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1487 if (parents.contains(child)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1488 throw new Exception("Cyclic dependency graphs are not supported (cycle detected for " + child.fqname + ")"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1489 checkForLoop(child, parents); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1490 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1491 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1492 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1493 |